langroid 0.1.256__py3-none-any.whl → 0.1.257__py3-none-any.whl

langroid/agent/base.py

@@ -284,7 +284,7 @@ class Agent(ABC):
         enabled_classes: List[Type[ToolMessage]] = list(self.llm_tools_map.values())
         # use at most 2 sample conversations, no need to be exhaustive;
         sample_convo = [
-            msg_cls().usage_example()  # type: ignore
+            msg_cls().usage_examples(random=True)  # type: ignore
             for i, msg_cls in enumerate(enabled_classes)
             if i < 2
         ]

langroid/agent/chat_agent.py

@@ -270,9 +270,9 @@ class ChatAgent(Agent):
             ):
                 # example will be shown in json_format_rules() when using TOOLs,
                 # so we don't need to show it here.
-                example = "" if self.config.use_tools else (msg_cls.usage_example())
+                example = "" if self.config.use_tools else (msg_cls.usage_examples())
                 if example != "":
-                    example = "EXAMPLE: " + example
+                    example = "EXAMPLES:\n" + example
                 class_instructions = msg_cls.instructions()
                 guidance = (
                     ""

langroid/agent/task.py

@@ -525,9 +525,11 @@ class Task:
             ):
                 raise InfiniteLoopException(
                     """Possible infinite loop detected!
-                    You can adjust infinite loop detection by changing the params
-                    in the TaskConfig passed to the Task constructor: 
-                    e.g. set inf_loop_cycle_len=0 to disable loop detection."""
+                    You can adjust infinite loop detection (or turn it off)
+                    by changing the params in the TaskConfig passed to the Task 
+                    constructor; see here:
+                    https://langroid.github.io/langroid/reference/agent/task/#langroid.agent.task.TaskConfig
+                    """
                 )
 
         final_result = self.result()
@@ -620,9 +622,11 @@ class Task:
             ):
                 raise InfiniteLoopException(
                     """Possible infinite loop detected!
-                    You can adjust infinite loop detection by changing the params
-                    in the TaskConfig passed to the Task constructor: 
-                    e.g. set inf_loop_cycle_len=0 to disable loop detection."""
+                    You can adjust infinite loop detection (or turn it off)
+                    by changing the params in the TaskConfig passed to the Task 
+                    constructor; see here:
+                    https://langroid.github.io/langroid/reference/agent/task/#langroid.agent.task.TaskConfig
+                    """
                 )
 
         final_result = self.result()

langroid/agent/tool_message.py

@@ -80,21 +80,34 @@ class ToolMessage(ABC, BaseModel):
         return []
 
     @classmethod
-    def usage_example(cls) -> str:
+    def usage_examples(cls, random: bool = False) -> str:
         """
-        Instruction to the LLM showing an example of how to use the message.
+        Instruction to the LLM showing examples of how to use the tool-message.
+
+        Args:
+            random (bool): whether to pick a random example from the list of examples.
+                Set to `true` when using this to illustrate a dialog between LLM and
+                user.
+                (if false, use ALL examples)
         Returns:
-            str: example of how to use the message
+            str: examples of how to use the tool/function-call
         """
         # pick a random example of the fields
         if len(cls.examples()) == 0:
             return ""
-        ex = choice(cls.examples())
-        if isinstance(ex, tuple):
-            # (description, example_instance)
-            return f"{ex[0]} => {ex[1].json_example()}"
+        if random:
+            examples = [choice(cls.examples())]
         else:
-            return ex.json_example()
+            examples = cls.examples()
+        examples_jsons = [
+            (
+                f"EXAMPLE {i}: (THOUGHT: {ex[0]}) => \n{ex[1].json_example()}"
+                if isinstance(ex, tuple)
+                else f"EXAMPLE {i}:\n {ex.json_example()}"
+            )
+            for i, ex in enumerate(examples, 1)
+        ]
+        return "\n\n".join(examples_jsons)
 
     def to_json(self) -> str:
         return self.json(indent=4, exclude={"result", "purpose"})
@@ -139,6 +152,9 @@ class ToolMessage(ABC, BaseModel):
             # cls.simple_schema() if tool else
             cls.llm_function_schema(request=True).parameters
         )
+        examples_str = ""
+        if cls.examples():
+            examples_str = "EXAMPLES:\n" + cls.usage_examples()
         return textwrap.dedent(
             f"""
             TOOL: {cls.default_value("request")}
@@ -146,7 +162,7 @@ class ToolMessage(ABC, BaseModel):
             JSON FORMAT: {
                 json.dumps(param_dict, indent=4)
             }
-            {"EXAMPLE: " + cls.usage_example() if cls.examples() else ""}
+            {examples_str}
             """.lstrip()
         )
 

langroid/agent/tools/segment_extract_tool.py

@@ -26,7 +26,12 @@ class SegmentExtractTool(ToolMessage):
 
     @classmethod
     def examples(cls) -> List["ToolMessage" | Tuple[str, "ToolMessage"]]:
-        return [cls(segment_list="1,3,5-7")]
+        return [
+            (
+                "I want to extract segments 1, 3, and 5 thru 7",
+                cls(segment_list="1,3,5-7"),
+            )
+        ]
 
     @classmethod
     def instructions(cls) -> str:

{langroid-0.1.256.dist-info → langroid-0.1.257.dist-info}/METADATA

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

{langroid-0.1.256.dist-info → langroid-0.1.257.dist-info}/RECORD

@@ -1,10 +1,10 @@
 langroid/__init__.py,sha256=z_fCOLQJPOw3LLRPBlFB5-2HyCjpPgQa4m4iY5Fvb8Y,1800
 langroid/agent/__init__.py,sha256=ll0Cubd2DZ-fsCMl7e10hf9ZjFGKzphfBco396IKITY,786
-langroid/agent/base.py,sha256=6EznAMa4zeRdouo3U3_UzOI_cblvdpNH4v5CAM-fgbA,37171
+langroid/agent/base.py,sha256=aSwWmOBg0d3QQHUSauscMNfnl8Wkv6nrk2nngKa9DjM,37183
 langroid/agent/batch.py,sha256=feRA_yRG768ElOQjrKEefcRv6Aefd_yY7qktuYUQDwc,10040
 langroid/agent/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/callbacks/chainlit.py,sha256=XWih3UkHsYgPIaBQ1cZ-8H2VU_URh-sSfw8fcq1hfMo,20841
-langroid/agent/chat_agent.py,sha256=_xsBfGBkwcHd8IRErsW7tKNz7qn0h2oKSg_BFleOPCs,39475
+langroid/agent/chat_agent.py,sha256=hnmeOxdi4i5w8WaL2kPjQOEpenoRW_hG5EfeMWuuVsQ,39478
 langroid/agent/chat_document.py,sha256=uwCq53SHRyxQw6qyhjzPYuJG48VHBgOf2122Ew3fk6c,9316
 langroid/agent/helpers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/junk,sha256=LxfuuW7Cijsg0szAzT81OjWWv1PMNI-6w_-DspVIO2s,339
@@ -32,8 +32,8 @@ langroid/agent/special/sql/utils/populate_metadata.py,sha256=1J22UsyEPKzwK0XlJZt
 langroid/agent/special/sql/utils/system_message.py,sha256=qKLHkvQWRQodTtPLPxr1GSLUYUFASZU8x-ybV67cB68,1885
 langroid/agent/special/sql/utils/tools.py,sha256=6uB2424SLtmapui9ggcEr0ZTiB6_dL1-JRGgN8RK9Js,1332
 langroid/agent/special/table_chat_agent.py,sha256=d9v2wsblaRx7oMnKhLV7uO_ujvk9gh59pSGvBXyeyNc,9659
-langroid/agent/task.py,sha256=J4GIvbPMlL9uXKr7J-bSPHGvFK5tI2R0INgbnbFRmqk,59265
-langroid/agent/tool_message.py,sha256=okhIK0M2ZIGTlQF3h6UGkYk6ibH17mzm0ypDeczlWc0,9137
+langroid/agent/task.py,sha256=8f65p79ID17mCT8l2F7VuldL7uEWOavLTOkX0B29sbo,59465
+langroid/agent/tool_message.py,sha256=7t-UGEbykosKHAvaLI0Rm59sgxvN31IO3-P7bg7gLug,9730
 langroid/agent/tools/__init__.py,sha256=8Pc9BlGCB5FQ2IDGKS_WPpHCoWp5jblMU8EHJwwikAY,303
 langroid/agent/tools/duckduckgo_search_tool.py,sha256=NhsCaGZkdv28nja7yveAhSK_w6l_Ftym8agbrdzqgfo,1935
 langroid/agent/tools/extract_tool.py,sha256=u5lL9rKBzaLBOrRyLnTAZ97pQ1uxyLP39XsWMnpaZpw,3789
@@ -43,7 +43,7 @@ langroid/agent/tools/metaphor_search_tool.py,sha256=qj4gt453cLEX3EGW7nVzVu6X7LCd
 langroid/agent/tools/recipient_tool.py,sha256=NrLxIeQT-kbMv7AeYX0uqvGeMK4Q3fIDvG15OVzlgk8,9624
 langroid/agent/tools/retrieval_tool.py,sha256=2q2pfoYbZNfbWQ0McxrtmfF0ekGglIgRl-6uF26pa-E,871
 langroid/agent/tools/run_python_code.py,sha256=BvoxYzzHijU-p4703n2iVlt5BCieR1oMSy50w0tQZAg,1787
-langroid/agent/tools/segment_extract_tool.py,sha256=WOwZdTTOqKaJUDIqI0jWDV126VM1UjJzIUandHsnC-g,1320
+langroid/agent/tools/segment_extract_tool.py,sha256=__srZ_VGYLVOdPrITUM8S0HpmX4q7r5FHWMDdHdEv8w,1440
 langroid/agent_config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/cachedb/__init__.py,sha256=icAT2s7Vhf-ZGUeqpDQGNU6ob6o0aFEyjwcxxUGRFjg,225
 langroid/cachedb/base.py,sha256=LKiJyOFQUN1NRzPIynfbYKGFfSanA6auDfBNEedBK7Y,1342
@@ -126,7 +126,7 @@ langroid/vector_store/meilisearch.py,sha256=6frB7GFWeWmeKzRfLZIvzRjllniZ1cYj3Hmh
 langroid/vector_store/momento.py,sha256=QaPzUnTwlswoawGB-paLtUPyLRvckFXLfLDfvbTzjNQ,10505
 langroid/vector_store/qdrant_cloud.py,sha256=3im4Mip0QXLkR6wiqVsjV1QvhSElfxdFSuDKddBDQ-4,188
 langroid/vector_store/qdrantdb.py,sha256=sk5Qb2ZNbooi0rorsMuqIMokF7WADw6PJ0D6goM2XBw,16802
-langroid-0.1.256.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.1.256.dist-info/METADATA,sha256=W7Daa9ruwJoQrUJqJc4g0Ios64ifd0w1LFYGHUOhuKo,51001
-langroid-0.1.256.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langroid-0.1.256.dist-info/RECORD,,
+langroid-0.1.257.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.1.257.dist-info/METADATA,sha256=xLoU_XiewcfxXk98d6dG8S8dkniC4r5hVrEFgS9_h8s,51001
+langroid-0.1.257.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langroid-0.1.257.dist-info/RECORD,,