attp-client 0.0.4__tar.gz → 0.0.6__tar.gz

{attp_client-0.0.4 → attp_client-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: attp-client
-Version: 0.0.4
+Version: 0.0.6
 Summary: A python-sdk client for interacting with AgentHub's ATTP protocol (Agent Tool Transport Protocol)
 License: MIT
 Author: Ascender Team

{attp_client-0.0.4 → attp_client-0.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "attp-client"
-version = "0.0.4"
+version = "0.0.6"
 description = "A python-sdk client for interacting with AgentHub's ATTP protocol (Agent Tool Transport Protocol)"
 authors = [
     {name = "Ascender Team"}

{attp_client-0.0.4 → attp_client-0.0.6}/src/attp_client/catalog.py

@@ -96,7 +96,7 @@ class AttpCatalog:
         self.tool_name_to_id_symlink[name] = str(assigned_id)
         return assigned_id
     
-    async def detatch_tool(
+    async def detach_tool(
         self,
         name: str
     ):
@@ -108,7 +108,7 @@ class AttpCatalog:
         await self.tool_manager.unregister(self.catalog_name, tool_id)
         return tool_id
     
-    async def detatch_all_tools(self):
+    async def detach_all_tools(self):
         for tool_id in list(self.attached_tools.keys()):
             await self.tool_manager.unregister(self.catalog_name, tool_id)
             del self.attached_tools[tool_id]

{attp_client-0.0.4 → attp_client-0.0.6}/src/attp_client/inference.py

@@ -67,11 +67,13 @@ class AttpInferenceAPI:
             raise ValueError("Cannot find agent by two identification specifiers, use only one!")
         
         response = await self.router.send("messages:inference:invoke", Serializable[dict[str, Any]]({
-            "agent_id": agent_id,
-            "agent_name": agent_name,
-            "input_configuration": input_configuration,
-            "messages": [message.model_dump(mode="json") for message in (messages or [])],
-            "stream": stream
+            "data": {
+                "agent_id": agent_id,
+                "agent_name": agent_name,
+                "input_configuration": input_configuration,
+                "messages": [message.model_dump(mode="json") for message in (messages or [])],
+                "stream": stream
+            }    
         }), timeout=timeout, expected_response=IMessageResponse)
         
         return response
@@ -107,9 +109,14 @@ class AttpInferenceAPI:
         
         response = await self.router.send(
             "messages:chat:invoke", 
-            Serializable[dict[str, Any]]({"chat_id": str(chat_id), "stream": stream}),
+            Serializable[dict[str, Any]]({
+                "data": {
+                    "chat_id": str(chat_id),
+                    "stream": stream
+                }
+            }),
             timeout=timeout,
-            # expected_response=IMessageResponse
+            expected_response=IMessageResponse
         )
         
         return response
@@ -125,6 +132,6 @@ class AttpInferenceAPI:
         """
         if isinstance(message, Sequence):
             for msg in message:
-                await self.router.emit("messages:append", msg)
+                await self.router.emit("messages:append", Serializable[dict[str, Any]]({"data": msg}))
         else:
-            await self.router.emit("messages:append", message)
+            await self.router.emit("messages:append", Serializable[dict[str, Any]]({"data": message}))

{attp_client-0.0.4 → attp_client-0.0.6}/src/attp_client/tools.py

@@ -25,16 +25,18 @@ class ToolsManager:
         response = await self.router.send(
             "tools:register",
             Serializable[dict[str, Any]]({
-                "catalog": catalog_name,
-                "tool": {
-                    "name": name,
-                    "description": description,
-                    "schema_id": schema_id,
-                    "return_direct": return_direct,
-                    "schema_ver": schema_ver,
-                    "schema": schema,
-                    "timeout_ms": timeout_ms,
-                    "idempotent": idempotent
+                "data": {
+                    "catalog": catalog_name,
+                    "tool": {
+                        "name": name,
+                        "description": description,
+                        "schema_id": schema_id,
+                        "return_direct": return_direct,
+                        "schema_ver": schema_ver,
+                        "schema": schema,
+                        "timeout_ms": timeout_ms,
+                        "idempotent": idempotent
+                    }
                 }
             }),
             timeout=30,
@@ -51,8 +53,10 @@ class ToolsManager:
         response = await self.router.send(
             "tool:unregister",
             Serializable[dict[str, Any]]({
-                "catalog": catalog_name,
-                "tool_id": tool_id
+                "data": {
+                    "catalog": catalog_name,
+                    "tool_id": tool_id
+                }
             }),
             timeout=30,
             expected_response=dict[str, Any]