robosystems-client 0.1.16__py3-none-any.whl → 0.1.18__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.

Potentially problematic release.

This version of robosystems-client might be problematic. Click here for more details.

Files changed (109) hide show
  1. robosystems_client/__init__.py +15 -4
  2. robosystems_client/api/agent/auto_select_agent.py +24 -17
  3. robosystems_client/api/agent/batch_process_queries.py +24 -17
  4. robosystems_client/api/agent/execute_specific_agent.py +24 -17
  5. robosystems_client/api/agent/get_agent_metadata.py +24 -17
  6. robosystems_client/api/agent/list_agents.py +20 -18
  7. robosystems_client/api/agent/recommend_agent.py +24 -17
  8. robosystems_client/api/auth/forgot_password.py +191 -0
  9. robosystems_client/api/auth/generate_sso_token.py +18 -0
  10. robosystems_client/api/auth/get_current_auth_user.py +4 -22
  11. robosystems_client/api/auth/logout_user.py +17 -17
  12. robosystems_client/api/auth/{refresh_session.py → refresh_auth_session.py} +21 -21
  13. robosystems_client/api/auth/resend_verification_email.py +222 -0
  14. robosystems_client/api/auth/reset_password.py +177 -0
  15. robosystems_client/api/auth/validate_reset_token.py +171 -0
  16. robosystems_client/api/auth/verify_email.py +177 -0
  17. robosystems_client/api/backup/create_backup.py +24 -17
  18. robosystems_client/api/backup/export_backup.py +24 -17
  19. robosystems_client/api/backup/get_backup_download_url.py +20 -18
  20. robosystems_client/api/backup/get_backup_stats.py +24 -17
  21. robosystems_client/api/backup/list_backups.py +20 -18
  22. robosystems_client/api/backup/restore_backup.py +24 -17
  23. robosystems_client/api/connections/create_connection.py +24 -17
  24. robosystems_client/api/connections/create_link_token.py +24 -17
  25. robosystems_client/api/connections/delete_connection.py +24 -17
  26. robosystems_client/api/connections/exchange_link_token.py +24 -17
  27. robosystems_client/api/connections/get_connection.py +24 -17
  28. robosystems_client/api/connections/get_connection_options.py +24 -17
  29. robosystems_client/api/connections/init_o_auth.py +24 -17
  30. robosystems_client/api/connections/list_connections.py +20 -18
  31. robosystems_client/api/connections/oauth_callback.py +28 -21
  32. robosystems_client/api/connections/sync_connection.py +24 -17
  33. robosystems_client/api/copy/copy_data_to_graph.py +24 -17
  34. robosystems_client/api/create/create_graph.py +24 -17
  35. robosystems_client/api/graph_analytics/get_graph_metrics.py +24 -17
  36. robosystems_client/api/graph_analytics/get_graph_usage_stats.py +20 -18
  37. robosystems_client/api/graph_billing/get_current_graph_bill.py +24 -17
  38. robosystems_client/api/graph_billing/get_graph_billing_history.py +20 -18
  39. robosystems_client/api/graph_billing/get_graph_monthly_bill.py +24 -17
  40. robosystems_client/api/graph_billing/get_graph_usage_details.py +20 -18
  41. robosystems_client/api/graph_credits/check_credit_balance.py +20 -18
  42. robosystems_client/api/graph_credits/check_storage_limits.py +24 -17
  43. robosystems_client/api/graph_credits/get_credit_summary.py +24 -17
  44. robosystems_client/api/graph_credits/get_storage_usage.py +20 -18
  45. robosystems_client/api/graph_credits/list_credit_transactions.py +20 -18
  46. robosystems_client/api/graph_health/get_database_health.py +24 -17
  47. robosystems_client/api/graph_info/get_database_info.py +24 -17
  48. robosystems_client/api/graph_limits/get_graph_limits.py +24 -17
  49. robosystems_client/api/mcp/call_mcp_tool.py +20 -18
  50. robosystems_client/api/mcp/list_mcp_tools.py +24 -17
  51. robosystems_client/api/operations/cancel_operation.py +24 -17
  52. robosystems_client/api/operations/get_operation_status.py +24 -17
  53. robosystems_client/api/operations/stream_operation_events.py +20 -18
  54. robosystems_client/api/query/execute_cypher_query.py +20 -18
  55. robosystems_client/api/schema/export_graph_schema.py +20 -18
  56. robosystems_client/api/schema/get_graph_schema_info.py +24 -17
  57. robosystems_client/api/schema/list_schema_extensions.py +24 -17
  58. robosystems_client/api/schema/validate_schema.py +24 -17
  59. robosystems_client/api/subgraphs/create_subgraph.py +24 -17
  60. robosystems_client/api/subgraphs/delete_subgraph.py +24 -17
  61. robosystems_client/api/subgraphs/get_subgraph_info.py +24 -17
  62. robosystems_client/api/subgraphs/get_subgraph_quota.py +24 -17
  63. robosystems_client/api/subgraphs/list_subgraphs.py +24 -17
  64. robosystems_client/api/user/create_user_api_key.py +24 -17
  65. robosystems_client/api/user/get_all_credit_summaries.py +24 -17
  66. robosystems_client/api/user/get_current_user.py +24 -17
  67. robosystems_client/api/user/get_user_graphs.py +24 -17
  68. robosystems_client/api/user/list_user_api_keys.py +24 -17
  69. robosystems_client/api/user/revoke_user_api_key.py +24 -17
  70. robosystems_client/api/user/select_user_graph.py +24 -17
  71. robosystems_client/api/user/update_user.py +24 -17
  72. robosystems_client/api/user/update_user_api_key.py +24 -17
  73. robosystems_client/api/user/update_user_password.py +24 -17
  74. robosystems_client/api/user_analytics/get_detailed_user_analytics.py +20 -18
  75. robosystems_client/api/user_analytics/get_user_usage_overview.py +24 -17
  76. robosystems_client/api/user_limits/get_all_shared_repository_limits.py +24 -17
  77. robosystems_client/api/user_limits/get_shared_repository_limits.py +24 -17
  78. robosystems_client/api/user_limits/get_user_limits.py +24 -17
  79. robosystems_client/api/user_limits/get_user_usage.py +24 -17
  80. robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +24 -17
  81. robosystems_client/api/user_subscriptions/get_repository_credits.py +24 -17
  82. robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +24 -17
  83. robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +20 -18
  84. robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +24 -17
  85. robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +24 -17
  86. robosystems_client/extensions/__init__.py +70 -0
  87. robosystems_client/extensions/auth_integration.py +14 -1
  88. robosystems_client/extensions/copy_client.py +32 -22
  89. robosystems_client/extensions/dataframe_utils.py +455 -0
  90. robosystems_client/extensions/extensions.py +16 -0
  91. robosystems_client/extensions/operation_client.py +43 -21
  92. robosystems_client/extensions/query_client.py +109 -12
  93. robosystems_client/extensions/tests/test_dataframe_utils.py +334 -0
  94. robosystems_client/extensions/tests/test_integration.py +1 -1
  95. robosystems_client/extensions/tests/test_token_utils.py +274 -0
  96. robosystems_client/extensions/token_utils.py +417 -0
  97. robosystems_client/extensions/utils.py +32 -2
  98. robosystems_client/models/__init__.py +16 -0
  99. robosystems_client/models/email_verification_request.py +60 -0
  100. robosystems_client/models/forgot_password_request.py +60 -0
  101. robosystems_client/models/forgot_password_response_forgotpassword.py +44 -0
  102. robosystems_client/models/resend_verification_email_response_resendverificationemail.py +44 -0
  103. robosystems_client/models/reset_password_request.py +68 -0
  104. robosystems_client/models/reset_password_validate_response.py +82 -0
  105. robosystems_client-0.1.18.dist-info/METADATA +89 -0
  106. {robosystems_client-0.1.16.dist-info → robosystems_client-0.1.18.dist-info}/RECORD +108 -92
  107. robosystems_client-0.1.18.dist-info/licenses/LICENSE +21 -0
  108. robosystems_client-0.1.16.dist-info/METADATA +0 -302
  109. {robosystems_client-0.1.16.dist-info → robosystems_client-0.1.18.dist-info}/WHEEL +0 -0
robosystems_client/__init__.py CHANGED
@@ -1,14 +1,25 @@
1
1
  """RoboSystems Python Client."""
2
2
 
3
- __version__ = "0.1.0"
4
-
5
3
  from .client import AuthenticatedClient, Client
6
4
 
5
+ # Convenience alias for the main SDK
6
+ RoboSystemsSDK = AuthenticatedClient
7
+
7
8
  __all__ = (
8
9
  "AuthenticatedClient",
9
10
  "Client",
10
11
  "RoboSystemsSDK",
11
12
  )
12
13
 
13
- # Convenience alias for the main SDK
14
- RoboSystemsSDK = AuthenticatedClient
14
+
15
+ def _get_version() -> str:
16
+ """Get version from package metadata."""
17
+ try:
18
+ from importlib.metadata import version
19
+
20
+ return version("robosystems-client")
21
+ except Exception:
22
+ return "0.0.0+development"
23
+
24
+
25
+ __version__ = _get_version()
robosystems_client/api/agent/auto_select_agent.py CHANGED
@@ -16,21 +16,28 @@ def _get_kwargs(
16
16
  graph_id: str,
17
17
  *,
18
18
  body: AgentRequest,
19
+ token: Union[None, Unset, str] = UNSET,
19
20
  authorization: Union[None, Unset, str] = UNSET,
20
- auth_token: Union[None, Unset, str] = UNSET,
21
21
  ) -> dict[str, Any]:
22
22
  headers: dict[str, Any] = {}
23
23
  if not isinstance(authorization, Unset):
24
24
  headers["authorization"] = authorization
25
25
 
26
- cookies = {}
27
- if auth_token is not UNSET:
28
- cookies["auth-token"] = auth_token
26
+ params: dict[str, Any] = {}
27
+
28
+ json_token: Union[None, Unset, str]
29
+ if isinstance(token, Unset):
30
+ json_token = UNSET
31
+ else:
32
+ json_token = token
33
+ params["token"] = json_token
34
+
35
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
29
36
 
30
37
  _kwargs: dict[str, Any] = {
31
38
  "method": "post",
32
39
  "url": f"/v1/{graph_id}/agent",
33
- "cookies": cookies,
40
+ "params": params,
34
41
  }
35
42
 
36
43
  _kwargs["json"] = body.to_dict()
@@ -87,8 +94,8 @@ def sync_detailed(
87
94
  *,
88
95
  client: AuthenticatedClient,
89
96
  body: AgentRequest,
97
+ token: Union[None, Unset, str] = UNSET,
90
98
  authorization: Union[None, Unset, str] = UNSET,
91
- auth_token: Union[None, Unset, str] = UNSET,
92
99
  ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
93
100
  """Auto-select agent for query
94
101
 
@@ -104,8 +111,8 @@ def sync_detailed(
104
111
 
105
112
  Args:
106
113
  graph_id (str):
114
+ token (Union[None, Unset, str]): JWT token for SSE authentication
107
115
  authorization (Union[None, Unset, str]):
108
- auth_token (Union[None, Unset, str]):
109
116
  body (AgentRequest): Request model for agent interactions.
110
117
 
111
118
  Raises:
@@ -119,8 +126,8 @@ def sync_detailed(
119
126
  kwargs = _get_kwargs(
120
127
  graph_id=graph_id,
121
128
  body=body,
129
+ token=token,
122
130
  authorization=authorization,
123
- auth_token=auth_token,
124
131
  )
125
132
 
126
133
  response = client.get_httpx_client().request(
@@ -135,8 +142,8 @@ def sync(
135
142
  *,
136
143
  client: AuthenticatedClient,
137
144
  body: AgentRequest,
145
+ token: Union[None, Unset, str] = UNSET,
138
146
  authorization: Union[None, Unset, str] = UNSET,
139
- auth_token: Union[None, Unset, str] = UNSET,
140
147
  ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
141
148
  """Auto-select agent for query
142
149
 
@@ -152,8 +159,8 @@ def sync(
152
159
 
153
160
  Args:
154
161
  graph_id (str):
162
+ token (Union[None, Unset, str]): JWT token for SSE authentication
155
163
  authorization (Union[None, Unset, str]):
156
- auth_token (Union[None, Unset, str]):
157
164
  body (AgentRequest): Request model for agent interactions.
158
165
 
159
166
  Raises:
@@ -168,8 +175,8 @@ def sync(
168
175
  graph_id=graph_id,
169
176
  client=client,
170
177
  body=body,
178
+ token=token,
171
179
  authorization=authorization,
172
- auth_token=auth_token,
173
180
  ).parsed
174
181
 
175
182
 
@@ -178,8 +185,8 @@ async def asyncio_detailed(
178
185
  *,
179
186
  client: AuthenticatedClient,
180
187
  body: AgentRequest,
188
+ token: Union[None, Unset, str] = UNSET,
181
189
  authorization: Union[None, Unset, str] = UNSET,
182
- auth_token: Union[None, Unset, str] = UNSET,
183
190
  ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
184
191
  """Auto-select agent for query
185
192
 
@@ -195,8 +202,8 @@ async def asyncio_detailed(
195
202
 
196
203
  Args:
197
204
  graph_id (str):
205
+ token (Union[None, Unset, str]): JWT token for SSE authentication
198
206
  authorization (Union[None, Unset, str]):
199
- auth_token (Union[None, Unset, str]):
200
207
  body (AgentRequest): Request model for agent interactions.
201
208
 
202
209
  Raises:
@@ -210,8 +217,8 @@ async def asyncio_detailed(
210
217
  kwargs = _get_kwargs(
211
218
  graph_id=graph_id,
212
219
  body=body,
220
+ token=token,
213
221
  authorization=authorization,
214
- auth_token=auth_token,
215
222
  )
216
223
 
217
224
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -224,8 +231,8 @@ async def asyncio(
224
231
  *,
225
232
  client: AuthenticatedClient,
226
233
  body: AgentRequest,
234
+ token: Union[None, Unset, str] = UNSET,
227
235
  authorization: Union[None, Unset, str] = UNSET,
228
- auth_token: Union[None, Unset, str] = UNSET,
229
236
  ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
230
237
  """Auto-select agent for query
231
238
 
@@ -241,8 +248,8 @@ async def asyncio(
241
248
 
242
249
  Args:
243
250
  graph_id (str):
251
+ token (Union[None, Unset, str]): JWT token for SSE authentication
244
252
  authorization (Union[None, Unset, str]):
245
- auth_token (Union[None, Unset, str]):
246
253
  body (AgentRequest): Request model for agent interactions.
247
254
 
248
255
  Raises:
@@ -258,7 +265,7 @@ async def asyncio(
258
265
  graph_id=graph_id,
259
266
  client=client,
260
267
  body=body,
268
+ token=token,
261
269
  authorization=authorization,
262
- auth_token=auth_token,
263
270
  )
264
271
  ).parsed
robosystems_client/api/agent/batch_process_queries.py CHANGED
@@ -15,21 +15,28 @@ def _get_kwargs(
15
15
  graph_id: str,
16
16
  *,
17
17
  body: BatchAgentRequest,
18
+ token: Union[None, Unset, str] = UNSET,
18
19
  authorization: Union[None, Unset, str] = UNSET,
19
- auth_token: Union[None, Unset, str] = UNSET,
20
20
  ) -> dict[str, Any]:
21
21
  headers: dict[str, Any] = {}
22
22
  if not isinstance(authorization, Unset):
23
23
  headers["authorization"] = authorization
24
24
 
25
- cookies = {}
26
- if auth_token is not UNSET:
27
- cookies["auth-token"] = auth_token
25
+ params: dict[str, Any] = {}
26
+
27
+ json_token: Union[None, Unset, str]
28
+ if isinstance(token, Unset):
29
+ json_token = UNSET
30
+ else:
31
+ json_token = token
32
+ params["token"] = json_token
33
+
34
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
28
35
 
29
36
  _kwargs: dict[str, Any] = {
30
37
  "method": "post",
31
38
  "url": f"/v1/{graph_id}/agent/batch",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["json"] = body.to_dict()
@@ -82,8 +89,8 @@ def sync_detailed(
82
89
  *,
83
90
  client: AuthenticatedClient,
84
91
  body: BatchAgentRequest,
92
+ token: Union[None, Unset, str] = UNSET,
85
93
  authorization: Union[None, Unset, str] = UNSET,
86
- auth_token: Union[None, Unset, str] = UNSET,
87
94
  ) -> Response[Union[Any, BatchAgentResponse, HTTPValidationError]]:
88
95
  """Batch process multiple queries
89
96
 
@@ -104,8 +111,8 @@ def sync_detailed(
104
111
 
105
112
  Args:
106
113
  graph_id (str):
114
+ token (Union[None, Unset, str]): JWT token for SSE authentication
107
115
  authorization (Union[None, Unset, str]):
108
- auth_token (Union[None, Unset, str]):
109
116
  body (BatchAgentRequest): Request for batch processing multiple queries.
110
117
 
111
118
  Raises:
@@ -119,8 +126,8 @@ def sync_detailed(
119
126
  kwargs = _get_kwargs(
120
127
  graph_id=graph_id,
121
128
  body=body,
129
+ token=token,
122
130
  authorization=authorization,
123
- auth_token=auth_token,
124
131
  )
125
132
 
126
133
  response = client.get_httpx_client().request(
@@ -135,8 +142,8 @@ def sync(
135
142
  *,
136
143
  client: AuthenticatedClient,
137
144
  body: BatchAgentRequest,
145
+ token: Union[None, Unset, str] = UNSET,
138
146
  authorization: Union[None, Unset, str] = UNSET,
139
- auth_token: Union[None, Unset, str] = UNSET,
140
147
  ) -> Optional[Union[Any, BatchAgentResponse, HTTPValidationError]]:
141
148
  """Batch process multiple queries
142
149
 
@@ -157,8 +164,8 @@ def sync(
157
164
 
158
165
  Args:
159
166
  graph_id (str):
167
+ token (Union[None, Unset, str]): JWT token for SSE authentication
160
168
  authorization (Union[None, Unset, str]):
161
- auth_token (Union[None, Unset, str]):
162
169
  body (BatchAgentRequest): Request for batch processing multiple queries.
163
170
 
164
171
  Raises:
@@ -173,8 +180,8 @@ def sync(
173
180
  graph_id=graph_id,
174
181
  client=client,
175
182
  body=body,
183
+ token=token,
176
184
  authorization=authorization,
177
- auth_token=auth_token,
178
185
  ).parsed
179
186
 
180
187
 
@@ -183,8 +190,8 @@ async def asyncio_detailed(
183
190
  *,
184
191
  client: AuthenticatedClient,
185
192
  body: BatchAgentRequest,
193
+ token: Union[None, Unset, str] = UNSET,
186
194
  authorization: Union[None, Unset, str] = UNSET,
187
- auth_token: Union[None, Unset, str] = UNSET,
188
195
  ) -> Response[Union[Any, BatchAgentResponse, HTTPValidationError]]:
189
196
  """Batch process multiple queries
190
197
 
@@ -205,8 +212,8 @@ async def asyncio_detailed(
205
212
 
206
213
  Args:
207
214
  graph_id (str):
215
+ token (Union[None, Unset, str]): JWT token for SSE authentication
208
216
  authorization (Union[None, Unset, str]):
209
- auth_token (Union[None, Unset, str]):
210
217
  body (BatchAgentRequest): Request for batch processing multiple queries.
211
218
 
212
219
  Raises:
@@ -220,8 +227,8 @@ async def asyncio_detailed(
220
227
  kwargs = _get_kwargs(
221
228
  graph_id=graph_id,
222
229
  body=body,
230
+ token=token,
223
231
  authorization=authorization,
224
- auth_token=auth_token,
225
232
  )
226
233
 
227
234
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -234,8 +241,8 @@ async def asyncio(
234
241
  *,
235
242
  client: AuthenticatedClient,
236
243
  body: BatchAgentRequest,
244
+ token: Union[None, Unset, str] = UNSET,
237
245
  authorization: Union[None, Unset, str] = UNSET,
238
- auth_token: Union[None, Unset, str] = UNSET,
239
246
  ) -> Optional[Union[Any, BatchAgentResponse, HTTPValidationError]]:
240
247
  """Batch process multiple queries
241
248
 
@@ -256,8 +263,8 @@ async def asyncio(
256
263
 
257
264
  Args:
258
265
  graph_id (str):
266
+ token (Union[None, Unset, str]): JWT token for SSE authentication
259
267
  authorization (Union[None, Unset, str]):
260
- auth_token (Union[None, Unset, str]):
261
268
  body (BatchAgentRequest): Request for batch processing multiple queries.
262
269
 
263
270
  Raises:
@@ -273,7 +280,7 @@ async def asyncio(
273
280
  graph_id=graph_id,
274
281
  client=client,
275
282
  body=body,
283
+ token=token,
276
284
  authorization=authorization,
277
- auth_token=auth_token,
278
285
  )
279
286
  ).parsed
robosystems_client/api/agent/execute_specific_agent.py CHANGED
@@ -17,21 +17,28 @@ def _get_kwargs(
17
17
  agent_type: str,
18
18
  *,
19
19
  body: AgentRequest,
20
+ token: Union[None, Unset, str] = UNSET,
20
21
  authorization: Union[None, Unset, str] = UNSET,
21
- auth_token: Union[None, Unset, str] = UNSET,
22
22
  ) -> dict[str, Any]:
23
23
  headers: dict[str, Any] = {}
24
24
  if not isinstance(authorization, Unset):
25
25
  headers["authorization"] = authorization
26
26
 
27
- cookies = {}
28
- if auth_token is not UNSET:
29
- cookies["auth-token"] = auth_token
27
+ params: dict[str, Any] = {}
28
+
29
+ json_token: Union[None, Unset, str]
30
+ if isinstance(token, Unset):
31
+ json_token = UNSET
32
+ else:
33
+ json_token = token
34
+ params["token"] = json_token
35
+
36
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
30
37
 
31
38
  _kwargs: dict[str, Any] = {
32
39
  "method": "post",
33
40
  "url": f"/v1/{graph_id}/agent/{agent_type}",
34
- "cookies": cookies,
41
+ "params": params,
35
42
  }
36
43
 
37
44
  _kwargs["json"] = body.to_dict()
@@ -92,8 +99,8 @@ def sync_detailed(
92
99
  *,
93
100
  client: AuthenticatedClient,
94
101
  body: AgentRequest,
102
+ token: Union[None, Unset, str] = UNSET,
95
103
  authorization: Union[None, Unset, str] = UNSET,
96
- auth_token: Union[None, Unset, str] = UNSET,
97
104
  ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
98
105
  """Execute specific agent
99
106
 
@@ -109,8 +116,8 @@ def sync_detailed(
109
116
  Args:
110
117
  graph_id (str):
111
118
  agent_type (str):
119
+ token (Union[None, Unset, str]): JWT token for SSE authentication
112
120
  authorization (Union[None, Unset, str]):
113
- auth_token (Union[None, Unset, str]):
114
121
  body (AgentRequest): Request model for agent interactions.
115
122
 
116
123
  Raises:
@@ -125,8 +132,8 @@ def sync_detailed(
125
132
  graph_id=graph_id,
126
133
  agent_type=agent_type,
127
134
  body=body,
135
+ token=token,
128
136
  authorization=authorization,
129
- auth_token=auth_token,
130
137
  )
131
138
 
132
139
  response = client.get_httpx_client().request(
@@ -142,8 +149,8 @@ def sync(
142
149
  *,
143
150
  client: AuthenticatedClient,
144
151
  body: AgentRequest,
152
+ token: Union[None, Unset, str] = UNSET,
145
153
  authorization: Union[None, Unset, str] = UNSET,
146
- auth_token: Union[None, Unset, str] = UNSET,
147
154
  ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
148
155
  """Execute specific agent
149
156
 
@@ -159,8 +166,8 @@ def sync(
159
166
  Args:
160
167
  graph_id (str):
161
168
  agent_type (str):
169
+ token (Union[None, Unset, str]): JWT token for SSE authentication
162
170
  authorization (Union[None, Unset, str]):
163
- auth_token (Union[None, Unset, str]):
164
171
  body (AgentRequest): Request model for agent interactions.
165
172
 
166
173
  Raises:
@@ -176,8 +183,8 @@ def sync(
176
183
  agent_type=agent_type,
177
184
  client=client,
178
185
  body=body,
186
+ token=token,
179
187
  authorization=authorization,
180
- auth_token=auth_token,
181
188
  ).parsed
182
189
 
183
190
 
@@ -187,8 +194,8 @@ async def asyncio_detailed(
187
194
  *,
188
195
  client: AuthenticatedClient,
189
196
  body: AgentRequest,
197
+ token: Union[None, Unset, str] = UNSET,
190
198
  authorization: Union[None, Unset, str] = UNSET,
191
- auth_token: Union[None, Unset, str] = UNSET,
192
199
  ) -> Response[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
193
200
  """Execute specific agent
194
201
 
@@ -204,8 +211,8 @@ async def asyncio_detailed(
204
211
  Args:
205
212
  graph_id (str):
206
213
  agent_type (str):
214
+ token (Union[None, Unset, str]): JWT token for SSE authentication
207
215
  authorization (Union[None, Unset, str]):
208
- auth_token (Union[None, Unset, str]):
209
216
  body (AgentRequest): Request model for agent interactions.
210
217
 
211
218
  Raises:
@@ -220,8 +227,8 @@ async def asyncio_detailed(
220
227
  graph_id=graph_id,
221
228
  agent_type=agent_type,
222
229
  body=body,
230
+ token=token,
223
231
  authorization=authorization,
224
- auth_token=auth_token,
225
232
  )
226
233
 
227
234
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -235,8 +242,8 @@ async def asyncio(
235
242
  *,
236
243
  client: AuthenticatedClient,
237
244
  body: AgentRequest,
245
+ token: Union[None, Unset, str] = UNSET,
238
246
  authorization: Union[None, Unset, str] = UNSET,
239
- auth_token: Union[None, Unset, str] = UNSET,
240
247
  ) -> Optional[Union[AgentResponse, Any, ErrorResponse, HTTPValidationError]]:
241
248
  """Execute specific agent
242
249
 
@@ -252,8 +259,8 @@ async def asyncio(
252
259
  Args:
253
260
  graph_id (str):
254
261
  agent_type (str):
262
+ token (Union[None, Unset, str]): JWT token for SSE authentication
255
263
  authorization (Union[None, Unset, str]):
256
- auth_token (Union[None, Unset, str]):
257
264
  body (AgentRequest): Request model for agent interactions.
258
265
 
259
266
  Raises:
@@ -270,7 +277,7 @@ async def asyncio(
270
277
  agent_type=agent_type,
271
278
  client=client,
272
279
  body=body,
280
+ token=token,
273
281
  authorization=authorization,
274
- auth_token=auth_token,
275
282
  )
276
283
  ).parsed
robosystems_client/api/agent/get_agent_metadata.py CHANGED
@@ -14,21 +14,28 @@ def _get_kwargs(
14
14
  graph_id: str,
15
15
  agent_type: str,
16
16
  *,
17
+ token: Union[None, Unset, str] = UNSET,
17
18
  authorization: Union[None, Unset, str] = UNSET,
18
- auth_token: Union[None, Unset, str] = UNSET,
19
19
  ) -> dict[str, Any]:
20
20
  headers: dict[str, Any] = {}
21
21
  if not isinstance(authorization, Unset):
22
22
  headers["authorization"] = authorization
23
23
 
24
- cookies = {}
25
- if auth_token is not UNSET:
26
- cookies["auth-token"] = auth_token
24
+ params: dict[str, Any] = {}
25
+
26
+ json_token: Union[None, Unset, str]
27
+ if isinstance(token, Unset):
28
+ json_token = UNSET
29
+ else:
30
+ json_token = token
31
+ params["token"] = json_token
32
+
33
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
27
34
 
28
35
  _kwargs: dict[str, Any] = {
29
36
  "method": "get",
30
37
  "url": f"/v1/{graph_id}/agent/{agent_type}/metadata",
31
- "cookies": cookies,
38
+ "params": params,
32
39
  }
33
40
 
34
41
  _kwargs["headers"] = headers
@@ -71,8 +78,8 @@ def sync_detailed(
71
78
  agent_type: str,
72
79
  *,
73
80
  client: AuthenticatedClient,
81
+ token: Union[None, Unset, str] = UNSET,
74
82
  authorization: Union[None, Unset, str] = UNSET,
75
- auth_token: Union[None, Unset, str] = UNSET,
76
83
  ) -> Response[Union[AgentMetadataResponse, Any, HTTPValidationError]]:
77
84
  """Get agent metadata
78
85
 
@@ -91,8 +98,8 @@ def sync_detailed(
91
98
  Args:
92
99
  graph_id (str): Graph database identifier
93
100
  agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag')
101
+ token (Union[None, Unset, str]): JWT token for SSE authentication
94
102
  authorization (Union[None, Unset, str]):
95
- auth_token (Union[None, Unset, str]):
96
103
 
97
104
  Raises:
98
105
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -105,8 +112,8 @@ def sync_detailed(
105
112
  kwargs = _get_kwargs(
106
113
  graph_id=graph_id,
107
114
  agent_type=agent_type,
115
+ token=token,
108
116
  authorization=authorization,
109
- auth_token=auth_token,
110
117
  )
111
118
 
112
119
  response = client.get_httpx_client().request(
@@ -121,8 +128,8 @@ def sync(
121
128
  agent_type: str,
122
129
  *,
123
130
  client: AuthenticatedClient,
131
+ token: Union[None, Unset, str] = UNSET,
124
132
  authorization: Union[None, Unset, str] = UNSET,
125
- auth_token: Union[None, Unset, str] = UNSET,
126
133
  ) -> Optional[Union[AgentMetadataResponse, Any, HTTPValidationError]]:
127
134
  """Get agent metadata
128
135
 
@@ -141,8 +148,8 @@ def sync(
141
148
  Args:
142
149
  graph_id (str): Graph database identifier
143
150
  agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag')
151
+ token (Union[None, Unset, str]): JWT token for SSE authentication
144
152
  authorization (Union[None, Unset, str]):
145
- auth_token (Union[None, Unset, str]):
146
153
 
147
154
  Raises:
148
155
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -156,8 +163,8 @@ def sync(
156
163
  graph_id=graph_id,
157
164
  agent_type=agent_type,
158
165
  client=client,
166
+ token=token,
159
167
  authorization=authorization,
160
- auth_token=auth_token,
161
168
  ).parsed
162
169
 
163
170
 
@@ -166,8 +173,8 @@ async def asyncio_detailed(
166
173
  agent_type: str,
167
174
  *,
168
175
  client: AuthenticatedClient,
176
+ token: Union[None, Unset, str] = UNSET,
169
177
  authorization: Union[None, Unset, str] = UNSET,
170
- auth_token: Union[None, Unset, str] = UNSET,
171
178
  ) -> Response[Union[AgentMetadataResponse, Any, HTTPValidationError]]:
172
179
  """Get agent metadata
173
180
 
@@ -186,8 +193,8 @@ async def asyncio_detailed(
186
193
  Args:
187
194
  graph_id (str): Graph database identifier
188
195
  agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag')
196
+ token (Union[None, Unset, str]): JWT token for SSE authentication
189
197
  authorization (Union[None, Unset, str]):
190
- auth_token (Union[None, Unset, str]):
191
198
 
192
199
  Raises:
193
200
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -200,8 +207,8 @@ async def asyncio_detailed(
200
207
  kwargs = _get_kwargs(
201
208
  graph_id=graph_id,
202
209
  agent_type=agent_type,
210
+ token=token,
203
211
  authorization=authorization,
204
- auth_token=auth_token,
205
212
  )
206
213
 
207
214
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -214,8 +221,8 @@ async def asyncio(
214
221
  agent_type: str,
215
222
  *,
216
223
  client: AuthenticatedClient,
224
+ token: Union[None, Unset, str] = UNSET,
217
225
  authorization: Union[None, Unset, str] = UNSET,
218
- auth_token: Union[None, Unset, str] = UNSET,
219
226
  ) -> Optional[Union[AgentMetadataResponse, Any, HTTPValidationError]]:
220
227
  """Get agent metadata
221
228
 
@@ -234,8 +241,8 @@ async def asyncio(
234
241
  Args:
235
242
  graph_id (str): Graph database identifier
236
243
  agent_type (str): Agent type identifier (e.g., 'financial', 'research', 'rag')
244
+ token (Union[None, Unset, str]): JWT token for SSE authentication
237
245
  authorization (Union[None, Unset, str]):
238
- auth_token (Union[None, Unset, str]):
239
246
 
240
247
  Raises:
241
248
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -250,7 +257,7 @@ async def asyncio(
250
257
  graph_id=graph_id,
251
258
  agent_type=agent_type,
252
259
  client=client,
260
+ token=token,
253
261
  authorization=authorization,
254
- auth_token=auth_token,
255
262
  )
256
263
  ).parsed