langroid 0.22.1__tar.gz → 0.22.3__tar.gz

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

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

{langroid-0.22.1 → langroid-0.22.3}/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.3}/langroid/agent/chat_agent.py

@@ -623,10 +623,27 @@ class ChatAgent(Agent):
             self.llm_tools_usable.discard(r)
             self.llm_functions_usable.discard(r)
 
+    def truncate_message(
+        self,
+        idx: int,
+        tokens: int = 5,
+        warning: str = "...[Contents truncated!]",
+    ) -> LLMMessage:
+        """Truncate message at idx in msg history to `tokens` tokens"""
+        llm_msg = self.message_history[idx]
+        orig_content = llm_msg.content
+        new_content = (
+            self.parser.truncate_tokens(orig_content, tokens)
+            if self.parser is not None
+            else orig_content[: tokens * 4]  # approx truncation
+        )
+        llm_msg.content = new_content + "\n" + warning
+        return llm_msg
+
     def _reduce_raw_tool_results(self, message: ChatDocument) -> None:
         """
-        If message is the result of a ToolMessage that had the
-        flag `_retain_raw_results = False`, then we replace contents
+        If message is the result of a ToolMessage that had
+        a `_max_retained_tokens` set to a non-None value, then we replace contents
         with a placeholder message.
         """
         parent_message: ChatDocument | None = message.parent
@@ -641,14 +658,9 @@ class ChatAgent(Agent):
                 and has been truncated to {max_tokens} tokens.
                 To obtain the full result, the tool needs to be re-used.
             """
-            llm_msg = self.message_history[message.metadata.msg_idx]
-            orig_content = llm_msg.content
-            new_content = (
-                self.parser.truncate_tokens(orig_content, max_tokens)
-                if self.parser is not None
-                else orig_content[: max_tokens * 4]  # approx truncation
+            self.truncate_message(
+                message.metadata.msg_idx, max_tokens, truncation_warning
             )
-            llm_msg.content = new_content + "\n\n" + truncation_warning
 
     def llm_response(
         self, message: Optional[str | ChatDocument] = None

{langroid-0.22.1 → langroid-0.22.3}/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.3}/pyproject.toml

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