agentkit.llm_api

Functions

get_query(LLM_name)

Get the query model for the specified LLM_name.

get_token_counts()

query_claude_chat(model)

query_gpt_chat(model)

query_ollama_chat(model)

agentkit.llm_api.get_query(LLM_name)

Get the query model for the specified LLM_name.

Currently supported LLMs:

  • GPT-4

  • GPT-3.5

  • Claude

  • Ollama

Parameters:

  • LLM_name (str) – Name of the LLM model

  • tokenmodel_path (str) – Path to the token model

Returns:

The query model for the specified LLM_name

Return type:

LLM_API_FUNCTION

Modules

agentkit.llm_api.base