Gemini Embedder 是一个类,允许您通过 Google Generative AI API 使用 Google 的 Gemini 嵌入模型嵌入文档。
参数 | 类型 | 描述 | 默认值 |
---|---|---|---|
id | str | 用于 embeddings 的模型 ID | "models/text-embedding-004" |
task_type | str | 生成 embedding 的任务类型 | "RETRIEVAL_QUERY" |
title | Optional[str] | 要嵌入内容的可选标题 | None |
dimensions | Optional[int] | embedding 的输出维度 | 768 |
api_key | Optional[str] | Google API 密钥 | 环境变量 GOOGLE_API_KEY |
request_params | Optional[Dict[str, Any]] | embedding 请求的附加参数 | None |
client_params | Optional[Dict[str, Any]] | 客户端初始化的附加参数 | None |
gemini_client | Optional[ModuleType] | 预先配置的 Gemini 客户端 | None |