robosystems-client 0.2.2__py3-none-any.whl → 0.2.4__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 (121) hide show
  1. robosystems_client/api/agent/auto_select_agent.py +1 -41
  2. robosystems_client/api/agent/batch_process_queries.py +1 -41
  3. robosystems_client/api/agent/execute_specific_agent.py +1 -41
  4. robosystems_client/api/agent/get_agent_metadata.py +5 -49
  5. robosystems_client/api/agent/list_agents.py +4 -42
  6. robosystems_client/api/agent/recommend_agent.py +5 -45
  7. robosystems_client/api/auth/generate_sso_token.py +0 -18
  8. robosystems_client/api/auth/get_current_auth_user.py +14 -74
  9. robosystems_client/api/auth/logout_user.py +14 -50
  10. robosystems_client/api/auth/refresh_auth_session.py +14 -50
  11. robosystems_client/api/auth/resend_verification_email.py +14 -74
  12. robosystems_client/api/backup/create_backup.py +5 -45
  13. robosystems_client/api/backup/get_backup_download_url.py +4 -42
  14. robosystems_client/api/backup/get_backup_stats.py +5 -49
  15. robosystems_client/api/backup/list_backups.py +4 -42
  16. robosystems_client/api/backup/restore_backup.py +5 -45
  17. robosystems_client/api/connections/create_connection.py +5 -45
  18. robosystems_client/api/connections/create_link_token.py +5 -45
  19. robosystems_client/api/connections/delete_connection.py +5 -49
  20. robosystems_client/api/connections/exchange_link_token.py +5 -45
  21. robosystems_client/api/connections/get_connection.py +5 -49
  22. robosystems_client/api/connections/get_connection_options.py +5 -49
  23. robosystems_client/api/connections/init_o_auth.py +5 -45
  24. robosystems_client/api/connections/list_connections.py +4 -42
  25. robosystems_client/api/connections/oauth_callback.py +5 -45
  26. robosystems_client/api/connections/sync_connection.py +5 -45
  27. robosystems_client/api/graph_analytics/get_graph_metrics.py +5 -49
  28. robosystems_client/api/graph_analytics/get_graph_usage_stats.py +4 -42
  29. robosystems_client/api/graph_billing/get_current_graph_bill.py +5 -49
  30. robosystems_client/api/graph_billing/get_graph_billing_history.py +4 -42
  31. robosystems_client/api/graph_billing/get_graph_monthly_bill.py +5 -49
  32. robosystems_client/api/graph_billing/get_graph_usage_details.py +4 -42
  33. robosystems_client/api/graph_credits/check_credit_balance.py +0 -38
  34. robosystems_client/api/graph_credits/check_storage_limits.py +1 -45
  35. robosystems_client/api/graph_credits/get_credit_summary.py +1 -45
  36. robosystems_client/api/graph_credits/get_storage_usage.py +0 -38
  37. robosystems_client/api/graph_credits/list_credit_transactions.py +4 -42
  38. robosystems_client/api/graph_health/get_database_health.py +5 -49
  39. robosystems_client/api/graph_info/get_database_info.py +5 -49
  40. robosystems_client/api/graph_limits/get_graph_limits.py +5 -49
  41. robosystems_client/api/graphs/create_graph.py +21 -57
  42. robosystems_client/api/graphs/get_available_extensions.py +131 -15
  43. robosystems_client/api/graphs/get_graphs.py +154 -79
  44. robosystems_client/api/graphs/select_graph.py +117 -49
  45. robosystems_client/api/mcp/call_mcp_tool.py +24 -47
  46. robosystems_client/api/mcp/list_mcp_tools.py +13 -61
  47. robosystems_client/api/operations/cancel_operation.py +1 -45
  48. robosystems_client/api/operations/get_operation_status.py +1 -45
  49. robosystems_client/api/query/execute_cypher_query.py +68 -57
  50. robosystems_client/api/schema/export_graph_schema.py +223 -65
  51. robosystems_client/api/schema/get_graph_schema.py +137 -79
  52. robosystems_client/api/schema/validate_schema.py +5 -45
  53. robosystems_client/api/subgraphs/create_subgraph.py +5 -45
  54. robosystems_client/api/subgraphs/delete_subgraph.py +5 -45
  55. robosystems_client/api/subgraphs/get_subgraph_info.py +5 -49
  56. robosystems_client/api/subgraphs/get_subgraph_quota.py +5 -49
  57. robosystems_client/api/subgraphs/list_subgraphs.py +5 -49
  58. robosystems_client/api/tables/delete_file.py +317 -0
  59. robosystems_client/api/tables/get_file_info.py +249 -0
  60. robosystems_client/api/tables/get_upload_url.py +352 -0
  61. robosystems_client/api/tables/ingest_tables.py +428 -0
  62. robosystems_client/api/tables/list_table_files.py +329 -0
  63. robosystems_client/api/tables/list_tables.py +316 -0
  64. robosystems_client/api/tables/query_tables.py +431 -0
  65. robosystems_client/api/tables/update_file_status.py +395 -0
  66. robosystems_client/api/user/create_user_api_key.py +1 -41
  67. robosystems_client/api/user/get_all_credit_summaries.py +14 -111
  68. robosystems_client/api/user/get_current_user.py +14 -75
  69. robosystems_client/api/user/list_user_api_keys.py +14 -75
  70. robosystems_client/api/user/revoke_user_api_key.py +1 -45
  71. robosystems_client/api/user/update_user.py +1 -41
  72. robosystems_client/api/user/update_user_api_key.py +1 -41
  73. robosystems_client/api/user/update_user_password.py +1 -41
  74. robosystems_client/api/user_analytics/get_detailed_user_analytics.py +0 -38
  75. robosystems_client/api/user_analytics/get_user_usage_overview.py +14 -75
  76. robosystems_client/api/user_limits/get_all_shared_repository_limits.py +14 -105
  77. robosystems_client/api/user_limits/get_shared_repository_limits.py +1 -45
  78. robosystems_client/api/user_limits/get_user_limits.py +14 -75
  79. robosystems_client/api/user_limits/get_user_usage.py +14 -75
  80. robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +1 -45
  81. robosystems_client/api/user_subscriptions/get_repository_credits.py +1 -45
  82. robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +14 -75
  83. robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +0 -38
  84. robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +1 -41
  85. robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +1 -41
  86. robosystems_client/extensions/__init__.py +8 -1
  87. robosystems_client/extensions/auth_integration.py +1 -2
  88. robosystems_client/extensions/graph_client.py +5 -0
  89. robosystems_client/extensions/query_client.py +3 -2
  90. robosystems_client/extensions/sse_client.py +1 -1
  91. robosystems_client/extensions/table_ingest_client.py +36 -40
  92. robosystems_client/extensions/utils.py +2 -2
  93. robosystems_client/models/__init__.py +17 -21
  94. robosystems_client/models/create_graph_request.py +15 -3
  95. robosystems_client/models/cypher_query_request.py +5 -22
  96. robosystems_client/models/{delete_file_v1_graphs_graph_id_tables_files_file_id_delete_response_delete_file_v1_graphs_graph_id_tables_files_file_id_delete.py → delete_file_response.py} +45 -9
  97. robosystems_client/models/file_info.py +169 -0
  98. robosystems_client/models/file_status_update.py +41 -0
  99. robosystems_client/models/get_file_info_response.py +205 -0
  100. robosystems_client/models/list_table_files_response.py +105 -0
  101. robosystems_client/models/schema_export_response.py +4 -2
  102. robosystems_client/models/schema_info_response.py +77 -0
  103. robosystems_client/models/{get_graph_schema_response_getgraphschema.py → schema_info_response_schema.py} +6 -6
  104. robosystems_client/models/schema_validation_response.py +7 -6
  105. robosystems_client/models/table_query_request.py +37 -2
  106. robosystems_client/models/{get_file_info_v1_graphs_graph_id_tables_files_file_id_get_response_get_file_info_v1_graphs_graph_id_tables_files_file_id_get.py → update_file_status_response_updatefilestatus.py} +5 -8
  107. {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/METADATA +2 -4
  108. {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/RECORD +110 -108
  109. robosystems_client/api/tables/delete_file_v1_graphs_graph_id_tables_files_file_id_delete.py +0 -287
  110. robosystems_client/api/tables/get_file_info_v1_graphs_graph_id_tables_files_file_id_get.py +0 -283
  111. robosystems_client/api/tables/get_upload_url_v1_graphs_graph_id_tables_table_name_files_post.py +0 -260
  112. robosystems_client/api/tables/ingest_tables_v1_graphs_graph_id_tables_ingest_post.py +0 -251
  113. robosystems_client/api/tables/list_table_files_v1_graphs_graph_id_tables_table_name_files_get.py +0 -283
  114. robosystems_client/api/tables/list_tables_v1_graphs_graph_id_tables_get.py +0 -224
  115. robosystems_client/api/tables/query_tables_v1_graphs_graph_id_tables_query_post.py +0 -247
  116. robosystems_client/api/tables/update_file_v1_graphs_graph_id_tables_files_file_id_patch.py +0 -306
  117. robosystems_client/models/file_update_request.py +0 -62
  118. robosystems_client/models/list_table_files_v1_graphs_graph_id_tables_table_name_files_get_response_list_table_files_v1_graphs_graph_id_tables_table_name_files_get.py +0 -47
  119. robosystems_client/models/update_file_v1_graphs_graph_id_tables_files_file_id_patch_response_update_file_v1_graphs_graph_id_tables_files_file_id_patch.py +0 -47
  120. {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/WHEEL +0 -0
  121. {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/licenses/LICENSE +0 -0
@@ -6,43 +6,21 @@ import httpx
6
6
  from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
8
  from ...models.credits_summary_response import CreditsSummaryResponse
9
- from ...models.http_validation_error import HTTPValidationError
10
- from ...types import UNSET, Response, Unset
9
+ from ...types import Response
11
10
 
12
11
 
13
- def _get_kwargs(
14
- *,
15
- token: Union[None, Unset, str] = UNSET,
16
- authorization: Union[None, Unset, str] = UNSET,
17
- ) -> dict[str, Any]:
18
- headers: dict[str, Any] = {}
19
- if not isinstance(authorization, Unset):
20
- headers["authorization"] = authorization
21
-
22
- params: dict[str, Any] = {}
23
-
24
- json_token: Union[None, Unset, str]
25
- if isinstance(token, Unset):
26
- json_token = UNSET
27
- else:
28
- json_token = token
29
- params["token"] = json_token
30
-
31
- params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
32
-
12
+ def _get_kwargs() -> dict[str, Any]:
33
13
  _kwargs: dict[str, Any] = {
34
14
  "method": "get",
35
15
  "url": "/v1/user/subscriptions/shared-repositories/credits",
36
- "params": params,
37
16
  }
38
17
 
39
- _kwargs["headers"] = headers
40
18
  return _kwargs
41
19
 
42
20
 
43
21
  def _parse_response(
44
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
45
- ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
23
+ ) -> Optional[Union[Any, CreditsSummaryResponse]]:
46
24
  if response.status_code == 200:
47
25
  response_200 = CreditsSummaryResponse.from_dict(response.json())
48
26
 
@@ -52,11 +30,6 @@ def _parse_response(
52
30
  response_401 = cast(Any, None)
53
31
  return response_401
54
32
 
55
- if response.status_code == 422:
56
- response_422 = HTTPValidationError.from_dict(response.json())
57
-
58
- return response_422
59
-
60
33
  if response.status_code == 500:
61
34
  response_500 = cast(Any, None)
62
35
  return response_500
@@ -69,7 +42,7 @@ def _parse_response(
69
42
 
70
43
  def _build_response(
71
44
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
72
- ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
45
+ ) -> Response[Union[Any, CreditsSummaryResponse]]:
73
46
  return Response(
74
47
  status_code=HTTPStatus(response.status_code),
75
48
  content=response.content,
@@ -81,29 +54,20 @@ def _build_response(
81
54
  def sync_detailed(
82
55
  *,
83
56
  client: AuthenticatedClient,
84
- token: Union[None, Unset, str] = UNSET,
85
- authorization: Union[None, Unset, str] = UNSET,
86
- ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
57
+ ) -> Response[Union[Any, CreditsSummaryResponse]]:
87
58
  """Get Credit Balances
88
59
 
89
60
  Retrieve credit balances for all shared repository subscriptions
90
61
 
91
- Args:
92
- token (Union[None, Unset, str]): JWT token for SSE authentication
93
- authorization (Union[None, Unset, str]):
94
-
95
62
  Raises:
96
63
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
97
64
  httpx.TimeoutException: If the request takes longer than Client.timeout.
98
65
 
99
66
  Returns:
100
- Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]
67
+ Response[Union[Any, CreditsSummaryResponse]]
101
68
  """
102
69
 
103
- kwargs = _get_kwargs(
104
- token=token,
105
- authorization=authorization,
106
- )
70
+ kwargs = _get_kwargs()
107
71
 
108
72
  response = client.get_httpx_client().request(
109
73
  **kwargs,
@@ -115,58 +79,41 @@ def sync_detailed(
115
79
  def sync(
116
80
  *,
117
81
  client: AuthenticatedClient,
118
- token: Union[None, Unset, str] = UNSET,
119
- authorization: Union[None, Unset, str] = UNSET,
120
- ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
82
+ ) -> Optional[Union[Any, CreditsSummaryResponse]]:
121
83
  """Get Credit Balances
122
84
 
123
85
  Retrieve credit balances for all shared repository subscriptions
124
86
 
125
- Args:
126
- token (Union[None, Unset, str]): JWT token for SSE authentication
127
- authorization (Union[None, Unset, str]):
128
-
129
87
  Raises:
130
88
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
131
89
  httpx.TimeoutException: If the request takes longer than Client.timeout.
132
90
 
133
91
  Returns:
134
- Union[Any, CreditsSummaryResponse, HTTPValidationError]
92
+ Union[Any, CreditsSummaryResponse]
135
93
  """
136
94
 
137
95
  return sync_detailed(
138
96
  client=client,
139
- token=token,
140
- authorization=authorization,
141
97
  ).parsed
142
98
 
143
99
 
144
100
  async def asyncio_detailed(
145
101
  *,
146
102
  client: AuthenticatedClient,
147
- token: Union[None, Unset, str] = UNSET,
148
- authorization: Union[None, Unset, str] = UNSET,
149
- ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
103
+ ) -> Response[Union[Any, CreditsSummaryResponse]]:
150
104
  """Get Credit Balances
151
105
 
152
106
  Retrieve credit balances for all shared repository subscriptions
153
107
 
154
- Args:
155
- token (Union[None, Unset, str]): JWT token for SSE authentication
156
- authorization (Union[None, Unset, str]):
157
-
158
108
  Raises:
159
109
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
160
110
  httpx.TimeoutException: If the request takes longer than Client.timeout.
161
111
 
162
112
  Returns:
163
- Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]
113
+ Response[Union[Any, CreditsSummaryResponse]]
164
114
  """
165
115
 
166
- kwargs = _get_kwargs(
167
- token=token,
168
- authorization=authorization,
169
- )
116
+ kwargs = _get_kwargs()
170
117
 
171
118
  response = await client.get_async_httpx_client().request(**kwargs)
172
119
 
@@ -176,29 +123,21 @@ async def asyncio_detailed(
176
123
  async def asyncio(
177
124
  *,
178
125
  client: AuthenticatedClient,
179
- token: Union[None, Unset, str] = UNSET,
180
- authorization: Union[None, Unset, str] = UNSET,
181
- ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
126
+ ) -> Optional[Union[Any, CreditsSummaryResponse]]:
182
127
  """Get Credit Balances
183
128
 
184
129
  Retrieve credit balances for all shared repository subscriptions
185
130
 
186
- Args:
187
- token (Union[None, Unset, str]): JWT token for SSE authentication
188
- authorization (Union[None, Unset, str]):
189
-
190
131
  Raises:
191
132
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
192
133
  httpx.TimeoutException: If the request takes longer than Client.timeout.
193
134
 
194
135
  Returns:
195
- Union[Any, CreditsSummaryResponse, HTTPValidationError]
136
+ Union[Any, CreditsSummaryResponse]
196
137
  """
197
138
 
198
139
  return (
199
140
  await asyncio_detailed(
200
141
  client=client,
201
- token=token,
202
- authorization=authorization,
203
142
  )
204
143
  ).parsed
@@ -13,24 +13,11 @@ from ...types import UNSET, Response, Unset
13
13
  def _get_kwargs(
14
14
  *,
15
15
  active_only: Union[Unset, bool] = True,
16
- token: Union[None, Unset, str] = UNSET,
17
- authorization: Union[None, Unset, str] = UNSET,
18
16
  ) -> dict[str, Any]:
19
- headers: dict[str, Any] = {}
20
- if not isinstance(authorization, Unset):
21
- headers["authorization"] = authorization
22
-
23
17
  params: dict[str, Any] = {}
24
18
 
25
19
  params["active_only"] = active_only
26
20
 
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
21
  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
35
22
 
36
23
  _kwargs: dict[str, Any] = {
@@ -39,7 +26,6 @@ def _get_kwargs(
39
26
  "params": params,
40
27
  }
41
28
 
42
- _kwargs["headers"] = headers
43
29
  return _kwargs
44
30
 
45
31
 
@@ -85,8 +71,6 @@ def sync_detailed(
85
71
  *,
86
72
  client: AuthenticatedClient,
87
73
  active_only: Union[Unset, bool] = True,
88
- token: Union[None, Unset, str] = UNSET,
89
- authorization: Union[None, Unset, str] = UNSET,
90
74
  ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
91
75
  """Get User Subscriptions
92
76
 
@@ -94,8 +78,6 @@ def sync_detailed(
94
78
 
95
79
  Args:
96
80
  active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
97
- token (Union[None, Unset, str]): JWT token for SSE authentication
98
- authorization (Union[None, Unset, str]):
99
81
 
100
82
  Raises:
101
83
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -107,8 +89,6 @@ def sync_detailed(
107
89
 
108
90
  kwargs = _get_kwargs(
109
91
  active_only=active_only,
110
- token=token,
111
- authorization=authorization,
112
92
  )
113
93
 
114
94
  response = client.get_httpx_client().request(
@@ -122,8 +102,6 @@ def sync(
122
102
  *,
123
103
  client: AuthenticatedClient,
124
104
  active_only: Union[Unset, bool] = True,
125
- token: Union[None, Unset, str] = UNSET,
126
- authorization: Union[None, Unset, str] = UNSET,
127
105
  ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
128
106
  """Get User Subscriptions
129
107
 
@@ -131,8 +109,6 @@ def sync(
131
109
 
132
110
  Args:
133
111
  active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
134
- token (Union[None, Unset, str]): JWT token for SSE authentication
135
- authorization (Union[None, Unset, str]):
136
112
 
137
113
  Raises:
138
114
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -145,8 +121,6 @@ def sync(
145
121
  return sync_detailed(
146
122
  client=client,
147
123
  active_only=active_only,
148
- token=token,
149
- authorization=authorization,
150
124
  ).parsed
151
125
 
152
126
 
@@ -154,8 +128,6 @@ async def asyncio_detailed(
154
128
  *,
155
129
  client: AuthenticatedClient,
156
130
  active_only: Union[Unset, bool] = True,
157
- token: Union[None, Unset, str] = UNSET,
158
- authorization: Union[None, Unset, str] = UNSET,
159
131
  ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
160
132
  """Get User Subscriptions
161
133
 
@@ -163,8 +135,6 @@ async def asyncio_detailed(
163
135
 
164
136
  Args:
165
137
  active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
166
- token (Union[None, Unset, str]): JWT token for SSE authentication
167
- authorization (Union[None, Unset, str]):
168
138
 
169
139
  Raises:
170
140
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -176,8 +146,6 @@ async def asyncio_detailed(
176
146
 
177
147
  kwargs = _get_kwargs(
178
148
  active_only=active_only,
179
- token=token,
180
- authorization=authorization,
181
149
  )
182
150
 
183
151
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -189,8 +157,6 @@ async def asyncio(
189
157
  *,
190
158
  client: AuthenticatedClient,
191
159
  active_only: Union[Unset, bool] = True,
192
- token: Union[None, Unset, str] = UNSET,
193
- authorization: Union[None, Unset, str] = UNSET,
194
160
  ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
195
161
  """Get User Subscriptions
196
162
 
@@ -198,8 +164,6 @@ async def asyncio(
198
164
 
199
165
  Args:
200
166
  active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
201
- token (Union[None, Unset, str]): JWT token for SSE authentication
202
- authorization (Union[None, Unset, str]):
203
167
 
204
168
  Raises:
205
169
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -213,7 +177,5 @@ async def asyncio(
213
177
  await asyncio_detailed(
214
178
  client=client,
215
179
  active_only=active_only,
216
- token=token,
217
- authorization=authorization,
218
180
  )
219
181
  ).parsed
@@ -8,34 +8,18 @@ from ...client import AuthenticatedClient, Client
8
8
  from ...models.http_validation_error import HTTPValidationError
9
9
  from ...models.subscription_request import SubscriptionRequest
10
10
  from ...models.subscription_response import SubscriptionResponse
11
- from ...types import UNSET, Response, Unset
11
+ from ...types import Response
12
12
 
13
13
 
14
14
  def _get_kwargs(
15
15
  *,
16
16
  body: SubscriptionRequest,
17
- token: Union[None, Unset, str] = UNSET,
18
- authorization: Union[None, Unset, str] = UNSET,
19
17
  ) -> dict[str, Any]:
20
18
  headers: dict[str, Any] = {}
21
- if not isinstance(authorization, Unset):
22
- headers["authorization"] = authorization
23
-
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}
34
19
 
35
20
  _kwargs: dict[str, Any] = {
36
21
  "method": "post",
37
22
  "url": "/v1/user/subscriptions/shared-repositories/subscribe",
38
- "params": params,
39
23
  }
40
24
 
41
25
  _kwargs["json"] = body.to_dict()
@@ -92,16 +76,12 @@ def sync_detailed(
92
76
  *,
93
77
  client: AuthenticatedClient,
94
78
  body: SubscriptionRequest,
95
- token: Union[None, Unset, str] = UNSET,
96
- authorization: Union[None, Unset, str] = UNSET,
97
79
  ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]:
98
80
  """Subscribe to Shared Repository
99
81
 
100
82
  Create a new subscription to a shared repository add-on with specified tier
101
83
 
102
84
  Args:
103
- token (Union[None, Unset, str]): JWT token for SSE authentication
104
- authorization (Union[None, Unset, str]):
105
85
  body (SubscriptionRequest): Request to create a new subscription.
106
86
 
107
87
  Raises:
@@ -114,8 +94,6 @@ def sync_detailed(
114
94
 
115
95
  kwargs = _get_kwargs(
116
96
  body=body,
117
- token=token,
118
- authorization=authorization,
119
97
  )
120
98
 
121
99
  response = client.get_httpx_client().request(
@@ -129,16 +107,12 @@ def sync(
129
107
  *,
130
108
  client: AuthenticatedClient,
131
109
  body: SubscriptionRequest,
132
- token: Union[None, Unset, str] = UNSET,
133
- authorization: Union[None, Unset, str] = UNSET,
134
110
  ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]:
135
111
  """Subscribe to Shared Repository
136
112
 
137
113
  Create a new subscription to a shared repository add-on with specified tier
138
114
 
139
115
  Args:
140
- token (Union[None, Unset, str]): JWT token for SSE authentication
141
- authorization (Union[None, Unset, str]):
142
116
  body (SubscriptionRequest): Request to create a new subscription.
143
117
 
144
118
  Raises:
@@ -152,8 +126,6 @@ def sync(
152
126
  return sync_detailed(
153
127
  client=client,
154
128
  body=body,
155
- token=token,
156
- authorization=authorization,
157
129
  ).parsed
158
130
 
159
131
 
@@ -161,16 +133,12 @@ async def asyncio_detailed(
161
133
  *,
162
134
  client: AuthenticatedClient,
163
135
  body: SubscriptionRequest,
164
- token: Union[None, Unset, str] = UNSET,
165
- authorization: Union[None, Unset, str] = UNSET,
166
136
  ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]:
167
137
  """Subscribe to Shared Repository
168
138
 
169
139
  Create a new subscription to a shared repository add-on with specified tier
170
140
 
171
141
  Args:
172
- token (Union[None, Unset, str]): JWT token for SSE authentication
173
- authorization (Union[None, Unset, str]):
174
142
  body (SubscriptionRequest): Request to create a new subscription.
175
143
 
176
144
  Raises:
@@ -183,8 +151,6 @@ async def asyncio_detailed(
183
151
 
184
152
  kwargs = _get_kwargs(
185
153
  body=body,
186
- token=token,
187
- authorization=authorization,
188
154
  )
189
155
 
190
156
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -196,16 +162,12 @@ async def asyncio(
196
162
  *,
197
163
  client: AuthenticatedClient,
198
164
  body: SubscriptionRequest,
199
- token: Union[None, Unset, str] = UNSET,
200
- authorization: Union[None, Unset, str] = UNSET,
201
165
  ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]:
202
166
  """Subscribe to Shared Repository
203
167
 
204
168
  Create a new subscription to a shared repository add-on with specified tier
205
169
 
206
170
  Args:
207
- token (Union[None, Unset, str]): JWT token for SSE authentication
208
- authorization (Union[None, Unset, str]):
209
171
  body (SubscriptionRequest): Request to create a new subscription.
210
172
 
211
173
  Raises:
@@ -220,7 +182,5 @@ async def asyncio(
220
182
  await asyncio_detailed(
221
183
  client=client,
222
184
  body=body,
223
- token=token,
224
- authorization=authorization,
225
185
  )
226
186
  ).parsed
@@ -7,35 +7,19 @@ from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
8
  from ...models.http_validation_error import HTTPValidationError
9
9
  from ...models.tier_upgrade_request import TierUpgradeRequest
10
- from ...types import UNSET, Response, Unset
10
+ from ...types import Response
11
11
 
12
12
 
13
13
  def _get_kwargs(
14
14
  subscription_id: str,
15
15
  *,
16
16
  body: TierUpgradeRequest,
17
- token: Union[None, Unset, str] = UNSET,
18
- authorization: Union[None, Unset, str] = UNSET,
19
17
  ) -> dict[str, Any]:
20
18
  headers: dict[str, Any] = {}
21
- if not isinstance(authorization, Unset):
22
- headers["authorization"] = authorization
23
-
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}
34
19
 
35
20
  _kwargs: dict[str, Any] = {
36
21
  "method": "put",
37
22
  "url": f"/v1/user/subscriptions/shared-repositories/{subscription_id}/upgrade",
38
- "params": params,
39
23
  }
40
24
 
41
25
  _kwargs["json"] = body.to_dict()
@@ -96,8 +80,6 @@ def sync_detailed(
96
80
  *,
97
81
  client: AuthenticatedClient,
98
82
  body: TierUpgradeRequest,
99
- token: Union[None, Unset, str] = UNSET,
100
- authorization: Union[None, Unset, str] = UNSET,
101
83
  ) -> Response[Union[Any, HTTPValidationError]]:
102
84
  """Upgrade Subscription Tier
103
85
 
@@ -105,8 +87,6 @@ def sync_detailed(
105
87
 
106
88
  Args:
107
89
  subscription_id (str):
108
- token (Union[None, Unset, str]): JWT token for SSE authentication
109
- authorization (Union[None, Unset, str]):
110
90
  body (TierUpgradeRequest): Request to upgrade subscription tier.
111
91
 
112
92
  Raises:
@@ -120,8 +100,6 @@ def sync_detailed(
120
100
  kwargs = _get_kwargs(
121
101
  subscription_id=subscription_id,
122
102
  body=body,
123
- token=token,
124
- authorization=authorization,
125
103
  )
126
104
 
127
105
  response = client.get_httpx_client().request(
@@ -136,8 +114,6 @@ def sync(
136
114
  *,
137
115
  client: AuthenticatedClient,
138
116
  body: TierUpgradeRequest,
139
- token: Union[None, Unset, str] = UNSET,
140
- authorization: Union[None, Unset, str] = UNSET,
141
117
  ) -> Optional[Union[Any, HTTPValidationError]]:
142
118
  """Upgrade Subscription Tier
143
119
 
@@ -145,8 +121,6 @@ def sync(
145
121
 
146
122
  Args:
147
123
  subscription_id (str):
148
- token (Union[None, Unset, str]): JWT token for SSE authentication
149
- authorization (Union[None, Unset, str]):
150
124
  body (TierUpgradeRequest): Request to upgrade subscription tier.
151
125
 
152
126
  Raises:
@@ -161,8 +135,6 @@ def sync(
161
135
  subscription_id=subscription_id,
162
136
  client=client,
163
137
  body=body,
164
- token=token,
165
- authorization=authorization,
166
138
  ).parsed
167
139
 
168
140
 
@@ -171,8 +143,6 @@ async def asyncio_detailed(
171
143
  *,
172
144
  client: AuthenticatedClient,
173
145
  body: TierUpgradeRequest,
174
- token: Union[None, Unset, str] = UNSET,
175
- authorization: Union[None, Unset, str] = UNSET,
176
146
  ) -> Response[Union[Any, HTTPValidationError]]:
177
147
  """Upgrade Subscription Tier
178
148
 
@@ -180,8 +150,6 @@ async def asyncio_detailed(
180
150
 
181
151
  Args:
182
152
  subscription_id (str):
183
- token (Union[None, Unset, str]): JWT token for SSE authentication
184
- authorization (Union[None, Unset, str]):
185
153
  body (TierUpgradeRequest): Request to upgrade subscription tier.
186
154
 
187
155
  Raises:
@@ -195,8 +163,6 @@ async def asyncio_detailed(
195
163
  kwargs = _get_kwargs(
196
164
  subscription_id=subscription_id,
197
165
  body=body,
198
- token=token,
199
- authorization=authorization,
200
166
  )
201
167
 
202
168
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -209,8 +175,6 @@ async def asyncio(
209
175
  *,
210
176
  client: AuthenticatedClient,
211
177
  body: TierUpgradeRequest,
212
- token: Union[None, Unset, str] = UNSET,
213
- authorization: Union[None, Unset, str] = UNSET,
214
178
  ) -> Optional[Union[Any, HTTPValidationError]]:
215
179
  """Upgrade Subscription Tier
216
180
 
@@ -218,8 +182,6 @@ async def asyncio(
218
182
 
219
183
  Args:
220
184
  subscription_id (str):
221
- token (Union[None, Unset, str]): JWT token for SSE authentication
222
- authorization (Union[None, Unset, str]):
223
185
  body (TierUpgradeRequest): Request to upgrade subscription tier.
224
186
 
225
187
  Raises:
@@ -235,7 +197,5 @@ async def asyncio(
235
197
  subscription_id=subscription_id,
236
198
  client=client,
237
199
  body=body,
238
- token=token,
239
- authorization=authorization,
240
200
  )
241
201
  ).parsed
@@ -94,6 +94,7 @@ except ImportError:
94
94
  HAS_PANDAS = False
95
95
  DataFrameQueryClient = None
96
96
  # Set placeholders for optional functions
97
+ query_result_to_dataframe = None
97
98
  parse_datetime_columns = None
98
99
  _stream_to_dataframe = None
99
100
  dataframe_to_cypher_params = None
@@ -189,14 +190,20 @@ def stream_query(graph_id: str, query: str, parameters=None, chunk_size=None):
189
190
 
190
191
 
191
192
  # DataFrame convenience functions (if pandas is available)
192
- if HAS_PANDAS:
193
+ if (
194
+ HAS_PANDAS
195
+ and query_result_to_dataframe is not None
196
+ and _stream_to_dataframe is not None
197
+ ):
193
198
 
194
199
  def query_to_dataframe(graph_id: str, query: str, parameters=None, **kwargs):
195
200
  """Execute query and return results as pandas DataFrame"""
201
+ assert query_result_to_dataframe is not None
196
202
  result = execute_query(graph_id, query, parameters)
197
203
  return query_result_to_dataframe(result, **kwargs)
198
204
 
199
205
  def stream_to_dataframe(graph_id: str, query: str, parameters=None, chunk_size=10000):
200
206
  """Stream query results and return as pandas DataFrame"""
207
+ assert _stream_to_dataframe is not None
201
208
  stream = stream_query(graph_id, query, parameters, chunk_size)
202
209
  return _stream_to_dataframe(stream, chunk_size)
@@ -60,12 +60,11 @@ class AuthenticatedExtensions(RoboSystemsExtensions):
60
60
 
61
61
  request = CypherQueryRequest(query=query, parameters=parameters or {})
62
62
 
63
- # Pass the token parameter along with the client
63
+ # Execute the query
64
64
  response = sync_detailed(
65
65
  graph_id=graph_id,
66
66
  client=self._authenticated_client,
67
67
  body=request,
68
- token=self._authenticated_client.token,
69
68
  )
70
69
 
71
70
  if response.parsed:
@@ -58,6 +58,7 @@ class GraphClient:
58
58
  self,
59
59
  metadata: GraphMetadata,
60
60
  initial_entity: Optional[InitialEntityData] = None,
61
+ create_entity: bool = True,
61
62
  timeout: int = 60,
62
63
  poll_interval: int = 2,
63
64
  on_progress: Optional[Callable[[str], None]] = None,
@@ -68,6 +69,9 @@ class GraphClient:
68
69
  Args:
69
70
  metadata: Graph metadata
70
71
  initial_entity: Optional initial entity data
72
+ create_entity: Whether to create the entity node and upload initial data.
73
+ Only applies when initial_entity is provided. Set to False to create
74
+ graph without populating entity data (useful for file-based ingestion).
71
75
  timeout: Maximum time to wait in seconds
72
76
  poll_interval: Time between status checks in seconds
73
77
  on_progress: Callback for progress updates
@@ -121,6 +125,7 @@ class GraphClient:
121
125
  graph_create = CreateGraphRequest(
122
126
  metadata=api_metadata,
123
127
  initial_entity=initial_entity_dict,
128
+ create_entity=create_entity,
124
129
  )
125
130
 
126
131
  if on_progress:
@@ -476,7 +476,7 @@ class QueryClient:
476
476
  self,
477
477
  graph_id: str,
478
478
  queries: List[str],
479
- parameters_list: Optional[List[Dict[str, Any]]] = None,
479
+ parameters_list: Optional[List[Optional[Dict[str, Any]]]] = None,
480
480
  parallel: bool = False,
481
481
  ) -> List[Union[QueryResult, Dict[str, Any]]]:
482
482
  """Execute multiple queries in batch
@@ -497,7 +497,8 @@ class QueryClient:
497
497
  ... ])
498
498
  """
499
499
  if parameters_list is None:
500
- parameters_list = [None] * len(queries)
500
+ # Create a list of None values for each query
501
+ parameters_list = [None for _ in queries]
501
502
 
502
503
  if len(queries) != len(parameters_list):
503
504
  raise ValueError("queries and parameters_list must have same length")