robosystems-client 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of robosystems-client might be problematic. Click here for more details.

Files changed (109) hide show
  1. robosystems_client/__init__.py +15 -4
  2. robosystems_client/api/agent/auto_select_agent.py +24 -17
  3. robosystems_client/api/agent/batch_process_queries.py +24 -17
  4. robosystems_client/api/agent/execute_specific_agent.py +24 -17
  5. robosystems_client/api/agent/get_agent_metadata.py +24 -17
  6. robosystems_client/api/agent/list_agents.py +20 -18
  7. robosystems_client/api/agent/recommend_agent.py +24 -17
  8. robosystems_client/api/auth/forgot_password.py +191 -0
  9. robosystems_client/api/auth/generate_sso_token.py +18 -0
  10. robosystems_client/api/auth/get_current_auth_user.py +4 -22
  11. robosystems_client/api/auth/logout_user.py +17 -17
  12. robosystems_client/api/auth/{refresh_session.py → refresh_auth_session.py} +21 -21
  13. robosystems_client/api/auth/resend_verification_email.py +222 -0
  14. robosystems_client/api/auth/reset_password.py +177 -0
  15. robosystems_client/api/auth/validate_reset_token.py +171 -0
  16. robosystems_client/api/auth/verify_email.py +177 -0
  17. robosystems_client/api/backup/create_backup.py +24 -17
  18. robosystems_client/api/backup/export_backup.py +24 -17
  19. robosystems_client/api/backup/get_backup_download_url.py +20 -18
  20. robosystems_client/api/backup/get_backup_stats.py +24 -17
  21. robosystems_client/api/backup/list_backups.py +20 -18
  22. robosystems_client/api/backup/restore_backup.py +24 -17
  23. robosystems_client/api/connections/create_connection.py +24 -17
  24. robosystems_client/api/connections/create_link_token.py +24 -17
  25. robosystems_client/api/connections/delete_connection.py +24 -17
  26. robosystems_client/api/connections/exchange_link_token.py +24 -17
  27. robosystems_client/api/connections/get_connection.py +24 -17
  28. robosystems_client/api/connections/get_connection_options.py +24 -17
  29. robosystems_client/api/connections/init_o_auth.py +24 -17
  30. robosystems_client/api/connections/list_connections.py +20 -18
  31. robosystems_client/api/connections/oauth_callback.py +28 -21
  32. robosystems_client/api/connections/sync_connection.py +24 -17
  33. robosystems_client/api/copy/copy_data_to_graph.py +24 -17
  34. robosystems_client/api/create/create_graph.py +24 -17
  35. robosystems_client/api/graph_analytics/get_graph_metrics.py +24 -17
  36. robosystems_client/api/graph_analytics/get_graph_usage_stats.py +20 -18
  37. robosystems_client/api/graph_billing/get_current_graph_bill.py +24 -17
  38. robosystems_client/api/graph_billing/get_graph_billing_history.py +20 -18
  39. robosystems_client/api/graph_billing/get_graph_monthly_bill.py +24 -17
  40. robosystems_client/api/graph_billing/get_graph_usage_details.py +20 -18
  41. robosystems_client/api/graph_credits/check_credit_balance.py +20 -18
  42. robosystems_client/api/graph_credits/check_storage_limits.py +24 -17
  43. robosystems_client/api/graph_credits/get_credit_summary.py +24 -17
  44. robosystems_client/api/graph_credits/get_storage_usage.py +20 -18
  45. robosystems_client/api/graph_credits/list_credit_transactions.py +20 -18
  46. robosystems_client/api/graph_health/get_database_health.py +24 -17
  47. robosystems_client/api/graph_info/get_database_info.py +24 -17
  48. robosystems_client/api/graph_limits/get_graph_limits.py +24 -17
  49. robosystems_client/api/mcp/call_mcp_tool.py +20 -18
  50. robosystems_client/api/mcp/list_mcp_tools.py +24 -17
  51. robosystems_client/api/operations/cancel_operation.py +24 -17
  52. robosystems_client/api/operations/get_operation_status.py +24 -17
  53. robosystems_client/api/operations/stream_operation_events.py +20 -18
  54. robosystems_client/api/query/execute_cypher_query.py +20 -18
  55. robosystems_client/api/schema/export_graph_schema.py +20 -18
  56. robosystems_client/api/schema/get_graph_schema_info.py +24 -17
  57. robosystems_client/api/schema/list_schema_extensions.py +24 -17
  58. robosystems_client/api/schema/validate_schema.py +24 -17
  59. robosystems_client/api/subgraphs/create_subgraph.py +24 -17
  60. robosystems_client/api/subgraphs/delete_subgraph.py +24 -17
  61. robosystems_client/api/subgraphs/get_subgraph_info.py +24 -17
  62. robosystems_client/api/subgraphs/get_subgraph_quota.py +24 -17
  63. robosystems_client/api/subgraphs/list_subgraphs.py +24 -17
  64. robosystems_client/api/user/create_user_api_key.py +24 -17
  65. robosystems_client/api/user/get_all_credit_summaries.py +24 -17
  66. robosystems_client/api/user/get_current_user.py +24 -17
  67. robosystems_client/api/user/get_user_graphs.py +24 -17
  68. robosystems_client/api/user/list_user_api_keys.py +24 -17
  69. robosystems_client/api/user/revoke_user_api_key.py +24 -17
  70. robosystems_client/api/user/select_user_graph.py +24 -17
  71. robosystems_client/api/user/update_user.py +24 -17
  72. robosystems_client/api/user/update_user_api_key.py +24 -17
  73. robosystems_client/api/user/update_user_password.py +24 -17
  74. robosystems_client/api/user_analytics/get_detailed_user_analytics.py +20 -18
  75. robosystems_client/api/user_analytics/get_user_usage_overview.py +24 -17
  76. robosystems_client/api/user_limits/get_all_shared_repository_limits.py +24 -17
  77. robosystems_client/api/user_limits/get_shared_repository_limits.py +24 -17
  78. robosystems_client/api/user_limits/get_user_limits.py +24 -17
  79. robosystems_client/api/user_limits/get_user_usage.py +24 -17
  80. robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +24 -17
  81. robosystems_client/api/user_subscriptions/get_repository_credits.py +24 -17
  82. robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +24 -17
  83. robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +20 -18
  84. robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +24 -17
  85. robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +24 -17
  86. robosystems_client/extensions/__init__.py +70 -0
  87. robosystems_client/extensions/auth_integration.py +14 -1
  88. robosystems_client/extensions/copy_client.py +32 -22
  89. robosystems_client/extensions/dataframe_utils.py +455 -0
  90. robosystems_client/extensions/extensions.py +16 -0
  91. robosystems_client/extensions/operation_client.py +43 -21
  92. robosystems_client/extensions/query_client.py +109 -12
  93. robosystems_client/extensions/tests/test_dataframe_utils.py +334 -0
  94. robosystems_client/extensions/tests/test_integration.py +1 -1
  95. robosystems_client/extensions/tests/test_token_utils.py +274 -0
  96. robosystems_client/extensions/token_utils.py +417 -0
  97. robosystems_client/extensions/utils.py +32 -2
  98. robosystems_client/models/__init__.py +16 -0
  99. robosystems_client/models/email_verification_request.py +60 -0
  100. robosystems_client/models/forgot_password_request.py +60 -0
  101. robosystems_client/models/forgot_password_response_forgotpassword.py +44 -0
  102. robosystems_client/models/resend_verification_email_response_resendverificationemail.py +44 -0
  103. robosystems_client/models/reset_password_request.py +68 -0
  104. robosystems_client/models/reset_password_validate_response.py +82 -0
  105. robosystems_client-0.1.18.dist-info/METADATA +89 -0
  106. {robosystems_client-0.1.16.dist-info → robosystems_client-0.1.18.dist-info}/RECORD +108 -92
  107. robosystems_client-0.1.18.dist-info/licenses/LICENSE +21 -0
  108. robosystems_client-0.1.16.dist-info/METADATA +0 -302
  109. {robosystems_client-0.1.16.dist-info → robosystems_client-0.1.18.dist-info}/WHEEL +0 -0
robosystems_client/api/graph_health/get_database_health.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}/health",
30
- "cookies": cookies,
37
+ "params": params,
31
38
  }
32
39
 
33
40
  _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[Any, DatabaseHealthResponse, HTTPValidationError]]:
81
88
  """Database Health Check
82
89
 
@@ -100,8 +107,8 @@ def sync_detailed(
100
107
 
101
108
  Args:
102
109
  graph_id (str): Graph database 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.
@@ -113,8 +120,8 @@ def sync_detailed(
113
120
 
114
121
  kwargs = _get_kwargs(
115
122
  graph_id=graph_id,
123
+ token=token,
116
124
  authorization=authorization,
117
- auth_token=auth_token,
118
125
  )
119
126
 
120
127
  response = client.get_httpx_client().request(
@@ -128,8 +135,8 @@ def sync(
128
135
  graph_id: str,
129
136
  *,
130
137
  client: AuthenticatedClient,
138
+ token: Union[None, Unset, str] = UNSET,
131
139
  authorization: Union[None, Unset, str] = UNSET,
132
- auth_token: Union[None, Unset, str] = UNSET,
133
140
  ) -> Optional[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
134
141
  """Database Health Check
135
142
 
@@ -153,8 +160,8 @@ def sync(
153
160
 
154
161
  Args:
155
162
  graph_id (str): Graph database identifier
163
+ token (Union[None, Unset, str]): JWT token for SSE authentication
156
164
  authorization (Union[None, Unset, str]):
157
- auth_token (Union[None, Unset, str]):
158
165
 
159
166
  Raises:
160
167
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -167,8 +174,8 @@ def sync(
167
174
  return sync_detailed(
168
175
  graph_id=graph_id,
169
176
  client=client,
177
+ token=token,
170
178
  authorization=authorization,
171
- auth_token=auth_token,
172
179
  ).parsed
173
180
 
174
181
 
@@ -176,8 +183,8 @@ async def asyncio_detailed(
176
183
  graph_id: str,
177
184
  *,
178
185
  client: AuthenticatedClient,
186
+ token: Union[None, Unset, str] = UNSET,
179
187
  authorization: Union[None, Unset, str] = UNSET,
180
- auth_token: Union[None, Unset, str] = UNSET,
181
188
  ) -> Response[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
182
189
  """Database Health Check
183
190
 
@@ -201,8 +208,8 @@ async def asyncio_detailed(
201
208
 
202
209
  Args:
203
210
  graph_id (str): Graph database identifier
211
+ token (Union[None, Unset, str]): JWT token for SSE authentication
204
212
  authorization (Union[None, Unset, str]):
205
- auth_token (Union[None, Unset, str]):
206
213
 
207
214
  Raises:
208
215
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -214,8 +221,8 @@ async def asyncio_detailed(
214
221
 
215
222
  kwargs = _get_kwargs(
216
223
  graph_id=graph_id,
224
+ token=token,
217
225
  authorization=authorization,
218
- auth_token=auth_token,
219
226
  )
220
227
 
221
228
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -227,8 +234,8 @@ async def asyncio(
227
234
  graph_id: str,
228
235
  *,
229
236
  client: AuthenticatedClient,
237
+ token: Union[None, Unset, str] = UNSET,
230
238
  authorization: Union[None, Unset, str] = UNSET,
231
- auth_token: Union[None, Unset, str] = UNSET,
232
239
  ) -> Optional[Union[Any, DatabaseHealthResponse, HTTPValidationError]]:
233
240
  """Database Health Check
234
241
 
@@ -252,8 +259,8 @@ async def asyncio(
252
259
 
253
260
  Args:
254
261
  graph_id (str): Graph database identifier
262
+ token (Union[None, Unset, str]): JWT token for SSE authentication
255
263
  authorization (Union[None, Unset, str]):
256
- auth_token (Union[None, Unset, str]):
257
264
 
258
265
  Raises:
259
266
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -267,7 +274,7 @@ async def asyncio(
267
274
  await asyncio_detailed(
268
275
  graph_id=graph_id,
269
276
  client=client,
277
+ token=token,
270
278
  authorization=authorization,
271
- auth_token=auth_token,
272
279
  )
273
280
  ).parsed
robosystems_client/api/graph_info/get_database_info.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}/info",
30
- "cookies": cookies,
37
+ "params": params,
31
38
  }
32
39
 
33
40
  _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[Any, DatabaseInfoResponse, HTTPValidationError]]:
81
88
  """Database Information
82
89
 
@@ -101,8 +108,8 @@ def sync_detailed(
101
108
 
102
109
  Args:
103
110
  graph_id (str): Graph database identifier
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[Union[Any, DatabaseInfoResponse, HTTPValidationError]]:
135
142
  """Database Information
136
143
 
@@ -155,8 +162,8 @@ def sync(
155
162
 
156
163
  Args:
157
164
  graph_id (str): Graph database identifier
165
+ token (Union[None, Unset, str]): JWT token for SSE authentication
158
166
  authorization (Union[None, Unset, str]):
159
- auth_token (Union[None, Unset, str]):
160
167
 
161
168
  Raises:
162
169
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -169,8 +176,8 @@ def sync(
169
176
  return sync_detailed(
170
177
  graph_id=graph_id,
171
178
  client=client,
179
+ token=token,
172
180
  authorization=authorization,
173
- auth_token=auth_token,
174
181
  ).parsed
175
182
 
176
183
 
@@ -178,8 +185,8 @@ async def asyncio_detailed(
178
185
  graph_id: str,
179
186
  *,
180
187
  client: AuthenticatedClient,
188
+ token: Union[None, Unset, str] = UNSET,
181
189
  authorization: Union[None, Unset, str] = UNSET,
182
- auth_token: Union[None, Unset, str] = UNSET,
183
190
  ) -> Response[Union[Any, DatabaseInfoResponse, HTTPValidationError]]:
184
191
  """Database Information
185
192
 
@@ -204,8 +211,8 @@ async def asyncio_detailed(
204
211
 
205
212
  Args:
206
213
  graph_id (str): Graph database identifier
214
+ token (Union[None, Unset, str]): JWT token for SSE authentication
207
215
  authorization (Union[None, Unset, str]):
208
- auth_token (Union[None, Unset, str]):
209
216
 
210
217
  Raises:
211
218
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -217,8 +224,8 @@ async def asyncio_detailed(
217
224
 
218
225
  kwargs = _get_kwargs(
219
226
  graph_id=graph_id,
227
+ token=token,
220
228
  authorization=authorization,
221
- auth_token=auth_token,
222
229
  )
223
230
 
224
231
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -230,8 +237,8 @@ async def asyncio(
230
237
  graph_id: str,
231
238
  *,
232
239
  client: AuthenticatedClient,
240
+ token: Union[None, Unset, str] = UNSET,
233
241
  authorization: Union[None, Unset, str] = UNSET,
234
- auth_token: Union[None, Unset, str] = UNSET,
235
242
  ) -> Optional[Union[Any, DatabaseInfoResponse, HTTPValidationError]]:
236
243
  """Database Information
237
244
 
@@ -256,8 +263,8 @@ async def asyncio(
256
263
 
257
264
  Args:
258
265
  graph_id (str): Graph database identifier
266
+ token (Union[None, Unset, str]): JWT token for SSE authentication
259
267
  authorization (Union[None, Unset, str]):
260
- auth_token (Union[None, Unset, str]):
261
268
 
262
269
  Raises:
263
270
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -271,7 +278,7 @@ async def asyncio(
271
278
  await asyncio_detailed(
272
279
  graph_id=graph_id,
273
280
  client=client,
281
+ token=token,
274
282
  authorization=authorization,
275
- auth_token=auth_token,
276
283
  )
277
284
  ).parsed
robosystems_client/api/graph_limits/get_graph_limits.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}/limits",
32
- "cookies": cookies,
39
+ "params": params,
33
40
  }
34
41
 
35
42
  _kwargs["headers"] = headers
@@ -77,8 +84,8 @@ def sync_detailed(
77
84
  graph_id: str,
78
85
  *,
79
86
  client: AuthenticatedClient,
87
+ token: Union[None, Unset, str] = UNSET,
80
88
  authorization: Union[None, Unset, str] = UNSET,
81
- auth_token: Union[None, Unset, str] = UNSET,
82
89
  ) -> Response[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]:
83
90
  """Get Graph Operational Limits
84
91
 
@@ -98,8 +105,8 @@ def sync_detailed(
98
105
 
99
106
  Args:
100
107
  graph_id (str): Graph database identifier (user graph or shared repository)
108
+ token (Union[None, Unset, str]): JWT token for SSE authentication
101
109
  authorization (Union[None, Unset, str]):
102
- auth_token (Union[None, Unset, str]):
103
110
 
104
111
  Raises:
105
112
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -111,8 +118,8 @@ def sync_detailed(
111
118
 
112
119
  kwargs = _get_kwargs(
113
120
  graph_id=graph_id,
121
+ token=token,
114
122
  authorization=authorization,
115
- auth_token=auth_token,
116
123
  )
117
124
 
118
125
  response = client.get_httpx_client().request(
@@ -126,8 +133,8 @@ def sync(
126
133
  graph_id: str,
127
134
  *,
128
135
  client: AuthenticatedClient,
136
+ token: Union[None, Unset, str] = UNSET,
129
137
  authorization: Union[None, Unset, str] = UNSET,
130
- auth_token: Union[None, Unset, str] = UNSET,
131
138
  ) -> Optional[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]:
132
139
  """Get Graph Operational Limits
133
140
 
@@ -147,8 +154,8 @@ def sync(
147
154
 
148
155
  Args:
149
156
  graph_id (str): Graph database identifier (user graph or shared repository)
157
+ token (Union[None, Unset, str]): JWT token for SSE authentication
150
158
  authorization (Union[None, Unset, str]):
151
- auth_token (Union[None, Unset, str]):
152
159
 
153
160
  Raises:
154
161
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -161,8 +168,8 @@ def sync(
161
168
  return sync_detailed(
162
169
  graph_id=graph_id,
163
170
  client=client,
171
+ token=token,
164
172
  authorization=authorization,
165
- auth_token=auth_token,
166
173
  ).parsed
167
174
 
168
175
 
@@ -170,8 +177,8 @@ async def asyncio_detailed(
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
  ) -> Response[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]:
176
183
  """Get Graph Operational Limits
177
184
 
@@ -191,8 +198,8 @@ async def asyncio_detailed(
191
198
 
192
199
  Args:
193
200
  graph_id (str): Graph database identifier (user graph or shared repository)
201
+ token (Union[None, Unset, str]): JWT token for SSE authentication
194
202
  authorization (Union[None, Unset, str]):
195
- auth_token (Union[None, Unset, str]):
196
203
 
197
204
  Raises:
198
205
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -204,8 +211,8 @@ async def asyncio_detailed(
204
211
 
205
212
  kwargs = _get_kwargs(
206
213
  graph_id=graph_id,
214
+ token=token,
207
215
  authorization=authorization,
208
- auth_token=auth_token,
209
216
  )
210
217
 
211
218
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -217,8 +224,8 @@ async def asyncio(
217
224
  graph_id: str,
218
225
  *,
219
226
  client: AuthenticatedClient,
227
+ token: Union[None, Unset, str] = UNSET,
220
228
  authorization: Union[None, Unset, str] = UNSET,
221
- auth_token: Union[None, Unset, str] = UNSET,
222
229
  ) -> Optional[Union[Any, GetGraphLimitsResponseGetgraphlimits, HTTPValidationError]]:
223
230
  """Get Graph Operational Limits
224
231
 
@@ -238,8 +245,8 @@ async def asyncio(
238
245
 
239
246
  Args:
240
247
  graph_id (str): Graph database identifier (user graph or shared repository)
248
+ token (Union[None, Unset, str]): JWT token for SSE authentication
241
249
  authorization (Union[None, Unset, str]):
242
- auth_token (Union[None, Unset, str]):
243
250
 
244
251
  Raises:
245
252
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -253,7 +260,7 @@ async def asyncio(
253
260
  await asyncio_detailed(
254
261
  graph_id=graph_id,
255
262
  client=client,
263
+ token=token,
256
264
  authorization=authorization,
257
- auth_token=auth_token,
258
265
  )
259
266
  ).parsed
robosystems_client/api/mcp/call_mcp_tool.py CHANGED
@@ -17,17 +17,13 @@ def _get_kwargs(
17
17
  body: MCPToolCall,
18
18
  format_: Union[None, Unset, str] = UNSET,
19
19
  test_mode: Union[Unset, bool] = False,
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
  json_format_: Union[None, Unset, str]
@@ -39,13 +35,19 @@ def _get_kwargs(
39
35
 
40
36
  params["test_mode"] = test_mode
41
37
 
38
+ json_token: Union[None, Unset, str]
39
+ if isinstance(token, Unset):
40
+ json_token = UNSET
41
+ else:
42
+ json_token = token
43
+ params["token"] = json_token
44
+
42
45
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
43
46
 
44
47
  _kwargs: dict[str, Any] = {
45
48
  "method": "post",
46
49
  "url": f"/v1/{graph_id}/mcp/call-tool",
47
50
  "params": params,
48
- "cookies": cookies,
49
51
  }
50
52
 
51
53
  _kwargs["json"] = body.to_dict()
@@ -121,8 +123,8 @@ def sync_detailed(
121
123
  body: MCPToolCall,
122
124
  format_: Union[None, Unset, str] = UNSET,
123
125
  test_mode: Union[Unset, bool] = False,
126
+ token: Union[None, Unset, str] = UNSET,
124
127
  authorization: Union[None, Unset, str] = UNSET,
125
- auth_token: Union[None, Unset, str] = UNSET,
126
128
  ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
127
129
  """Execute MCP Tool
128
130
 
@@ -165,8 +167,8 @@ def sync_detailed(
165
167
  graph_id (str): Graph database identifier
166
168
  format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson)
167
169
  test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False.
170
+ token (Union[None, Unset, str]): JWT token for SSE authentication
168
171
  authorization (Union[None, Unset, str]):
169
- auth_token (Union[None, Unset, str]):
170
172
  body (MCPToolCall): Request model for MCP tool execution.
171
173
 
172
174
  Raises:
@@ -182,8 +184,8 @@ def sync_detailed(
182
184
  body=body,
183
185
  format_=format_,
184
186
  test_mode=test_mode,
187
+ token=token,
185
188
  authorization=authorization,
186
- auth_token=auth_token,
187
189
  )
188
190
 
189
191
  response = client.get_httpx_client().request(
@@ -200,8 +202,8 @@ def sync(
200
202
  body: MCPToolCall,
201
203
  format_: Union[None, Unset, str] = UNSET,
202
204
  test_mode: Union[Unset, bool] = False,
205
+ token: Union[None, Unset, str] = UNSET,
203
206
  authorization: Union[None, Unset, str] = UNSET,
204
- auth_token: Union[None, Unset, str] = UNSET,
205
207
  ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
206
208
  """Execute MCP Tool
207
209
 
@@ -244,8 +246,8 @@ def sync(
244
246
  graph_id (str): Graph database identifier
245
247
  format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson)
246
248
  test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False.
249
+ token (Union[None, Unset, str]): JWT token for SSE authentication
247
250
  authorization (Union[None, Unset, str]):
248
- auth_token (Union[None, Unset, str]):
249
251
  body (MCPToolCall): Request model for MCP tool execution.
250
252
 
251
253
  Raises:
@@ -262,8 +264,8 @@ def sync(
262
264
  body=body,
263
265
  format_=format_,
264
266
  test_mode=test_mode,
267
+ token=token,
265
268
  authorization=authorization,
266
- auth_token=auth_token,
267
269
  ).parsed
268
270
 
269
271
 
@@ -274,8 +276,8 @@ async def asyncio_detailed(
274
276
  body: MCPToolCall,
275
277
  format_: Union[None, Unset, str] = UNSET,
276
278
  test_mode: Union[Unset, bool] = False,
279
+ token: Union[None, Unset, str] = UNSET,
277
280
  authorization: Union[None, Unset, str] = UNSET,
278
- auth_token: Union[None, Unset, str] = UNSET,
279
281
  ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
280
282
  """Execute MCP Tool
281
283
 
@@ -318,8 +320,8 @@ async def asyncio_detailed(
318
320
  graph_id (str): Graph database identifier
319
321
  format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson)
320
322
  test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False.
323
+ token (Union[None, Unset, str]): JWT token for SSE authentication
321
324
  authorization (Union[None, Unset, str]):
322
- auth_token (Union[None, Unset, str]):
323
325
  body (MCPToolCall): Request model for MCP tool execution.
324
326
 
325
327
  Raises:
@@ -335,8 +337,8 @@ async def asyncio_detailed(
335
337
  body=body,
336
338
  format_=format_,
337
339
  test_mode=test_mode,
340
+ token=token,
338
341
  authorization=authorization,
339
- auth_token=auth_token,
340
342
  )
341
343
 
342
344
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -351,8 +353,8 @@ async def asyncio(
351
353
  body: MCPToolCall,
352
354
  format_: Union[None, Unset, str] = UNSET,
353
355
  test_mode: Union[Unset, bool] = False,
356
+ token: Union[None, Unset, str] = UNSET,
354
357
  authorization: Union[None, Unset, str] = UNSET,
355
- auth_token: Union[None, Unset, str] = UNSET,
356
358
  ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
357
359
  """Execute MCP Tool
358
360
 
@@ -395,8 +397,8 @@ async def asyncio(
395
397
  graph_id (str): Graph database identifier
396
398
  format_ (Union[None, Unset, str]): Response format override (json, sse, ndjson)
397
399
  test_mode (Union[Unset, bool]): Enable test mode for debugging Default: False.
400
+ token (Union[None, Unset, str]): JWT token for SSE authentication
398
401
  authorization (Union[None, Unset, str]):
399
- auth_token (Union[None, Unset, str]):
400
402
  body (MCPToolCall): Request model for MCP tool execution.
401
403
 
402
404
  Raises:
@@ -414,7 +416,7 @@ async def asyncio(
414
416
  body=body,
415
417
  format_=format_,
416
418
  test_mode=test_mode,
419
+ token=token,
417
420
  authorization=authorization,
418
- auth_token=auth_token,
419
421
  )
420
422
  ).parsed