Add api key as field in Gemini Embedding (run-llama#14061)

feat(gemini-embedding): add api key as field
gladiaio · Jun 11, 2024 · 60129df · 60129df
1 parent a80fb09
commit 60129df
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/...tegrations/embeddings/llama-index-embeddings-gemini/llama_index/embeddings/gemini/base.py b/...tegrations/embeddings/llama-index-embeddings-gemini/llama_index/embeddings/gemini/base.py
@@ -33,6 +33,10 @@ class GeminiEmbedding(BaseEmbedding):
         default="retrieval_document",
         description="The task for embedding model.",
     )
+    api_key: Optional[str] = Field(
+        default=None,
+        description="API key to access the model. Defaults to None.",
+    )
 
     def __init__(
         self,
@@ -60,6 +64,7 @@ def __init__(
         self._model = gemini
 
         super().__init__(
+            api_key=api_key,
             model_name=model_name,
             embed_batch_size=embed_batch_size,
             callback_manager=callback_manager,

diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-gemini/pyproject.toml b/llama-index-integrations/embeddings/llama-index-embeddings-gemini/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-embeddings-gemini"
 readme = "README.md"
-version = "0.1.7"
+version = "0.1.8"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"