24 lines
553 B
YAML
24 lines
553 B
YAML
|
# models_config.yaml
|
||
|
|
|
||
|
|
GEMINI:
|
||
|
|
api_key_env: GEMINI_API_KEY
|
||
|
|
base_url: https://generativelanguage.googleapis.com/v1beta
|
||
|
|
supports_switching: true
|
||
|
|
switch_options:
|
||
|
|
small:
|
||
|
|
name: gemini-pro
|
||
|
|
max_tokens: 2048
|
||
|
|
large:
|
||
|
|
name: gemini-1.5-pro-latest
|
||
|
|
max_tokens: 8192
|
||
|
|
OPENAI:
|
||
|
|
api_key_env: OPENAI_API_KEY
|
||
|
|
base_url: null # Indicates to use the default OpenAI API base URL
|
||
|
|
supports_switching: true
|
||
|
|
switch_options:
|
||
|
|
small:
|
||
|
|
name: gpt-3.5-turbo
|
||
|
|
max_tokens: null
|
||
|
|
large:
|
||
|
|
name: gpt-4
|
||
|
|
max_tokens: null
|