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/connections/exchange_link_token.py CHANGED
@@ -15,21 +15,28 @@ def _get_kwargs(
15
15
  graph_id: str,
16
16
  *,
17
17
  body: ExchangeTokenRequest,
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
25
+ params: dict[str, Any] = {}
26
+
27
+ json_token: Union[None, Unset, str]
28
+ if isinstance(token, Unset):
29
+ json_token = UNSET
30
+ else:
31
+ json_token = token
32
+ params["token"] = json_token
33
+
34
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
28
35
 
29
36
  _kwargs: dict[str, Any] = {
30
37
  "method": "post",
31
38
  "url": f"/v1/{graph_id}/connections/link/exchange",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["json"] = body.to_dict()
@@ -84,8 +91,8 @@ def sync_detailed(
84
91
  *,
85
92
  client: AuthenticatedClient,
86
93
  body: ExchangeTokenRequest,
94
+ token: Union[None, Unset, str] = UNSET,
87
95
  authorization: Union[None, Unset, str] = UNSET,
88
- auth_token: Union[None, Unset, str] = UNSET,
89
96
  ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
90
97
  """Exchange Link Token
91
98
 
@@ -111,8 +118,8 @@ def sync_detailed(
111
118
 
112
119
  Args:
113
120
  graph_id (str): Graph database identifier
121
+ token (Union[None, Unset, str]): JWT token for SSE authentication
114
122
  authorization (Union[None, Unset, str]):
115
- auth_token (Union[None, Unset, str]):
116
123
  body (ExchangeTokenRequest): Exchange temporary token for permanent credentials.
117
124
 
118
125
  Raises:
@@ -126,8 +133,8 @@ def sync_detailed(
126
133
  kwargs = _get_kwargs(
127
134
  graph_id=graph_id,
128
135
  body=body,
136
+ token=token,
129
137
  authorization=authorization,
130
- auth_token=auth_token,
131
138
  )
132
139
 
133
140
  response = client.get_httpx_client().request(
@@ -142,8 +149,8 @@ def sync(
142
149
  *,
143
150
  client: AuthenticatedClient,
144
151
  body: ExchangeTokenRequest,
152
+ token: Union[None, Unset, str] = UNSET,
145
153
  authorization: Union[None, Unset, str] = UNSET,
146
- auth_token: Union[None, Unset, str] = UNSET,
147
154
  ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
148
155
  """Exchange Link Token
149
156
 
@@ -169,8 +176,8 @@ def sync(
169
176
 
170
177
  Args:
171
178
  graph_id (str): Graph database identifier
179
+ token (Union[None, Unset, str]): JWT token for SSE authentication
172
180
  authorization (Union[None, Unset, str]):
173
- auth_token (Union[None, Unset, str]):
174
181
  body (ExchangeTokenRequest): Exchange temporary token for permanent credentials.
175
182
 
176
183
  Raises:
@@ -185,8 +192,8 @@ def sync(
185
192
  graph_id=graph_id,
186
193
  client=client,
187
194
  body=body,
195
+ token=token,
188
196
  authorization=authorization,
189
- auth_token=auth_token,
190
197
  ).parsed
191
198
 
192
199
 
@@ -195,8 +202,8 @@ async def asyncio_detailed(
195
202
  *,
196
203
  client: AuthenticatedClient,
197
204
  body: ExchangeTokenRequest,
205
+ token: Union[None, Unset, str] = UNSET,
198
206
  authorization: Union[None, Unset, str] = UNSET,
199
- auth_token: Union[None, Unset, str] = UNSET,
200
207
  ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
201
208
  """Exchange Link Token
202
209
 
@@ -222,8 +229,8 @@ async def asyncio_detailed(
222
229
 
223
230
  Args:
224
231
  graph_id (str): Graph database identifier
232
+ token (Union[None, Unset, str]): JWT token for SSE authentication
225
233
  authorization (Union[None, Unset, str]):
226
- auth_token (Union[None, Unset, str]):
227
234
  body (ExchangeTokenRequest): Exchange temporary token for permanent credentials.
228
235
 
229
236
  Raises:
@@ -237,8 +244,8 @@ async def asyncio_detailed(
237
244
  kwargs = _get_kwargs(
238
245
  graph_id=graph_id,
239
246
  body=body,
247
+ token=token,
240
248
  authorization=authorization,
241
- auth_token=auth_token,
242
249
  )
243
250
 
244
251
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -251,8 +258,8 @@ async def asyncio(
251
258
  *,
252
259
  client: AuthenticatedClient,
253
260
  body: ExchangeTokenRequest,
261
+ token: Union[None, Unset, str] = UNSET,
254
262
  authorization: Union[None, Unset, str] = UNSET,
255
- auth_token: Union[None, Unset, str] = UNSET,
256
263
  ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
257
264
  """Exchange Link Token
258
265
 
@@ -278,8 +285,8 @@ async def asyncio(
278
285
 
279
286
  Args:
280
287
  graph_id (str): Graph database identifier
288
+ token (Union[None, Unset, str]): JWT token for SSE authentication
281
289
  authorization (Union[None, Unset, str]):
282
- auth_token (Union[None, Unset, str]):
283
290
  body (ExchangeTokenRequest): Exchange temporary token for permanent credentials.
284
291
 
285
292
  Raises:
@@ -295,7 +302,7 @@ async def asyncio(
295
302
  graph_id=graph_id,
296
303
  client=client,
297
304
  body=body,
305
+ token=token,
298
306
  authorization=authorization,
299
- auth_token=auth_token,
300
307
  )
301
308
  ).parsed
robosystems_client/api/connections/get_connection.py CHANGED
@@ -15,21 +15,28 @@ def _get_kwargs(
15
15
  graph_id: str,
16
16
  connection_id: str,
17
17
  *,
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
25
+ params: dict[str, Any] = {}
26
+
27
+ json_token: Union[None, Unset, str]
28
+ if isinstance(token, Unset):
29
+ json_token = UNSET
30
+ else:
31
+ json_token = token
32
+ params["token"] = json_token
33
+
34
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
28
35
 
29
36
  _kwargs: dict[str, Any] = {
30
37
  "method": "get",
31
38
  "url": f"/v1/{graph_id}/connections/{connection_id}",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["headers"] = headers
@@ -81,8 +88,8 @@ def sync_detailed(
81
88
  connection_id: str,
82
89
  *,
83
90
  client: AuthenticatedClient,
91
+ token: Union[None, Unset, str] = UNSET,
84
92
  authorization: Union[None, Unset, str] = UNSET,
85
- auth_token: Union[None, Unset, str] = UNSET,
86
93
  ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
87
94
  """Get Connection
88
95
 
@@ -100,8 +107,8 @@ def sync_detailed(
100
107
  Args:
101
108
  graph_id (str): Graph database identifier
102
109
  connection_id (str): Unique connection identifier
110
+ token (Union[None, Unset, str]): JWT token for SSE authentication
103
111
  authorization (Union[None, Unset, str]):
104
- auth_token (Union[None, Unset, str]):
105
112
 
106
113
  Raises:
107
114
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -114,8 +121,8 @@ def sync_detailed(
114
121
  kwargs = _get_kwargs(
115
122
  graph_id=graph_id,
116
123
  connection_id=connection_id,
124
+ token=token,
117
125
  authorization=authorization,
118
- auth_token=auth_token,
119
126
  )
120
127
 
121
128
  response = client.get_httpx_client().request(
@@ -130,8 +137,8 @@ def sync(
130
137
  connection_id: str,
131
138
  *,
132
139
  client: AuthenticatedClient,
140
+ token: Union[None, Unset, str] = UNSET,
133
141
  authorization: Union[None, Unset, str] = UNSET,
134
- auth_token: Union[None, Unset, str] = UNSET,
135
142
  ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
136
143
  """Get Connection
137
144
 
@@ -149,8 +156,8 @@ def sync(
149
156
  Args:
150
157
  graph_id (str): Graph database identifier
151
158
  connection_id (str): Unique connection identifier
159
+ token (Union[None, Unset, str]): JWT token for SSE authentication
152
160
  authorization (Union[None, Unset, str]):
153
- auth_token (Union[None, Unset, str]):
154
161
 
155
162
  Raises:
156
163
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -164,8 +171,8 @@ def sync(
164
171
  graph_id=graph_id,
165
172
  connection_id=connection_id,
166
173
  client=client,
174
+ token=token,
167
175
  authorization=authorization,
168
- auth_token=auth_token,
169
176
  ).parsed
170
177
 
171
178
 
@@ -174,8 +181,8 @@ async def asyncio_detailed(
174
181
  connection_id: str,
175
182
  *,
176
183
  client: AuthenticatedClient,
184
+ token: Union[None, Unset, str] = UNSET,
177
185
  authorization: Union[None, Unset, str] = UNSET,
178
- auth_token: Union[None, Unset, str] = UNSET,
179
186
  ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
180
187
  """Get Connection
181
188
 
@@ -193,8 +200,8 @@ async def asyncio_detailed(
193
200
  Args:
194
201
  graph_id (str): Graph database identifier
195
202
  connection_id (str): Unique connection identifier
203
+ token (Union[None, Unset, str]): JWT token for SSE authentication
196
204
  authorization (Union[None, Unset, str]):
197
- auth_token (Union[None, Unset, str]):
198
205
 
199
206
  Raises:
200
207
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -207,8 +214,8 @@ async def asyncio_detailed(
207
214
  kwargs = _get_kwargs(
208
215
  graph_id=graph_id,
209
216
  connection_id=connection_id,
217
+ token=token,
210
218
  authorization=authorization,
211
- auth_token=auth_token,
212
219
  )
213
220
 
214
221
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -221,8 +228,8 @@ async def asyncio(
221
228
  connection_id: str,
222
229
  *,
223
230
  client: AuthenticatedClient,
231
+ token: Union[None, Unset, str] = UNSET,
224
232
  authorization: Union[None, Unset, str] = UNSET,
225
- auth_token: Union[None, Unset, str] = UNSET,
226
233
  ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
227
234
  """Get Connection
228
235
 
@@ -240,8 +247,8 @@ async def asyncio(
240
247
  Args:
241
248
  graph_id (str): Graph database identifier
242
249
  connection_id (str): Unique connection identifier
250
+ token (Union[None, Unset, str]): JWT token for SSE authentication
243
251
  authorization (Union[None, Unset, str]):
244
- auth_token (Union[None, Unset, str]):
245
252
 
246
253
  Raises:
247
254
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -256,7 +263,7 @@ async def asyncio(
256
263
  graph_id=graph_id,
257
264
  connection_id=connection_id,
258
265
  client=client,
266
+ token=token,
259
267
  authorization=authorization,
260
- auth_token=auth_token,
261
268
  )
262
269
  ).parsed
robosystems_client/api/connections/get_connection_options.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}/connections/options",
31
- "cookies": cookies,
38
+ "params": params,
32
39
  }
33
40
 
34
41
  _kwargs["headers"] = headers
@@ -75,8 +82,8 @@ def sync_detailed(
75
82
  graph_id: str,
76
83
  *,
77
84
  client: AuthenticatedClient,
85
+ token: Union[None, Unset, str] = UNSET,
78
86
  authorization: Union[None, Unset, str] = UNSET,
79
- auth_token: Union[None, Unset, str] = UNSET,
80
87
  ) -> Response[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
81
88
  """List Connection Options
82
89
 
@@ -103,8 +110,8 @@ def sync_detailed(
103
110
 
104
111
  Args:
105
112
  graph_id (str): Graph database identifier
113
+ token (Union[None, Unset, str]): JWT token for SSE authentication
106
114
  authorization (Union[None, Unset, str]):
107
- auth_token (Union[None, Unset, str]):
108
115
 
109
116
  Raises:
110
117
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -116,8 +123,8 @@ def sync_detailed(
116
123
 
117
124
  kwargs = _get_kwargs(
118
125
  graph_id=graph_id,
126
+ token=token,
119
127
  authorization=authorization,
120
- auth_token=auth_token,
121
128
  )
122
129
 
123
130
  response = client.get_httpx_client().request(
@@ -131,8 +138,8 @@ def sync(
131
138
  graph_id: str,
132
139
  *,
133
140
  client: AuthenticatedClient,
141
+ token: Union[None, Unset, str] = UNSET,
134
142
  authorization: Union[None, Unset, str] = UNSET,
135
- auth_token: Union[None, Unset, str] = UNSET,
136
143
  ) -> Optional[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
137
144
  """List Connection Options
138
145
 
@@ -159,8 +166,8 @@ def sync(
159
166
 
160
167
  Args:
161
168
  graph_id (str): Graph database identifier
169
+ token (Union[None, Unset, str]): JWT token for SSE authentication
162
170
  authorization (Union[None, Unset, str]):
163
- auth_token (Union[None, Unset, str]):
164
171
 
165
172
  Raises:
166
173
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -173,8 +180,8 @@ def sync(
173
180
  return sync_detailed(
174
181
  graph_id=graph_id,
175
182
  client=client,
183
+ token=token,
176
184
  authorization=authorization,
177
- auth_token=auth_token,
178
185
  ).parsed
179
186
 
180
187
 
@@ -182,8 +189,8 @@ async def asyncio_detailed(
182
189
  graph_id: str,
183
190
  *,
184
191
  client: AuthenticatedClient,
192
+ token: Union[None, Unset, str] = UNSET,
185
193
  authorization: Union[None, Unset, str] = UNSET,
186
- auth_token: Union[None, Unset, str] = UNSET,
187
194
  ) -> Response[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
188
195
  """List Connection Options
189
196
 
@@ -210,8 +217,8 @@ async def asyncio_detailed(
210
217
 
211
218
  Args:
212
219
  graph_id (str): Graph database identifier
220
+ token (Union[None, Unset, str]): JWT token for SSE authentication
213
221
  authorization (Union[None, Unset, str]):
214
- auth_token (Union[None, Unset, str]):
215
222
 
216
223
  Raises:
217
224
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -223,8 +230,8 @@ async def asyncio_detailed(
223
230
 
224
231
  kwargs = _get_kwargs(
225
232
  graph_id=graph_id,
233
+ token=token,
226
234
  authorization=authorization,
227
- auth_token=auth_token,
228
235
  )
229
236
 
230
237
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -236,8 +243,8 @@ async def asyncio(
236
243
  graph_id: str,
237
244
  *,
238
245
  client: AuthenticatedClient,
246
+ token: Union[None, Unset, str] = UNSET,
239
247
  authorization: Union[None, Unset, str] = UNSET,
240
- auth_token: Union[None, Unset, str] = UNSET,
241
248
  ) -> Optional[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
242
249
  """List Connection Options
243
250
 
@@ -264,8 +271,8 @@ async def asyncio(
264
271
 
265
272
  Args:
266
273
  graph_id (str): Graph database identifier
274
+ token (Union[None, Unset, str]): JWT token for SSE authentication
267
275
  authorization (Union[None, Unset, str]):
268
- auth_token (Union[None, Unset, str]):
269
276
 
270
277
  Raises:
271
278
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -279,7 +286,7 @@ async def asyncio(
279
286
  await asyncio_detailed(
280
287
  graph_id=graph_id,
281
288
  client=client,
289
+ token=token,
282
290
  authorization=authorization,
283
- auth_token=auth_token,
284
291
  )
285
292
  ).parsed
robosystems_client/api/connections/init_o_auth.py CHANGED
@@ -15,21 +15,28 @@ def _get_kwargs(
15
15
  graph_id: str,
16
16
  *,
17
17
  body: OAuthInitRequest,
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
25
+ params: dict[str, Any] = {}
26
+
27
+ json_token: Union[None, Unset, str]
28
+ if isinstance(token, Unset):
29
+ json_token = UNSET
30
+ else:
31
+ json_token = token
32
+ params["token"] = json_token
33
+
34
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
28
35
 
29
36
  _kwargs: dict[str, Any] = {
30
37
  "method": "post",
31
38
  "url": f"/v1/{graph_id}/connections/oauth/init",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["json"] = body.to_dict()
@@ -73,8 +80,8 @@ def sync_detailed(
73
80
  *,
74
81
  client: AuthenticatedClient,
75
82
  body: OAuthInitRequest,
83
+ token: Union[None, Unset, str] = UNSET,
76
84
  authorization: Union[None, Unset, str] = UNSET,
77
- auth_token: Union[None, Unset, str] = UNSET,
78
85
  ) -> Response[Union[HTTPValidationError, OAuthInitResponse]]:
79
86
  """Init Oauth
80
87
 
@@ -85,8 +92,8 @@ def sync_detailed(
85
92
 
86
93
  Args:
87
94
  graph_id (str): Graph database identifier
95
+ token (Union[None, Unset, str]): JWT token for SSE authentication
88
96
  authorization (Union[None, Unset, str]):
89
- auth_token (Union[None, Unset, str]):
90
97
  body (OAuthInitRequest): Request to initiate OAuth flow.
91
98
 
92
99
  Raises:
@@ -100,8 +107,8 @@ def sync_detailed(
100
107
  kwargs = _get_kwargs(
101
108
  graph_id=graph_id,
102
109
  body=body,
110
+ token=token,
103
111
  authorization=authorization,
104
- auth_token=auth_token,
105
112
  )
106
113
 
107
114
  response = client.get_httpx_client().request(
@@ -116,8 +123,8 @@ def sync(
116
123
  *,
117
124
  client: AuthenticatedClient,
118
125
  body: OAuthInitRequest,
126
+ token: Union[None, Unset, str] = UNSET,
119
127
  authorization: Union[None, Unset, str] = UNSET,
120
- auth_token: Union[None, Unset, str] = UNSET,
121
128
  ) -> Optional[Union[HTTPValidationError, OAuthInitResponse]]:
122
129
  """Init Oauth
123
130
 
@@ -128,8 +135,8 @@ def sync(
128
135
 
129
136
  Args:
130
137
  graph_id (str): Graph database identifier
138
+ token (Union[None, Unset, str]): JWT token for SSE authentication
131
139
  authorization (Union[None, Unset, str]):
132
- auth_token (Union[None, Unset, str]):
133
140
  body (OAuthInitRequest): Request to initiate OAuth flow.
134
141
 
135
142
  Raises:
@@ -144,8 +151,8 @@ def sync(
144
151
  graph_id=graph_id,
145
152
  client=client,
146
153
  body=body,
154
+ token=token,
147
155
  authorization=authorization,
148
- auth_token=auth_token,
149
156
  ).parsed
150
157
 
151
158
 
@@ -154,8 +161,8 @@ async def asyncio_detailed(
154
161
  *,
155
162
  client: AuthenticatedClient,
156
163
  body: OAuthInitRequest,
164
+ token: Union[None, Unset, str] = UNSET,
157
165
  authorization: Union[None, Unset, str] = UNSET,
158
- auth_token: Union[None, Unset, str] = UNSET,
159
166
  ) -> Response[Union[HTTPValidationError, OAuthInitResponse]]:
160
167
  """Init Oauth
161
168
 
@@ -166,8 +173,8 @@ async def asyncio_detailed(
166
173
 
167
174
  Args:
168
175
  graph_id (str): Graph database identifier
176
+ token (Union[None, Unset, str]): JWT token for SSE authentication
169
177
  authorization (Union[None, Unset, str]):
170
- auth_token (Union[None, Unset, str]):
171
178
  body (OAuthInitRequest): Request to initiate OAuth flow.
172
179
 
173
180
  Raises:
@@ -181,8 +188,8 @@ async def asyncio_detailed(
181
188
  kwargs = _get_kwargs(
182
189
  graph_id=graph_id,
183
190
  body=body,
191
+ token=token,
184
192
  authorization=authorization,
185
- auth_token=auth_token,
186
193
  )
187
194
 
188
195
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -195,8 +202,8 @@ async def asyncio(
195
202
  *,
196
203
  client: AuthenticatedClient,
197
204
  body: OAuthInitRequest,
205
+ token: Union[None, Unset, str] = UNSET,
198
206
  authorization: Union[None, Unset, str] = UNSET,
199
- auth_token: Union[None, Unset, str] = UNSET,
200
207
  ) -> Optional[Union[HTTPValidationError, OAuthInitResponse]]:
201
208
  """Init Oauth
202
209
 
@@ -207,8 +214,8 @@ async def asyncio(
207
214
 
208
215
  Args:
209
216
  graph_id (str): Graph database identifier
217
+ token (Union[None, Unset, str]): JWT token for SSE authentication
210
218
  authorization (Union[None, Unset, str]):
211
- auth_token (Union[None, Unset, str]):
212
219
  body (OAuthInitRequest): Request to initiate OAuth flow.
213
220
 
214
221
  Raises:
@@ -224,7 +231,7 @@ async def asyncio(
224
231
  graph_id=graph_id,
225
232
  client=client,
226
233
  body=body,
234
+ token=token,
227
235
  authorization=authorization,
228
- auth_token=auth_token,
229
236
  )
230
237
  ).parsed