attp-client 0.0.5__py3-none-any.whl → 0.0.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
attp_client/inference.py CHANGED
@@ -67,11 +67,13 @@ class AttpInferenceAPI:
67
67
  raise ValueError("Cannot find agent by two identification specifiers, use only one!")
68
68
 
69
69
  response = await self.router.send("messages:inference:invoke", Serializable[dict[str, Any]]({
70
- "agent_id": agent_id,
71
- "agent_name": agent_name,
72
- "input_configuration": input_configuration,
73
- "messages": [message.model_dump(mode="json") for message in (messages or [])],
74
- "stream": stream
70
+ "data": {
71
+ "agent_id": agent_id,
72
+ "agent_name": agent_name,
73
+ "input_configuration": input_configuration,
74
+ "messages": [message.model_dump(mode="json") for message in (messages or [])],
75
+ "stream": stream
76
+ }
75
77
  }), timeout=timeout, expected_response=IMessageResponse)
76
78
 
77
79
  return response
@@ -107,9 +109,14 @@ class AttpInferenceAPI:
107
109
 
108
110
  response = await self.router.send(
109
111
  "messages:chat:invoke",
110
- Serializable[dict[str, Any]]({"chat_id": str(chat_id), "stream": stream}),
112
+ Serializable[dict[str, Any]]({
113
+ "data": {
114
+ "chat_id": str(chat_id),
115
+ "stream": stream
116
+ }
117
+ }),
111
118
  timeout=timeout,
112
- # expected_response=IMessageResponse
119
+ expected_response=IMessageResponse
113
120
  )
114
121
 
115
122
  return response
@@ -125,6 +132,6 @@ class AttpInferenceAPI:
125
132
  """
126
133
  if isinstance(message, Sequence):
127
134
  for msg in message:
128
- await self.router.emit("messages:append", msg)
135
+ await self.router.emit("messages:append", Serializable[dict[str, Any]]({"data": msg}))
129
136
  else:
130
- await self.router.emit("messages:append", message)
137
+ await self.router.emit("messages:append", Serializable[dict[str, Any]]({"data": message}))
attp_client/session.py CHANGED
@@ -303,7 +303,7 @@ class SessionDriver:
303
303
 
304
304
  else:
305
305
  if self.is_authenticated:
306
- self.logger.debug("cyan]ATTP[/] ┆ Handing incoming message to a route handler.")
306
+ self.logger.debug("[cyan]ATTP[/] ┆ Handing incoming message to a route handler.")
307
307
  self.messages.put_nowait(event)
308
308
 
309
309
  async def start_listener(self):
attp_client/tools.py CHANGED
@@ -25,16 +25,18 @@ class ToolsManager:
25
25
  response = await self.router.send(
26
26
  "tools:register",
27
27
  Serializable[dict[str, Any]]({
28
- "catalog": catalog_name,
29
- "tool": {
30
- "name": name,
31
- "description": description,
32
- "schema_id": schema_id,
33
- "return_direct": return_direct,
34
- "schema_ver": schema_ver,
35
- "schema": schema,
36
- "timeout_ms": timeout_ms,
37
- "idempotent": idempotent
28
+ "data": {
29
+ "catalog": catalog_name,
30
+ "tool": {
31
+ "name": name,
32
+ "description": description,
33
+ "schema_id": schema_id,
34
+ "return_direct": return_direct,
35
+ "schema_ver": schema_ver,
36
+ "schema": schema,
37
+ "timeout_ms": timeout_ms,
38
+ "idempotent": idempotent
39
+ }
38
40
  }
39
41
  }),
40
42
  timeout=30,
@@ -51,8 +53,10 @@ class ToolsManager:
51
53
  response = await self.router.send(
52
54
  "tool:unregister",
53
55
  Serializable[dict[str, Any]]({
54
- "catalog": catalog_name,
55
- "tool_id": tool_id
56
+ "data": {
57
+ "catalog": catalog_name,
58
+ "tool_id": tool_id
59
+ }
56
60
  }),
57
61
  timeout=30,
58
62
  expected_response=dict[str, Any]
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: attp-client
3
- Version: 0.0.5
3
+ Version: 0.0.7
4
4
  Summary: A python-sdk client for interacting with AgentHub's ATTP protocol (Agent Tool Transport Protocol)
5
5
  License: MIT
6
6
  Author: Ascender Team
@@ -8,7 +8,7 @@ attp_client/errors/dead_session.py,sha256=BI-EuTqxVP7j23wcD0GfhteyPsKR2xqUsMNm9S
8
8
  attp_client/errors/not_found.py,sha256=Y-Y_Mki1hQYihJttvO0ugHFu9-73--1wqwdOomp2IEM,39
9
9
  attp_client/errors/serialization_error.py,sha256=Pa8PRzFJrrikA1Ikj0q-0euvXVUMb_qj-NRIp55SfOk,198
10
10
  attp_client/errors/unauthenticated_error.py,sha256=F0V1FjO0qVLMl6Y120y3AXKZnwb5iDD17c4GEMbL5aI,46
11
- attp_client/inference.py,sha256=gYCCaUI_DPgCkmxSNNdEte-5dp9MDEazjV6D79ycKEU,4611
11
+ attp_client/inference.py,sha256=MUr7DAzTCUlGRf4YYQw1YO98M2rCbC-OeiyqCdlSHbs,4847
12
12
  attp_client/interfaces/catalogs/catalog.py,sha256=3PxlRwR3y2tbQVfXAkhDIv07AJPraMfH0c_pyi7Y6z8,146
13
13
  attp_client/interfaces/catalogs/tools/envelope.py,sha256=6aUx06ou9If9OYv4BODKiBybrgBL2YWfSHZ6ukIR1K0,693
14
14
  attp_client/interfaces/error.py,sha256=fIrk5XlAhMs6mbYQ5PzgwS0v-LIbtne3OlQue4CjWXs,139
@@ -24,12 +24,12 @@ attp_client/interfaces/route_mappings.py,sha256=j6hEdkCP5xPpHS16EWmlkdTlnHa7z6e8
24
24
  attp_client/misc/fixed_basemodel.py,sha256=0MTVmlTrA75Oxv0pVfLdXFTSp5AmBzgiNwvDiLFGF_w,1853
25
25
  attp_client/misc/serializable.py,sha256=tU08TsjiLiafAhU1jKd5BxajlHdEDcdKeEiKPqhMSTI,2102
26
26
  attp_client/router.py,sha256=UDHU2xsvTjgSIzMtV0jkPvOYK9R7GFjKfIrjjBHws-Q,4575
27
- attp_client/session.py,sha256=YDrd1KYmwKsZW6Xdkt6UprXZzGz6pFFSkk6O7XS6ZsA,11409
28
- attp_client/tools.py,sha256=8GppeVYtnWNqc8M6a-s-S43QYpZROU4aYk67R9388yc,1839
27
+ attp_client/session.py,sha256=oZx3Lgsr-y4onaL4yrM3Sj2caQck9wc14QcakhS42Js,11410
28
+ attp_client/tools.py,sha256=w8Vnd2wVBrq7MxsHG-Uw91A1j39m27T2Cu5rV5CHfds,1979
29
29
  attp_client/types/route_mapping.py,sha256=Kb9ZX88lqihRZr8IryfH1Vg_YAobW699Yjl6Raz1rdg,375
30
30
  attp_client/utils/context_awaiter.py,sha256=oCptu5g8mY43j5cr-W4fOe85OCCaqQI9r_Pn92NgZSY,1035
31
31
  attp_client/utils/route_mapper.py,sha256=uJNhKp6ipCSUxoiZS0Liix2lHOgUAnJM0kfgXWAh9RQ,554
32
32
  attp_client/utils/serializer.py,sha256=O1tWYbQS9jC9aus-ISKtliKCgGmOEIb9hxykVrmMKGY,636
33
- attp_client-0.0.5.dist-info/METADATA,sha256=6WV1-0Kgz_Ssf8NRG9BDSI5TdYw2mZAoSgP06Var97s,7137
34
- attp_client-0.0.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
35
- attp_client-0.0.5.dist-info/RECORD,,
33
+ attp_client-0.0.7.dist-info/METADATA,sha256=fSGJfOrXhRYiYJWVDcFT7ECgNDYypve8ncrW6elXM6M,7137
34
+ attp_client-0.0.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
35
+ attp_client-0.0.7.dist-info/RECORD,,