langroid 0.41.3__tar.gz → 0.41.4__tar.gz

{langroid-0.41.3 → langroid-0.41.4}/PKG-INFO

@@ -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 → langroid-0.41.4}/README.md

@@ -138,7 +138,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).
@@ -446,7 +446,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 → langroid-0.41.4}/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 → langroid-0.41.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langroid"
-version = "0.41.3"
+version = "0.41.4"
 authors = [
     {name = "Prasad Chalasani", email = "pchalasani@gmail.com"},
 ]
@@ -24,7 +24,7 @@ dependencies = [
     "fire<1.0.0,>=0.5.0",
     "gitpython<4.0.0,>=3.1.43",
     "google-api-python-client<3.0.0,>=2.95.0",
-    "google-generativeai<1.0.0,>=0.5.2",
+    "google-genai>=1.0.0",
     "groq<1.0.0,>=0.13.0",
     "grpcio<2.0.0,>=1.62.1",
     "halo<1.0.0,>=0.0.31",
@@ -236,8 +236,12 @@ momento = [
 fastembed = [
     "fastembed<0.4.0,>=0.3.1",
 ]
+google-genai = [
+    "google-genai>=1.0.0",
+]
+
 google-generativeai = [
-    "google-generativeai>=0.8.4",
+    "google-genai>=1.0.0",
 ]
 
 pinecone = [