robosystems-client 0.1.17__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 (92) hide show
  1. robosystems_client/__init__.py +15 -4
  2. robosystems_client/api/agent/auto_select_agent.py +25 -0
  3. robosystems_client/api/agent/batch_process_queries.py +25 -0
  4. robosystems_client/api/agent/execute_specific_agent.py +25 -0
  5. robosystems_client/api/agent/get_agent_metadata.py +25 -0
  6. robosystems_client/api/agent/list_agents.py +20 -0
  7. robosystems_client/api/agent/recommend_agent.py +25 -0
  8. robosystems_client/api/backup/create_backup.py +25 -0
  9. robosystems_client/api/backup/export_backup.py +25 -0
  10. robosystems_client/api/backup/get_backup_download_url.py +20 -0
  11. robosystems_client/api/backup/get_backup_stats.py +25 -0
  12. robosystems_client/api/backup/list_backups.py +20 -0
  13. robosystems_client/api/backup/restore_backup.py +25 -0
  14. robosystems_client/api/connections/create_connection.py +25 -0
  15. robosystems_client/api/connections/create_link_token.py +25 -0
  16. robosystems_client/api/connections/delete_connection.py +25 -0
  17. robosystems_client/api/connections/exchange_link_token.py +25 -0
  18. robosystems_client/api/connections/get_connection.py +25 -0
  19. robosystems_client/api/connections/get_connection_options.py +25 -0
  20. robosystems_client/api/connections/init_o_auth.py +25 -0
  21. robosystems_client/api/connections/list_connections.py +20 -0
  22. robosystems_client/api/connections/oauth_callback.py +25 -0
  23. robosystems_client/api/connections/sync_connection.py +25 -0
  24. robosystems_client/api/copy/copy_data_to_graph.py +25 -0
  25. robosystems_client/api/create/create_graph.py +25 -0
  26. robosystems_client/api/graph_analytics/get_graph_metrics.py +25 -0
  27. robosystems_client/api/graph_analytics/get_graph_usage_stats.py +20 -0
  28. robosystems_client/api/graph_billing/get_current_graph_bill.py +25 -0
  29. robosystems_client/api/graph_billing/get_graph_billing_history.py +20 -0
  30. robosystems_client/api/graph_billing/get_graph_monthly_bill.py +25 -0
  31. robosystems_client/api/graph_billing/get_graph_usage_details.py +20 -0
  32. robosystems_client/api/graph_credits/check_credit_balance.py +20 -0
  33. robosystems_client/api/graph_credits/check_storage_limits.py +25 -0
  34. robosystems_client/api/graph_credits/get_credit_summary.py +25 -0
  35. robosystems_client/api/graph_credits/get_storage_usage.py +20 -0
  36. robosystems_client/api/graph_credits/list_credit_transactions.py +20 -0
  37. robosystems_client/api/graph_health/get_database_health.py +25 -0
  38. robosystems_client/api/graph_info/get_database_info.py +25 -0
  39. robosystems_client/api/graph_limits/get_graph_limits.py +25 -0
  40. robosystems_client/api/mcp/call_mcp_tool.py +20 -0
  41. robosystems_client/api/mcp/list_mcp_tools.py +25 -0
  42. robosystems_client/api/operations/cancel_operation.py +25 -0
  43. robosystems_client/api/operations/get_operation_status.py +25 -0
  44. robosystems_client/api/operations/stream_operation_events.py +20 -0
  45. robosystems_client/api/query/execute_cypher_query.py +20 -0
  46. robosystems_client/api/schema/export_graph_schema.py +20 -0
  47. robosystems_client/api/schema/get_graph_schema_info.py +25 -0
  48. robosystems_client/api/schema/list_schema_extensions.py +25 -0
  49. robosystems_client/api/schema/validate_schema.py +25 -0
  50. robosystems_client/api/subgraphs/create_subgraph.py +25 -0
  51. robosystems_client/api/subgraphs/delete_subgraph.py +25 -0
  52. robosystems_client/api/subgraphs/get_subgraph_info.py +25 -0
  53. robosystems_client/api/subgraphs/get_subgraph_quota.py +25 -0
  54. robosystems_client/api/subgraphs/list_subgraphs.py +25 -0
  55. robosystems_client/api/user/create_user_api_key.py +25 -0
  56. robosystems_client/api/user/get_all_credit_summaries.py +25 -0
  57. robosystems_client/api/user/get_current_user.py +25 -0
  58. robosystems_client/api/user/get_user_graphs.py +25 -0
  59. robosystems_client/api/user/list_user_api_keys.py +25 -0
  60. robosystems_client/api/user/revoke_user_api_key.py +25 -0
  61. robosystems_client/api/user/select_user_graph.py +25 -0
  62. robosystems_client/api/user/update_user.py +25 -0
  63. robosystems_client/api/user/update_user_api_key.py +25 -0
  64. robosystems_client/api/user/update_user_password.py +25 -0
  65. robosystems_client/api/user_analytics/get_detailed_user_analytics.py +20 -0
  66. robosystems_client/api/user_analytics/get_user_usage_overview.py +25 -0
  67. robosystems_client/api/user_limits/get_all_shared_repository_limits.py +25 -0
  68. robosystems_client/api/user_limits/get_shared_repository_limits.py +25 -0
  69. robosystems_client/api/user_limits/get_user_limits.py +25 -0
  70. robosystems_client/api/user_limits/get_user_usage.py +25 -0
  71. robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +25 -0
  72. robosystems_client/api/user_subscriptions/get_repository_credits.py +25 -0
  73. robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +25 -0
  74. robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +20 -0
  75. robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +25 -0
  76. robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +25 -0
  77. robosystems_client/extensions/__init__.py +70 -0
  78. robosystems_client/extensions/auth_integration.py +14 -1
  79. robosystems_client/extensions/copy_client.py +32 -22
  80. robosystems_client/extensions/dataframe_utils.py +455 -0
  81. robosystems_client/extensions/extensions.py +16 -0
  82. robosystems_client/extensions/operation_client.py +43 -21
  83. robosystems_client/extensions/query_client.py +109 -12
  84. robosystems_client/extensions/tests/test_dataframe_utils.py +334 -0
  85. robosystems_client/extensions/tests/test_integration.py +1 -1
  86. robosystems_client/extensions/tests/test_token_utils.py +274 -0
  87. robosystems_client/extensions/token_utils.py +417 -0
  88. robosystems_client/extensions/utils.py +32 -2
  89. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.18.dist-info}/METADATA +1 -1
  90. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.18.dist-info}/RECORD +92 -88
  91. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.18.dist-info}/WHEEL +0 -0
  92. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.18.dist-info}/licenses/LICENSE +0 -0
@@ -16,15 +16,28 @@ from ...types import UNSET, Response, Unset
16
16
  def _get_kwargs(
17
17
  graph_id: str,
18
18
  *,
19
+ token: Union[None, Unset, str] = UNSET,
19
20
  authorization: Union[None, Unset, str] = UNSET,
20
21
  ) -> dict[str, Any]:
21
22
  headers: dict[str, Any] = {}
22
23
  if not isinstance(authorization, Unset):
23
24
  headers["authorization"] = authorization
24
25
 
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}
36
+
25
37
  _kwargs: dict[str, Any] = {
26
38
  "method": "get",
27
39
  "url": f"/v1/{graph_id}/billing/current",
40
+ "params": params,
28
41
  }
29
42
 
30
43
  _kwargs["headers"] = headers
@@ -85,6 +98,7 @@ def sync_detailed(
85
98
  graph_id: str,
86
99
  *,
87
100
  client: AuthenticatedClient,
101
+ token: Union[None, Unset, str] = UNSET,
88
102
  authorization: Union[None, Unset, str] = UNSET,
89
103
  ) -> Response[
90
104
  Union[
@@ -108,6 +122,7 @@ def sync_detailed(
108
122
 
109
123
  Args:
110
124
  graph_id (str): Graph database identifier
125
+ token (Union[None, Unset, str]): JWT token for SSE authentication
111
126
  authorization (Union[None, Unset, str]):
112
127
 
113
128
  Raises:
@@ -120,6 +135,7 @@ def sync_detailed(
120
135
 
121
136
  kwargs = _get_kwargs(
122
137
  graph_id=graph_id,
138
+ token=token,
123
139
  authorization=authorization,
124
140
  )
125
141
 
@@ -134,6 +150,7 @@ def sync(
134
150
  graph_id: str,
135
151
  *,
136
152
  client: AuthenticatedClient,
153
+ token: Union[None, Unset, str] = UNSET,
137
154
  authorization: Union[None, Unset, str] = UNSET,
138
155
  ) -> Optional[
139
156
  Union[
@@ -157,6 +174,7 @@ def sync(
157
174
 
158
175
  Args:
159
176
  graph_id (str): Graph database identifier
177
+ token (Union[None, Unset, str]): JWT token for SSE authentication
160
178
  authorization (Union[None, Unset, str]):
161
179
 
162
180
  Raises:
@@ -170,6 +188,7 @@ def sync(
170
188
  return sync_detailed(
171
189
  graph_id=graph_id,
172
190
  client=client,
191
+ token=token,
173
192
  authorization=authorization,
174
193
  ).parsed
175
194
 
@@ -178,6 +197,7 @@ async def asyncio_detailed(
178
197
  graph_id: str,
179
198
  *,
180
199
  client: AuthenticatedClient,
200
+ token: Union[None, Unset, str] = UNSET,
181
201
  authorization: Union[None, Unset, str] = UNSET,
182
202
  ) -> Response[
183
203
  Union[
@@ -201,6 +221,7 @@ async def asyncio_detailed(
201
221
 
202
222
  Args:
203
223
  graph_id (str): Graph database identifier
224
+ token (Union[None, Unset, str]): JWT token for SSE authentication
204
225
  authorization (Union[None, Unset, str]):
205
226
 
206
227
  Raises:
@@ -213,6 +234,7 @@ async def asyncio_detailed(
213
234
 
214
235
  kwargs = _get_kwargs(
215
236
  graph_id=graph_id,
237
+ token=token,
216
238
  authorization=authorization,
217
239
  )
218
240
 
@@ -225,6 +247,7 @@ async def asyncio(
225
247
  graph_id: str,
226
248
  *,
227
249
  client: AuthenticatedClient,
250
+ token: Union[None, Unset, str] = UNSET,
228
251
  authorization: Union[None, Unset, str] = UNSET,
229
252
  ) -> Optional[
230
253
  Union[
@@ -248,6 +271,7 @@ async def asyncio(
248
271
 
249
272
  Args:
250
273
  graph_id (str): Graph database identifier
274
+ token (Union[None, Unset, str]): JWT token for SSE authentication
251
275
  authorization (Union[None, Unset, str]):
252
276
 
253
277
  Raises:
@@ -262,6 +286,7 @@ async def asyncio(
262
286
  await asyncio_detailed(
263
287
  graph_id=graph_id,
264
288
  client=client,
289
+ token=token,
265
290
  authorization=authorization,
266
291
  )
267
292
  ).parsed
@@ -17,6 +17,7 @@ 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
22
  ) -> dict[str, Any]:
22
23
  headers: dict[str, Any] = {}
@@ -27,6 +28,13 @@ def _get_kwargs(
27
28
 
28
29
  params["months"] = months
29
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
+
30
38
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
31
39
 
32
40
  _kwargs: dict[str, Any] = {
@@ -98,6 +106,7 @@ def sync_detailed(
98
106
  *,
99
107
  client: AuthenticatedClient,
100
108
  months: Union[Unset, int] = 6,
109
+ token: Union[None, Unset, str] = UNSET,
101
110
  authorization: Union[None, Unset, str] = UNSET,
102
111
  ) -> Response[
103
112
  Union[
@@ -127,6 +136,7 @@ def sync_detailed(
127
136
  Args:
128
137
  graph_id (str): Graph database identifier
129
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
130
140
  authorization (Union[None, Unset, str]):
131
141
 
132
142
  Raises:
@@ -140,6 +150,7 @@ def sync_detailed(
140
150
  kwargs = _get_kwargs(
141
151
  graph_id=graph_id,
142
152
  months=months,
153
+ token=token,
143
154
  authorization=authorization,
144
155
  )
145
156
 
@@ -155,6 +166,7 @@ def sync(
155
166
  *,
156
167
  client: AuthenticatedClient,
157
168
  months: Union[Unset, int] = 6,
169
+ token: Union[None, Unset, str] = UNSET,
158
170
  authorization: Union[None, Unset, str] = UNSET,
159
171
  ) -> Optional[
160
172
  Union[
@@ -184,6 +196,7 @@ def sync(
184
196
  Args:
185
197
  graph_id (str): Graph database identifier
186
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
187
200
  authorization (Union[None, Unset, str]):
188
201
 
189
202
  Raises:
@@ -198,6 +211,7 @@ def sync(
198
211
  graph_id=graph_id,
199
212
  client=client,
200
213
  months=months,
214
+ token=token,
201
215
  authorization=authorization,
202
216
  ).parsed
203
217
 
@@ -207,6 +221,7 @@ async def asyncio_detailed(
207
221
  *,
208
222
  client: AuthenticatedClient,
209
223
  months: Union[Unset, int] = 6,
224
+ token: Union[None, Unset, str] = UNSET,
210
225
  authorization: Union[None, Unset, str] = UNSET,
211
226
  ) -> Response[
212
227
  Union[
@@ -236,6 +251,7 @@ async def asyncio_detailed(
236
251
  Args:
237
252
  graph_id (str): Graph database identifier
238
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
239
255
  authorization (Union[None, Unset, str]):
240
256
 
241
257
  Raises:
@@ -249,6 +265,7 @@ async def asyncio_detailed(
249
265
  kwargs = _get_kwargs(
250
266
  graph_id=graph_id,
251
267
  months=months,
268
+ token=token,
252
269
  authorization=authorization,
253
270
  )
254
271
 
@@ -262,6 +279,7 @@ async def asyncio(
262
279
  *,
263
280
  client: AuthenticatedClient,
264
281
  months: Union[Unset, int] = 6,
282
+ token: Union[None, Unset, str] = UNSET,
265
283
  authorization: Union[None, Unset, str] = UNSET,
266
284
  ) -> Optional[
267
285
  Union[
@@ -291,6 +309,7 @@ async def asyncio(
291
309
  Args:
292
310
  graph_id (str): Graph database identifier
293
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
294
313
  authorization (Union[None, Unset, str]):
295
314
 
296
315
  Raises:
@@ -306,6 +325,7 @@ async def asyncio(
306
325
  graph_id=graph_id,
307
326
  client=client,
308
327
  months=months,
328
+ token=token,
309
329
  authorization=authorization,
310
330
  )
311
331
  ).parsed
@@ -18,15 +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
23
  ) -> dict[str, Any]:
23
24
  headers: dict[str, Any] = {}
24
25
  if not isinstance(authorization, Unset):
25
26
  headers["authorization"] = authorization
26
27
 
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}
38
+
27
39
  _kwargs: dict[str, Any] = {
28
40
  "method": "get",
29
41
  "url": f"/v1/{graph_id}/billing/history/{year}/{month}",
42
+ "params": params,
30
43
  }
31
44
 
32
45
  _kwargs["headers"] = headers
@@ -93,6 +106,7 @@ def sync_detailed(
93
106
  month: int,
94
107
  *,
95
108
  client: AuthenticatedClient,
109
+ token: Union[None, Unset, str] = UNSET,
96
110
  authorization: Union[None, Unset, str] = UNSET,
97
111
  ) -> Response[
98
112
  Union[
@@ -118,6 +132,7 @@ def sync_detailed(
118
132
  graph_id (str): Graph database identifier
119
133
  year (int): Year (2024-2030)
120
134
  month (int): Month (1-12)
135
+ token (Union[None, Unset, str]): JWT token for SSE authentication
121
136
  authorization (Union[None, Unset, str]):
122
137
 
123
138
  Raises:
@@ -132,6 +147,7 @@ def sync_detailed(
132
147
  graph_id=graph_id,
133
148
  year=year,
134
149
  month=month,
150
+ token=token,
135
151
  authorization=authorization,
136
152
  )
137
153
 
@@ -148,6 +164,7 @@ def sync(
148
164
  month: int,
149
165
  *,
150
166
  client: AuthenticatedClient,
167
+ token: Union[None, Unset, str] = UNSET,
151
168
  authorization: Union[None, Unset, str] = UNSET,
152
169
  ) -> Optional[
153
170
  Union[
@@ -173,6 +190,7 @@ def sync(
173
190
  graph_id (str): Graph database identifier
174
191
  year (int): Year (2024-2030)
175
192
  month (int): Month (1-12)
193
+ token (Union[None, Unset, str]): JWT token for SSE authentication
176
194
  authorization (Union[None, Unset, str]):
177
195
 
178
196
  Raises:
@@ -188,6 +206,7 @@ def sync(
188
206
  year=year,
189
207
  month=month,
190
208
  client=client,
209
+ token=token,
191
210
  authorization=authorization,
192
211
  ).parsed
193
212
 
@@ -198,6 +217,7 @@ async def asyncio_detailed(
198
217
  month: int,
199
218
  *,
200
219
  client: AuthenticatedClient,
220
+ token: Union[None, Unset, str] = UNSET,
201
221
  authorization: Union[None, Unset, str] = UNSET,
202
222
  ) -> Response[
203
223
  Union[
@@ -223,6 +243,7 @@ async def asyncio_detailed(
223
243
  graph_id (str): Graph database identifier
224
244
  year (int): Year (2024-2030)
225
245
  month (int): Month (1-12)
246
+ token (Union[None, Unset, str]): JWT token for SSE authentication
226
247
  authorization (Union[None, Unset, str]):
227
248
 
228
249
  Raises:
@@ -237,6 +258,7 @@ async def asyncio_detailed(
237
258
  graph_id=graph_id,
238
259
  year=year,
239
260
  month=month,
261
+ token=token,
240
262
  authorization=authorization,
241
263
  )
242
264
 
@@ -251,6 +273,7 @@ async def asyncio(
251
273
  month: int,
252
274
  *,
253
275
  client: AuthenticatedClient,
276
+ token: Union[None, Unset, str] = UNSET,
254
277
  authorization: Union[None, Unset, str] = UNSET,
255
278
  ) -> Optional[
256
279
  Union[
@@ -276,6 +299,7 @@ async def asyncio(
276
299
  graph_id (str): Graph database identifier
277
300
  year (int): Year (2024-2030)
278
301
  month (int): Month (1-12)
302
+ token (Union[None, Unset, str]): JWT token for SSE authentication
279
303
  authorization (Union[None, Unset, str]):
280
304
 
281
305
  Raises:
@@ -292,6 +316,7 @@ async def asyncio(
292
316
  year=year,
293
317
  month=month,
294
318
  client=client,
319
+ token=token,
295
320
  authorization=authorization,
296
321
  )
297
322
  ).parsed
@@ -18,6 +18,7 @@ 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
23
  ) -> dict[str, Any]:
23
24
  headers: dict[str, Any] = {}
@@ -40,6 +41,13 @@ def _get_kwargs(
40
41
  json_month = month
41
42
  params["month"] = json_month
42
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
+
43
51
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
44
52
 
45
53
  _kwargs: dict[str, Any] = {
@@ -112,6 +120,7 @@ def sync_detailed(
112
120
  client: AuthenticatedClient,
113
121
  year: Union[None, Unset, int] = UNSET,
114
122
  month: Union[None, Unset, int] = UNSET,
123
+ token: Union[None, Unset, str] = UNSET,
115
124
  authorization: Union[None, Unset, str] = UNSET,
116
125
  ) -> Response[
117
126
  Union[
@@ -141,6 +150,7 @@ def sync_detailed(
141
150
  graph_id (str): Graph database identifier
142
151
  year (Union[None, Unset, int]): Year (defaults to current)
143
152
  month (Union[None, Unset, int]): Month (defaults to current)
153
+ token (Union[None, Unset, str]): JWT token for SSE authentication
144
154
  authorization (Union[None, Unset, str]):
145
155
 
146
156
  Raises:
@@ -155,6 +165,7 @@ def sync_detailed(
155
165
  graph_id=graph_id,
156
166
  year=year,
157
167
  month=month,
168
+ token=token,
158
169
  authorization=authorization,
159
170
  )
160
171
 
@@ -171,6 +182,7 @@ def sync(
171
182
  client: AuthenticatedClient,
172
183
  year: Union[None, Unset, int] = UNSET,
173
184
  month: Union[None, Unset, int] = UNSET,
185
+ token: Union[None, Unset, str] = UNSET,
174
186
  authorization: Union[None, Unset, str] = UNSET,
175
187
  ) -> Optional[
176
188
  Union[
@@ -200,6 +212,7 @@ def sync(
200
212
  graph_id (str): Graph database identifier
201
213
  year (Union[None, Unset, int]): Year (defaults to current)
202
214
  month (Union[None, Unset, int]): Month (defaults to current)
215
+ token (Union[None, Unset, str]): JWT token for SSE authentication
203
216
  authorization (Union[None, Unset, str]):
204
217
 
205
218
  Raises:
@@ -215,6 +228,7 @@ def sync(
215
228
  client=client,
216
229
  year=year,
217
230
  month=month,
231
+ token=token,
218
232
  authorization=authorization,
219
233
  ).parsed
220
234
 
@@ -225,6 +239,7 @@ async def asyncio_detailed(
225
239
  client: AuthenticatedClient,
226
240
  year: Union[None, Unset, int] = UNSET,
227
241
  month: Union[None, Unset, int] = UNSET,
242
+ token: Union[None, Unset, str] = UNSET,
228
243
  authorization: Union[None, Unset, str] = UNSET,
229
244
  ) -> Response[
230
245
  Union[
@@ -254,6 +269,7 @@ async def asyncio_detailed(
254
269
  graph_id (str): Graph database identifier
255
270
  year (Union[None, Unset, int]): Year (defaults to current)
256
271
  month (Union[None, Unset, int]): Month (defaults to current)
272
+ token (Union[None, Unset, str]): JWT token for SSE authentication
257
273
  authorization (Union[None, Unset, str]):
258
274
 
259
275
  Raises:
@@ -268,6 +284,7 @@ async def asyncio_detailed(
268
284
  graph_id=graph_id,
269
285
  year=year,
270
286
  month=month,
287
+ token=token,
271
288
  authorization=authorization,
272
289
  )
273
290
 
@@ -282,6 +299,7 @@ async def asyncio(
282
299
  client: AuthenticatedClient,
283
300
  year: Union[None, Unset, int] = UNSET,
284
301
  month: Union[None, Unset, int] = UNSET,
302
+ token: Union[None, Unset, str] = UNSET,
285
303
  authorization: Union[None, Unset, str] = UNSET,
286
304
  ) -> Optional[
287
305
  Union[
@@ -311,6 +329,7 @@ async def asyncio(
311
329
  graph_id (str): Graph database identifier
312
330
  year (Union[None, Unset, int]): Year (defaults to current)
313
331
  month (Union[None, Unset, int]): Month (defaults to current)
332
+ token (Union[None, Unset, str]): JWT token for SSE authentication
314
333
  authorization (Union[None, Unset, str]):
315
334
 
316
335
  Raises:
@@ -327,6 +346,7 @@ async def asyncio(
327
346
  client=client,
328
347
  year=year,
329
348
  month=month,
349
+ token=token,
330
350
  authorization=authorization,
331
351
  )
332
352
  ).parsed
@@ -18,6 +18,7 @@ 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
23
  ) -> dict[str, Any]:
23
24
  headers: dict[str, Any] = {}
@@ -35,6 +36,13 @@ def _get_kwargs(
35
36
  json_base_cost = base_cost
36
37
  params["base_cost"] = json_base_cost
37
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
+
38
46
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
39
47
 
40
48
  _kwargs: dict[str, Any] = {
@@ -103,6 +111,7 @@ def sync_detailed(
103
111
  client: AuthenticatedClient,
104
112
  operation_type: str,
105
113
  base_cost: Union[None, Unset, float, str] = UNSET,
114
+ token: Union[None, Unset, str] = UNSET,
106
115
  authorization: Union[None, Unset, str] = UNSET,
107
116
  ) -> Response[
108
117
  Union[
@@ -128,6 +137,7 @@ def sync_detailed(
128
137
  operation_type (str): Type of operation to check
129
138
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
130
139
  provided)
140
+ token (Union[None, Unset, str]): JWT token for SSE authentication
131
141
  authorization (Union[None, Unset, str]):
132
142
 
133
143
  Raises:
@@ -142,6 +152,7 @@ def sync_detailed(
142
152
  graph_id=graph_id,
143
153
  operation_type=operation_type,
144
154
  base_cost=base_cost,
155
+ token=token,
145
156
  authorization=authorization,
146
157
  )
147
158
 
@@ -158,6 +169,7 @@ def sync(
158
169
  client: AuthenticatedClient,
159
170
  operation_type: str,
160
171
  base_cost: Union[None, Unset, float, str] = UNSET,
172
+ token: Union[None, Unset, str] = UNSET,
161
173
  authorization: Union[None, Unset, str] = UNSET,
162
174
  ) -> Optional[
163
175
  Union[
@@ -183,6 +195,7 @@ def sync(
183
195
  operation_type (str): Type of operation to check
184
196
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
185
197
  provided)
198
+ token (Union[None, Unset, str]): JWT token for SSE authentication
186
199
  authorization (Union[None, Unset, str]):
187
200
 
188
201
  Raises:
@@ -198,6 +211,7 @@ def sync(
198
211
  client=client,
199
212
  operation_type=operation_type,
200
213
  base_cost=base_cost,
214
+ token=token,
201
215
  authorization=authorization,
202
216
  ).parsed
203
217
 
@@ -208,6 +222,7 @@ async def asyncio_detailed(
208
222
  client: AuthenticatedClient,
209
223
  operation_type: str,
210
224
  base_cost: Union[None, Unset, float, str] = UNSET,
225
+ token: Union[None, Unset, str] = UNSET,
211
226
  authorization: Union[None, Unset, str] = UNSET,
212
227
  ) -> Response[
213
228
  Union[
@@ -233,6 +248,7 @@ async def asyncio_detailed(
233
248
  operation_type (str): Type of operation to check
234
249
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
235
250
  provided)
251
+ token (Union[None, Unset, str]): JWT token for SSE authentication
236
252
  authorization (Union[None, Unset, str]):
237
253
 
238
254
  Raises:
@@ -247,6 +263,7 @@ async def asyncio_detailed(
247
263
  graph_id=graph_id,
248
264
  operation_type=operation_type,
249
265
  base_cost=base_cost,
266
+ token=token,
250
267
  authorization=authorization,
251
268
  )
252
269
 
@@ -261,6 +278,7 @@ async def asyncio(
261
278
  client: AuthenticatedClient,
262
279
  operation_type: str,
263
280
  base_cost: Union[None, Unset, float, str] = UNSET,
281
+ token: Union[None, Unset, str] = UNSET,
264
282
  authorization: Union[None, Unset, str] = UNSET,
265
283
  ) -> Optional[
266
284
  Union[
@@ -286,6 +304,7 @@ async def asyncio(
286
304
  operation_type (str): Type of operation to check
287
305
  base_cost (Union[None, Unset, float, str]): Custom base cost (uses default if not
288
306
  provided)
307
+ token (Union[None, Unset, str]): JWT token for SSE authentication
289
308
  authorization (Union[None, Unset, str]):
290
309
 
291
310
  Raises:
@@ -302,6 +321,7 @@ async def asyncio(
302
321
  client=client,
303
322
  operation_type=operation_type,
304
323
  base_cost=base_cost,
324
+ token=token,
305
325
  authorization=authorization,
306
326
  )
307
327
  ).parsed
@@ -14,15 +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
19
  ) -> dict[str, Any]:
19
20
  headers: dict[str, Any] = {}
20
21
  if not isinstance(authorization, Unset):
21
22
  headers["authorization"] = authorization
22
23
 
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}
34
+
23
35
  _kwargs: dict[str, Any] = {
24
36
  "method": "get",
25
37
  "url": f"/v1/{graph_id}/credits/storage/limits",
38
+ "params": params,
26
39
  }
27
40
 
28
41
  _kwargs["headers"] = headers
@@ -73,6 +86,7 @@ def sync_detailed(
73
86
  graph_id: str,
74
87
  *,
75
88
  client: AuthenticatedClient,
89
+ token: Union[None, Unset, str] = UNSET,
76
90
  authorization: Union[None, Unset, str] = UNSET,
77
91
  ) -> Response[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]:
78
92
  """Check Storage Limits
@@ -90,6 +104,7 @@ def sync_detailed(
90
104
 
91
105
  Args:
92
106
  graph_id (str): Graph database identifier
107
+ token (Union[None, Unset, str]): JWT token for SSE authentication
93
108
  authorization (Union[None, Unset, str]):
94
109
 
95
110
  Raises:
@@ -102,6 +117,7 @@ def sync_detailed(
102
117
 
103
118
  kwargs = _get_kwargs(
104
119
  graph_id=graph_id,
120
+ token=token,
105
121
  authorization=authorization,
106
122
  )
107
123
 
@@ -116,6 +132,7 @@ def sync(
116
132
  graph_id: str,
117
133
  *,
118
134
  client: AuthenticatedClient,
135
+ token: Union[None, Unset, str] = UNSET,
119
136
  authorization: Union[None, Unset, str] = UNSET,
120
137
  ) -> Optional[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]:
121
138
  """Check Storage Limits
@@ -133,6 +150,7 @@ def sync(
133
150
 
134
151
  Args:
135
152
  graph_id (str): Graph database identifier
153
+ token (Union[None, Unset, str]): JWT token for SSE authentication
136
154
  authorization (Union[None, Unset, str]):
137
155
 
138
156
  Raises:
@@ -146,6 +164,7 @@ def sync(
146
164
  return sync_detailed(
147
165
  graph_id=graph_id,
148
166
  client=client,
167
+ token=token,
149
168
  authorization=authorization,
150
169
  ).parsed
151
170
 
@@ -154,6 +173,7 @@ async def asyncio_detailed(
154
173
  graph_id: str,
155
174
  *,
156
175
  client: AuthenticatedClient,
176
+ token: Union[None, Unset, str] = UNSET,
157
177
  authorization: Union[None, Unset, str] = UNSET,
158
178
  ) -> Response[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]:
159
179
  """Check Storage Limits
@@ -171,6 +191,7 @@ async def asyncio_detailed(
171
191
 
172
192
  Args:
173
193
  graph_id (str): Graph database identifier
194
+ token (Union[None, Unset, str]): JWT token for SSE authentication
174
195
  authorization (Union[None, Unset, str]):
175
196
 
176
197
  Raises:
@@ -183,6 +204,7 @@ async def asyncio_detailed(
183
204
 
184
205
  kwargs = _get_kwargs(
185
206
  graph_id=graph_id,
207
+ token=token,
186
208
  authorization=authorization,
187
209
  )
188
210
 
@@ -195,6 +217,7 @@ async def asyncio(
195
217
  graph_id: str,
196
218
  *,
197
219
  client: AuthenticatedClient,
220
+ token: Union[None, Unset, str] = UNSET,
198
221
  authorization: Union[None, Unset, str] = UNSET,
199
222
  ) -> Optional[Union[ErrorResponse, HTTPValidationError, StorageLimitResponse]]:
200
223
  """Check Storage Limits
@@ -212,6 +235,7 @@ async def asyncio(
212
235
 
213
236
  Args:
214
237
  graph_id (str): Graph database identifier
238
+ token (Union[None, Unset, str]): JWT token for SSE authentication
215
239
  authorization (Union[None, Unset, str]):
216
240
 
217
241
  Raises:
@@ -226,6 +250,7 @@ async def asyncio(
226
250
  await asyncio_detailed(
227
251
  graph_id=graph_id,
228
252
  client=client,
253
+ token=token,
229
254
  authorization=authorization,
230
255
  )
231
256
  ).parsed