langroid 0.22.2__py3-none-any.whl → 0.22.3__py3-none-any.whl

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.2.dist-info → langroid-0.22.3.dist-info}/METADATA

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

{langroid-0.22.2.dist-info → langroid-0.22.3.dist-info}/RECORD

@@ -4,7 +4,7 @@ langroid/agent/base.py,sha256=IatVtrdstxMcreeVSndNoChFKxrJIEkYRpNruBxRQ7M,67112
 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
-langroid/agent/chat_agent.py,sha256=2aUDqM6jsdeU8xt-P86JU2TuZhTpPkUVx9mkXFMy8Rs,51646
+langroid/agent/chat_agent.py,sha256=DU_wtNFAxSJJFRkxpWNNqvI1P5QHkdi529As-22-Aog,51955
 langroid/agent/chat_document.py,sha256=FZ_PkeKU5OVp1IUlMvspfqxIXzlyd7J_F32DSYrxQ7E,17651
 langroid/agent/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/junk,sha256=LxfuuW7Cijsg0szAzT81OjWWv1PMNI-6w_-DspVIO2s,339
@@ -142,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=SqUjrPzZZgrSr-AojnyP3Du81tNyxCEsP_N1m-smqiA,7488
-langroid-0.22.2.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.22.2.dist-info/METADATA,sha256=xoC9bs2-ntnP7OWrkvrGRmFWt2opCRmNCaeCy2hF7Tc,57107
-langroid-0.22.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-langroid-0.22.2.dist-info/RECORD,,
+pyproject.toml,sha256=blIMIWJ0_N8r8YBAJV1BoWriHqWE3erxmjQb-2npkbU,7488
+langroid-0.22.3.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.22.3.dist-info/METADATA,sha256=_1rwLApadQ9D1HTsVSTq03sPgyiALup286Snmx2Ei7I,57107
+langroid-0.22.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+langroid-0.22.3.dist-info/RECORD,,

pyproject.toml

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