janus.llm.models_info#

Attributes#

Functions#

load_model(model_name)

Module Contents#

janus.llm.models_info.log#
janus.llm.models_info.openai_model_reroutes#
janus.llm.models_info.openai_models = ['gpt-4-0613', 'gpt-4-1106-preview', 'gpt-4-0125-preview', 'gpt-4o-2024-05-13', 'gpt-3.5-turbo-0125']#
janus.llm.models_info.claude_models = ['bedrock-claude-v2', 'bedrock-claude-instant-v1', 'bedrock-claude-haiku', 'bedrock-claude-sonnet']#
janus.llm.models_info.llama2_models = ['bedrock-llama2-70b', 'bedrock-llama2-70b-chat', 'bedrock-llama2-13b', 'bedrock-llama2-13b-chat']#
janus.llm.models_info.llama3_models = ['bedrock-llama3-8b-instruct', 'bedrock-llama3-70b-instruct']#
janus.llm.models_info.titan_models = ['bedrock-titan-text-lite', 'bedrock-titan-text-express', 'bedrock-jurassic-2-mid',...#
janus.llm.models_info.cohere_models = ['bedrock-command-r-plus']#
janus.llm.models_info.mistral_models = ['bedrock-mistral-7b-instruct', 'bedrock-mistral-large', 'bedrock-mixtral']#
janus.llm.models_info.bedrock_models#
janus.llm.models_info.all_models#
janus.llm.models_info.MODEL_TYPE_CONSTRUCTORS: dict[str, Callable[[Any], langchain_core.language_models.BaseLanguageModel]]#
janus.llm.models_info.MODEL_PROMPT_ENGINES: dict[str, Callable[Ellipsis, janus.prompts.prompt.PromptEngine]]#
janus.llm.models_info.model_identifiers#
janus.llm.models_info.MODEL_DEFAULT_ARGUMENTS: dict[str, dict[str, str]]#
janus.llm.models_info.DEFAULT_MODELS#
janus.llm.models_info.MODEL_CONFIG_DIR#
janus.llm.models_info.MODEL_TYPES: dict[str, janus.prompts.prompt.PromptEngine]#
janus.llm.models_info.TOKEN_LIMITS: dict[str, int]#
janus.llm.models_info.load_model(model_name)#
Parameters:

model_name (str) –

Return type:

tuple[langchain_core.language_models.BaseLanguageModel, int, dict[str, float]]