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