langroid 0.22.1__tar.gz → 0.22.2__tar.gz

{langroid-0.22.1 → langroid-0.22.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.22.1
+Version: 0.22.2
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani

{langroid-0.22.1 → langroid-0.22.2}/langroid/agent/base.py

@@ -703,7 +703,7 @@ class Agent(ABC):
         if self.llm is None:
             return False
 
-        if message is not None and len(self.get_tool_messages(message)) > 0:
+        if message is not None and len(self.try_get_tool_messages(message)) > 0:
             # if there is a valid "tool" message (either JSON or via `function_call`)
             # then LLM cannot respond to it
             return False
@@ -717,7 +717,7 @@ class Agent(ABC):
         Args:
             message (str|ChatDocument): message or ChatDocument object to respond to.
         """
-        tools = self.get_tool_messages(message)
+        tools = self.try_get_tool_messages(message)
         if len(tools) == 0 and self.config.respond_tools_only:
             return False
         if message is not None and self.has_only_unhandled_tools(message):
@@ -915,11 +915,21 @@ class Agent(ABC):
         """
         if msg is None:
             return False
-        tools = self.get_tool_messages(msg, all_tools=True)
+        tools = self.try_get_tool_messages(msg, all_tools=True)
         if len(tools) == 0:
             return False
         return all(not self._tool_recipient_match(t) for t in tools)
 
+    def try_get_tool_messages(
+        self,
+        msg: str | ChatDocument | None,
+        all_tools: bool = False,
+    ) -> List[ToolMessage]:
+        try:
+            return self.get_tool_messages(msg, all_tools)
+        except ValidationError:
+            return []
+
     def get_tool_messages(
         self,
         msg: str | ChatDocument | None,
@@ -1417,7 +1427,7 @@ class Agent(ABC):
         if content_any is not None and isinstance(content_any, output_type):
             return cast(T, content_any)
 
-        tools = self.get_tool_messages(msg, all_tools=True)
+        tools = self.try_get_tool_messages(msg, all_tools=True)
 
         if get_origin(output_type) is list:
             list_element_type = get_args(output_type)[0]

{langroid-0.22.1 → langroid-0.22.2}/langroid/agent/task.py

@@ -1314,7 +1314,7 @@ class Task:
             actual_turns = e.turns if e.turns > 0 else turns
             e.agent.callbacks.set_parent_agent(self.agent)
             # e.callbacks.set_parent_agent(self.agent)
-            pending_tools = self.agent.get_tool_messages(self.pending_message)
+            pending_tools = self.agent.try_get_tool_messages(self.pending_message)
             # TODO disable this
             if (
                 len(pending_tools) > 1
@@ -1332,7 +1332,7 @@ class Task:
                 )
                 # update result.tool_messages if any
                 if isinstance(result, ChatDocument):
-                    self.agent.get_tool_messages(result)
+                    self.agent.try_get_tool_messages(result)
                 if result is not None:
                     content, id2result, oai_tool_id = self.agent.process_tool_results(
                         result.content,
@@ -1363,7 +1363,7 @@ class Task:
             result = response_fn(self.pending_message)
             # update result.tool_messages if any
             if isinstance(result, ChatDocument):
-                self.agent.get_tool_messages(result)
+                self.agent.try_get_tool_messages(result)
 
         result_chat_doc = self.agent.to_ChatDocument(
             result,
@@ -1446,7 +1446,7 @@ class Task:
         if isinstance(e, Task):
             actual_turns = e.turns if e.turns > 0 else turns
             e.agent.callbacks.set_parent_agent(self.agent)
-            pending_tools = self.agent.get_tool_messages(self.pending_message)
+            pending_tools = self.agent.try_get_tool_messages(self.pending_message)
             # TODO disable this
             if (
                 len(pending_tools) > 1

{langroid-0.22.1 → langroid-0.22.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.22.1"
+version = "0.22.2"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"