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/query/execute_cypher_query.py CHANGED
@@ -18,17 +18,13 @@ def _get_kwargs(
18
18
  mode: Union[None, ResponseMode, Unset] = UNSET,
19
19
  chunk_size: Union[Unset, int] = 1000,
20
20
  test_mode: Union[Unset, bool] = False,
21
+ token: Union[None, Unset, str] = UNSET,
21
22
  authorization: Union[None, Unset, str] = UNSET,
22
- auth_token: Union[None, Unset, str] = UNSET,
23
23
  ) -> dict[str, Any]:
24
24
  headers: dict[str, Any] = {}
25
25
  if not isinstance(authorization, Unset):
26
26
  headers["authorization"] = authorization
27
27
 
28
- cookies = {}
29
- if auth_token is not UNSET:
30
- cookies["auth-token"] = auth_token
31
-
32
28
  params: dict[str, Any] = {}
33
29
 
34
30
  json_mode: Union[None, Unset, str]
@@ -44,13 +40,19 @@ def _get_kwargs(
44
40
 
45
41
  params["test_mode"] = test_mode
46
42
 
43
+ json_token: Union[None, Unset, str]
44
+ if isinstance(token, Unset):
45
+ json_token = UNSET
46
+ else:
47
+ json_token = token
48
+ params["token"] = json_token
49
+
47
50
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
48
51
 
49
52
  _kwargs: dict[str, Any] = {
50
53
  "method": "post",
51
54
  "url": f"/v1/{graph_id}/query",
52
55
  "params": params,
53
- "cookies": cookies,
54
56
  }
55
57
 
56
58
  _kwargs["json"] = body.to_dict()
@@ -117,8 +119,8 @@ def sync_detailed(
117
119
  mode: Union[None, ResponseMode, Unset] = UNSET,
118
120
  chunk_size: Union[Unset, int] = 1000,
119
121
  test_mode: Union[Unset, bool] = False,
122
+ token: Union[None, Unset, str] = UNSET,
120
123
  authorization: Union[None, Unset, str] = UNSET,
121
- auth_token: Union[None, Unset, str] = UNSET,
122
124
  ) -> Response[Union[Any, HTTPValidationError]]:
123
125
  """Execute Cypher Query
124
126
 
@@ -175,8 +177,8 @@ def sync_detailed(
175
177
  mode (Union[None, ResponseMode, Unset]): Response mode override
176
178
  chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000.
177
179
  test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False.
180
+ token (Union[None, Unset, str]): JWT token for SSE authentication
178
181
  authorization (Union[None, Unset, str]):
179
- auth_token (Union[None, Unset, str]):
180
182
  body (CypherQueryRequest): Request model for Cypher query execution.
181
183
 
182
184
  Raises:
@@ -193,8 +195,8 @@ def sync_detailed(
193
195
  mode=mode,
194
196
  chunk_size=chunk_size,
195
197
  test_mode=test_mode,
198
+ token=token,
196
199
  authorization=authorization,
197
- auth_token=auth_token,
198
200
  )
199
201
 
200
202
  response = client.get_httpx_client().request(
@@ -212,8 +214,8 @@ def sync(
212
214
  mode: Union[None, ResponseMode, Unset] = UNSET,
213
215
  chunk_size: Union[Unset, int] = 1000,
214
216
  test_mode: Union[Unset, bool] = False,
217
+ token: Union[None, Unset, str] = UNSET,
215
218
  authorization: Union[None, Unset, str] = UNSET,
216
- auth_token: Union[None, Unset, str] = UNSET,
217
219
  ) -> Optional[Union[Any, HTTPValidationError]]:
218
220
  """Execute Cypher Query
219
221
 
@@ -270,8 +272,8 @@ def sync(
270
272
  mode (Union[None, ResponseMode, Unset]): Response mode override
271
273
  chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000.
272
274
  test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False.
275
+ token (Union[None, Unset, str]): JWT token for SSE authentication
273
276
  authorization (Union[None, Unset, str]):
274
- auth_token (Union[None, Unset, str]):
275
277
  body (CypherQueryRequest): Request model for Cypher query execution.
276
278
 
277
279
  Raises:
@@ -289,8 +291,8 @@ def sync(
289
291
  mode=mode,
290
292
  chunk_size=chunk_size,
291
293
  test_mode=test_mode,
294
+ token=token,
292
295
  authorization=authorization,
293
- auth_token=auth_token,
294
296
  ).parsed
295
297
 
296
298
 
@@ -302,8 +304,8 @@ async def asyncio_detailed(
302
304
  mode: Union[None, ResponseMode, Unset] = UNSET,
303
305
  chunk_size: Union[Unset, int] = 1000,
304
306
  test_mode: Union[Unset, bool] = False,
307
+ token: Union[None, Unset, str] = UNSET,
305
308
  authorization: Union[None, Unset, str] = UNSET,
306
- auth_token: Union[None, Unset, str] = UNSET,
307
309
  ) -> Response[Union[Any, HTTPValidationError]]:
308
310
  """Execute Cypher Query
309
311
 
@@ -360,8 +362,8 @@ async def asyncio_detailed(
360
362
  mode (Union[None, ResponseMode, Unset]): Response mode override
361
363
  chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000.
362
364
  test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False.
365
+ token (Union[None, Unset, str]): JWT token for SSE authentication
363
366
  authorization (Union[None, Unset, str]):
364
- auth_token (Union[None, Unset, str]):
365
367
  body (CypherQueryRequest): Request model for Cypher query execution.
366
368
 
367
369
  Raises:
@@ -378,8 +380,8 @@ async def asyncio_detailed(
378
380
  mode=mode,
379
381
  chunk_size=chunk_size,
380
382
  test_mode=test_mode,
383
+ token=token,
381
384
  authorization=authorization,
382
- auth_token=auth_token,
383
385
  )
384
386
 
385
387
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -395,8 +397,8 @@ async def asyncio(
395
397
  mode: Union[None, ResponseMode, Unset] = UNSET,
396
398
  chunk_size: Union[Unset, int] = 1000,
397
399
  test_mode: Union[Unset, bool] = False,
400
+ token: Union[None, Unset, str] = UNSET,
398
401
  authorization: Union[None, Unset, str] = UNSET,
399
- auth_token: Union[None, Unset, str] = UNSET,
400
402
  ) -> Optional[Union[Any, HTTPValidationError]]:
401
403
  """Execute Cypher Query
402
404
 
@@ -453,8 +455,8 @@ async def asyncio(
453
455
  mode (Union[None, ResponseMode, Unset]): Response mode override
454
456
  chunk_size (Union[Unset, int]): Rows per chunk for streaming Default: 1000.
455
457
  test_mode (Union[Unset, bool]): Enable test mode for better debugging Default: False.
458
+ token (Union[None, Unset, str]): JWT token for SSE authentication
456
459
  authorization (Union[None, Unset, str]):
457
- auth_token (Union[None, Unset, str]):
458
460
  body (CypherQueryRequest): Request model for Cypher query execution.
459
461
 
460
462
  Raises:
@@ -473,7 +475,7 @@ async def asyncio(
473
475
  mode=mode,
474
476
  chunk_size=chunk_size,
475
477
  test_mode=test_mode,
478
+ token=token,
476
479
  authorization=authorization,
477
- auth_token=auth_token,
478
480
  )
479
481
  ).parsed
robosystems_client/api/schema/export_graph_schema.py CHANGED
@@ -15,30 +15,32 @@ def _get_kwargs(
15
15
  *,
16
16
  format_: Union[Unset, str] = "json",
17
17
  include_data_stats: Union[Unset, bool] = False,
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["format"] = format_
32
28
 
33
29
  params["include_data_stats"] = include_data_stats
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}/schema/export",
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
  format_: Union[Unset, str] = "json",
81
83
  include_data_stats: Union[Unset, bool] = False,
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[HTTPValidationError, SchemaExportResponse]]:
85
87
  """Export Graph Schema
86
88
 
@@ -91,8 +93,8 @@ def sync_detailed(
91
93
  format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'.
92
94
  include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph
93
95
  Default: False.
96
+ token (Union[None, Unset, str]): JWT token for SSE authentication
94
97
  authorization (Union[None, Unset, str]):
95
- auth_token (Union[None, Unset, str]):
96
98
 
97
99
  Raises:
98
100
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -106,8 +108,8 @@ def sync_detailed(
106
108
  graph_id=graph_id,
107
109
  format_=format_,
108
110
  include_data_stats=include_data_stats,
111
+ token=token,
109
112
  authorization=authorization,
110
- auth_token=auth_token,
111
113
  )
112
114
 
113
115
  response = client.get_httpx_client().request(
@@ -123,8 +125,8 @@ def sync(
123
125
  client: AuthenticatedClient,
124
126
  format_: Union[Unset, str] = "json",
125
127
  include_data_stats: Union[Unset, bool] = False,
128
+ token: Union[None, Unset, str] = UNSET,
126
129
  authorization: Union[None, Unset, str] = UNSET,
127
- auth_token: Union[None, Unset, str] = UNSET,
128
130
  ) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]:
129
131
  """Export Graph Schema
130
132
 
@@ -135,8 +137,8 @@ def sync(
135
137
  format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'.
136
138
  include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph
137
139
  Default: False.
140
+ token (Union[None, Unset, str]): JWT token for SSE authentication
138
141
  authorization (Union[None, Unset, str]):
139
- auth_token (Union[None, Unset, str]):
140
142
 
141
143
  Raises:
142
144
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -151,8 +153,8 @@ def sync(
151
153
  client=client,
152
154
  format_=format_,
153
155
  include_data_stats=include_data_stats,
156
+ token=token,
154
157
  authorization=authorization,
155
- auth_token=auth_token,
156
158
  ).parsed
157
159
 
158
160
 
@@ -162,8 +164,8 @@ async def asyncio_detailed(
162
164
  client: AuthenticatedClient,
163
165
  format_: Union[Unset, str] = "json",
164
166
  include_data_stats: Union[Unset, bool] = False,
167
+ token: Union[None, Unset, str] = UNSET,
165
168
  authorization: Union[None, Unset, str] = UNSET,
166
- auth_token: Union[None, Unset, str] = UNSET,
167
169
  ) -> Response[Union[HTTPValidationError, SchemaExportResponse]]:
168
170
  """Export Graph Schema
169
171
 
@@ -174,8 +176,8 @@ async def asyncio_detailed(
174
176
  format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'.
175
177
  include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph
176
178
  Default: False.
179
+ token (Union[None, Unset, str]): JWT token for SSE authentication
177
180
  authorization (Union[None, Unset, str]):
178
- auth_token (Union[None, Unset, str]):
179
181
 
180
182
  Raises:
181
183
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -189,8 +191,8 @@ async def asyncio_detailed(
189
191
  graph_id=graph_id,
190
192
  format_=format_,
191
193
  include_data_stats=include_data_stats,
194
+ token=token,
192
195
  authorization=authorization,
193
- auth_token=auth_token,
194
196
  )
195
197
 
196
198
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -204,8 +206,8 @@ async def asyncio(
204
206
  client: AuthenticatedClient,
205
207
  format_: Union[Unset, str] = "json",
206
208
  include_data_stats: Union[Unset, bool] = False,
209
+ token: Union[None, Unset, str] = UNSET,
207
210
  authorization: Union[None, Unset, str] = UNSET,
208
- auth_token: Union[None, Unset, str] = UNSET,
209
211
  ) -> Optional[Union[HTTPValidationError, SchemaExportResponse]]:
210
212
  """Export Graph Schema
211
213
 
@@ -216,8 +218,8 @@ async def asyncio(
216
218
  format_ (Union[Unset, str]): Export format: json, yaml, or cypher Default: 'json'.
217
219
  include_data_stats (Union[Unset, bool]): Include statistics about actual data in the graph
218
220
  Default: False.
221
+ token (Union[None, Unset, str]): JWT token for SSE authentication
219
222
  authorization (Union[None, Unset, str]):
220
- auth_token (Union[None, Unset, str]):
221
223
 
222
224
  Raises:
223
225
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -233,7 +235,7 @@ async def asyncio(
233
235
  client=client,
234
236
  format_=format_,
235
237
  include_data_stats=include_data_stats,
238
+ token=token,
236
239
  authorization=authorization,
237
- auth_token=auth_token,
238
240
  )
239
241
  ).parsed
robosystems_client/api/schema/get_graph_schema_info.py CHANGED
@@ -15,21 +15,28 @@ from ...types import UNSET, Response, Unset
15
15
  def _get_kwargs(
16
16
  graph_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}/schema/info",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["headers"] = headers
@@ -80,8 +87,8 @@ def sync_detailed(
80
87
  graph_id: str,
81
88
  *,
82
89
  client: AuthenticatedClient,
90
+ token: Union[None, Unset, str] = UNSET,
83
91
  authorization: Union[None, Unset, str] = UNSET,
84
- auth_token: Union[None, Unset, str] = UNSET,
85
92
  ) -> Response[
86
93
  Union[Any, GetGraphSchemaInfoResponseGetgraphschemainfo, HTTPValidationError]
87
94
  ]:
@@ -101,8 +108,8 @@ def sync_detailed(
101
108
 
102
109
  Args:
103
110
  graph_id (str): The graph database to get schema for
111
+ token (Union[None, Unset, str]): JWT token for SSE authentication
104
112
  authorization (Union[None, Unset, str]):
105
- auth_token (Union[None, Unset, str]):
106
113
 
107
114
  Raises:
108
115
  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
 
115
122
  kwargs = _get_kwargs(
116
123
  graph_id=graph_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(
@@ -129,8 +136,8 @@ def sync(
129
136
  graph_id: str,
130
137
  *,
131
138
  client: AuthenticatedClient,
139
+ token: Union[None, Unset, str] = UNSET,
132
140
  authorization: Union[None, Unset, str] = UNSET,
133
- auth_token: Union[None, Unset, str] = UNSET,
134
141
  ) -> Optional[
135
142
  Union[Any, GetGraphSchemaInfoResponseGetgraphschemainfo, HTTPValidationError]
136
143
  ]:
@@ -150,8 +157,8 @@ def sync(
150
157
 
151
158
  Args:
152
159
  graph_id (str): The graph database to get schema for
160
+ token (Union[None, Unset, str]): JWT token for SSE authentication
153
161
  authorization (Union[None, Unset, str]):
154
- auth_token (Union[None, Unset, str]):
155
162
 
156
163
  Raises:
157
164
  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
  return sync_detailed(
165
172
  graph_id=graph_id,
166
173
  client=client,
174
+ token=token,
167
175
  authorization=authorization,
168
- auth_token=auth_token,
169
176
  ).parsed
170
177
 
171
178
 
@@ -173,8 +180,8 @@ async def asyncio_detailed(
173
180
  graph_id: str,
174
181
  *,
175
182
  client: AuthenticatedClient,
183
+ token: Union[None, Unset, str] = UNSET,
176
184
  authorization: Union[None, Unset, str] = UNSET,
177
- auth_token: Union[None, Unset, str] = UNSET,
178
185
  ) -> Response[
179
186
  Union[Any, GetGraphSchemaInfoResponseGetgraphschemainfo, HTTPValidationError]
180
187
  ]:
@@ -194,8 +201,8 @@ async def asyncio_detailed(
194
201
 
195
202
  Args:
196
203
  graph_id (str): The graph database to get schema for
204
+ token (Union[None, Unset, str]): JWT token for SSE authentication
197
205
  authorization (Union[None, Unset, str]):
198
- auth_token (Union[None, Unset, str]):
199
206
 
200
207
  Raises:
201
208
  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
 
208
215
  kwargs = _get_kwargs(
209
216
  graph_id=graph_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)
@@ -220,8 +227,8 @@ async def asyncio(
220
227
  graph_id: str,
221
228
  *,
222
229
  client: AuthenticatedClient,
230
+ token: Union[None, Unset, str] = UNSET,
223
231
  authorization: Union[None, Unset, str] = UNSET,
224
- auth_token: Union[None, Unset, str] = UNSET,
225
232
  ) -> Optional[
226
233
  Union[Any, GetGraphSchemaInfoResponseGetgraphschemainfo, HTTPValidationError]
227
234
  ]:
@@ -241,8 +248,8 @@ async def asyncio(
241
248
 
242
249
  Args:
243
250
  graph_id (str): The graph database to get schema for
251
+ token (Union[None, Unset, str]): JWT token for SSE authentication
244
252
  authorization (Union[None, Unset, str]):
245
- auth_token (Union[None, Unset, str]):
246
253
 
247
254
  Raises:
248
255
  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
  await asyncio_detailed(
257
264
  graph_id=graph_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/schema/list_schema_extensions.py CHANGED
@@ -15,21 +15,28 @@ from ...types import UNSET, Response, Unset
15
15
  def _get_kwargs(
16
16
  graph_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}/schema/extensions",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["headers"] = headers
@@ -74,8 +81,8 @@ def sync_detailed(
74
81
  graph_id: str,
75
82
  *,
76
83
  client: AuthenticatedClient,
84
+ token: Union[None, Unset, str] = UNSET,
77
85
  authorization: Union[None, Unset, str] = UNSET,
78
- auth_token: Union[None, Unset, str] = UNSET,
79
86
  ) -> Response[
80
87
  Union[HTTPValidationError, ListSchemaExtensionsResponseListschemaextensions]
81
88
  ]:
@@ -85,8 +92,8 @@ def sync_detailed(
85
92
 
86
93
  Args:
87
94
  graph_id (str): The graph ID to list extensions for
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
 
91
98
  Raises:
92
99
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -98,8 +105,8 @@ def sync_detailed(
98
105
 
99
106
  kwargs = _get_kwargs(
100
107
  graph_id=graph_id,
108
+ token=token,
101
109
  authorization=authorization,
102
- auth_token=auth_token,
103
110
  )
104
111
 
105
112
  response = client.get_httpx_client().request(
@@ -113,8 +120,8 @@ def sync(
113
120
  graph_id: str,
114
121
  *,
115
122
  client: AuthenticatedClient,
123
+ token: Union[None, Unset, str] = UNSET,
116
124
  authorization: Union[None, Unset, str] = UNSET,
117
- auth_token: Union[None, Unset, str] = UNSET,
118
125
  ) -> Optional[
119
126
  Union[HTTPValidationError, ListSchemaExtensionsResponseListschemaextensions]
120
127
  ]:
@@ -124,8 +131,8 @@ def sync(
124
131
 
125
132
  Args:
126
133
  graph_id (str): The graph ID to list extensions for
134
+ token (Union[None, Unset, str]): JWT token for SSE authentication
127
135
  authorization (Union[None, Unset, str]):
128
- auth_token (Union[None, Unset, str]):
129
136
 
130
137
  Raises:
131
138
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -138,8 +145,8 @@ def sync(
138
145
  return sync_detailed(
139
146
  graph_id=graph_id,
140
147
  client=client,
148
+ token=token,
141
149
  authorization=authorization,
142
- auth_token=auth_token,
143
150
  ).parsed
144
151
 
145
152
 
@@ -147,8 +154,8 @@ async def asyncio_detailed(
147
154
  graph_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[
153
160
  Union[HTTPValidationError, ListSchemaExtensionsResponseListschemaextensions]
154
161
  ]:
@@ -158,8 +165,8 @@ async def asyncio_detailed(
158
165
 
159
166
  Args:
160
167
  graph_id (str): The graph ID to list extensions for
168
+ token (Union[None, Unset, str]): JWT token for SSE authentication
161
169
  authorization (Union[None, Unset, str]):
162
- auth_token (Union[None, Unset, str]):
163
170
 
164
171
  Raises:
165
172
  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
 
172
179
  kwargs = _get_kwargs(
173
180
  graph_id=graph_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)
@@ -184,8 +191,8 @@ async def asyncio(
184
191
  graph_id: str,
185
192
  *,
186
193
  client: AuthenticatedClient,
194
+ token: Union[None, Unset, str] = UNSET,
187
195
  authorization: Union[None, Unset, str] = UNSET,
188
- auth_token: Union[None, Unset, str] = UNSET,
189
196
  ) -> Optional[
190
197
  Union[HTTPValidationError, ListSchemaExtensionsResponseListschemaextensions]
191
198
  ]:
@@ -195,8 +202,8 @@ async def asyncio(
195
202
 
196
203
  Args:
197
204
  graph_id (str): The graph ID to list extensions for
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.
@@ -210,7 +217,7 @@ async def asyncio(
210
217
  await asyncio_detailed(
211
218
  graph_id=graph_id,
212
219
  client=client,
220
+ token=token,
213
221
  authorization=authorization,
214
- auth_token=auth_token,
215
222
  )
216
223
  ).parsed