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/api/mcp/list_mcp_tools.py CHANGED
@@ -14,21 +14,28 @@ from ...types import UNSET, Response, Unset
14
14
  def _get_kwargs(
15
15
  graph_id: 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}/mcp/tools",
31
- "cookies": cookies,
38
+ "params": params,
32
39
  }
33
40
 
34
41
  _kwargs["headers"] = headers
@@ -75,8 +82,8 @@ def sync_detailed(
75
82
  graph_id: str,
76
83
  *,
77
84
  client: AuthenticatedClient,
85
+ token: Union[None, Unset, str] = UNSET,
78
86
  authorization: Union[None, Unset, str] = UNSET,
79
- auth_token: Union[None, Unset, str] = UNSET,
80
87
  ) -> Response[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]:
81
88
  """List MCP Tools
82
89
 
@@ -98,8 +105,8 @@ def sync_detailed(
98
105
 
99
106
  Args:
100
107
  graph_id (str): Graph database identifier
108
+ token (Union[None, Unset, str]): JWT token for SSE authentication
101
109
  authorization (Union[None, Unset, str]):
102
- auth_token (Union[None, Unset, str]):
103
110
 
104
111
  Raises:
105
112
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -111,8 +118,8 @@ def sync_detailed(
111
118
 
112
119
  kwargs = _get_kwargs(
113
120
  graph_id=graph_id,
121
+ token=token,
114
122
  authorization=authorization,
115
- auth_token=auth_token,
116
123
  )
117
124
 
118
125
  response = client.get_httpx_client().request(
@@ -126,8 +133,8 @@ def sync(
126
133
  graph_id: str,
127
134
  *,
128
135
  client: AuthenticatedClient,
136
+ token: Union[None, Unset, str] = UNSET,
129
137
  authorization: Union[None, Unset, str] = UNSET,
130
- auth_token: Union[None, Unset, str] = UNSET,
131
138
  ) -> Optional[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]:
132
139
  """List MCP Tools
133
140
 
@@ -149,8 +156,8 @@ def sync(
149
156
 
150
157
  Args:
151
158
  graph_id (str): Graph database identifier
159
+ token (Union[None, Unset, str]): JWT token for SSE authentication
152
160
  authorization (Union[None, Unset, str]):
153
- auth_token (Union[None, Unset, str]):
154
161
 
155
162
  Raises:
156
163
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -163,8 +170,8 @@ def sync(
163
170
  return sync_detailed(
164
171
  graph_id=graph_id,
165
172
  client=client,
173
+ token=token,
166
174
  authorization=authorization,
167
- auth_token=auth_token,
168
175
  ).parsed
169
176
 
170
177
 
@@ -172,8 +179,8 @@ async def asyncio_detailed(
172
179
  graph_id: str,
173
180
  *,
174
181
  client: AuthenticatedClient,
182
+ token: Union[None, Unset, str] = UNSET,
175
183
  authorization: Union[None, Unset, str] = UNSET,
176
- auth_token: Union[None, Unset, str] = UNSET,
177
184
  ) -> Response[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]:
178
185
  """List MCP Tools
179
186
 
@@ -195,8 +202,8 @@ async def asyncio_detailed(
195
202
 
196
203
  Args:
197
204
  graph_id (str): Graph database identifier
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
 
201
208
  Raises:
202
209
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -208,8 +215,8 @@ async def asyncio_detailed(
208
215
 
209
216
  kwargs = _get_kwargs(
210
217
  graph_id=graph_id,
218
+ token=token,
211
219
  authorization=authorization,
212
- auth_token=auth_token,
213
220
  )
214
221
 
215
222
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -221,8 +228,8 @@ async def asyncio(
221
228
  graph_id: str,
222
229
  *,
223
230
  client: AuthenticatedClient,
231
+ token: Union[None, Unset, str] = UNSET,
224
232
  authorization: Union[None, Unset, str] = UNSET,
225
- auth_token: Union[None, Unset, str] = UNSET,
226
233
  ) -> Optional[Union[ErrorResponse, HTTPValidationError, MCPToolsResponse]]:
227
234
  """List MCP Tools
228
235
 
@@ -244,8 +251,8 @@ async def asyncio(
244
251
 
245
252
  Args:
246
253
  graph_id (str): Graph database identifier
254
+ token (Union[None, Unset, str]): JWT token for SSE authentication
247
255
  authorization (Union[None, Unset, str]):
248
- auth_token (Union[None, Unset, str]):
249
256
 
250
257
  Raises:
251
258
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -259,7 +266,7 @@ async def asyncio(
259
266
  await asyncio_detailed(
260
267
  graph_id=graph_id,
261
268
  client=client,
269
+ token=token,
262
270
  authorization=authorization,
263
- auth_token=auth_token,
264
271
  )
265
272
  ).parsed
robosystems_client/api/operations/cancel_operation.py CHANGED
@@ -15,21 +15,28 @@ from ...types import UNSET, Response, Unset
15
15
  def _get_kwargs(
16
16
  operation_id: str,
17
17
  *,
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": "delete",
31
38
  "url": f"/v1/operations/{operation_id}",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["headers"] = headers
@@ -80,8 +87,8 @@ def sync_detailed(
80
87
  operation_id: str,
81
88
  *,
82
89
  client: AuthenticatedClient,
90
+ token: Union[None, Unset, str] = UNSET,
83
91
  authorization: Union[None, Unset, str] = UNSET,
84
- auth_token: Union[None, Unset, str] = UNSET,
85
92
  ) -> Response[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]:
86
93
  """Cancel Operation
87
94
 
@@ -97,8 +104,8 @@ def sync_detailed(
97
104
 
98
105
  Args:
99
106
  operation_id (str): Operation identifier
107
+ token (Union[None, Unset, str]): JWT token for SSE authentication
100
108
  authorization (Union[None, Unset, str]):
101
- auth_token (Union[None, Unset, str]):
102
109
 
103
110
  Raises:
104
111
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -110,8 +117,8 @@ def sync_detailed(
110
117
 
111
118
  kwargs = _get_kwargs(
112
119
  operation_id=operation_id,
120
+ token=token,
113
121
  authorization=authorization,
114
- auth_token=auth_token,
115
122
  )
116
123
 
117
124
  response = client.get_httpx_client().request(
@@ -125,8 +132,8 @@ def sync(
125
132
  operation_id: str,
126
133
  *,
127
134
  client: AuthenticatedClient,
135
+ token: Union[None, Unset, str] = UNSET,
128
136
  authorization: Union[None, Unset, str] = UNSET,
129
- auth_token: Union[None, Unset, str] = UNSET,
130
137
  ) -> Optional[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]:
131
138
  """Cancel Operation
132
139
 
@@ -142,8 +149,8 @@ def sync(
142
149
 
143
150
  Args:
144
151
  operation_id (str): Operation identifier
152
+ token (Union[None, Unset, str]): JWT token for SSE authentication
145
153
  authorization (Union[None, Unset, str]):
146
- auth_token (Union[None, Unset, str]):
147
154
 
148
155
  Raises:
149
156
  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
  return sync_detailed(
157
164
  operation_id=operation_id,
158
165
  client=client,
166
+ token=token,
159
167
  authorization=authorization,
160
- auth_token=auth_token,
161
168
  ).parsed
162
169
 
163
170
 
@@ -165,8 +172,8 @@ async def asyncio_detailed(
165
172
  operation_id: str,
166
173
  *,
167
174
  client: AuthenticatedClient,
175
+ token: Union[None, Unset, str] = UNSET,
168
176
  authorization: Union[None, Unset, str] = UNSET,
169
- auth_token: Union[None, Unset, str] = UNSET,
170
177
  ) -> Response[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]:
171
178
  """Cancel Operation
172
179
 
@@ -182,8 +189,8 @@ async def asyncio_detailed(
182
189
 
183
190
  Args:
184
191
  operation_id (str): Operation identifier
192
+ token (Union[None, Unset, str]): JWT token for SSE authentication
185
193
  authorization (Union[None, Unset, str]):
186
- auth_token (Union[None, Unset, str]):
187
194
 
188
195
  Raises:
189
196
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -195,8 +202,8 @@ async def asyncio_detailed(
195
202
 
196
203
  kwargs = _get_kwargs(
197
204
  operation_id=operation_id,
205
+ token=token,
198
206
  authorization=authorization,
199
- auth_token=auth_token,
200
207
  )
201
208
 
202
209
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -208,8 +215,8 @@ async def asyncio(
208
215
  operation_id: str,
209
216
  *,
210
217
  client: AuthenticatedClient,
218
+ token: Union[None, Unset, str] = UNSET,
211
219
  authorization: Union[None, Unset, str] = UNSET,
212
- auth_token: Union[None, Unset, str] = UNSET,
213
220
  ) -> Optional[Union[Any, CancelOperationResponseCanceloperation, HTTPValidationError]]:
214
221
  """Cancel Operation
215
222
 
@@ -225,8 +232,8 @@ async def asyncio(
225
232
 
226
233
  Args:
227
234
  operation_id (str): Operation identifier
235
+ token (Union[None, Unset, str]): JWT token for SSE authentication
228
236
  authorization (Union[None, Unset, str]):
229
- auth_token (Union[None, Unset, str]):
230
237
 
231
238
  Raises:
232
239
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -240,7 +247,7 @@ async def asyncio(
240
247
  await asyncio_detailed(
241
248
  operation_id=operation_id,
242
249
  client=client,
250
+ token=token,
243
251
  authorization=authorization,
244
- auth_token=auth_token,
245
252
  )
246
253
  ).parsed
robosystems_client/api/operations/get_operation_status.py CHANGED
@@ -15,21 +15,28 @@ from ...types import UNSET, Response, Unset
15
15
  def _get_kwargs(
16
16
  operation_id: str,
17
17
  *,
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": "get",
31
38
  "url": f"/v1/operations/{operation_id}/status",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["headers"] = headers
@@ -83,8 +90,8 @@ def sync_detailed(
83
90
  operation_id: str,
84
91
  *,
85
92
  client: AuthenticatedClient,
93
+ token: Union[None, Unset, str] = UNSET,
86
94
  authorization: Union[None, Unset, str] = UNSET,
87
- auth_token: Union[None, Unset, str] = UNSET,
88
95
  ) -> Response[
89
96
  Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError]
90
97
  ]:
@@ -106,8 +113,8 @@ def sync_detailed(
106
113
 
107
114
  Args:
108
115
  operation_id (str): Operation identifier
116
+ token (Union[None, Unset, str]): JWT token for SSE authentication
109
117
  authorization (Union[None, Unset, str]):
110
- auth_token (Union[None, Unset, str]):
111
118
 
112
119
  Raises:
113
120
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -119,8 +126,8 @@ def sync_detailed(
119
126
 
120
127
  kwargs = _get_kwargs(
121
128
  operation_id=operation_id,
129
+ token=token,
122
130
  authorization=authorization,
123
- auth_token=auth_token,
124
131
  )
125
132
 
126
133
  response = client.get_httpx_client().request(
@@ -134,8 +141,8 @@ def sync(
134
141
  operation_id: str,
135
142
  *,
136
143
  client: AuthenticatedClient,
144
+ token: Union[None, Unset, str] = UNSET,
137
145
  authorization: Union[None, Unset, str] = UNSET,
138
- auth_token: Union[None, Unset, str] = UNSET,
139
146
  ) -> Optional[
140
147
  Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError]
141
148
  ]:
@@ -157,8 +164,8 @@ def sync(
157
164
 
158
165
  Args:
159
166
  operation_id (str): Operation identifier
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
 
163
170
  Raises:
164
171
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -171,8 +178,8 @@ def sync(
171
178
  return sync_detailed(
172
179
  operation_id=operation_id,
173
180
  client=client,
181
+ token=token,
174
182
  authorization=authorization,
175
- auth_token=auth_token,
176
183
  ).parsed
177
184
 
178
185
 
@@ -180,8 +187,8 @@ async def asyncio_detailed(
180
187
  operation_id: str,
181
188
  *,
182
189
  client: AuthenticatedClient,
190
+ token: Union[None, Unset, str] = UNSET,
183
191
  authorization: Union[None, Unset, str] = UNSET,
184
- auth_token: Union[None, Unset, str] = UNSET,
185
192
  ) -> Response[
186
193
  Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError]
187
194
  ]:
@@ -203,8 +210,8 @@ async def asyncio_detailed(
203
210
 
204
211
  Args:
205
212
  operation_id (str): Operation identifier
213
+ token (Union[None, Unset, str]): JWT token for SSE authentication
206
214
  authorization (Union[None, Unset, str]):
207
- auth_token (Union[None, Unset, str]):
208
215
 
209
216
  Raises:
210
217
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -216,8 +223,8 @@ async def asyncio_detailed(
216
223
 
217
224
  kwargs = _get_kwargs(
218
225
  operation_id=operation_id,
226
+ token=token,
219
227
  authorization=authorization,
220
- auth_token=auth_token,
221
228
  )
222
229
 
223
230
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -229,8 +236,8 @@ async def asyncio(
229
236
  operation_id: str,
230
237
  *,
231
238
  client: AuthenticatedClient,
239
+ token: Union[None, Unset, str] = UNSET,
232
240
  authorization: Union[None, Unset, str] = UNSET,
233
- auth_token: Union[None, Unset, str] = UNSET,
234
241
  ) -> Optional[
235
242
  Union[Any, GetOperationStatusResponseGetoperationstatus, HTTPValidationError]
236
243
  ]:
@@ -252,8 +259,8 @@ async def asyncio(
252
259
 
253
260
  Args:
254
261
  operation_id (str): Operation identifier
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
 
258
265
  Raises:
259
266
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -267,7 +274,7 @@ async def asyncio(
267
274
  await asyncio_detailed(
268
275
  operation_id=operation_id,
269
276
  client=client,
277
+ token=token,
270
278
  authorization=authorization,
271
- auth_token=auth_token,
272
279
  )
273
280
  ).parsed
robosystems_client/api/operations/stream_operation_events.py CHANGED
@@ -13,28 +13,30 @@ def _get_kwargs(
13
13
  operation_id: str,
14
14
  *,
15
15
  from_sequence: Union[Unset, int] = 0,
16
+ token: Union[None, Unset, str] = UNSET,
16
17
  authorization: Union[None, Unset, str] = UNSET,
17
- auth_token: Union[None, Unset, str] = UNSET,
18
18
  ) -> dict[str, Any]:
19
19
  headers: dict[str, Any] = {}
20
20
  if not isinstance(authorization, Unset):
21
21
  headers["authorization"] = authorization
22
22
 
23
- cookies = {}
24
- if auth_token is not UNSET:
25
- cookies["auth-token"] = auth_token
26
-
27
23
  params: dict[str, Any] = {}
28
24
 
29
25
  params["from_sequence"] = from_sequence
30
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
+
31
34
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
32
35
 
33
36
  _kwargs: dict[str, Any] = {
34
37
  "method": "get",
35
38
  "url": f"/v1/operations/{operation_id}/stream",
36
39
  "params": params,
37
- "cookies": cookies,
38
40
  }
39
41
 
40
42
  _kwargs["headers"] = headers
@@ -82,8 +84,8 @@ def sync_detailed(
82
84
  *,
83
85
  client: AuthenticatedClient,
84
86
  from_sequence: Union[Unset, int] = 0,
87
+ token: Union[None, Unset, str] = UNSET,
85
88
  authorization: Union[None, Unset, str] = UNSET,
86
- auth_token: Union[None, Unset, str] = UNSET,
87
89
  ) -> Response[Union[Any, HTTPValidationError]]:
88
90
  """Stream Operation Events
89
91
 
@@ -139,8 +141,8 @@ def sync_detailed(
139
141
  operation_id (str): Operation identifier from initial submission
140
142
  from_sequence (Union[Unset, int]): Start streaming from this sequence number (0 = from
141
143
  beginning) Default: 0.
144
+ token (Union[None, Unset, str]): JWT token for SSE authentication
142
145
  authorization (Union[None, Unset, str]):
143
- auth_token (Union[None, Unset, str]):
144
146
 
145
147
  Raises:
146
148
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -153,8 +155,8 @@ def sync_detailed(
153
155
  kwargs = _get_kwargs(
154
156
  operation_id=operation_id,
155
157
  from_sequence=from_sequence,
158
+ token=token,
156
159
  authorization=authorization,
157
- auth_token=auth_token,
158
160
  )
159
161
 
160
162
  response = client.get_httpx_client().request(
@@ -169,8 +171,8 @@ def sync(
169
171
  *,
170
172
  client: AuthenticatedClient,
171
173
  from_sequence: Union[Unset, int] = 0,
174
+ token: Union[None, Unset, str] = UNSET,
172
175
  authorization: Union[None, Unset, str] = UNSET,
173
- auth_token: Union[None, Unset, str] = UNSET,
174
176
  ) -> Optional[Union[Any, HTTPValidationError]]:
175
177
  """Stream Operation Events
176
178
 
@@ -226,8 +228,8 @@ def sync(
226
228
  operation_id (str): Operation identifier from initial submission
227
229
  from_sequence (Union[Unset, int]): Start streaming from this sequence number (0 = from
228
230
  beginning) Default: 0.
231
+ token (Union[None, Unset, str]): JWT token for SSE authentication
229
232
  authorization (Union[None, Unset, str]):
230
- auth_token (Union[None, Unset, str]):
231
233
 
232
234
  Raises:
233
235
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -241,8 +243,8 @@ def sync(
241
243
  operation_id=operation_id,
242
244
  client=client,
243
245
  from_sequence=from_sequence,
246
+ token=token,
244
247
  authorization=authorization,
245
- auth_token=auth_token,
246
248
  ).parsed
247
249
 
248
250
 
@@ -251,8 +253,8 @@ async def asyncio_detailed(
251
253
  *,
252
254
  client: AuthenticatedClient,
253
255
  from_sequence: Union[Unset, int] = 0,
256
+ token: Union[None, Unset, str] = UNSET,
254
257
  authorization: Union[None, Unset, str] = UNSET,
255
- auth_token: Union[None, Unset, str] = UNSET,
256
258
  ) -> Response[Union[Any, HTTPValidationError]]:
257
259
  """Stream Operation Events
258
260
 
@@ -308,8 +310,8 @@ async def asyncio_detailed(
308
310
  operation_id (str): Operation identifier from initial submission
309
311
  from_sequence (Union[Unset, int]): Start streaming from this sequence number (0 = from
310
312
  beginning) Default: 0.
313
+ token (Union[None, Unset, str]): JWT token for SSE authentication
311
314
  authorization (Union[None, Unset, str]):
312
- auth_token (Union[None, Unset, str]):
313
315
 
314
316
  Raises:
315
317
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -322,8 +324,8 @@ async def asyncio_detailed(
322
324
  kwargs = _get_kwargs(
323
325
  operation_id=operation_id,
324
326
  from_sequence=from_sequence,
327
+ token=token,
325
328
  authorization=authorization,
326
- auth_token=auth_token,
327
329
  )
328
330
 
329
331
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -336,8 +338,8 @@ async def asyncio(
336
338
  *,
337
339
  client: AuthenticatedClient,
338
340
  from_sequence: Union[Unset, int] = 0,
341
+ token: Union[None, Unset, str] = UNSET,
339
342
  authorization: Union[None, Unset, str] = UNSET,
340
- auth_token: Union[None, Unset, str] = UNSET,
341
343
  ) -> Optional[Union[Any, HTTPValidationError]]:
342
344
  """Stream Operation Events
343
345
 
@@ -393,8 +395,8 @@ async def asyncio(
393
395
  operation_id (str): Operation identifier from initial submission
394
396
  from_sequence (Union[Unset, int]): Start streaming from this sequence number (0 = from
395
397
  beginning) Default: 0.
398
+ token (Union[None, Unset, str]): JWT token for SSE authentication
396
399
  authorization (Union[None, Unset, str]):
397
- auth_token (Union[None, Unset, str]):
398
400
 
399
401
  Raises:
400
402
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -409,7 +411,7 @@ async def asyncio(
409
411
  operation_id=operation_id,
410
412
  client=client,
411
413
  from_sequence=from_sequence,
414
+ token=token,
412
415
  authorization=authorization,
413
- auth_token=auth_token,
414
416
  )
415
417
  ).parsed