langroid 0.37.3__tar.gz → 0.37.5__tar.gz

{langroid-0.37.3 → langroid-0.37.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.37.3
+Version: 0.37.5
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT

{langroid-0.37.3 → langroid-0.37.5}/langroid/agent/chat_agent.py

@@ -1023,7 +1023,7 @@ class ChatAgent(Agent):
         all_tools: bool = False,
     ) -> List[ToolMessage]:
         """
-        Extracts messages and tracks whether any errors occured. If strict mode
+        Extracts messages and tracks whether any errors occurred. If strict mode
         was enabled, disables it for the tool, else triggers strict recovery.
         """
         self.tool_error = False

{langroid-0.37.3 → langroid-0.37.5}/langroid/agent/special/sql/sql_chat_agent.py

@@ -427,15 +427,7 @@ class SQLChatAgent(ChatAgent):
         return error_message_template
 
     def _available_tool_names(self) -> str:
-        return ",".join(
-            tool.name()  # type: ignore
-            for tool in [
-                RunQueryTool,
-                GetTableNamesTool,
-                GetTableSchemaTool,
-                GetColumnDescriptionsTool,
-            ]
-        )
+        return ",".join(self.llm_tools_usable)
 
     def _tool_result_llm_answer_prompt(self) -> str:
         """
@@ -510,7 +502,7 @@ class SQLChatAgent(ChatAgent):
         {self._tool_result_llm_answer_prompt()}
         OTHERWISE:
              continue using one of your available TOOLs:
-             {self._available_tool_names()}
+             {",".join(self.llm_tools_usable)}
         """
         return final_message
 

{langroid-0.37.3 → langroid-0.37.5}/langroid/language_models/openai_gpt.py

@@ -1897,7 +1897,12 @@ class OpenAIGPT(LanguageModel):
             }
         }
         """
-        message = response["choices"][0]["message"]
+        if response.get("choices") is None:
+            message = {}
+        else:
+            message = response["choices"][0].get("message", {})
+        if message is None:
+            message = {}
         msg = message.get("content", "")
         reasoning = message.get("reasoning_content", "")
 

{langroid-0.37.3 → langroid-0.37.5}/langroid/vector_store/base.py

@@ -8,7 +8,7 @@ import pandas as pd
 
 from langroid.embedding_models.base import EmbeddingModel, EmbeddingModelsConfig
 from langroid.embedding_models.models import OpenAIEmbeddingsConfig
-from langroid.mytypes import DocMetaData, Document
+from langroid.mytypes import DocMetaData, Document, EmbeddingFunction
 from langroid.pydantic_v1 import BaseSettings
 from langroid.utils.algorithms.graph import components, topological_sort
 from langroid.utils.configuration import settings
@@ -51,6 +51,7 @@ class VectorStore(ABC):
             self.embedding_model = EmbeddingModel.create(config.embedding)
         else:
             self.embedding_model = config.embedding_model
+        self.embedding_fn: EmbeddingFunction = self.embedding_model.embedding_fn()
 
     @staticmethod
     def create(config: VectorStoreConfig) -> Optional["VectorStore"]:
@@ -86,6 +87,10 @@ class VectorStore(ABC):
             )
             return None
 
+    @property
+    def embedding_dim(self) -> int:
+        return len(self.embedding_fn(["test"])[0])
+
     @abstractmethod
     def clear_empty_collections(self) -> int:
         """Clear all empty collections in the vector store.

{langroid-0.37.3 → langroid-0.37.5}/langroid/vector_store/chromadb.py

@@ -35,7 +35,6 @@ class ChromaDB(VectorStore):
         except ImportError:
             raise LangroidImportError("chromadb", "chromadb")
         self.config = config
-        self.embedding_fn = self.embedding_model.embedding_fn()
         self.client = chromadb.Client(
             chromadb.config.Settings(
                 # chroma_db_impl="duckdb+parquet",

{langroid-0.37.3 → langroid-0.37.5}/langroid/vector_store/lancedb.py

@@ -26,7 +26,7 @@ from langroid.embedding_models.base import (
 )
 from langroid.embedding_models.models import OpenAIEmbeddingsConfig
 from langroid.exceptions import LangroidImportError
-from langroid.mytypes import Document, EmbeddingFunction
+from langroid.mytypes import Document
 from langroid.utils.configuration import settings
 from langroid.utils.pydantic_utils import (
     dataframe_to_document_model,
@@ -60,8 +60,6 @@ class LanceDB(VectorStore):
             raise LangroidImportError("lancedb", "lancedb")
 
         self.config: LanceDBConfig = config
-        self.embedding_fn: EmbeddingFunction = self.embedding_model.embedding_fn()
-        self.embedding_dim = self.embedding_model.embedding_dims
         self.host = config.host
         self.port = config.port
         self.is_from_dataframe = False  # were docs ingested from a dataframe?

{langroid-0.37.3 → langroid-0.37.5}/langroid/vector_store/momento.py

@@ -37,7 +37,7 @@ from langroid.embedding_models.base import (
     EmbeddingModelsConfig,
 )
 from langroid.embedding_models.models import OpenAIEmbeddingsConfig
-from langroid.mytypes import Document, EmbeddingFunction
+from langroid.mytypes import Document
 from langroid.utils.configuration import settings
 from langroid.utils.pydantic_utils import (
     flatten_pydantic_instance,
@@ -61,8 +61,6 @@ class MomentoVI(VectorStore):
             raise LangroidImportError("momento", "momento")
         self.distance = SimilarityMetric.COSINE_SIMILARITY
         self.config: MomentoVIConfig = config
-        self.embedding_fn: EmbeddingFunction = self.embedding_model.embedding_fn()
-        self.embedding_dim = self.embedding_model.embedding_dims
         self.host = config.host
         self.port = config.port
         load_dotenv()

{langroid-0.37.3 → langroid-0.37.5}/langroid/vector_store/qdrantdb.py

@@ -27,7 +27,7 @@ from langroid.embedding_models.base import (
     EmbeddingModelsConfig,
 )
 from langroid.embedding_models.models import OpenAIEmbeddingsConfig
-from langroid.mytypes import Document, EmbeddingFunction, Embeddings
+from langroid.mytypes import Document, Embeddings
 from langroid.utils.configuration import settings
 from langroid.vector_store.base import VectorStore, VectorStoreConfig
 
@@ -77,8 +77,6 @@ class QdrantDB(VectorStore):
     def __init__(self, config: QdrantDBConfig = QdrantDBConfig()):
         super().__init__(config)
         self.config: QdrantDBConfig = config
-        self.embedding_fn: EmbeddingFunction = self.embedding_model.embedding_fn()
-        self.embedding_dim = len(self.embedding_fn(["test"])[0])
         if self.config.use_sparse_embeddings:
             try:
                 from transformers import AutoModelForMaskedLM, AutoTokenizer

{langroid-0.37.3 → langroid-0.37.5}/langroid/vector_store/weaviatedb.py

@@ -10,7 +10,7 @@ from langroid.embedding_models.base import (
 )
 from langroid.embedding_models.models import OpenAIEmbeddingsConfig
 from langroid.exceptions import LangroidImportError
-from langroid.mytypes import DocMetaData, Document, EmbeddingFunction
+from langroid.mytypes import DocMetaData, Document
 from langroid.utils.configuration import settings
 from langroid.vector_store.base import VectorStore, VectorStoreConfig
 
@@ -38,8 +38,6 @@ class WeaviateDB(VectorStore):
     def __init__(self, config: WeaviateDBConfig = WeaviateDBConfig()):
         super().__init__(config)
         self.config: WeaviateDBConfig = config
-        self.embedding_fn: EmbeddingFunction = self.embedding_model.embedding_fn()
-        self.embedding_dim = self.embedding_model.embedding_dims
         load_dotenv()
         key = os.getenv("WEAVIATE_API_KEY")
         url = os.getenv("WEAVIATE_API_URL")

{langroid-0.37.3 → langroid-0.37.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langroid"
-version = "0.37.3"
+version = "0.37.5"
 authors = [
     {name = "Prasad Chalasani", email = "pchalasani@gmail.com"},
 ]