Skip to main content

Google AI Studio

google 提供商兼容 Google AI Studio(以前称为 PaLM),提供对 Gemini 模型的访问。

您可以通过指定其中一个 可用模型 来使用它。目前支持以下模型:

  • google:gemini-pro
  • google:gemini-pro-vision
  • google:aqa(归属问答)
  • google:chat-bison-001
tip

如果您使用的是 Google Vertex,请参阅 vertex 提供商

支持的环境变量:

  • GOOGLE_API_KEY(必需)- Google AI Studio/PaLM API 令牌
  • GOOGLE_API_HOST - 用于覆盖 Google API 主机,默认为 generativelanguage.googleapis.com

PaLM 提供商支持各种 配置选项,例如 safetySettingsstopSequencestemperaturemaxOutputTokenstopPtopK,可用于自定义模型的行为,如下所示:

providers:
- id: google:gemini-pro
config:
temperature: 0
maxOutputTokens: 1024

您还可以传入一个 responseSchema 文件:

providers:
- id: google:gemini-pro
config:
responseSchema: file://test.json

如果您想要更精细地控制 generationConfig

providers:
- id: google:gemini-1.5-flash-002
config:
generationConfig:
response_mime_type: application/json
response_schema:
type: object
properties:
foo:
type: string