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

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

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

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

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

@@ -67,13 +67,11 @@ 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]]({
-            "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
-            }    
+            "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
@@ -110,10 +108,8 @@ class AttpInferenceAPI:
         response = await self.router.send(
             "messages:chat:invoke", 
             Serializable[dict[str, Any]]({
-                "data": {
-                    "chat_id": str(chat_id),
-                    "stream": stream
-                }
+                "chat_id": str(chat_id),
+                "stream": stream
             }),
             timeout=timeout,
             expected_response=IMessageResponse
@@ -132,6 +128,6 @@ class AttpInferenceAPI:
         """
         if isinstance(message, Sequence):
             for msg in message:
-                await self.router.emit("messages:append", Serializable[dict[str, Any]]({"data": msg}))
+                await self.router.emit("messages:append", msg)
         else:
-            await self.router.emit("messages:append", Serializable[dict[str, Any]]({"data": message}))
+            await self.router.emit("messages:append", message)

{attp_client-0.0.6 → attp_client-0.0.8}/src/attp_client/session.py

@@ -303,7 +303,7 @@ class SessionDriver:
             
             else:
                 if self.is_authenticated:
-                    self.logger.debug("cyan]ATTP[/] ┆ Handing incoming message to a route handler.")
+                    self.logger.debug("[cyan]ATTP[/] ┆ Handing incoming message to a route handler.")
                     self.messages.put_nowait(event)
 
     async def start_listener(self):

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

@@ -20,23 +20,23 @@ class ToolsManager:
         return_direct: bool = False,
         schema_ver: str = "1.0",
         timeout_ms: float = 20000,
-        idempotent: bool = False
+        idempotent: bool = False,
+        configs: Any | None = None
     ) -> UUID:
         response = await self.router.send(
             "tools:register",
             Serializable[dict[str, Any]]({
-                "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
-                    }
+                "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,
+                    "config": configs or {}
                 }
             }),
             timeout=30,
@@ -53,10 +53,8 @@ class ToolsManager:
         response = await self.router.send(
             "tool:unregister",
             Serializable[dict[str, Any]]({
-                "data": {
-                    "catalog": catalog_name,
-                    "tool_id": tool_id
-                }
+                "catalog": catalog_name,
+                "tool_id": tool_id
             }),
             timeout=30,
             expected_response=dict[str, Any]