# 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