langroid 0.1.226__py3-none-any.whl → 0.1.227__py3-none-any.whl

langroid/agent/openai_assistant.py

@@ -8,6 +8,10 @@ from enum import Enum
 from typing import Any, Dict, List, Optional, Tuple, Type, cast, no_type_check
 
 from openai.types.beta import Assistant, Thread
+from openai.types.beta.assistant_update_params import (
+    ToolResources,
+    ToolResourcesCodeInterpreter,
+)
 from openai.types.beta.threads import Message, Run
 from openai.types.beta.threads.runs import RunStep
 from pydantic import BaseModel
@@ -29,7 +33,7 @@ logger = logging.getLogger(__name__)
 
 
 class ToolType(str, Enum):
-    RETRIEVAL = "retrieval"
+    RETRIEVAL = "file_search"
     CODE_INTERPRETER = "code_interpreter"
     FUNCTION = "function"
 
@@ -127,7 +131,11 @@ class OpenAIAssistant(ChatAgent):
         self.config.files = list(set(self.config.files + files))
         self.assistant = self.assistants.update(
             self.assistant.id,
-            file_ids=[f.id for f in self.files],
+            tool_resources=ToolResources(
+                code_interpreter=ToolResourcesCodeInterpreter(
+                    file_ids=[f.id for f in self.files],
+                ),
+            ),
         )
 
     def add_assistant_tools(self, tools: List[AssistantTool]) -> None:

langroid/agent/special/doc_chat_agent.py

@@ -169,7 +169,8 @@ class DocChatAgentConfig(ChatAgentConfig):
         dims=1536,
     )
 
-    vecdb: VectorStoreConfig = LanceDBConfig(
+    # Allow vecdb to be None in case we want to explicitly set it later
+    vecdb: Optional[VectorStoreConfig] = LanceDBConfig(
         collection_name="doc-chat-lancedb",
         replace_collection=True,
         storage_path=".lancedb/data/",
@@ -209,12 +210,13 @@ class DocChatAgent(ChatAgent):
 
     def clear(self) -> None:
         """Clear the document collection and the specific collection in vecdb"""
-        if self.vecdb is None:
-            raise ValueError("VecDB not set")
         self.original_docs = []
         self.original_docs_length = 0
         self.chunked_docs = []
         self.chunked_docs_clean = []
+        if self.vecdb is None:
+            logger.warning("Attempting to clear VecDB, but VecDB not set.")
+            return
         collection_name = self.vecdb.config.collection_name
         if collection_name is None:
             return

langroid/vector_store/qdrantdb.py

@@ -192,8 +192,7 @@ class QdrantDB(VectorStore):
                 with the same name. Defaults to False.
         """
         self.config.collection_name = collection_name
-        collections = self.list_collections()
-        if collection_name in collections:
+        if self.client.collection_exists(collection_name=collection_name):
             coll = self.client.get_collection(collection_name=collection_name)
             if (
                 coll.status == CollectionStatus.GREEN
@@ -205,7 +204,8 @@ class QdrantDB(VectorStore):
                     return
                 else:
                     logger.warning("Recreating fresh collection")
-        self.client.recreate_collection(
+            self.client.delete_collection(collection_name=collection_name)
+        self.client.create_collection(
             collection_name=collection_name,
             vectors_config=VectorParams(
                 size=self.embedding_dim,
@@ -214,7 +214,7 @@ class QdrantDB(VectorStore):
         )
         collection_info = self.client.get_collection(collection_name=collection_name)
         assert collection_info.status == CollectionStatus.GREEN
-        assert collection_info.vectors_count == 0
+        assert collection_info.vectors_count in [0, None]
         if settings.debug:
             level = logger.getEffectiveLevel()
             logger.setLevel(logging.INFO)

{langroid-0.1.226.dist-info → langroid-0.1.227.dist-info}/METADATA

@@ -1,14 +1,13 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.1.226
+Version: 0.1.227
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
 Author-email: pchalasani@gmail.com
-Requires-Python: >=3.9.1,<3.12
+Requires-Python: >=3.11,<3.12
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Provides-Extra: chainlit
 Provides-Extra: chromadb

{langroid-0.1.226.dist-info → langroid-0.1.227.dist-info}/RECORD

@@ -8,9 +8,9 @@ langroid/agent/chat_agent.py,sha256=X5uVMm9qdw3j-FRf4hbN8k8ByaSdtQCTuU8olKE0sbs,
 langroid/agent/chat_document.py,sha256=PL8iA1ZYjXNFVa3kMO8T4sbbM3rzHWpRAY6PN_7n5LQ,7969
 langroid/agent/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/junk,sha256=LxfuuW7Cijsg0szAzT81OjWWv1PMNI-6w_-DspVIO2s,339
-langroid/agent/openai_assistant.py,sha256=QTLBgnH6Btf2GWzN-WApvra-vPQWvYcXcAOULuIy4Ig,32702
+langroid/agent/openai_assistant.py,sha256=xWSDR4SyMcZhCUkgaM4-mu77rbEDF_xpt7N8m8CkiA4,32962
 langroid/agent/special/__init__.py,sha256=XPE076zD-roskxNBn-A1hnh4AHoMiQN9gk1UDjPaBaU,1201
-langroid/agent/special/doc_chat_agent.py,sha256=OpPWLFHBi3C5V6H0xRgmXUubmvy80EJRTgS6YkYfBJA,53242
+langroid/agent/special/doc_chat_agent.py,sha256=Ur57J3xolTaoTgzr2DfP8TY7BViIMNFGNEmmRfHbkL4,53373
 langroid/agent/special/lance_doc_chat_agent.py,sha256=USp0U3eTaJzwF_3bdqE7CedSLbaqAi2tm-VzygcyLaA,10175
 langroid/agent/special/lance_rag/__init__.py,sha256=QTbs0IVE2ZgDg8JJy1zN97rUUg4uEPH7SLGctFNumk4,174
 langroid/agent/special/lance_rag/critic_agent.py,sha256=pi_9eMBxEycbWTddtq_yz-mOb2V4SgGm3zfsOH1HU-Q,5775
@@ -120,8 +120,8 @@ langroid/vector_store/lancedb.py,sha256=lbl8wZuV6GNw0LnIwOSriSNwoMEba90umQTcQHtM
 langroid/vector_store/meilisearch.py,sha256=d2huA9P-NoYRuAQ9ZeXJmMKr7ry8u90RUSR28k2ecQg,11340
 langroid/vector_store/momento.py,sha256=9cui31TTrILid2KIzUpBkN2Ey3g_CZWOQVdaFsA4Ors,10045
 langroid/vector_store/qdrant_cloud.py,sha256=3im4Mip0QXLkR6wiqVsjV1QvhSElfxdFSuDKddBDQ-4,188
-langroid/vector_store/qdrantdb.py,sha256=_egbsP9SWBwmI827EDYSSOqfIQSmwNsmJfFTxrLpWYE,13457
-langroid-0.1.226.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.1.226.dist-info/METADATA,sha256=Dssqa7k6VMEBWxwulUiub5XM9TyreczBTC-xrNGXeK4,48003
-langroid-0.1.226.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langroid-0.1.226.dist-info/RECORD,,
+langroid/vector_store/qdrantdb.py,sha256=foKRxRv0BBony6S4Vt0Vav9Rn9HMxZvcIh1cE7nosFE,13524
+langroid-0.1.227.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.1.227.dist-info/METADATA,sha256=pm5XedC6otRVliKik-XnZd7l0DUn9QpSXnA_tFTiDfI,47951
+langroid-0.1.227.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langroid-0.1.227.dist-info/RECORD,,