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/graph_billing/get_graph_billing_history.py CHANGED
@@ -17,28 +17,30 @@ def _get_kwargs(
17
17
  graph_id: str,
18
18
  *,
19
19
  months: Union[Unset, int] = 6,
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
30
-
31
27
  params: dict[str, Any] = {}
32
28
 
33
29
  params["months"] = months
34
30
 
31
+ json_token: Union[None, Unset, str]
32
+ if isinstance(token, Unset):
33
+ json_token = UNSET
34
+ else:
35
+ json_token = token
36
+ params["token"] = json_token
37
+
35
38
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
36
39
 
37
40
  _kwargs: dict[str, Any] = {
38
41
  "method": "get",
39
42
  "url": f"/v1/{graph_id}/billing/history",
40
43
  "params": params,
41
- "cookies": cookies,
42
44
  }
43
45
 
44
46
  _kwargs["headers"] = headers
@@ -104,8 +106,8 @@ def sync_detailed(
104
106
  *,
105
107
  client: AuthenticatedClient,
106
108
  months: Union[Unset, int] = 6,
109
+ token: Union[None, Unset, str] = UNSET,
107
110
  authorization: Union[None, Unset, str] = UNSET,
108
- auth_token: Union[None, Unset, str] = UNSET,
109
111
  ) -> Response[
110
112
  Union[
111
113
  ErrorResponse,
@@ -134,8 +136,8 @@ def sync_detailed(
134
136
  Args:
135
137
  graph_id (str): Graph database identifier
136
138
  months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6.
139
+ token (Union[None, Unset, str]): JWT token for SSE authentication
137
140
  authorization (Union[None, Unset, str]):
138
- auth_token (Union[None, Unset, str]):
139
141
 
140
142
  Raises:
141
143
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -148,8 +150,8 @@ def sync_detailed(
148
150
  kwargs = _get_kwargs(
149
151
  graph_id=graph_id,
150
152
  months=months,
153
+ token=token,
151
154
  authorization=authorization,
152
- auth_token=auth_token,
153
155
  )
154
156
 
155
157
  response = client.get_httpx_client().request(
@@ -164,8 +166,8 @@ def sync(
164
166
  *,
165
167
  client: AuthenticatedClient,
166
168
  months: Union[Unset, int] = 6,
169
+ token: Union[None, Unset, str] = UNSET,
167
170
  authorization: Union[None, Unset, str] = UNSET,
168
- auth_token: Union[None, Unset, str] = UNSET,
169
171
  ) -> Optional[
170
172
  Union[
171
173
  ErrorResponse,
@@ -194,8 +196,8 @@ def sync(
194
196
  Args:
195
197
  graph_id (str): Graph database identifier
196
198
  months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6.
199
+ token (Union[None, Unset, str]): JWT token for SSE authentication
197
200
  authorization (Union[None, Unset, str]):
198
- auth_token (Union[None, Unset, str]):
199
201
 
200
202
  Raises:
201
203
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -209,8 +211,8 @@ def sync(
209
211
  graph_id=graph_id,
210
212
  client=client,
211
213
  months=months,
214
+ token=token,
212
215
  authorization=authorization,
213
- auth_token=auth_token,
214
216
  ).parsed
215
217
 
216
218
 
@@ -219,8 +221,8 @@ async def asyncio_detailed(
219
221
  *,
220
222
  client: AuthenticatedClient,
221
223
  months: Union[Unset, int] = 6,
224
+ token: Union[None, Unset, str] = UNSET,
222
225
  authorization: Union[None, Unset, str] = UNSET,
223
- auth_token: Union[None, Unset, str] = UNSET,
224
226
  ) -> Response[
225
227
  Union[
226
228
  ErrorResponse,
@@ -249,8 +251,8 @@ async def asyncio_detailed(
249
251
  Args:
250
252
  graph_id (str): Graph database identifier
251
253
  months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6.
254
+ token (Union[None, Unset, str]): JWT token for SSE authentication
252
255
  authorization (Union[None, Unset, str]):
253
- auth_token (Union[None, Unset, str]):
254
256
 
255
257
  Raises:
256
258
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -263,8 +265,8 @@ async def asyncio_detailed(
263
265
  kwargs = _get_kwargs(
264
266
  graph_id=graph_id,
265
267
  months=months,
268
+ token=token,
266
269
  authorization=authorization,
267
- auth_token=auth_token,
268
270
  )
269
271
 
270
272
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -277,8 +279,8 @@ async def asyncio(
277
279
  *,
278
280
  client: AuthenticatedClient,
279
281
  months: Union[Unset, int] = 6,
282
+ token: Union[None, Unset, str] = UNSET,
280
283
  authorization: Union[None, Unset, str] = UNSET,
281
- auth_token: Union[None, Unset, str] = UNSET,
282
284
  ) -> Optional[
283
285
  Union[
284
286
  ErrorResponse,
@@ -307,8 +309,8 @@ async def asyncio(
307
309
  Args:
308
310
  graph_id (str): Graph database identifier
309
311
  months (Union[Unset, int]): Number of months to retrieve (1-24) Default: 6.
312
+ token (Union[None, Unset, str]): JWT token for SSE authentication
310
313
  authorization (Union[None, Unset, str]):
311
- auth_token (Union[None, Unset, str]):
312
314
 
313
315
  Raises:
314
316
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -323,7 +325,7 @@ async def asyncio(
323
325
  graph_id=graph_id,
324
326
  client=client,
325
327
  months=months,
328
+ token=token,
326
329
  authorization=authorization,
327
- auth_token=auth_token,
328
330
  )
329
331
  ).parsed
robosystems_client/api/graph_billing/get_graph_monthly_bill.py CHANGED
@@ -18,21 +18,28 @@ def _get_kwargs(
18
18
  year: int,
19
19
  month: int,
20
20
  *,
21
+ token: Union[None, Unset, str] = UNSET,
21
22
  authorization: Union[None, Unset, str] = UNSET,
22
- auth_token: Union[None, Unset, str] = UNSET,
23
23
  ) -> dict[str, Any]:
24
24
  headers: dict[str, Any] = {}
25
25
  if not isinstance(authorization, Unset):
26
26
  headers["authorization"] = authorization
27
27
 
28
- cookies = {}
29
- if auth_token is not UNSET:
30
- cookies["auth-token"] = auth_token
28
+ params: dict[str, Any] = {}
29
+
30
+ json_token: Union[None, Unset, str]
31
+ if isinstance(token, Unset):
32
+ json_token = UNSET
33
+ else:
34
+ json_token = token
35
+ params["token"] = json_token
36
+
37
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
31
38
 
32
39
  _kwargs: dict[str, Any] = {
33
40
  "method": "get",
34
41
  "url": f"/v1/{graph_id}/billing/history/{year}/{month}",
35
- "cookies": cookies,
42
+ "params": params,
36
43
  }
37
44
 
38
45
  _kwargs["headers"] = headers
@@ -99,8 +106,8 @@ def sync_detailed(
99
106
  month: int,
100
107
  *,
101
108
  client: AuthenticatedClient,
109
+ token: Union[None, Unset, str] = UNSET,
102
110
  authorization: Union[None, Unset, str] = UNSET,
103
- auth_token: Union[None, Unset, str] = UNSET,
104
111
  ) -> Response[
105
112
  Union[
106
113
  ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError
@@ -125,8 +132,8 @@ def sync_detailed(
125
132
  graph_id (str): Graph database identifier
126
133
  year (int): Year (2024-2030)
127
134
  month (int): Month (1-12)
135
+ token (Union[None, Unset, str]): JWT token for SSE authentication
128
136
  authorization (Union[None, Unset, str]):
129
- auth_token (Union[None, Unset, str]):
130
137
 
131
138
  Raises:
132
139
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -140,8 +147,8 @@ def sync_detailed(
140
147
  graph_id=graph_id,
141
148
  year=year,
142
149
  month=month,
150
+ token=token,
143
151
  authorization=authorization,
144
- auth_token=auth_token,
145
152
  )
146
153
 
147
154
  response = client.get_httpx_client().request(
@@ -157,8 +164,8 @@ def sync(
157
164
  month: int,
158
165
  *,
159
166
  client: AuthenticatedClient,
167
+ token: Union[None, Unset, str] = UNSET,
160
168
  authorization: Union[None, Unset, str] = UNSET,
161
- auth_token: Union[None, Unset, str] = UNSET,
162
169
  ) -> Optional[
163
170
  Union[
164
171
  ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError
@@ -183,8 +190,8 @@ def sync(
183
190
  graph_id (str): Graph database identifier
184
191
  year (int): Year (2024-2030)
185
192
  month (int): Month (1-12)
193
+ token (Union[None, Unset, str]): JWT token for SSE authentication
186
194
  authorization (Union[None, Unset, str]):
187
- auth_token (Union[None, Unset, str]):
188
195
 
189
196
  Raises:
190
197
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -199,8 +206,8 @@ def sync(
199
206
  year=year,
200
207
  month=month,
201
208
  client=client,
209
+ token=token,
202
210
  authorization=authorization,
203
- auth_token=auth_token,
204
211
  ).parsed
205
212
 
206
213
 
@@ -210,8 +217,8 @@ async def asyncio_detailed(
210
217
  month: int,
211
218
  *,
212
219
  client: AuthenticatedClient,
220
+ token: Union[None, Unset, str] = UNSET,
213
221
  authorization: Union[None, Unset, str] = UNSET,
214
- auth_token: Union[None, Unset, str] = UNSET,
215
222
  ) -> Response[
216
223
  Union[
217
224
  ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError
@@ -236,8 +243,8 @@ async def asyncio_detailed(
236
243
  graph_id (str): Graph database identifier
237
244
  year (int): Year (2024-2030)
238
245
  month (int): Month (1-12)
246
+ token (Union[None, Unset, str]): JWT token for SSE authentication
239
247
  authorization (Union[None, Unset, str]):
240
- auth_token (Union[None, Unset, str]):
241
248
 
242
249
  Raises:
243
250
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -251,8 +258,8 @@ async def asyncio_detailed(
251
258
  graph_id=graph_id,
252
259
  year=year,
253
260
  month=month,
261
+ token=token,
254
262
  authorization=authorization,
255
- auth_token=auth_token,
256
263
  )
257
264
 
258
265
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -266,8 +273,8 @@ async def asyncio(
266
273
  month: int,
267
274
  *,
268
275
  client: AuthenticatedClient,
276
+ token: Union[None, Unset, str] = UNSET,
269
277
  authorization: Union[None, Unset, str] = UNSET,
270
- auth_token: Union[None, Unset, str] = UNSET,
271
278
  ) -> Optional[
272
279
  Union[
273
280
  ErrorResponse, GetGraphMonthlyBillResponseGetgraphmonthlybill, HTTPValidationError
@@ -292,8 +299,8 @@ async def asyncio(
292
299
  graph_id (str): Graph database identifier
293
300
  year (int): Year (2024-2030)
294
301
  month (int): Month (1-12)
302
+ token (Union[None, Unset, str]): JWT token for SSE authentication
295
303
  authorization (Union[None, Unset, str]):
296
- auth_token (Union[None, Unset, str]):
297
304
 
298
305
  Raises:
299
306
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -309,7 +316,7 @@ async def asyncio(
309
316
  year=year,
310
317
  month=month,
311
318
  client=client,
319
+ token=token,
312
320
  authorization=authorization,
313
- auth_token=auth_token,
314
321
  )
315
322
  ).parsed
robosystems_client/api/graph_billing/get_graph_usage_details.py CHANGED
@@ -18,17 +18,13 @@ def _get_kwargs(
18
18
  *,
19
19
  year: Union[None, Unset, int] = UNSET,
20
20
  month: Union[None, Unset, int] = UNSET,
21
+ token: Union[None, Unset, str] = UNSET,
21
22
  authorization: Union[None, Unset, str] = UNSET,
22
- auth_token: Union[None, Unset, str] = UNSET,
23
23
  ) -> dict[str, Any]:
24
24
  headers: dict[str, Any] = {}
25
25
  if not isinstance(authorization, Unset):
26
26
  headers["authorization"] = authorization
27
27
 
28
- cookies = {}
29
- if auth_token is not UNSET:
30
- cookies["auth-token"] = auth_token
31
-
32
28
  params: dict[str, Any] = {}
33
29
 
34
30
  json_year: Union[None, Unset, int]
@@ -45,13 +41,19 @@ def _get_kwargs(
45
41
  json_month = month
46
42
  params["month"] = json_month
47
43
 
44
+ json_token: Union[None, Unset, str]
45
+ if isinstance(token, Unset):
46
+ json_token = UNSET
47
+ else:
48
+ json_token = token
49
+ params["token"] = json_token
50
+
48
51
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
49
52
 
50
53
  _kwargs: dict[str, Any] = {
51
54
  "method": "get",
52
55
  "url": f"/v1/{graph_id}/billing/usage",
53
56
  "params": params,
54
- "cookies": cookies,
55
57
  }
56
58
 
57
59
  _kwargs["headers"] = headers
@@ -118,8 +120,8 @@ def sync_detailed(
118
120
  client: AuthenticatedClient,
119
121
  year: Union[None, Unset, int] = UNSET,
120
122
  month: Union[None, Unset, int] = UNSET,
123
+ token: Union[None, Unset, str] = UNSET,
121
124
  authorization: Union[None, Unset, str] = UNSET,
122
- auth_token: Union[None, Unset, str] = UNSET,
123
125
  ) -> Response[
124
126
  Union[
125
127
  ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError
@@ -148,8 +150,8 @@ def sync_detailed(
148
150
  graph_id (str): Graph database identifier
149
151
  year (Union[None, Unset, int]): Year (defaults to current)
150
152
  month (Union[None, Unset, int]): Month (defaults to current)
153
+ token (Union[None, Unset, str]): JWT token for SSE authentication
151
154
  authorization (Union[None, Unset, str]):
152
- auth_token (Union[None, Unset, str]):
153
155
 
154
156
  Raises:
155
157
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -163,8 +165,8 @@ def sync_detailed(
163
165
  graph_id=graph_id,
164
166
  year=year,
165
167
  month=month,
168
+ token=token,
166
169
  authorization=authorization,
167
- auth_token=auth_token,
168
170
  )
169
171
 
170
172
  response = client.get_httpx_client().request(
@@ -180,8 +182,8 @@ def sync(
180
182
  client: AuthenticatedClient,
181
183
  year: Union[None, Unset, int] = UNSET,
182
184
  month: Union[None, Unset, int] = UNSET,
185
+ token: Union[None, Unset, str] = UNSET,
183
186
  authorization: Union[None, Unset, str] = UNSET,
184
- auth_token: Union[None, Unset, str] = UNSET,
185
187
  ) -> Optional[
186
188
  Union[
187
189
  ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError
@@ -210,8 +212,8 @@ def sync(
210
212
  graph_id (str): Graph database identifier
211
213
  year (Union[None, Unset, int]): Year (defaults to current)
212
214
  month (Union[None, Unset, int]): Month (defaults to current)
215
+ token (Union[None, Unset, str]): JWT token for SSE authentication
213
216
  authorization (Union[None, Unset, str]):
214
- auth_token (Union[None, Unset, str]):
215
217
 
216
218
  Raises:
217
219
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -226,8 +228,8 @@ def sync(
226
228
  client=client,
227
229
  year=year,
228
230
  month=month,
231
+ token=token,
229
232
  authorization=authorization,
230
- auth_token=auth_token,
231
233
  ).parsed
232
234
 
233
235
 
@@ -237,8 +239,8 @@ async def asyncio_detailed(
237
239
  client: AuthenticatedClient,
238
240
  year: Union[None, Unset, int] = UNSET,
239
241
  month: Union[None, Unset, int] = UNSET,
242
+ token: Union[None, Unset, str] = UNSET,
240
243
  authorization: Union[None, Unset, str] = UNSET,
241
- auth_token: Union[None, Unset, str] = UNSET,
242
244
  ) -> Response[
243
245
  Union[
244
246
  ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError
@@ -267,8 +269,8 @@ async def asyncio_detailed(
267
269
  graph_id (str): Graph database identifier
268
270
  year (Union[None, Unset, int]): Year (defaults to current)
269
271
  month (Union[None, Unset, int]): Month (defaults to current)
272
+ token (Union[None, Unset, str]): JWT token for SSE authentication
270
273
  authorization (Union[None, Unset, str]):
271
- auth_token (Union[None, Unset, str]):
272
274
 
273
275
  Raises:
274
276
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -282,8 +284,8 @@ async def asyncio_detailed(
282
284
  graph_id=graph_id,
283
285
  year=year,
284
286
  month=month,
287
+ token=token,
285
288
  authorization=authorization,
286
- auth_token=auth_token,
287
289
  )
288
290
 
289
291
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -297,8 +299,8 @@ async def asyncio(
297
299
  client: AuthenticatedClient,
298
300
  year: Union[None, Unset, int] = UNSET,
299
301
  month: Union[None, Unset, int] = UNSET,
302
+ token: Union[None, Unset, str] = UNSET,
300
303
  authorization: Union[None, Unset, str] = UNSET,
301
- auth_token: Union[None, Unset, str] = UNSET,
302
304
  ) -> Optional[
303
305
  Union[
304
306
  ErrorResponse, GetGraphUsageDetailsResponseGetgraphusagedetails, HTTPValidationError
@@ -327,8 +329,8 @@ async def asyncio(
327
329
  graph_id (str): Graph database identifier
328
330
  year (Union[None, Unset, int]): Year (defaults to current)
329
331
  month (Union[None, Unset, int]): Month (defaults to current)
332
+ token (Union[None, Unset, str]): JWT token for SSE authentication
330
333
  authorization (Union[None, Unset, str]):
331
- auth_token (Union[None, Unset, str]):
332
334
 
333
335
  Raises:
334
336
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -344,7 +346,7 @@ async def asyncio(
344
346
  client=client,
345
347
  year=year,
346
348
  month=month,
349
+ token=token,
347
350
  authorization=authorization,
348
- auth_token=auth_token,
349
351
  )
350
352
  ).parsed
robosystems_client/api/graph_credits/check_credit_balance.py CHANGED
@@ -18,17 +18,13 @@ def _get_kwargs(
18
18
  *,
19
19
  operation_type: str,
20
20
  base_cost: Union[None, Unset, float, str] = UNSET,
21
+ token: Union[None, Unset, str] = UNSET,
21
22
  authorization: Union[None, Unset, str] = UNSET,
22
- auth_token: Union[None, Unset, str] = UNSET,
23
23
  ) -> dict[str, Any]:
24
24
  headers: dict[str, Any] = {}
25
25
  if not isinstance(authorization, Unset):
26
26
  headers["authorization"] = authorization
27
27
 
28
- cookies = {}
29
- if auth_token is not UNSET:
30
- cookies["auth-token"] = auth_token
31
-
32
28
  params: dict[str, Any] = {}
33
29
 
34
30
  params["operation_type"] = operation_type
@@ -40,13 +36,19 @@ def _get_kwargs(
40
36
  json_base_cost = base_cost
41
37
  params["base_cost"] = json_base_cost
42
38
 
39
+ json_token: Union[None, Unset, str]
40
+ if isinstance(token, Unset):
41
+ json_token = UNSET
42
+ else:
43
+ json_token = token
44
+ params["token"] = json_token
45
+
43
46
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
44
47
 
45
48
  _kwargs: dict[str, Any] = {
46
49
  "method": "get",
47
50
  "url": f"/v1/{graph_id}/credits/balance/check",
48
51
  "params": params,
49
- "cookies": cookies,
50
52
  }
51
53
 
52
54
  _kwargs["headers"] = headers
@@ -109,8 +111,8 @@ def sync_detailed(
109
111
  client: AuthenticatedClient,
110
112
  operation_type: str,
111
113
  base_cost: Union[None, Unset, float, str] = UNSET,
114
+ token: Union[None, Unset, str] = UNSET,
112
115
  authorization: Union[None, Unset, str] = UNSET,
113
- auth_token: Union[None, Unset, str] = UNSET,
114
116
  ) -> Response[
115
117
  Union[
116
118
  CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError
@@ -135,8 +137,8 @@ def sync_detailed(
135
137
  operation_type (str): Type of operation to check
136
138
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
137
139
  provided)
140
+ token (Union[None, Unset, str]): JWT token for SSE authentication
138
141
  authorization (Union[None, Unset, str]):
139
- auth_token (Union[None, Unset, str]):
140
142
 
141
143
  Raises:
142
144
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -150,8 +152,8 @@ def sync_detailed(
150
152
  graph_id=graph_id,
151
153
  operation_type=operation_type,
152
154
  base_cost=base_cost,
155
+ token=token,
153
156
  authorization=authorization,
154
- auth_token=auth_token,
155
157
  )
156
158
 
157
159
  response = client.get_httpx_client().request(
@@ -167,8 +169,8 @@ def sync(
167
169
  client: AuthenticatedClient,
168
170
  operation_type: str,
169
171
  base_cost: Union[None, Unset, float, str] = UNSET,
172
+ token: Union[None, Unset, str] = UNSET,
170
173
  authorization: Union[None, Unset, str] = UNSET,
171
- auth_token: Union[None, Unset, str] = UNSET,
172
174
  ) -> Optional[
173
175
  Union[
174
176
  CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError
@@ -193,8 +195,8 @@ def sync(
193
195
  operation_type (str): Type of operation to check
194
196
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
195
197
  provided)
198
+ token (Union[None, Unset, str]): JWT token for SSE authentication
196
199
  authorization (Union[None, Unset, str]):
197
- auth_token (Union[None, Unset, str]):
198
200
 
199
201
  Raises:
200
202
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -209,8 +211,8 @@ def sync(
209
211
  client=client,
210
212
  operation_type=operation_type,
211
213
  base_cost=base_cost,
214
+ token=token,
212
215
  authorization=authorization,
213
- auth_token=auth_token,
214
216
  ).parsed
215
217
 
216
218
 
@@ -220,8 +222,8 @@ async def asyncio_detailed(
220
222
  client: AuthenticatedClient,
221
223
  operation_type: str,
222
224
  base_cost: Union[None, Unset, float, str] = UNSET,
225
+ token: Union[None, Unset, str] = UNSET,
223
226
  authorization: Union[None, Unset, str] = UNSET,
224
- auth_token: Union[None, Unset, str] = UNSET,
225
227
  ) -> Response[
226
228
  Union[
227
229
  CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError
@@ -246,8 +248,8 @@ async def asyncio_detailed(
246
248
  operation_type (str): Type of operation to check
247
249
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
248
250
  provided)
251
+ token (Union[None, Unset, str]): JWT token for SSE authentication
249
252
  authorization (Union[None, Unset, str]):
250
- auth_token (Union[None, Unset, str]):
251
253
 
252
254
  Raises:
253
255
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -261,8 +263,8 @@ async def asyncio_detailed(
261
263
  graph_id=graph_id,
262
264
  operation_type=operation_type,
263
265
  base_cost=base_cost,
266
+ token=token,
264
267
  authorization=authorization,
265
- auth_token=auth_token,
266
268
  )
267
269
 
268
270
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -276,8 +278,8 @@ async def asyncio(
276
278
  client: AuthenticatedClient,
277
279
  operation_type: str,
278
280
  base_cost: Union[None, Unset, float, str] = UNSET,
281
+ token: Union[None, Unset, str] = UNSET,
279
282
  authorization: Union[None, Unset, str] = UNSET,
280
- auth_token: Union[None, Unset, str] = UNSET,
281
283
  ) -> Optional[
282
284
  Union[
283
285
  CheckCreditBalanceResponseCheckcreditbalance, ErrorResponse, HTTPValidationError
@@ -302,8 +304,8 @@ async def asyncio(
302
304
  operation_type (str): Type of operation to check
303
305
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
304
306
  provided)
307
+ token (Union[None, Unset, str]): JWT token for SSE authentication
305
308
  authorization (Union[None, Unset, str]):
306
- auth_token (Union[None, Unset, str]):
307
309
 
308
310
  Raises:
309
311
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -319,7 +321,7 @@ async def asyncio(
319
321
  client=client,
320
322
  operation_type=operation_type,
321
323
  base_cost=base_cost,
324
+ token=token,
322
325
  authorization=authorization,
323
- auth_token=auth_token,
324
326
  )
325
327
  ).parsed