Gemini Embedder 是一个类,允许您通过 Google Generative AI API 使用 Google 的 Gemini 嵌入模型嵌入文档。

参数

参数类型描述默认值
idstr用于 embeddings 的模型 ID"models/text-embedding-004"
task_typestr生成 embedding 的任务类型"RETRIEVAL_QUERY"
titleOptional[str]要嵌入内容的可选标题None
dimensionsOptional[int]embedding 的输出维度768
api_keyOptional[str]Google API 密钥环境变量 GOOGLE_API_KEY
request_paramsOptional[Dict[str, Any]]embedding 请求的附加参数None
client_paramsOptional[Dict[str, Any]]客户端初始化的附加参数None
gemini_clientOptional[ModuleType]预先配置的 Gemini 客户端None