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/backup/export_backup.py CHANGED
@@ -13,21 +13,28 @@ def _get_kwargs(
13
13
  graph_id: str,
14
14
  backup_id: str,
15
15
  *,
16
+ token: Union[None, Unset, str] = UNSET,
16
17
  authorization: Union[None, Unset, str] = UNSET,
17
- auth_token: Union[None, Unset, str] = UNSET,
18
18
  ) -> dict[str, Any]:
19
19
  headers: dict[str, Any] = {}
20
20
  if not isinstance(authorization, Unset):
21
21
  headers["authorization"] = authorization
22
22
 
23
- cookies = {}
24
- if auth_token is not UNSET:
25
- cookies["auth-token"] = auth_token
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}
26
33
 
27
34
  _kwargs: dict[str, Any] = {
28
35
  "method": "post",
29
36
  "url": f"/v1/{graph_id}/backups/{backup_id}/export",
30
- "cookies": cookies,
37
+ "params": params,
31
38
  }
32
39
 
33
40
  _kwargs["headers"] = headers
@@ -72,8 +79,8 @@ def sync_detailed(
72
79
  backup_id: str,
73
80
  *,
74
81
  client: AuthenticatedClient,
82
+ token: Union[None, Unset, str] = UNSET,
75
83
  authorization: Union[None, Unset, str] = UNSET,
76
- auth_token: Union[None, Unset, str] = UNSET,
77
84
  ) -> Response[Union[Any, HTTPValidationError]]:
78
85
  """Export Kuzu backup for download
79
86
 
@@ -82,8 +89,8 @@ def sync_detailed(
82
89
  Args:
83
90
  graph_id (str): Graph database identifier
84
91
  backup_id (str): Backup identifier
92
+ token (Union[None, Unset, str]): JWT token for SSE authentication
85
93
  authorization (Union[None, Unset, str]):
86
- auth_token (Union[None, Unset, str]):
87
94
 
88
95
  Raises:
89
96
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -96,8 +103,8 @@ def sync_detailed(
96
103
  kwargs = _get_kwargs(
97
104
  graph_id=graph_id,
98
105
  backup_id=backup_id,
106
+ token=token,
99
107
  authorization=authorization,
100
- auth_token=auth_token,
101
108
  )
102
109
 
103
110
  response = client.get_httpx_client().request(
@@ -112,8 +119,8 @@ def sync(
112
119
  backup_id: str,
113
120
  *,
114
121
  client: AuthenticatedClient,
122
+ token: Union[None, Unset, str] = UNSET,
115
123
  authorization: Union[None, Unset, str] = UNSET,
116
- auth_token: Union[None, Unset, str] = UNSET,
117
124
  ) -> Optional[Union[Any, HTTPValidationError]]:
118
125
  """Export Kuzu backup for download
119
126
 
@@ -122,8 +129,8 @@ def sync(
122
129
  Args:
123
130
  graph_id (str): Graph database identifier
124
131
  backup_id (str): Backup identifier
132
+ token (Union[None, Unset, str]): JWT token for SSE authentication
125
133
  authorization (Union[None, Unset, str]):
126
- auth_token (Union[None, Unset, str]):
127
134
 
128
135
  Raises:
129
136
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -137,8 +144,8 @@ def sync(
137
144
  graph_id=graph_id,
138
145
  backup_id=backup_id,
139
146
  client=client,
147
+ token=token,
140
148
  authorization=authorization,
141
- auth_token=auth_token,
142
149
  ).parsed
143
150
 
144
151
 
@@ -147,8 +154,8 @@ async def asyncio_detailed(
147
154
  backup_id: str,
148
155
  *,
149
156
  client: AuthenticatedClient,
157
+ token: Union[None, Unset, str] = UNSET,
150
158
  authorization: Union[None, Unset, str] = UNSET,
151
- auth_token: Union[None, Unset, str] = UNSET,
152
159
  ) -> Response[Union[Any, HTTPValidationError]]:
153
160
  """Export Kuzu backup for download
154
161
 
@@ -157,8 +164,8 @@ async def asyncio_detailed(
157
164
  Args:
158
165
  graph_id (str): Graph database identifier
159
166
  backup_id (str): Backup identifier
167
+ token (Union[None, Unset, str]): JWT token for SSE authentication
160
168
  authorization (Union[None, Unset, str]):
161
- auth_token (Union[None, Unset, str]):
162
169
 
163
170
  Raises:
164
171
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -171,8 +178,8 @@ async def asyncio_detailed(
171
178
  kwargs = _get_kwargs(
172
179
  graph_id=graph_id,
173
180
  backup_id=backup_id,
181
+ token=token,
174
182
  authorization=authorization,
175
- auth_token=auth_token,
176
183
  )
177
184
 
178
185
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -185,8 +192,8 @@ async def asyncio(
185
192
  backup_id: str,
186
193
  *,
187
194
  client: AuthenticatedClient,
195
+ token: Union[None, Unset, str] = UNSET,
188
196
  authorization: Union[None, Unset, str] = UNSET,
189
- auth_token: Union[None, Unset, str] = UNSET,
190
197
  ) -> Optional[Union[Any, HTTPValidationError]]:
191
198
  """Export Kuzu backup for download
192
199
 
@@ -195,8 +202,8 @@ async def asyncio(
195
202
  Args:
196
203
  graph_id (str): Graph database identifier
197
204
  backup_id (str): Backup identifier
205
+ token (Union[None, Unset, str]): JWT token for SSE authentication
198
206
  authorization (Union[None, Unset, str]):
199
- auth_token (Union[None, Unset, str]):
200
207
 
201
208
  Raises:
202
209
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -211,7 +218,7 @@ async def asyncio(
211
218
  graph_id=graph_id,
212
219
  backup_id=backup_id,
213
220
  client=client,
221
+ token=token,
214
222
  authorization=authorization,
215
- auth_token=auth_token,
216
223
  )
217
224
  ).parsed
robosystems_client/api/backup/get_backup_download_url.py CHANGED
@@ -17,28 +17,30 @@ def _get_kwargs(
17
17
  backup_id: str,
18
18
  *,
19
19
  expires_in: Union[Unset, int] = 3600,
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["expires_in"] = expires_in
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}/backups/{backup_id}/download",
40
43
  "params": params,
41
- "cookies": cookies,
42
44
  }
43
45
 
44
46
  _kwargs["headers"] = headers
@@ -94,8 +96,8 @@ def sync_detailed(
94
96
  *,
95
97
  client: AuthenticatedClient,
96
98
  expires_in: Union[Unset, int] = 3600,
99
+ token: Union[None, Unset, str] = UNSET,
97
100
  authorization: Union[None, Unset, str] = UNSET,
98
- auth_token: Union[None, Unset, str] = UNSET,
99
101
  ) -> Response[
100
102
  Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
101
103
  ]:
@@ -107,8 +109,8 @@ def sync_detailed(
107
109
  graph_id (str): Graph database identifier
108
110
  backup_id (str): Backup identifier
109
111
  expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
112
+ token (Union[None, Unset, str]): JWT token for SSE authentication
110
113
  authorization (Union[None, Unset, str]):
111
- auth_token (Union[None, Unset, str]):
112
114
 
113
115
  Raises:
114
116
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -122,8 +124,8 @@ def sync_detailed(
122
124
  graph_id=graph_id,
123
125
  backup_id=backup_id,
124
126
  expires_in=expires_in,
127
+ token=token,
125
128
  authorization=authorization,
126
- auth_token=auth_token,
127
129
  )
128
130
 
129
131
  response = client.get_httpx_client().request(
@@ -139,8 +141,8 @@ def sync(
139
141
  *,
140
142
  client: AuthenticatedClient,
141
143
  expires_in: Union[Unset, int] = 3600,
144
+ token: Union[None, Unset, str] = UNSET,
142
145
  authorization: Union[None, Unset, str] = UNSET,
143
- auth_token: Union[None, Unset, str] = UNSET,
144
146
  ) -> Optional[
145
147
  Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
146
148
  ]:
@@ -152,8 +154,8 @@ def sync(
152
154
  graph_id (str): Graph database identifier
153
155
  backup_id (str): Backup identifier
154
156
  expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
157
+ token (Union[None, Unset, str]): JWT token for SSE authentication
155
158
  authorization (Union[None, Unset, str]):
156
- auth_token (Union[None, Unset, str]):
157
159
 
158
160
  Raises:
159
161
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -168,8 +170,8 @@ def sync(
168
170
  backup_id=backup_id,
169
171
  client=client,
170
172
  expires_in=expires_in,
173
+ token=token,
171
174
  authorization=authorization,
172
- auth_token=auth_token,
173
175
  ).parsed
174
176
 
175
177
 
@@ -179,8 +181,8 @@ async def asyncio_detailed(
179
181
  *,
180
182
  client: AuthenticatedClient,
181
183
  expires_in: Union[Unset, int] = 3600,
184
+ token: Union[None, Unset, str] = UNSET,
182
185
  authorization: Union[None, Unset, str] = UNSET,
183
- auth_token: Union[None, Unset, str] = UNSET,
184
186
  ) -> Response[
185
187
  Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
186
188
  ]:
@@ -192,8 +194,8 @@ async def asyncio_detailed(
192
194
  graph_id (str): Graph database identifier
193
195
  backup_id (str): Backup identifier
194
196
  expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
197
+ token (Union[None, Unset, str]): JWT token for SSE authentication
195
198
  authorization (Union[None, Unset, str]):
196
- auth_token (Union[None, Unset, str]):
197
199
 
198
200
  Raises:
199
201
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -207,8 +209,8 @@ async def asyncio_detailed(
207
209
  graph_id=graph_id,
208
210
  backup_id=backup_id,
209
211
  expires_in=expires_in,
212
+ token=token,
210
213
  authorization=authorization,
211
- auth_token=auth_token,
212
214
  )
213
215
 
214
216
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -222,8 +224,8 @@ async def asyncio(
222
224
  *,
223
225
  client: AuthenticatedClient,
224
226
  expires_in: Union[Unset, int] = 3600,
227
+ token: Union[None, Unset, str] = UNSET,
225
228
  authorization: Union[None, Unset, str] = UNSET,
226
- auth_token: Union[None, Unset, str] = UNSET,
227
229
  ) -> Optional[
228
230
  Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
229
231
  ]:
@@ -235,8 +237,8 @@ async def asyncio(
235
237
  graph_id (str): Graph database identifier
236
238
  backup_id (str): Backup identifier
237
239
  expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
240
+ token (Union[None, Unset, str]): JWT token for SSE authentication
238
241
  authorization (Union[None, Unset, str]):
239
- auth_token (Union[None, Unset, str]):
240
242
 
241
243
  Raises:
242
244
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -252,7 +254,7 @@ async def asyncio(
252
254
  backup_id=backup_id,
253
255
  client=client,
254
256
  expires_in=expires_in,
257
+ token=token,
255
258
  authorization=authorization,
256
- auth_token=auth_token,
257
259
  )
258
260
  ).parsed
robosystems_client/api/backup/get_backup_stats.py CHANGED
@@ -13,21 +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
- auth_token: Union[None, Unset, str] = UNSET,
18
18
  ) -> dict[str, Any]:
19
19
  headers: dict[str, Any] = {}
20
20
  if not isinstance(authorization, Unset):
21
21
  headers["authorization"] = authorization
22
22
 
23
- cookies = {}
24
- if auth_token is not UNSET:
25
- cookies["auth-token"] = auth_token
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}
26
33
 
27
34
  _kwargs: dict[str, Any] = {
28
35
  "method": "get",
29
36
  "url": f"/v1/{graph_id}/backups/stats",
30
- "cookies": cookies,
37
+ "params": params,
31
38
  }
32
39
 
33
40
  _kwargs["headers"] = headers
@@ -66,8 +73,8 @@ def sync_detailed(
66
73
  graph_id: str,
67
74
  *,
68
75
  client: AuthenticatedClient,
76
+ token: Union[None, Unset, str] = UNSET,
69
77
  authorization: Union[None, Unset, str] = UNSET,
70
- auth_token: Union[None, Unset, str] = UNSET,
71
78
  ) -> Response[Union[BackupStatsResponse, HTTPValidationError]]:
72
79
  """Get backup statistics
73
80
 
@@ -75,8 +82,8 @@ def sync_detailed(
75
82
 
76
83
  Args:
77
84
  graph_id (str): Graph database identifier
85
+ token (Union[None, Unset, str]): JWT token for SSE authentication
78
86
  authorization (Union[None, Unset, str]):
79
- auth_token (Union[None, Unset, str]):
80
87
 
81
88
  Raises:
82
89
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -88,8 +95,8 @@ def sync_detailed(
88
95
 
89
96
  kwargs = _get_kwargs(
90
97
  graph_id=graph_id,
98
+ token=token,
91
99
  authorization=authorization,
92
- auth_token=auth_token,
93
100
  )
94
101
 
95
102
  response = client.get_httpx_client().request(
@@ -103,8 +110,8 @@ def sync(
103
110
  graph_id: str,
104
111
  *,
105
112
  client: AuthenticatedClient,
113
+ token: Union[None, Unset, str] = UNSET,
106
114
  authorization: Union[None, Unset, str] = UNSET,
107
- auth_token: Union[None, Unset, str] = UNSET,
108
115
  ) -> Optional[Union[BackupStatsResponse, HTTPValidationError]]:
109
116
  """Get backup statistics
110
117
 
@@ -112,8 +119,8 @@ def sync(
112
119
 
113
120
  Args:
114
121
  graph_id (str): Graph database identifier
122
+ token (Union[None, Unset, str]): JWT token for SSE authentication
115
123
  authorization (Union[None, Unset, str]):
116
- auth_token (Union[None, Unset, str]):
117
124
 
118
125
  Raises:
119
126
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -126,8 +133,8 @@ def sync(
126
133
  return sync_detailed(
127
134
  graph_id=graph_id,
128
135
  client=client,
136
+ token=token,
129
137
  authorization=authorization,
130
- auth_token=auth_token,
131
138
  ).parsed
132
139
 
133
140
 
@@ -135,8 +142,8 @@ async def asyncio_detailed(
135
142
  graph_id: str,
136
143
  *,
137
144
  client: AuthenticatedClient,
145
+ token: Union[None, Unset, str] = UNSET,
138
146
  authorization: Union[None, Unset, str] = UNSET,
139
- auth_token: Union[None, Unset, str] = UNSET,
140
147
  ) -> Response[Union[BackupStatsResponse, HTTPValidationError]]:
141
148
  """Get backup statistics
142
149
 
@@ -144,8 +151,8 @@ async def asyncio_detailed(
144
151
 
145
152
  Args:
146
153
  graph_id (str): Graph database identifier
154
+ token (Union[None, Unset, str]): JWT token for SSE authentication
147
155
  authorization (Union[None, Unset, str]):
148
- auth_token (Union[None, Unset, str]):
149
156
 
150
157
  Raises:
151
158
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -157,8 +164,8 @@ async def asyncio_detailed(
157
164
 
158
165
  kwargs = _get_kwargs(
159
166
  graph_id=graph_id,
167
+ token=token,
160
168
  authorization=authorization,
161
- auth_token=auth_token,
162
169
  )
163
170
 
164
171
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -170,8 +177,8 @@ async def asyncio(
170
177
  graph_id: str,
171
178
  *,
172
179
  client: AuthenticatedClient,
180
+ token: Union[None, Unset, str] = UNSET,
173
181
  authorization: Union[None, Unset, str] = UNSET,
174
- auth_token: Union[None, Unset, str] = UNSET,
175
182
  ) -> Optional[Union[BackupStatsResponse, HTTPValidationError]]:
176
183
  """Get backup statistics
177
184
 
@@ -179,8 +186,8 @@ async def asyncio(
179
186
 
180
187
  Args:
181
188
  graph_id (str): Graph database identifier
189
+ token (Union[None, Unset, str]): JWT token for SSE authentication
182
190
  authorization (Union[None, Unset, str]):
183
- auth_token (Union[None, Unset, str]):
184
191
 
185
192
  Raises:
186
193
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -194,7 +201,7 @@ async def asyncio(
194
201
  await asyncio_detailed(
195
202
  graph_id=graph_id,
196
203
  client=client,
204
+ token=token,
197
205
  authorization=authorization,
198
- auth_token=auth_token,
199
206
  )
200
207
  ).parsed
robosystems_client/api/backup/list_backups.py CHANGED
@@ -15,30 +15,32 @@ def _get_kwargs(
15
15
  *,
16
16
  limit: Union[Unset, int] = 50,
17
17
  offset: Union[Unset, int] = 0,
18
+ token: Union[None, Unset, str] = UNSET,
18
19
  authorization: Union[None, Unset, str] = UNSET,
19
- auth_token: Union[None, Unset, str] = UNSET,
20
20
  ) -> dict[str, Any]:
21
21
  headers: dict[str, Any] = {}
22
22
  if not isinstance(authorization, Unset):
23
23
  headers["authorization"] = authorization
24
24
 
25
- cookies = {}
26
- if auth_token is not UNSET:
27
- cookies["auth-token"] = auth_token
28
-
29
25
  params: dict[str, Any] = {}
30
26
 
31
27
  params["limit"] = limit
32
28
 
33
29
  params["offset"] = offset
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}/backups",
40
43
  "params": params,
41
- "cookies": cookies,
42
44
  }
43
45
 
44
46
  _kwargs["headers"] = headers
@@ -79,8 +81,8 @@ def sync_detailed(
79
81
  client: AuthenticatedClient,
80
82
  limit: Union[Unset, int] = 50,
81
83
  offset: Union[Unset, int] = 0,
84
+ token: Union[None, Unset, str] = UNSET,
82
85
  authorization: Union[None, Unset, str] = UNSET,
83
- auth_token: Union[None, Unset, str] = UNSET,
84
86
  ) -> Response[Union[BackupListResponse, HTTPValidationError]]:
85
87
  """List Kuzu graph backups
86
88
 
@@ -90,8 +92,8 @@ def sync_detailed(
90
92
  graph_id (str): Graph database identifier
91
93
  limit (Union[Unset, int]): Maximum number of backups to return Default: 50.
92
94
  offset (Union[Unset, int]): Number of backups to skip Default: 0.
95
+ token (Union[None, Unset, str]): JWT token for SSE authentication
93
96
  authorization (Union[None, Unset, str]):
94
- auth_token (Union[None, Unset, str]):
95
97
 
96
98
  Raises:
97
99
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -105,8 +107,8 @@ def sync_detailed(
105
107
  graph_id=graph_id,
106
108
  limit=limit,
107
109
  offset=offset,
110
+ token=token,
108
111
  authorization=authorization,
109
- auth_token=auth_token,
110
112
  )
111
113
 
112
114
  response = client.get_httpx_client().request(
@@ -122,8 +124,8 @@ def sync(
122
124
  client: AuthenticatedClient,
123
125
  limit: Union[Unset, int] = 50,
124
126
  offset: Union[Unset, int] = 0,
127
+ token: Union[None, Unset, str] = UNSET,
125
128
  authorization: Union[None, Unset, str] = UNSET,
126
- auth_token: Union[None, Unset, str] = UNSET,
127
129
  ) -> Optional[Union[BackupListResponse, HTTPValidationError]]:
128
130
  """List Kuzu graph backups
129
131
 
@@ -133,8 +135,8 @@ def sync(
133
135
  graph_id (str): Graph database identifier
134
136
  limit (Union[Unset, int]): Maximum number of backups to return Default: 50.
135
137
  offset (Union[Unset, int]): Number of backups to skip Default: 0.
138
+ token (Union[None, Unset, str]): JWT token for SSE authentication
136
139
  authorization (Union[None, Unset, str]):
137
- auth_token (Union[None, Unset, str]):
138
140
 
139
141
  Raises:
140
142
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -149,8 +151,8 @@ def sync(
149
151
  client=client,
150
152
  limit=limit,
151
153
  offset=offset,
154
+ token=token,
152
155
  authorization=authorization,
153
- auth_token=auth_token,
154
156
  ).parsed
155
157
 
156
158
 
@@ -160,8 +162,8 @@ async def asyncio_detailed(
160
162
  client: AuthenticatedClient,
161
163
  limit: Union[Unset, int] = 50,
162
164
  offset: Union[Unset, int] = 0,
165
+ token: Union[None, Unset, str] = UNSET,
163
166
  authorization: Union[None, Unset, str] = UNSET,
164
- auth_token: Union[None, Unset, str] = UNSET,
165
167
  ) -> Response[Union[BackupListResponse, HTTPValidationError]]:
166
168
  """List Kuzu graph backups
167
169
 
@@ -171,8 +173,8 @@ async def asyncio_detailed(
171
173
  graph_id (str): Graph database identifier
172
174
  limit (Union[Unset, int]): Maximum number of backups to return Default: 50.
173
175
  offset (Union[Unset, int]): Number of backups to skip Default: 0.
176
+ token (Union[None, Unset, str]): JWT token for SSE authentication
174
177
  authorization (Union[None, Unset, str]):
175
- auth_token (Union[None, Unset, str]):
176
178
 
177
179
  Raises:
178
180
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -186,8 +188,8 @@ async def asyncio_detailed(
186
188
  graph_id=graph_id,
187
189
  limit=limit,
188
190
  offset=offset,
191
+ token=token,
189
192
  authorization=authorization,
190
- auth_token=auth_token,
191
193
  )
192
194
 
193
195
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -201,8 +203,8 @@ async def asyncio(
201
203
  client: AuthenticatedClient,
202
204
  limit: Union[Unset, int] = 50,
203
205
  offset: Union[Unset, int] = 0,
206
+ token: Union[None, Unset, str] = UNSET,
204
207
  authorization: Union[None, Unset, str] = UNSET,
205
- auth_token: Union[None, Unset, str] = UNSET,
206
208
  ) -> Optional[Union[BackupListResponse, HTTPValidationError]]:
207
209
  """List Kuzu graph backups
208
210
 
@@ -212,8 +214,8 @@ async def asyncio(
212
214
  graph_id (str): Graph database identifier
213
215
  limit (Union[Unset, int]): Maximum number of backups to return Default: 50.
214
216
  offset (Union[Unset, int]): Number of backups to skip Default: 0.
217
+ token (Union[None, Unset, str]): JWT token for SSE authentication
215
218
  authorization (Union[None, Unset, str]):
216
- auth_token (Union[None, Unset, str]):
217
219
 
218
220
  Raises:
219
221
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -229,7 +231,7 @@ async def asyncio(
229
231
  client=client,
230
232
  limit=limit,
231
233
  offset=offset,
234
+ token=token,
232
235
  authorization=authorization,
233
- auth_token=auth_token,
234
236
  )
235
237
  ).parsed