fix ut

Author: ponytailer

pydantic_client/base.py

@@ -93,18 +93,19 @@ def _request(self, request_info: RequestInfo) -> Any:
     def register_agno_tools(self, agent):
         """
         Register all agno tools of this client instance to the given agent.
-        Each tool will be registered with its description, parameters, and a call bound to this instance.
-
-        :param agent: The agent instance which supports .register_tool(name, description, parameters, call)
+        Compatible with agno agent's set_tools/add_tool API.
         """
+        tools = []
         for tool_info in self.get_agno_tools():
             tool_name = tool_info['name']
-            agent.register_tool(
+            tool = dict(
                 name=tool_name,
                 description=tool_info.get('description', ''),
                 parameters=tool_info.get('parameters', {}),
                 call=lambda params, tool_name=tool_name: getattr(self, tool_name)(**params)
             )
+            tools.append(tool)
+        agent.set_tools(tools)
 
     @classmethod
     def get_agno_tools(cls) -> List[Dict[str, Any]]:

tests/test_base_client_cov.py

@@ -7,8 +7,8 @@ class DummyAgent:
     def __init__(self):
         self.tools = []
 
-    def register_tool(self, name, description, parameters, call):
-        self.tools.append((name, description, parameters, call))
+    def set_tools(self, tools):
+        self.tools = tools
 
 
 class TestClient(BaseWebClient):
@@ -59,6 +59,6 @@ def test_register_agno_tools_and_get_agno_tools():
     client = TestClient(base_url="http://a")
     agent = DummyAgent()
     client.register_agno_tools(agent)
-    assert any(t[0] == 'tool_method' for t in agent.tools)
+    assert any(t["name"] == 'tool_method' for t in agent.tools)
     tools = TestClient.get_agno_tools()
     assert any(t['name'] == 'tool_method' for t in tools)