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
robosystems_client/api/graph_credits/check_storage_limits.py CHANGED
@@ -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
- "url": f"/v1/{graph_id}/credits/storage/limits",
37
+ "url": f"/v1/graphs/{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
robosystems_client/api/graph_credits/get_credit_summary.py CHANGED
@@ -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
- "url": f"/v1/{graph_id}/credits/summary",
37
+ "url": f"/v1/graphs/{graph_id}/credits/summary",
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[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]:
78
92
  """Get Credit Summary
@@ -89,6 +103,7 @@ def sync_detailed(
89
103
 
90
104
  Args:
91
105
  graph_id (str): Graph database identifier
106
+ token (Union[None, Unset, str]): JWT token for SSE authentication
92
107
  authorization (Union[None, Unset, str]):
93
108
 
94
109
  Raises:
@@ -101,6 +116,7 @@ def sync_detailed(
101
116
 
102
117
  kwargs = _get_kwargs(
103
118
  graph_id=graph_id,
119
+ token=token,
104
120
  authorization=authorization,
105
121
  )
106
122
 
@@ -115,6 +131,7 @@ def sync(
115
131
  graph_id: str,
116
132
  *,
117
133
  client: AuthenticatedClient,
134
+ token: Union[None, Unset, str] = UNSET,
118
135
  authorization: Union[None, Unset, str] = UNSET,
119
136
  ) -> Optional[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]:
120
137
  """Get Credit Summary
@@ -131,6 +148,7 @@ def sync(
131
148
 
132
149
  Args:
133
150
  graph_id (str): Graph database identifier
151
+ token (Union[None, Unset, str]): JWT token for SSE authentication
134
152
  authorization (Union[None, Unset, str]):
135
153
 
136
154
  Raises:
@@ -144,6 +162,7 @@ def sync(
144
162
  return sync_detailed(
145
163
  graph_id=graph_id,
146
164
  client=client,
165
+ token=token,
147
166
  authorization=authorization,
148
167
  ).parsed
149
168
 
@@ -152,6 +171,7 @@ async def asyncio_detailed(
152
171
  graph_id: str,
153
172
  *,
154
173
  client: AuthenticatedClient,
174
+ token: Union[None, Unset, str] = UNSET,
155
175
  authorization: Union[None, Unset, str] = UNSET,
156
176
  ) -> Response[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]:
157
177
  """Get Credit Summary
@@ -168,6 +188,7 @@ async def asyncio_detailed(
168
188
 
169
189
  Args:
170
190
  graph_id (str): Graph database identifier
191
+ token (Union[None, Unset, str]): JWT token for SSE authentication
171
192
  authorization (Union[None, Unset, str]):
172
193
 
173
194
  Raises:
@@ -180,6 +201,7 @@ async def asyncio_detailed(
180
201
 
181
202
  kwargs = _get_kwargs(
182
203
  graph_id=graph_id,
204
+ token=token,
183
205
  authorization=authorization,
184
206
  )
185
207
 
@@ -192,6 +214,7 @@ async def asyncio(
192
214
  graph_id: str,
193
215
  *,
194
216
  client: AuthenticatedClient,
217
+ token: Union[None, Unset, str] = UNSET,
195
218
  authorization: Union[None, Unset, str] = UNSET,
196
219
  ) -> Optional[Union[CreditSummaryResponse, ErrorResponse, HTTPValidationError]]:
197
220
  """Get Credit Summary
@@ -208,6 +231,7 @@ async def asyncio(
208
231
 
209
232
  Args:
210
233
  graph_id (str): Graph database identifier
234
+ token (Union[None, Unset, str]): JWT token for SSE authentication
211
235
  authorization (Union[None, Unset, str]):
212
236
 
213
237
  Raises:
@@ -222,6 +246,7 @@ async def asyncio(
222
246
  await asyncio_detailed(
223
247
  graph_id=graph_id,
224
248
  client=client,
249
+ token=token,
225
250
  authorization=authorization,
226
251
  )
227
252
  ).parsed
robosystems_client/api/graph_credits/get_storage_usage.py CHANGED
@@ -17,6 +17,7 @@ def _get_kwargs(
17
17
  graph_id: str,
18
18
  *,
19
19
  days: Union[Unset, int] = 30,
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,11 +28,18 @@ def _get_kwargs(
27
28
 
28
29
  params["days"] = days
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] = {
33
41
  "method": "get",
34
- "url": f"/v1/{graph_id}/credits/storage/usage",
42
+ "url": f"/v1/graphs/{graph_id}/credits/storage/usage",
35
43
  "params": params,
36
44
  }
37
45
 
@@ -84,6 +92,7 @@ def sync_detailed(
84
92
  *,
85
93
  client: AuthenticatedClient,
86
94
  days: Union[Unset, int] = 30,
95
+ token: Union[None, Unset, str] = UNSET,
87
96
  authorization: Union[None, Unset, str] = UNSET,
88
97
  ) -> Response[
89
98
  Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError]
@@ -104,6 +113,7 @@ def sync_detailed(
104
113
  Args:
105
114
  graph_id (str): Graph database identifier
106
115
  days (Union[Unset, int]): Number of days of history to return Default: 30.
116
+ token (Union[None, Unset, str]): JWT token for SSE authentication
107
117
  authorization (Union[None, Unset, str]):
108
118
 
109
119
  Raises:
@@ -117,6 +127,7 @@ def sync_detailed(
117
127
  kwargs = _get_kwargs(
118
128
  graph_id=graph_id,
119
129
  days=days,
130
+ token=token,
120
131
  authorization=authorization,
121
132
  )
122
133
 
@@ -132,6 +143,7 @@ def sync(
132
143
  *,
133
144
  client: AuthenticatedClient,
134
145
  days: Union[Unset, int] = 30,
146
+ token: Union[None, Unset, str] = UNSET,
135
147
  authorization: Union[None, Unset, str] = UNSET,
136
148
  ) -> Optional[
137
149
  Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError]
@@ -152,6 +164,7 @@ def sync(
152
164
  Args:
153
165
  graph_id (str): Graph database identifier
154
166
  days (Union[Unset, int]): Number of days of history to return Default: 30.
167
+ token (Union[None, Unset, str]): JWT token for SSE authentication
155
168
  authorization (Union[None, Unset, str]):
156
169
 
157
170
  Raises:
@@ -166,6 +179,7 @@ def sync(
166
179
  graph_id=graph_id,
167
180
  client=client,
168
181
  days=days,
182
+ token=token,
169
183
  authorization=authorization,
170
184
  ).parsed
171
185
 
@@ -175,6 +189,7 @@ async def asyncio_detailed(
175
189
  *,
176
190
  client: AuthenticatedClient,
177
191
  days: Union[Unset, int] = 30,
192
+ token: Union[None, Unset, str] = UNSET,
178
193
  authorization: Union[None, Unset, str] = UNSET,
179
194
  ) -> Response[
180
195
  Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError]
@@ -195,6 +210,7 @@ async def asyncio_detailed(
195
210
  Args:
196
211
  graph_id (str): Graph database identifier
197
212
  days (Union[Unset, int]): Number of days of history to return Default: 30.
213
+ token (Union[None, Unset, str]): JWT token for SSE authentication
198
214
  authorization (Union[None, Unset, str]):
199
215
 
200
216
  Raises:
@@ -208,6 +224,7 @@ async def asyncio_detailed(
208
224
  kwargs = _get_kwargs(
209
225
  graph_id=graph_id,
210
226
  days=days,
227
+ token=token,
211
228
  authorization=authorization,
212
229
  )
213
230
 
@@ -221,6 +238,7 @@ async def asyncio(
221
238
  *,
222
239
  client: AuthenticatedClient,
223
240
  days: Union[Unset, int] = 30,
241
+ token: Union[None, Unset, str] = UNSET,
224
242
  authorization: Union[None, Unset, str] = UNSET,
225
243
  ) -> Optional[
226
244
  Union[ErrorResponse, GetStorageUsageResponseGetstorageusage, HTTPValidationError]
@@ -241,6 +259,7 @@ async def asyncio(
241
259
  Args:
242
260
  graph_id (str): Graph database identifier
243
261
  days (Union[Unset, int]): Number of days of history to return Default: 30.
262
+ token (Union[None, Unset, str]): JWT token for SSE authentication
244
263
  authorization (Union[None, Unset, str]):
245
264
 
246
265
  Raises:
@@ -256,6 +275,7 @@ async def asyncio(
256
275
  graph_id=graph_id,
257
276
  client=client,
258
277
  days=days,
278
+ token=token,
259
279
  authorization=authorization,
260
280
  )
261
281
  ).parsed
robosystems_client/api/graph_credits/list_credit_transactions.py CHANGED
@@ -20,6 +20,7 @@ def _get_kwargs(
20
20
  end_date: Union[None, Unset, str] = UNSET,
21
21
  limit: Union[Unset, int] = 100,
22
22
  offset: Union[Unset, int] = 0,
23
+ token: Union[None, Unset, str] = UNSET,
23
24
  authorization: Union[None, Unset, str] = UNSET,
24
25
  ) -> dict[str, Any]:
25
26
  headers: dict[str, Any] = {}
@@ -60,11 +61,18 @@ def _get_kwargs(
60
61
 
61
62
  params["offset"] = offset
62
63
 
64
+ json_token: Union[None, Unset, str]
65
+ if isinstance(token, Unset):
66
+ json_token = UNSET
67
+ else:
68
+ json_token = token
69
+ params["token"] = json_token
70
+
63
71
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
64
72
 
65
73
  _kwargs: dict[str, Any] = {
66
74
  "method": "get",
67
- "url": f"/v1/{graph_id}/credits/transactions",
75
+ "url": f"/v1/graphs/{graph_id}/credits/transactions",
68
76
  "params": params,
69
77
  }
70
78
 
@@ -122,6 +130,7 @@ def sync_detailed(
122
130
  end_date: Union[None, Unset, str] = UNSET,
123
131
  limit: Union[Unset, int] = 100,
124
132
  offset: Union[Unset, int] = 0,
133
+ token: Union[None, Unset, str] = UNSET,
125
134
  authorization: Union[None, Unset, str] = UNSET,
126
135
  ) -> Response[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]:
127
136
  """List Credit Transactions
@@ -152,6 +161,7 @@ def sync_detailed(
152
161
  end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD)
153
162
  limit (Union[Unset, int]): Maximum number of transactions to return Default: 100.
154
163
  offset (Union[Unset, int]): Number of transactions to skip Default: 0.
164
+ token (Union[None, Unset, str]): JWT token for SSE authentication
155
165
  authorization (Union[None, Unset, str]):
156
166
 
157
167
  Raises:
@@ -170,6 +180,7 @@ def sync_detailed(
170
180
  end_date=end_date,
171
181
  limit=limit,
172
182
  offset=offset,
183
+ token=token,
173
184
  authorization=authorization,
174
185
  )
175
186
 
@@ -190,6 +201,7 @@ def sync(
190
201
  end_date: Union[None, Unset, str] = UNSET,
191
202
  limit: Union[Unset, int] = 100,
192
203
  offset: Union[Unset, int] = 0,
204
+ token: Union[None, Unset, str] = UNSET,
193
205
  authorization: Union[None, Unset, str] = UNSET,
194
206
  ) -> Optional[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]:
195
207
  """List Credit Transactions
@@ -220,6 +232,7 @@ def sync(
220
232
  end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD)
221
233
  limit (Union[Unset, int]): Maximum number of transactions to return Default: 100.
222
234
  offset (Union[Unset, int]): Number of transactions to skip Default: 0.
235
+ token (Union[None, Unset, str]): JWT token for SSE authentication
223
236
  authorization (Union[None, Unset, str]):
224
237
 
225
238
  Raises:
@@ -239,6 +252,7 @@ def sync(
239
252
  end_date=end_date,
240
253
  limit=limit,
241
254
  offset=offset,
255
+ token=token,
242
256
  authorization=authorization,
243
257
  ).parsed
244
258
 
@@ -253,6 +267,7 @@ async def asyncio_detailed(
253
267
  end_date: Union[None, Unset, str] = UNSET,
254
268
  limit: Union[Unset, int] = 100,
255
269
  offset: Union[Unset, int] = 0,
270
+ token: Union[None, Unset, str] = UNSET,
256
271
  authorization: Union[None, Unset, str] = UNSET,
257
272
  ) -> Response[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]:
258
273
  """List Credit Transactions
@@ -283,6 +298,7 @@ async def asyncio_detailed(
283
298
  end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD)
284
299
  limit (Union[Unset, int]): Maximum number of transactions to return Default: 100.
285
300
  offset (Union[Unset, int]): Number of transactions to skip Default: 0.
301
+ token (Union[None, Unset, str]): JWT token for SSE authentication
286
302
  authorization (Union[None, Unset, str]):
287
303
 
288
304
  Raises:
@@ -301,6 +317,7 @@ async def asyncio_detailed(
301
317
  end_date=end_date,
302
318
  limit=limit,
303
319
  offset=offset,
320
+ token=token,
304
321
  authorization=authorization,
305
322
  )
306
323
 
@@ -319,6 +336,7 @@ async def asyncio(
319
336
  end_date: Union[None, Unset, str] = UNSET,
320
337
  limit: Union[Unset, int] = 100,
321
338
  offset: Union[Unset, int] = 0,
339
+ token: Union[None, Unset, str] = UNSET,
322
340
  authorization: Union[None, Unset, str] = UNSET,
323
341
  ) -> Optional[Union[DetailedTransactionsResponse, ErrorResponse, HTTPValidationError]]:
324
342
  """List Credit Transactions
@@ -349,6 +367,7 @@ async def asyncio(
349
367
  end_date (Union[None, Unset, str]): End date for filtering (ISO format: YYYY-MM-DD)
350
368
  limit (Union[Unset, int]): Maximum number of transactions to return Default: 100.
351
369
  offset (Union[Unset, int]): Number of transactions to skip Default: 0.
370
+ token (Union[None, Unset, str]): JWT token for SSE authentication
352
371
  authorization (Union[None, Unset, str]):
353
372
 
354
373
  Raises:
@@ -369,6 +388,7 @@ async def asyncio(
369
388
  end_date=end_date,
370
389
  limit=limit,
371
390
  offset=offset,
391
+ token=token,
372
392
  authorization=authorization,
373
393
  )
374
394
  ).parsed
robosystems_client/api/graph_health/get_database_health.py CHANGED
@@ -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}/health",
36
+ "url": f"/v1/graphs/{graph_id}/health",
37
+ "params": params,
25
38
  }
26
39
 
27
40
  _kwargs["headers"] = headers
@@ -69,6 +82,7 @@ def sync_detailed(
69
82
  graph_id: str,
70
83
  *,
71
84
  client: AuthenticatedClient,
85
+ token: Union[None, Unset, str] = UNSET,
72
86
  authorization: Union[None, Unset, str] = UNSET,
73
87
  ) -> Response[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
74
88
  """Database Health Check
@@ -93,6 +107,7 @@ def sync_detailed(
93
107
 
94
108
  Args:
95
109
  graph_id (str): Graph database identifier
110
+ token (Union[None, Unset, str]): JWT token for SSE authentication
96
111
  authorization (Union[None, Unset, str]):
97
112
 
98
113
  Raises:
@@ -105,6 +120,7 @@ def sync_detailed(
105
120
 
106
121
  kwargs = _get_kwargs(
107
122
  graph_id=graph_id,
123
+ token=token,
108
124
  authorization=authorization,
109
125
  )
110
126
 
@@ -119,6 +135,7 @@ def sync(
119
135
  graph_id: str,
120
136
  *,
121
137
  client: AuthenticatedClient,
138
+ token: Union[None, Unset, str] = UNSET,
122
139
  authorization: Union[None, Unset, str] = UNSET,
123
140
  ) -> Optional[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
124
141
  """Database Health Check
@@ -143,6 +160,7 @@ def sync(
143
160
 
144
161
  Args:
145
162
  graph_id (str): Graph database identifier
163
+ token (Union[None, Unset, str]): JWT token for SSE authentication
146
164
  authorization (Union[None, Unset, str]):
147
165
 
148
166
  Raises:
@@ -156,6 +174,7 @@ def sync(
156
174
  return sync_detailed(
157
175
  graph_id=graph_id,
158
176
  client=client,
177
+ token=token,
159
178
  authorization=authorization,
160
179
  ).parsed
161
180
 
@@ -164,6 +183,7 @@ async def asyncio_detailed(
164
183
  graph_id: str,
165
184
  *,
166
185
  client: AuthenticatedClient,
186
+ token: Union[None, Unset, str] = UNSET,
167
187
  authorization: Union[None, Unset, str] = UNSET,
168
188
  ) -> Response[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
169
189
  """Database Health Check
@@ -188,6 +208,7 @@ async def asyncio_detailed(
188
208
 
189
209
  Args:
190
210
  graph_id (str): Graph database identifier
211
+ token (Union[None, Unset, str]): JWT token for SSE authentication
191
212
  authorization (Union[None, Unset, str]):
192
213
 
193
214
  Raises:
@@ -200,6 +221,7 @@ async def asyncio_detailed(
200
221
 
201
222
  kwargs = _get_kwargs(
202
223
  graph_id=graph_id,
224
+ token=token,
203
225
  authorization=authorization,
204
226
  )
205
227
 
@@ -212,6 +234,7 @@ async def asyncio(
212
234
  graph_id: str,
213
235
  *,
214
236
  client: AuthenticatedClient,
237
+ token: Union[None, Unset, str] = UNSET,
215
238
  authorization: Union[None, Unset, str] = UNSET,
216
239
  ) -> Optional[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
217
240
  """Database Health Check
@@ -236,6 +259,7 @@ async def asyncio(
236
259
 
237
260
  Args:
238
261
  graph_id (str): Graph database identifier
262
+ token (Union[None, Unset, str]): JWT token for SSE authentication
239
263
  authorization (Union[None, Unset, str]):
240
264
 
241
265
  Raises:
@@ -250,6 +274,7 @@ async def asyncio(
250
274
  await asyncio_detailed(
251
275
  graph_id=graph_id,
252
276
  client=client,
277
+ token=token,
253
278
  authorization=authorization,
254
279
  )
255
280
  ).parsed