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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: attp-client
-Version: 0.0.5
+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.5 → attp_client-0.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "attp-client"
-version = "0.0.5"
+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.5 → 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.5 → 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]