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