langroid 0.55.1__py3-none-any.whl → 0.56.1__py3-none-any.whl

langroid/agent/chat_document.py

@@ -268,7 +268,7 @@ class ChatDocument(Document):
     @staticmethod
     def _clean_fn_call(fc: LLMFunctionCall | None) -> None:
         # Sometimes an OpenAI LLM (esp gpt-4o) may generate a function-call
-        # with odditities:
+        # with oddities:
         # (a) the `name` is set, as well as `arguments.request` is set,
         #  and in langroid we use the `request` value as the `name`.
         #  In this case we override the `name` with the `request` value.

langroid/agent/done_sequence_parser.py

@@ -13,8 +13,6 @@ Examples:
 import re
 from typing import List, Union
 
-from langroid.pydantic_v1 import BaseModel
-
 from .task import AgentEvent, DoneSequence, EventType
 
 

langroid/agent/tools/task_tool.py

@@ -0,0 +1,174 @@
+"""
+TaskTool: A tool that allows agents to delegate a task to a sub-agent with
+    specific tools enabled.
+"""
+
+from typing import List, Optional
+
+import langroid.language_models as lm
+from langroid import ChatDocument
+from langroid.agent.chat_agent import ChatAgent, ChatAgentConfig
+from langroid.agent.task import Task
+from langroid.agent.tool_message import ToolMessage
+from langroid.agent.tools.orchestration import DoneTool
+from langroid.pydantic_v1 import Field
+
+
+class TaskTool(ToolMessage):
+    """
+    Tool that spawns a sub-agent with specified tools to handle a task.
+    """
+
+    request: str = "task_tool"
+    purpose: str = """
+        <HowToUse>
+        Use this tool to delegate a task to a sub-agent with specific tools enabled.
+        The sub-agent will be created with the specified tools and will run the task
+        non-interactively. Here is how to set the fields:
+        
+        - `system_message`: 
+        
+
+    """
+
+    # Parameters for the agent tool
+
+    system_message: Optional[str] = Field(
+        ...,
+        description="""
+        Optional system message to configure the sub-agent's general behavior.
+            A good system message will have these components:
+            - Inform the sub-agent of its role, e.g. "You are a financial analyst."
+            - Clear spec of the task
+            - Any additional general context needed for the task, such as a
+              (part of a) document, or data items, etc.
+            - Specify when to use certain tools, e.g. 
+                "You MUST use the 'stock_data' tool to extract stock information.
+        """,
+    )
+
+    prompt: str = Field(
+        ...,
+        description="""
+            The prompt to run the sub-agent with. This differs from the agent's
+            system message: Whereas the system message configures the sub-agent's
+            GENERAL role and goals, the `prompt` is the SPECIFIC input that the 
+            sub-agent will process. In LLM terms, the system message is sent to the 
+            LLM as the first message, with role = "system" or "developer", and 
+            the prompt is sent as a message with role = "user".
+            EXAMPLE: system_message = "You are a financial analyst, when the 
+                user asks about the share-price of a company, 
+                you must use your tools to do the research, and 
+                return the final answer to the user."
+            
+            prompt = "What is the share-price of Apple Inc.?"
+            """,
+    )
+
+    tools: List[str] = Field(
+        ...,
+        description="""
+        A list of tool names to enable for the sub-agent.
+        This must be a list of strings referring to the names of tools
+        that are known to you. 
+        If you want to enable all tools, you can set this field
+         to a singleton list containing 'ALL'
+        To disable all tools, set it to a singleton list containing 'NONE'
+        """,
+    )
+    model: str = Field(
+        default=None,
+        description="""
+            Optional name of the LLM model to use for the sub-agent, e.g. 'gpt-4.1'
+            If omitted, the sub-agent will use the same model as yours.
+            """,
+    )
+    max_iterations: Optional[int] = Field(
+        default=None,
+        description="Optional max iterations for the sub-agent to run the task",
+    )
+
+    def _set_up_task(self, agent: ChatAgent) -> Task:
+        """
+        Helper method to set up a task for the sub-agent.
+
+        Args:
+            agent: The parent ChatAgent that is handling this tool
+        """
+        # Create chat agent config with system message if provided
+        # TODO: Maybe we just copy the parent agent's config and override chat_model?
+        #   -- but what if parent agent has a MockLMConfig?
+        llm_config = lm.OpenAIGPTConfig(
+            chat_model=self.model or "gpt-4.1-mini",  # Default model if not specified
+        )
+        config = ChatAgentConfig(
+            llm=llm_config,
+            system_message=f"""
+                {self.system_message}
+                
+                When you are finished with your task, you MUST
+                use the TOOL `{DoneTool.name()}` to end the task
+                and return the result.                
+            """,
+        )
+
+        # Create the sub-agent
+        sub_agent = ChatAgent(config)
+
+        # Enable the specified tools for the sub-agent
+        # Convert tool names to actual tool classes using parent agent's tools_map
+        if self.tools == ["ALL"]:
+            # Enable all tools from the parent agent:
+            # This is the list of all tools KNOWN (whether usable or handle-able or not)
+            tool_classes = [
+                agent.llm_tools_map[t]
+                for t in agent.llm_tools_known
+                if t in agent.llm_tools_map and t != self.request
+                # Exclude the TaskTool itself!
+            ]
+        elif self.tools == ["NONE"]:
+            # No tools enabled
+            tool_classes = []
+        else:
+            # Enable only specified tools
+            tool_classes = [
+                agent.llm_tools_map[tool_name]
+                for tool_name in self.tools
+                if tool_name in agent.llm_tools_map
+            ]
+
+        # always enable the DoneTool to signal task completion
+        sub_agent.enable_message(tool_classes + [DoneTool], use=True, handle=True)
+
+        # Create a non-interactive task
+        task = Task(sub_agent, interactive=False)
+
+        return task
+
+    def handle(self, agent: ChatAgent) -> Optional[ChatDocument]:
+        """
+
+        Handle the TaskTool by creating a sub-agent with specified tools
+        and running the task non-interactively.
+
+        Args:
+            agent: The parent ChatAgent that is handling this tool
+        """
+
+        task = self._set_up_task(agent)
+        # Run the task on the prompt, and return the result
+        result = task.run(self.prompt, turns=self.max_iterations or 10)
+        return result
+
+    async def handle_async(self, agent: ChatAgent) -> Optional[ChatDocument]:
+        """
+        Async method to handle the TaskTool by creating a sub-agent with specified tools
+        and running the task non-interactively.
+
+        Args:
+            agent: The parent ChatAgent that is handling this tool
+        """
+        task = self._set_up_task(agent)
+        # Run the task on the prompt, and return the result
+        result = await task.run_async(self.prompt, turns=self.max_iterations or 10)
+        return result

langroid/mcp/__init__.py

@@ -0,0 +1 @@
+"""MCP (Model Context Protocol) integration for Langroid."""

langroid/mcp/server/__init__.py

@@ -0,0 +1 @@
+"""MCP server implementation for Langroid."""

{langroid-0.55.1.dist-info → langroid-0.56.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.55.1
+Version: 0.56.1
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT

{langroid-0.55.1.dist-info → langroid-0.56.1.dist-info}/RECORD

@@ -6,8 +6,8 @@ langroid/agent/__init__.py,sha256=ll0Cubd2DZ-fsCMl7e10hf9ZjFGKzphfBco396IKITY,78
 langroid/agent/base.py,sha256=a45iqoWet2W60h4wUIOyHDYlotgS0asqIjfbONT4fZQ,85706
 langroid/agent/batch.py,sha256=wpE9RqCNDVDhAXkCB7wEqfCIEAi6qKcrhaZ-Zr9T4C0,21375
 langroid/agent/chat_agent.py,sha256=2HIYzYxkrGkRIS97ioKfIqjaW3RbX89M39LjzBobBEY,88381
-langroid/agent/chat_document.py,sha256=6O20Fp4QrquykaF2jFtwNHkvcoDte1LLwVZNk9mVH9c,18057
-langroid/agent/done_sequence_parser.py,sha256=GiLTVtzzKb_YQcPpwqOajKu97RSUzZC6ae_ZagDmt_A,4425
+langroid/agent/chat_document.py,sha256=0e6zYkqIorMIVbCsxOul9ziwAPPOWDsBsRV9E8ux-WI,18055
+langroid/agent/done_sequence_parser.py,sha256=oUPzQCkkAo-5qos3ndSV47Lre7O_LoGWwTybjE9sCwc,4381
 langroid/agent/openai_assistant.py,sha256=JkAcs02bIrgPNVvUWVR06VCthc5-ulla2QMBzux_q6o,34340
 langroid/agent/task.py,sha256=PbHVcyFgyxiMZXHOB5xr5t8qaeACYfrjNF_lZQc8d8Y,101308
 langroid/agent/tool_message.py,sha256=BhjP-_TfQ2tgxuY4Yo_JHLOwwt0mJ4BwjPnREvEY4vk,14744
@@ -54,6 +54,7 @@ langroid/agent/tools/recipient_tool.py,sha256=dr0yTxgNEIoxUYxH6TtaExC4G_8WdJ0xGo
 langroid/agent/tools/retrieval_tool.py,sha256=zcAV20PP_6VzSd-UE-IJcabaBseFL_QNz59Bnig8-lE,946
 langroid/agent/tools/rewind_tool.py,sha256=XAXL3BpNhCmBGYq_qi_sZfHJuIw7NY2jp4wnojJ7WRs,5606
 langroid/agent/tools/segment_extract_tool.py,sha256=__srZ_VGYLVOdPrITUM8S0HpmX4q7r5FHWMDdHdEv8w,1440
+langroid/agent/tools/task_tool.py,sha256=O-3-tALocGuwfxXikBzRWFenFKZWIlOoRVvnZ5mf7io,6514
 langroid/agent/tools/tavily_search_tool.py,sha256=soI-j0HdgVQLf09wRQScaEK4b5RpAX9C4cwOivRFWWI,1903
 langroid/agent/tools/mcp/__init__.py,sha256=DJNM0VeFnFS3pJKCyFGggT8JVjVu0rBzrGzasT1HaSM,387
 langroid/agent/tools/mcp/decorators.py,sha256=h7dterhsmvWJ8q4mp_OopmuG2DF71ty8cZwOyzdDZuk,1127
@@ -83,6 +84,8 @@ langroid/language_models/prompt_formatter/__init__.py,sha256=2-5cdE24XoFDhifOLl8
 langroid/language_models/prompt_formatter/base.py,sha256=eDS1sgRNZVnoajwV_ZIha6cba5Dt8xjgzdRbPITwx3Q,1221
 langroid/language_models/prompt_formatter/hf_formatter.py,sha256=PVJppmjRvD-2DF-XNC6mE05vTZ9wbu37SmXwZBQhad0,5055
 langroid/language_models/prompt_formatter/llama2_formatter.py,sha256=YdcO88qyBeuMENVIVvVqSYuEpvYSTndUe_jd6hVTko4,2899
+langroid/mcp/__init__.py,sha256=HGtfHjcLVruCFJavOBAmVpk5u2h5UwzgqxOiOYIjXj0,61
+langroid/mcp/server/__init__.py,sha256=HIju6Y3npEAC6lx8svT3q_vrMN1vT8fGUTzChBMIM4U,46
 langroid/parsing/__init__.py,sha256=2oUWJJAxIavq9Wtw5RGlkXLq3GF3zgXeVLLW4j7yeb8,1138
 langroid/parsing/agent_chats.py,sha256=sbZRV9ujdM5QXvvuHVjIi2ysYSYlap-uqfMMUKulrW0,1068
 langroid/parsing/code_parser.py,sha256=5ze0MBytrGGkU69pA_bJDjRm6QZz_QYfPcIwkagUa7U,3796
@@ -134,7 +137,7 @@ langroid/vector_store/pineconedb.py,sha256=otxXZNaBKb9f_H75HTaU3lMHiaR2NUp5MqwLZ
 langroid/vector_store/postgres.py,sha256=wHPtIi2qM4fhO4pMQr95pz1ZCe7dTb2hxl4VYspGZoA,16104
 langroid/vector_store/qdrantdb.py,sha256=O6dSBoDZ0jzfeVBd7LLvsXu083xs2fxXtPa9gGX3JX4,18443
 langroid/vector_store/weaviatedb.py,sha256=Yn8pg139gOy3zkaPfoTbMXEEBCiLiYa1MU5d_3UA1K4,11847
-langroid-0.55.1.dist-info/METADATA,sha256=iBKanly7KGZOhgDMib-Ll2ZqvLqNlpR2g2M2_KLBQRk,65366
-langroid-0.55.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.55.1.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.55.1.dist-info/RECORD,,
+langroid-0.56.1.dist-info/METADATA,sha256=Yz9pHxVllXS-Sihx0AqWVjuXom9AjDbc_2DsU0wCgks,65366
+langroid-0.56.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.56.1.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.56.1.dist-info/RECORD,,