langroid 0.19.5__py3-none-any.whl → 0.20.1__py3-none-any.whl

langroid/agent/special/neo4j/system_messages.py

@@ -0,0 +1,120 @@
+from langroid.agent.special.neo4j.tools import (
+    cypher_creation_tool_name,
+    cypher_retrieval_tool_name,
+    graph_schema_tool_name,
+)
+from langroid.agent.tools.orchestration import DoneTool
+
+done_tool_name = DoneTool.default_value("request")
+
+graph_schema_tool_description = f"""
+`{graph_schema_tool_name}` tool/function-call to get all the node labels, relationship 
+ types, and property keys available in your Neo4j database. You MUST use
+ this tool BEFORE attempting to use the `{cypher_retrieval_tool_name}` tool,
+ to ensure that you are using the correct node labels, relationship types, and
+ property keys in your `{cypher_retrieval_tool_name}` tool/function-call.
+"""
+
+cypher_retrieval_tool_description = f"""
+`{cypher_retrieval_tool_name}` tool/function-call to retrieve information from the 
+     graph database to answer questions.
+"""
+
+cypher_creation_tool_description = f"""
+`{cypher_creation_tool_name}` tool/function-call to execute cypher query that creates
+   entities/relationships in the graph database.
+"""
+
+cypher_query_instructions = """
+You must be smart about using the right node labels, relationship types, and property
+keys based on the english description. If you are thinking of using a node label,
+relationship type, or property key that does not exist, you are probably on the wrong 
+track, so you should try your best to answer based on an existing table or column.
+DO NOT assume any nodes or relationships other than those above.
+"""
+
+
+# sys msg to use when schema already provided initially,
+# so agent does not need to use schema tool, at least initially,
+# but may do so later if the db evolves, or if needs to bring in the schema
+# to more recent context.
+SCHEMA_PROVIDED_SYS_MSG = f"""You are a data scientist and expert in Knowledge Graphs, 
+with expertise in answering questions by interacting with a Neo4j graph database.
+
+The schema below describes the Neo4j database structure, node labels, 
+relationship types, and property keys available in your Neo4j database.
+
+=== SCHEMA ===
+{{schema}}
+=== END SCHEMA ===
+
+To help with the user's question or database update/creation request, 
+you have access to these tools:
+
+- {cypher_retrieval_tool_description}
+
+- {cypher_creation_tool_description}
+
+Since the schema has been provided, you may not need to use the tool below,
+but you may use it if you need to remind yourself about the schema:
+
+- {graph_schema_tool_description}
+ 
+"""
+
+# sys msg to use when schema is not initially provided,
+# and we want agent to use schema tool to get schema
+SCHEMA_TOOLS_SYS_MSG = f"""You are a data scientist and expert in Knowledge Graphs, 
+with expertise in answering questions by querying Neo4j database.
+You have access to the following tools:
+
+- {graph_schema_tool_description}
+
+- {cypher_retrieval_tool_description}
+
+- {cypher_creation_tool_description}
+
+"""
+
+DEFAULT_NEO4J_CHAT_SYSTEM_MESSAGE = f"""
+{{mode}}
+
+You do not need to be able to answer a question with just one query. 
+You could make a sequence of Cypher queries to find the answer to the question.
+
+{cypher_query_instructions}
+
+
+
+RETRY-SUGGESTIONS:
+If you receive a null or other unexpected result,
+(a) make sure you use the available TOOLs correctly,
+(b) USE `{graph_schema_tool_name}` tool/function-call to get all the node labels, 
+    relationship types, and property keys available in your Neo4j database. 
+(c) LABELS are CASE-SENSITIVE -- make sure you adhere to the exact label name
+   you found in the schema.
+(d) see if you have made an assumption in your Neo4j query, and try another way, 
+   or use `{cypher_retrieval_tool_name}` to explore the database contents before 
+   submitting your final query. 
+(e) USE `{cypher_creation_tool_name}` tool/function-call to execute cypher query that 
+    creates entities/relationships in the graph database.
+(f) Try APPROXIMATE or PARTIAL MATCHES to strings in the user's query, 
+    e.g. user may ask about "Godfather" instead of "The Godfather",
+    or try using CASE-INSENSITIVE MATCHES.
+
+Start by asking what the user needs help with.
+"""
+
+ADDRESSING_INSTRUCTION = """
+IMPORTANT - Whenever you are NOT writing a CYPHER query, make sure you address the 
+user using {prefix}User. You MUST use the EXACT syntax {prefix} !!!
+
+In other words, you ALWAYS EITHER:
+ - write a CYPHER query using one of the tools, 
+ - OR address the user using {prefix}User.
+"""
+
+DONE_INSTRUCTION = f"""
+When you finally have the answer to a user's query or request, 
+use the `{done_tool_name}` with `content` set to the answer or result.
+"""

langroid/agent/special/neo4j/tools.py

@@ -0,0 +1,32 @@
+from langroid.agent import ToolMessage
+
+
+class CypherRetrievalTool(ToolMessage):
+    request: str = "cypher_retrieval_tool"
+    purpose: str = """To send the <cypher_query> to retrieve 
+        data from the graph database based on provided text description and schema.
+        """
+    cypher_query: str
+
+
+cypher_retrieval_tool_name = CypherRetrievalTool.default_value("request")
+
+
+class CypherCreationTool(ToolMessage):
+    request: str = "cypher_creation_tool"
+    purpose: str = """
+        To send the <cypher_query> to create 
+        entities/relationships in the graph database.
+        """
+    cypher_query: str
+
+
+cypher_creation_tool_name = CypherCreationTool.default_value("request")
+
+
+class GraphSchemaTool(ToolMessage):
+    request: str = "graph_schema_tool"
+    purpose: str = """To get the schema of the graph database."""
+
+
+graph_schema_tool_name = GraphSchemaTool.default_value("request")

langroid/agent/special/sql/sql_chat_agent.py

@@ -14,7 +14,7 @@ from rich import print
 from rich.console import Console
 
 from langroid.exceptions import LangroidImportError
-from langroid.utils.constants import DONE
+from langroid.utils.constants import DONE, SEND_TO
 
 try:
     from sqlalchemy import MetaData, Row, create_engine, inspect, text
@@ -96,7 +96,7 @@ class SQLChatAgentConfig(ChatAgentConfig):
     context_descriptions: Dict[str, Dict[str, Union[str, Dict[str, str]]]] = {}
     use_schema_tools: bool = False
     multi_schema: bool = False
-    addressing_prefix: str = ""
+    addressing_prefix: str = SEND_TO
 
     """
     Optional, but strongly recommended, context descriptions for tables, columns, 
@@ -257,6 +257,7 @@ class SQLChatAgent(ChatAgent):
         self, message: Optional[str | ChatDocument] = None
     ) -> Optional[ChatDocument]:
         self.llm_responded = True
+        self.used_run_query = False
         return super().llm_response(message)
 
     def user_response(
@@ -270,7 +271,11 @@ class SQLChatAgent(ChatAgent):
     def handle_message_fallback(
         self, msg: str | ChatDocument
     ) -> str | ChatDocument | None:
-
+        """
+        Handle the scenario where current msg is not a tool.
+        Special handling is only needed if the message was from the LLM
+        (as indicated by self.llm_responded).
+        """
         if not self.llm_responded:
             return None
         if self.used_run_query:

langroid/agent/task.py

@@ -1226,7 +1226,6 @@ class Task:
             # reset stuck counter since we made progress
             self.n_stalled_steps = 0
 
-
         if self.pending_message is not None:
             if (
                 self._is_done_response(result, r)
@@ -1331,6 +1330,9 @@ class Task:
                     max_cost=self.max_cost,
                     max_tokens=self.max_tokens,
                 )
+                # update result.tool_messages if any
+                if isinstance(result, ChatDocument):
+                    self.agent.get_tool_messages(result)
                 if result is not None:
                     content, id2result, oai_tool_id = self.agent.process_tool_results(
                         result.content,
@@ -1359,6 +1361,9 @@ class Task:
         else:
             response_fn = self._entity_responder_map[cast(Entity, e)]
             result = response_fn(self.pending_message)
+            # update result.tool_messages if any
+            if isinstance(result, ChatDocument):
+                self.agent.get_tool_messages(result)
 
         result_chat_doc = self.agent.to_ChatDocument(
             result,
@@ -1389,7 +1394,7 @@ class Task:
             # ignore all string-based signaling/routing
             return result
         # parse various routing/addressing strings in result
-        is_pass, recipient, content = parse_routing(
+        is_pass, recipient, content = self._parse_routing(
             result,
             addressing_prefix=self.config.addressing_prefix,
         )
@@ -1522,7 +1527,7 @@ class Task:
         oai_tool_id2result = result_msg.oai_tool_id2result if result_msg else None
         fun_call = result_msg.function_call if result_msg else None
         tool_messages = result_msg.tool_messages if result_msg else []
-        # if there is an LLMDoneTool or AgentDoneTool among these,
+        # if there is an DoneTool or AgentDoneTool among these,
         # we extract content and tools from here, and ignore all others
         for t in tool_messages:
             if isinstance(t, FinalResultTool):
@@ -1534,6 +1539,8 @@ class Task:
                 # there shouldn't be multiple tools like this; just take the first
                 content = to_string(t.content)
                 content_any = t.content
+                fun_call = None
+                oai_tool_calls = None
                 if isinstance(t, AgentDoneTool):
                     # AgentDoneTool may have tools, unlike DoneTool
                     tool_messages = t.tools
@@ -1906,6 +1913,7 @@ class Task:
             # user, then wait for user response
             self.pending_message is not None
             and self.pending_message.metadata.recipient == Entity.USER
+            and not self.agent.has_tool_message_attempt(self.pending_message)
         )
 
     def _can_respond(self, e: Responder) -> bool:
@@ -1940,58 +1948,72 @@ class Task:
         """
         self.color_log = enable
 
+    def _parse_routing(
+        self,
+        msg: ChatDocument | str,
+        addressing_prefix: str = "",
+    ) -> Tuple[bool | None, str | None, str | None]:
+        """
+        Parse routing instruction if any, of the form:
+        PASS:<recipient>  (pass current pending msg to recipient)
+        SEND:<recipient> <content> (send content to recipient)
+        @<recipient> <content> (send content to recipient)
+        Args:
+            msg (ChatDocument|str|None): message to parse
+            addressing_prefix (str): prefix to address other agents or entities,
+                 (e.g. "@". See documentation of `TaskConfig` for details).
+        Returns:
+            Tuple[bool|None, str|None, str|None]:
+                bool: true=PASS, false=SEND, or None if neither
+                str: recipient, or None
+                str: content to send, or None
+        """
+        # handle routing instruction-strings in result if any,
+        # such as PASS, PASS_TO, or SEND
 
-def parse_routing(
-    msg: ChatDocument | str,
-    addressing_prefix: str = "",
-) -> Tuple[bool | None, str | None, str | None]:
-    """
-    Parse routing instruction if any, of the form:
-    PASS:<recipient>  (pass current pending msg to recipient)
-    SEND:<recipient> <content> (send content to recipient)
-    @<recipient> <content> (send content to recipient)
-    Args:
-        msg (ChatDocument|str|None): message to parse
-        addressing_prefix (str): prefix to address other agents or entities,
-             (e.g. "@". See documentation of `TaskConfig` for details).
-    Returns:
-        Tuple[bool|None, str|None, str|None]:
-            bool: true=PASS, false=SEND, or None if neither
-            str: recipient, or None
-            str: content to send, or None
-    """
-    # handle routing instruction in result if any,
-    # of the form PASS=<recipient>
-    content = msg.content if isinstance(msg, ChatDocument) else msg
-    content = content.strip()
-    if PASS in content and PASS_TO not in content:
-        return True, None, None
-    if PASS_TO in content and content.split(":")[1] != "":
-        return True, content.split(":")[1], None
-    if (
-        SEND_TO in content
-        and (addressee_content := parse_addressed_message(content, SEND_TO))[0]
-        is not None
-    ):
-        (addressee, content_to_send) = addressee_content
-        # if no content then treat same as PASS_TO
-        if content_to_send == "":
-            return True, addressee, None
-        else:
-            return False, addressee, content_to_send
-    if (
-        addressing_prefix != ""
-        and addressing_prefix in content
-        and (addressee_content := parse_addressed_message(content, addressing_prefix))[
-            0
-        ]
-        is not None
-    ):
-        (addressee, content_to_send) = addressee_content
-        # if no content then treat same as PASS_TO
-        if content_to_send == "":
-            return True, addressee, None
-        else:
-            return False, addressee, content_to_send
+        msg_str = msg.content if isinstance(msg, ChatDocument) else msg
+        if (
+            self.agent.has_tool_message_attempt(msg)
+            and not msg_str.startswith(PASS)
+            and not msg_str.startswith(PASS_TO)
+            and not msg_str.startswith(SEND_TO)
+        ):
+            # if there's an attempted tool-call, we ignore any routing strings,
+            # unless they are at the start of the msg
+            return None, None, None
+
+        content = msg.content if isinstance(msg, ChatDocument) else msg
+        content = content.strip()
+        if PASS in content and PASS_TO not in content:
+            return True, None, None
+        if PASS_TO in content and content.split(":")[1] != "":
+            return True, content.split(":")[1], None
+        if (
+            SEND_TO in content
+            and (addressee_content := parse_addressed_message(content, SEND_TO))[0]
+            is not None
+        ):
+            # Note this will discard any portion of content BEFORE SEND_TO.
+            # TODO maybe make this configurable.
+            (addressee, content_to_send) = addressee_content
+            # if no content then treat same as PASS_TO
+            if content_to_send == "":
+                return True, addressee, None
+            else:
+                return False, addressee, content_to_send
+        if (
+            addressing_prefix != ""
+            and addressing_prefix in content
+            and (
+                addressee_content := parse_addressed_message(content, addressing_prefix)
+            )[0]
+            is not None
+        ):
+            (addressee, content_to_send) = addressee_content
+            # if no content then treat same as PASS_TO
+            if content_to_send == "":
+                return True, addressee, None
+            else:
+                return False, addressee, content_to_send
 
-    return None, None, None
+        return None, None, None

langroid/agent/tools/orchestration.py

@@ -41,11 +41,11 @@ class DoneTool(ToolMessage):
     """Tool for Agent Entity (i.e. agent_response) or LLM entity (i.e. llm_response) to
     signal the current task is done, with some content as the result."""
 
-    purpose = """
+    purpose: str = """
     To signal the current task is done, along with an optional message <content>
     of arbitrary type (default None).
     """
-    request = "done_tool"
+    request: str = "done_tool"
     content: str = ""
 
     def response(self, agent: ChatAgent) -> ChatDocument:
@@ -77,7 +77,7 @@ class ResultTool(ToolMessage):
     Note:
         - when defining a tool handler or agent_response, you can directly return
             ResultTool(field1 = val1, ...),
-            where the values can be aribitrary data structures, including nested
+            where the values can be arbitrary data structures, including nested
             Pydantic objs, or you can define a subclass of ResultTool with the
             fields you want to return.
         - This is a special ToolMessage that is NOT meant to be used or handled
@@ -143,10 +143,10 @@ class PassTool(ToolMessage):
     Similar to ForwardTool, but without specifying the recipient agent.
     """
 
-    purpose = """
+    purpose: str = """
     To pass the current message so that other agents can handle it.
     """
-    request = "pass_tool"
+    request: str = "pass_tool"
 
     def response(self, agent: ChatAgent, chat_doc: ChatDocument) -> ChatDocument:
         """When this tool is enabled for an Agent, this will result in a method
@@ -178,10 +178,10 @@ class DonePassTool(PassTool):
     Similar to PassTool, except we append a DoneTool to the result tool_messages.
     """
 
-    purpose = """
+    purpose: str = """
     To signal the current task is done, with results set to the current/incoming msg.
     """
-    request = "done_pass_tool"
+    request: str = "done_pass_tool"
 
     def response(self, agent: ChatAgent, chat_doc: ChatDocument) -> ChatDocument:
         # use PassTool to get the right ChatDocument to pass...

langroid/parsing/parser.py

@@ -66,6 +66,12 @@ class Parser:
         tokens = self.tokenizer.encode(text)
         return len(tokens)
 
+    def truncate_tokens(self, text: str, max_tokens: int) -> str:
+        tokens = self.tokenizer.encode(text)
+        if len(tokens) <= max_tokens:
+            return text
+        return self.tokenizer.decode(tokens[:max_tokens])
+
     def add_window_ids(self, chunks: List[Document]) -> None:
         """Chunks may belong to multiple docs, but for each doc,
         they appear consecutively. Add window_ids in metadata"""

{langroid-0.19.5.dist-info → langroid-0.20.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.19.5
+Version: 0.20.1
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: all
+Provides-Extra: arango
 Provides-Extra: chainlit
 Provides-Extra: chromadb
 Provides-Extra: db
@@ -35,7 +36,9 @@ Provides-Extra: sql
 Provides-Extra: transformers
 Provides-Extra: unstructured
 Provides-Extra: vecdbs
+Requires-Dist: adb-cloud-connector (>=1.0.2,<2.0.0)
 Requires-Dist: aiohttp (>=3.9.1,<4.0.0)
+Requires-Dist: arango-datasets (>=1.2.2,<2.0.0) ; extra == "all" or extra == "arango"
 Requires-Dist: async-generator (>=1.10,<2.0)
 Requires-Dist: bs4 (>=0.0.1,<0.0.2)
 Requires-Dist: cerebras-cloud-sdk (>=1.1.0,<2.0.0)
@@ -82,6 +85,7 @@ Requires-Dist: pymysql (>=1.1.0,<2.0.0) ; extra == "db" or extra == "all" or ext
 Requires-Dist: pyparsing (>=3.0.9,<4.0.0)
 Requires-Dist: pypdf (>=3.12.2,<4.0.0) ; extra == "doc-chat" or extra == "all" or extra == "pdf-parsers"
 Requires-Dist: pytesseract (>=0.3.10,<0.4.0) ; extra == "doc-chat" or extra == "all" or extra == "pdf-parsers"
+Requires-Dist: python-arango (>=8.1.2,<9.0.0) ; extra == "all" or extra == "arango"
 Requires-Dist: python-docx (>=1.1.0,<2.0.0) ; extra == "doc-chat" or extra == "all" or extra == "docx"
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
 Requires-Dist: python-magic (>=0.4.27,<0.5.0)

{langroid-0.19.5.dist-info → langroid-0.20.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 langroid/__init__.py,sha256=z_fCOLQJPOw3LLRPBlFB5-2HyCjpPgQa4m4iY5Fvb8Y,1800
 langroid/agent/__init__.py,sha256=ll0Cubd2DZ-fsCMl7e10hf9ZjFGKzphfBco396IKITY,786
-langroid/agent/base.py,sha256=swttuRRJKdDt-GUFsBNKHn1MQRj29b2f2U7Tuw5V5pk,64793
+langroid/agent/base.py,sha256=sOZapdzHaB4kbCLu8vI_zZx78jIhv9fmWn0EWV4yTAE,65371
 langroid/agent/batch.py,sha256=QZdlt1563hx4l3AXrCaGovE-PNG93M3DsvQAbDzdiS8,13705
 langroid/agent/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/callbacks/chainlit.py,sha256=JJXI3UGTyTDg2FFath4rqY1GyUo_0pbVBt8CZpvdtn4,23289
@@ -10,6 +10,11 @@ langroid/agent/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/junk,sha256=LxfuuW7Cijsg0szAzT81OjWWv1PMNI-6w_-DspVIO2s,339
 langroid/agent/openai_assistant.py,sha256=2rjCZw45ysNBEGNzQM4uf0bTC4KkatGYAWcVcW4xcek,34337
 langroid/agent/special/__init__.py,sha256=gik_Xtm_zV7U9s30Mn8UX3Gyuy4jTjQe9zjiE3HWmEo,1273
+langroid/agent/special/arangodb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langroid/agent/special/arangodb/arangodb_agent.py,sha256=y9zp2ZQcKkNsUegvDZryn79exth3T-hsmkR-z4gSU-w,25522
+langroid/agent/special/arangodb/system_messages.py,sha256=Uni6uDOSZpD9miCNzx7CEQjnyxDF8Egbe-hFvkraSt4,6739
+langroid/agent/special/arangodb/tools.py,sha256=WasFERC1cToLOWi1cWqUs-TujU0A68gZWbhbP128obo,3499
+langroid/agent/special/arangodb/utils.py,sha256=LIevtkayIdVVXyj3jlbKH2WgdZTtH5-JLgbXOHC7uxs,1420
 langroid/agent/special/doc_chat_agent.py,sha256=xIqBOyLax_jMU0UevxqXf_aQUrRkW6MQUKpKnKvaqkQ,59281
 langroid/agent/special/lance_doc_chat_agent.py,sha256=s8xoRs0gGaFtDYFUSIRchsgDVbS5Q3C2b2mr3V1Fd-Q,10419
 langroid/agent/special/lance_rag/__init__.py,sha256=QTbs0IVE2ZgDg8JJy1zN97rUUg4uEPH7SLGctFNumk4,174
@@ -19,13 +24,13 @@ langroid/agent/special/lance_rag/query_planner_agent.py,sha256=5YPeliCjlRk1LEDe5
 langroid/agent/special/lance_tools.py,sha256=qS8x4wi8mrqfbYV2ztFzrcxyhHQ0ZWOc-zkYiH7awj0,2105
 langroid/agent/special/neo4j/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/special/neo4j/csv_kg_chat.py,sha256=dRsAgMBa1H_EMI2YYgJR2Xyv1D7e4o3G9M64mTewq_c,6409
-langroid/agent/special/neo4j/neo4j_chat_agent.py,sha256=2r9kZ1AH3AxvJ6BZn0BZLhz-Bw8b7xUBlRv86gSEE8s,13542
-langroid/agent/special/neo4j/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langroid/agent/special/neo4j/utils/system_message.py,sha256=_2NKX2Sx_7nLGNk_0rUyDkgTHZJuKm_DgtG7M_BFPkY,2920
+langroid/agent/special/neo4j/neo4j_chat_agent.py,sha256=1RMKupJra0KZ-hA7AiiR662STJyYDZi8ZnAnnXF8oCA,16726
+langroid/agent/special/neo4j/system_messages.py,sha256=m2jsVayey6E_88F5B_gW2WbWKBJvIeDUoVCRBbNs97o,4522
+langroid/agent/special/neo4j/tools.py,sha256=Vw3HvtDfG2c4_bUHgt4_ZbJq48lpIQstbjjwhh1BjrQ,905
 langroid/agent/special/relevance_extractor_agent.py,sha256=zIx8GUdVo1aGW6ASla0NPQjYYIpmriK_TYMijqAx3F8,4796
 langroid/agent/special/retriever_agent.py,sha256=lvMvf-u9rSosg4YASuFdUbGLgkzLPknXAbJZfZ1LZCc,1868
 langroid/agent/special/sql/__init__.py,sha256=mWfmm1QpXCezpFOS2eI57M0L_Ok3q5_ukG8tXBnBrEA,319
-langroid/agent/special/sql/sql_chat_agent.py,sha256=PArNz9up0atw0cJet1LOfWlpW6xgofVBDR7Z3xvpzHk,16592
+langroid/agent/special/sql/sql_chat_agent.py,sha256=LVTuOlaY6m5DpsMFLFv00eOz04y6vKeO4mfGLIUp6AY,16844
 langroid/agent/special/sql/utils/__init__.py,sha256=JFif6CRTrN-bc91uuAI4K9fe2ndIWSNMVxJ0WA68--M,446
 langroid/agent/special/sql/utils/description_extractors.py,sha256=cX8TIpmTPXZXQTMpIi3OUFwFsPywxFFdurpx717Kq0I,6529
 langroid/agent/special/sql/utils/populate_metadata.py,sha256=1J22UsyEPKzwK0XlJZtYn9r6kYc0FXIr8-lZrndYlhc,3131
@@ -33,14 +38,14 @@ langroid/agent/special/sql/utils/system_message.py,sha256=qKLHkvQWRQodTtPLPxr1GS
 langroid/agent/special/sql/utils/tools.py,sha256=vFYysk6Vi7HJjII8B4RitA3pt_z3gkSglDNdhNVMiFc,1332
 langroid/agent/special/table_chat_agent.py,sha256=d9v2wsblaRx7oMnKhLV7uO_ujvk9gh59pSGvBXyeyNc,9659
 langroid/agent/structured_message.py,sha256=y7pud1EgRNeTFZlJmBkLmwME3yQJ_IYik-Xds9kdZbY,282
-langroid/agent/task.py,sha256=DUitErf1VIhaPNirP3fdOYU2MZEGjO1FPfEniIzFoks,85356
+langroid/agent/task.py,sha256=f7clh6p6Md0G4YGHqbFeeT88U4XoP0i3eatekV21hHE,86643
 langroid/agent/tool_message.py,sha256=jkN7uq7YwUC_wBcSCNUYjrB_His2YCfQay_lqIa4Tww,10498
 langroid/agent/tools/__init__.py,sha256=IMgCte-_ZIvCkozGQmvMqxIw7_nKLKzD78ccJL1bnQU,804
 langroid/agent/tools/duckduckgo_search_tool.py,sha256=NhsCaGZkdv28nja7yveAhSK_w6l_Ftym8agbrdzqgfo,1935
 langroid/agent/tools/file_tools.py,sha256=GjPB5YDILucYapElnvvoYpGJuZQ25ecLs2REv7edPEo,7292
 langroid/agent/tools/google_search_tool.py,sha256=y7b-3FtgXf0lfF4AYxrZ3K5pH2dhidvibUOAGBE--WI,1456
 langroid/agent/tools/metaphor_search_tool.py,sha256=qj4gt453cLEX3EGW7nVzVu6X7LCdrwjSlcNY0qJW104,2489
-langroid/agent/tools/orchestration.py,sha256=vp2Qx-DYPtDnACosxKqwHGy6DeD1QnEllWz0Ht81Cyc,10880
+langroid/agent/tools/orchestration.py,sha256=EDv1EMVGYqX82x3bCRbTn9gFNs66oosiUM8WTSZkUJg,10909
 langroid/agent/tools/recipient_tool.py,sha256=dr0yTxgNEIoxUYxH6TtaExC4G_8WdJ0xGohIa4dFLhY,9808
 langroid/agent/tools/retrieval_tool.py,sha256=2q2pfoYbZNfbWQ0McxrtmfF0ekGglIgRl-6uF26pa-E,871
 langroid/agent/tools/rewind_tool.py,sha256=XAXL3BpNhCmBGYq_qi_sZfHJuIw7NY2jp4wnojJ7WRs,5606
@@ -86,7 +91,7 @@ langroid/parsing/document_parser.py,sha256=ZGxgG4ytnCIah4HWk3ZrK3EGAMXDjAzI9KaPE
 langroid/parsing/image_text.py,sha256=sbLIQ5nHe2UnYUksBaQsmZGaX-X0qgEpPd7CEzi_z5M,910
 langroid/parsing/para_sentence_split.py,sha256=AJBzZojP3zpB-_IMiiHismhqcvkrVBQ3ZINoQyx_bE4,2000
 langroid/parsing/parse_json.py,sha256=aADo38bAHQhC8on4aWZZzVzSDy-dK35vRLZsFI2ewh8,4756
-langroid/parsing/parser.py,sha256=AgtmlVUvrkSG1l7-YZPX8rlldgXjh_HqXAMqpXkBxUo,11746
+langroid/parsing/parser.py,sha256=bTG5TO2CEwGdLf9979j9_dFntKX5FloGF8vhts6ObU0,11978
 langroid/parsing/repo_loader.py,sha256=3GjvPJS6Vf5L6gV2zOU8s-Tf1oq_fZm-IB_RL_7CTsY,29373
 langroid/parsing/routing.py,sha256=-FcnlqldzL4ZoxuDwXjQPNHgBe9F9-F4R6q7b_z9CvI,1232
 langroid/parsing/search.py,sha256=0i_r0ESb5HEQfagA2g7_uMQyxYPADWVbdcN9ixZhS4E,8992
@@ -137,8 +142,8 @@ langroid/vector_store/meilisearch.py,sha256=6frB7GFWeWmeKzRfLZIvzRjllniZ1cYj3Hmh
 langroid/vector_store/momento.py,sha256=qR-zBF1RKVHQZPZQYW_7g-XpTwr46p8HJuYPCkfJbM4,10534
 langroid/vector_store/qdrant_cloud.py,sha256=3im4Mip0QXLkR6wiqVsjV1QvhSElfxdFSuDKddBDQ-4,188
 langroid/vector_store/qdrantdb.py,sha256=v88lqFkepADvlN6lByUj9I4NEKa9X9lWH16uTPPbYrE,17457
-pyproject.toml,sha256=tKD9OebCwUNxXivb-EoM3ZvOjGBmFx4m-Bl9bztRkyg,7251
-langroid-0.19.5.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.19.5.dist-info/METADATA,sha256=jStMm7ROZxiIDxUHVKXHhc7UNXJ6LkK5QRpr7pTiUAU,56513
-langroid-0.19.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-langroid-0.19.5.dist-info/RECORD,,
+pyproject.toml,sha256=m6gT-kRBlRmBQ50VGLcLDXOmo3MmaYPAHxbySfpGd1E,7488
+langroid-0.20.1.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.20.1.dist-info/METADATA,sha256=8qg7AF3qK12XRvtEqpR2P8eyidON7E4womWSiY3QW_E,56758
+langroid-0.20.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+langroid-0.20.1.dist-info/RECORD,,

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.19.5"
+version = "0.20.1"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
@@ -90,6 +90,9 @@ python-magic = "^0.4.27"
 json-repair = "^0.29.9"
 cerebras-cloud-sdk = "^1.1.0"
 gitpython = "^3.1.43"
+python-arango = {version="^8.1.2", optional=true}
+arango-datasets = {version="^1.2.2", optional=true}
+adb-cloud-connector = "^1.0.2"
 
 
 [tool.poetry.extras]
@@ -113,6 +116,7 @@ all = [
 #    "lancedb", "tantivy", "pyarrow",
     "chromadb",
     "metaphor-python", "neo4j",
+    "python-arango", "arango-datasets",
     "litellm",
     "chainlit", "python-socketio",
     "fastembed"
@@ -130,6 +134,7 @@ mysql = ["pymysql"]
 sql = ["sqlalchemy", "pymysql",  "psycopg2"]
 litellm = ["litellm"]
 neo4j = ["neo4j"]
+arango = ["python-arango", "arango-datasets"]
 metaphor = ["metaphor-python"]
 chainlit = ["chainlit", "python-socketio"]
 chromadb = ["chromadb"]
@@ -166,6 +171,7 @@ coverage = "^7.2.5"
 pytest-xdist = "^3.6.1"
 pytest-timeout = "^2.3.1"
 pytest-cov = "^5.0.0"
+docker = "^7.1.0"
 
 [tool.poetry.group.docs]
 optional = true