langroid 0.1.254__tar.gz → 0.1.257__tar.gz

{langroid-0.1.254 → langroid-0.1.257}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.1.254
+Version: 0.1.257
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
@@ -88,6 +88,7 @@ Requires-Dist: python-docx (>=1.1.0,<2.0.0) ; extra == "doc-chat" or extra == "d
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
 Requires-Dist: python-magic (>=0.4.27,<0.5.0)
 Requires-Dist: python-socketio (>=5.11.0,<6.0.0) ; extra == "chainlit"
+Requires-Dist: pyyaml (>=6.0.1,<7.0.0)
 Requires-Dist: qdrant-client (>=1.8.0,<2.0.0)
 Requires-Dist: rank-bm25 (>=0.2.2,<0.3.0)
 Requires-Dist: redis (>=5.0.1,<6.0.0)

{langroid-0.1.254 → langroid-0.1.257}/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-0.1.254 → langroid-0.1.257}/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-0.1.254 → langroid-0.1.257}/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-0.1.254 → langroid-0.1.257}/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-0.1.254 → langroid-0.1.257}/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.254 → langroid-0.1.257}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.1.254"
+version = "0.1.257"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
@@ -49,6 +49,7 @@ python-docx = {version="^1.1.0", optional=true}
 scrapy = {version="^2.11.0", optional=true}
 
 # ====CORE================================
+pyyaml = "^6.0.1"
 onnxruntime = "^1.16.1"
 fire = "^0.5.0"
 bs4 = "^0.0.1"
@@ -97,6 +98,7 @@ python-magic = "^0.4.27"
 
 
 # ================= EXTRAS =================
+
 [tool.poetry.extras]
 # install these using, e.g.,
 # `poetry install -E [...]` where [...] is one of the extras below