langroid 0.56.2__py3-none-any.whl → 0.56.4__py3-none-any.whl

langroid/agent/base.py

@@ -265,8 +265,10 @@ class Agent(ABC):
         """
         sig = inspect.signature(handler_method)
         params = list(sig.parameters.values())
-        # Remove 'self' parameter
-        params = [p for p in params if p.name != "self"]
+        # Remove the first 'self' parameter
+        params = params[1:]
+        # Don't use name
+        # [p for p in params if p.name != "self"]
 
         agent_param = None
         chat_doc_param = None
@@ -277,18 +279,30 @@ class Agent(ABC):
             if param.annotation != inspect.Parameter.empty:
                 ann_str = str(param.annotation)
                 # Check for Agent-like types
+                print(param, inspect.isclass(param.annotation))
+                print(param, issubclass(param.annotation, Agent))
+                print(param, param.annotation is ChatDocument)
                 if (
-                    param.annotation == self.__class__
-                    or "Agent" in ann_str
-                    or (
-                        hasattr(param.annotation, "__name__")
-                        and "Agent" in param.annotation.__name__
+                    inspect.isclass(param.annotation)
+                    and issubclass(param.annotation, Agent)
+                ) or (
+                    not inspect.isclass(param.annotation)
+                    and (
+                        "Agent" in ann_str
+                        or (
+                            hasattr(param.annotation, "__name__")
+                            and "Agent" in param.annotation.__name__
+                        )
                     )
                 ):
                     agent_param = param.name
                     has_annotations = True
                 # Check for ChatDocument-like types
-                elif "ChatDocument" in ann_str or "ChatDoc" in ann_str:
+                elif (
+                    param.annotation is ChatDocument
+                    or "ChatDocument" in ann_str
+                    or "ChatDoc" in ann_str
+                ):
                     chat_doc_param = param.name
                     has_annotations = True
 
@@ -298,12 +312,12 @@ class Agent(ABC):
             elif param.name == "chat_doc":
                 chat_doc_param = param.name
 
+        print(has_annotations, agent_param, chat_doc_param)
         return has_annotations, agent_param, chat_doc_param
 
     @no_type_check
     def _create_handler_wrapper(
         self,
-        message_class: Type[ToolMessage],
         handler_method: Any,
         is_async: bool = False,
     ) -> Any:
@@ -320,7 +334,8 @@ class Agent(ABC):
         """
         sig = inspect.signature(handler_method)
         params = list(sig.parameters.values())
-        params = [p for p in params if p.name != "self"]
+        params = params[1:]
+        # params = [p for p in params if p.name != "self"]
 
         has_annotations, agent_param, chat_doc_param = self._analyze_handler_params(
             handler_method,
@@ -471,7 +486,6 @@ class Agent(ABC):
             See `tests/main/test_stateless_tool_messages.py` for an example.
             """
             wrapper = self._create_handler_wrapper(
-                message_class,
                 message_class.handle,
                 is_async=False,
             )
@@ -521,7 +535,6 @@ class Agent(ABC):
             and not hasattr(self, async_handler_name)
         ):
             wrapper = self._create_handler_wrapper(
-                message_class,
                 message_class.handle_async,
                 is_async=True,
             )

langroid/agent/tools/task_tool.py

@@ -3,6 +3,7 @@ TaskTool: A tool that allows agents to delegate a task to a sub-agent with
     specific tools enabled.
 """
 
+import uuid
 from typing import List, Optional
 
 import langroid.language_models as lm
@@ -17,6 +18,10 @@ from langroid.pydantic_v1 import Field
 class TaskTool(ToolMessage):
     """
     Tool that spawns a sub-agent with specified tools to handle a task.
+
+    The sub-agent can be given a custom name for identification in logs.
+    If no name is provided, a random unique name starting with 'agent'
+    will be generated.
     """
 
     request: str = "task_tool"
@@ -24,11 +29,7 @@ class TaskTool(ToolMessage):
         <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`: 
-        
-
+        non-interactively.
     """
 
     # Parameters for the agent tool
@@ -87,6 +88,14 @@ class TaskTool(ToolMessage):
         default=None,
         description="Optional max iterations for the sub-agent to run the task",
     )
+    agent_name: Optional[str] = Field(
+        default=None,
+        description="""
+            Optional name for the sub-agent. This will be used as the agent's name
+            in logs and for identification purposes. If not provided, a random unique
+            name starting with 'agent' will be generated.
+            """,
+    )
 
     def _set_up_task(self, agent: ChatAgent) -> Task:
         """
@@ -95,6 +104,9 @@ class TaskTool(ToolMessage):
         Args:
             agent: The parent ChatAgent that is handling this tool
         """
+        # Generate a random name if not provided
+        agent_name = self.agent_name or f"agent-{str(uuid.uuid4())[:8]}"
+
         # 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?
@@ -102,6 +114,7 @@ class TaskTool(ToolMessage):
             chat_model=self.model or "gpt-4.1-mini",  # Default model if not specified
         )
         config = ChatAgentConfig(
+            name=agent_name,
             llm=llm_config,
             system_message=f"""
                 {self.system_message}

{langroid-0.56.2.dist-info → langroid-0.56.4.dist-info}/METADATA

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

{langroid-0.56.2.dist-info → langroid-0.56.4.dist-info}/RECORD

@@ -3,7 +3,7 @@ langroid/exceptions.py,sha256=OPjece_8cwg94DLPcOGA1ddzy5bGh65pxzcHMnssTz8,2995
 langroid/mytypes.py,sha256=HIcYAqGeA9OK0Hlscym2FI5Oax9QFljDZoVgRlomhRk,4014
 langroid/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/agent/__init__.py,sha256=ll0Cubd2DZ-fsCMl7e10hf9ZjFGKzphfBco396IKITY,786
-langroid/agent/base.py,sha256=a45iqoWet2W60h4wUIOyHDYlotgS0asqIjfbONT4fZQ,85706
+langroid/agent/base.py,sha256=EMZGPVSqd4_optr9_FG1TLRF8-noEAloen8J4c7Ayto,86262
 langroid/agent/batch.py,sha256=wpE9RqCNDVDhAXkCB7wEqfCIEAi6qKcrhaZ-Zr9T4C0,21375
 langroid/agent/chat_agent.py,sha256=2HIYzYxkrGkRIS97ioKfIqjaW3RbX89M39LjzBobBEY,88381
 langroid/agent/chat_document.py,sha256=0e6zYkqIorMIVbCsxOul9ziwAPPOWDsBsRV9E8ux-WI,18055
@@ -54,7 +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/task_tool.py,sha256=VOHWv8uFPRczf6qp8YZPpUirkCqE6YVhN6jgXCvdpe0,7102
 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
@@ -137,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.56.2.dist-info/METADATA,sha256=xViaD-RgD0cPpfoT2q50ORTwar8EgXJjRVWuifS7dKw,65744
-langroid-0.56.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.56.2.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.56.2.dist-info/RECORD,,
+langroid-0.56.4.dist-info/METADATA,sha256=punxDL35bHKw4J-PDHEMYfWvwb7gu3ccpuMlTPmU1GU,65744
+langroid-0.56.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.56.4.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.56.4.dist-info/RECORD,,