langroid 0.41.3__py3-none-any.whl → 0.41.4__py3-none-any.whl

langroid/embedding_models/models.py

@@ -450,9 +450,9 @@ class LlamaCppServerEmbeddings(EmbeddingModel):
 class GeminiEmbeddings(EmbeddingModel):
     def __init__(self, config: GeminiEmbeddingsConfig = GeminiEmbeddingsConfig()):
         try:
-            import google.generativeai as genai
+            from google import genai
         except ImportError as e:
-            raise LangroidImportError(extra="google-generativeai", error=str(e))
+            raise LangroidImportError(extra="google-genai", error=str(e))
         super().__init__()
         self.config = config
         load_dotenv()
@@ -464,29 +464,30 @@ class GeminiEmbeddings(EmbeddingModel):
                 GEMINI_API_KEY env variable must be set to use GeminiEmbeddings.
                 """
             )
-        genai.configure(api_key=self.config.api_key)  # type: ignore[attr-defined]
-        self.client = genai
+        self.client = genai.Client(api_key=self.config.api_key)
 
     def embedding_fn(self) -> Callable[[List[str]], Embeddings]:
         return EmbeddingFunctionCallable(self, self.config.batch_size)
 
-    def generate_embeddings(self, texts: List[str]) -> Embeddings:
-        all_embeddings = []  # More precise type hint
+    def generate_embeddings(self, texts: List[str]) -> List[List[float]]:
+        """Generates embeddings for a list of input texts."""
+        all_embeddings: List[List[float]] = []
+
         for batch in batched(texts, self.config.batch_size):
-            result = self.client.embed_content(  # type: ignore[attr-defined]
+            result = self.client.models.embed_content(  # type: ignore[attr-defined]
                 model=self.config.model_name,
-                content=batch,
-                task_type="RETRIEVAL_DOCUMENT",
+                contents=batch,
             )
 
-            embeddings = result["embedding"]
-            if not isinstance(embeddings, list):
-                raise ValueError("Unexpected format for embeddings: not a list")
+            if not hasattr(result, "embeddings") or not isinstance(
+                result.embeddings, list
+            ):
+                raise ValueError(
+                    "Unexpected format for embeddings: missing or incorrect type"
+                )
 
-            if embeddings and isinstance(embeddings[0], list):
-                all_embeddings.extend(embeddings)
-            else:
-                all_embeddings.append(embeddings)
+            # Extract .values from ContentEmbedding objects
+            all_embeddings.extend([emb.values for emb in result.embeddings])
 
         return all_embeddings
 

{langroid-0.41.3.dist-info → langroid-0.41.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.41.3
+Version: 0.41.4
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT
@@ -21,7 +21,7 @@ Requires-Dist: fakeredis<3.0.0,>=2.12.1
 Requires-Dist: fire<1.0.0,>=0.5.0
 Requires-Dist: gitpython<4.0.0,>=3.1.43
 Requires-Dist: google-api-python-client<3.0.0,>=2.95.0
-Requires-Dist: google-generativeai<1.0.0,>=0.5.2
+Requires-Dist: google-genai>=1.0.0
 Requires-Dist: groq<1.0.0,>=0.13.0
 Requires-Dist: grpcio<2.0.0,>=1.62.1
 Requires-Dist: halo<1.0.0,>=0.0.31
@@ -112,8 +112,10 @@ Provides-Extra: exa
 Requires-Dist: exa-py>=1.8.7; extra == 'exa'
 Provides-Extra: fastembed
 Requires-Dist: fastembed<0.4.0,>=0.3.1; extra == 'fastembed'
+Provides-Extra: google-genai
+Requires-Dist: google-genai>=1.0.0; extra == 'google-genai'
 Provides-Extra: google-generativeai
-Requires-Dist: google-generativeai>=0.8.4; extra == 'google-generativeai'
+Requires-Dist: google-genai>=1.0.0; extra == 'google-generativeai'
 Provides-Extra: hf-embeddings
 Requires-Dist: sentence-transformers<3.0.0,>=2.2.2; extra == 'hf-embeddings'
 Requires-Dist: torch<3.0.0,>=2.0.0; extra == 'hf-embeddings'
@@ -319,7 +321,7 @@ teacher_task.run()
 - **Jan 2025:**
   - [0.36.0](https://github.com/langroid/langroid/releases/tag/0.36.0): Weaviate vector-db support (thanks @abab-dev).
   - [0.35.0](https://github.com/langroid/langroid/releases/tag/0.35.0): Capture/Stream reasoning content from 
-    Reasoning LLMs (e.g. DeepSeek, OpenAI o1) in addition to final answer.
+    Reasoning LLMs (e.g. DeepSeek-R1, OpenAI o1) in addition to final answer.
   - [0.34.0](https://github.com/langroid/langroid/releases/tag/0.34.0): DocChatAgent 
     chunk enrichment to improve retrieval. (collaboration with @dfm88). 
   - [0.33.0](https://github.com/langroid/langroid/releases/tag/0.33.3) Move from Poetry to uv! (thanks @abab-dev).
@@ -627,7 +629,7 @@ For many practical scenarios, you may need additional optional dependencies:
 - For "chat with databases", use the `db` extra:
     ```bash
     pip install "langroid[db]"
-    ``
+    ```
 - You can specify multiple extras by separating them with commas, e.g.:
     ```bash
     pip install "langroid[doc-chat,db]"

{langroid-0.41.3.dist-info → langroid-0.41.4.dist-info}/RECORD

@@ -59,7 +59,7 @@ langroid/cachedb/momento_cachedb.py,sha256=YEOJ62hEcV6iIeMr5aGgRYgWQqFYaej9gEDEc
 langroid/cachedb/redis_cachedb.py,sha256=7kgnbf4b5CKsCrlL97mHWKvdvlLt8zgn7lc528jEpiE,5141
 langroid/embedding_models/__init__.py,sha256=KyYxR3jDFUCfYjSuCL86qjAmrq6mXXjOT4lFNOKVj6Y,955
 langroid/embedding_models/base.py,sha256=Ml7oA6PzQm0wZmIYn3fhF7dvZCi-amviWUwOeBegH3A,2562
-langroid/embedding_models/models.py,sha256=fKQBiaaG7uYoeELDbAiNxwLdn-CWN8dyiVEZcdk_bjI,18959
+langroid/embedding_models/models.py,sha256=93rQjvcltGLDYbgT1TIW8PZiAbNPAvPAqxCLlLS9GAo,18955
 langroid/embedding_models/remote_embeds.py,sha256=6_kjXByVbqhY9cGwl9R83ZcYC2km-nGieNNAo1McHaY,5151
 langroid/embedding_models/protoc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/embedding_models/protoc/embeddings.proto,sha256=_O-SgFpTaylQeOTgSpxhEJ7CUw7PeCQQJLaPqpPYKJg,321
@@ -128,7 +128,7 @@ langroid/vector_store/pineconedb.py,sha256=otxXZNaBKb9f_H75HTaU3lMHiaR2NUp5MqwLZ
 langroid/vector_store/postgres.py,sha256=DQHd6dt-OcV_QVNm-ymn28rlTfhI6hqgcpLTPCsm0jI,15990
 langroid/vector_store/qdrantdb.py,sha256=v7TAsIoj_vxeKDYS9tpwJLBZA8fuTweTYxHo0X_uawM,17949
 langroid/vector_store/weaviatedb.py,sha256=ONEr2iGS0Ii73oMe7tRk6bB-BEXQUa70fYSrdI8d3yo,11481
-langroid-0.41.3.dist-info/METADATA,sha256=hQDjq_z2rXui7guy7-_TweJvkSkGWhQ-iQZb5i0vyIk,61259
-langroid-0.41.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.41.3.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.41.3.dist-info/RECORD,,
+langroid-0.41.4.dist-info/METADATA,sha256=hU062qh537keZYiNQZhdKWffOldsn_lz5BWYew1TQbg,61331
+langroid-0.41.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.41.4.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.41.4.dist-info/RECORD,,