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.
- robosystems_client/api/agent/auto_select_agent.py +1 -41
- robosystems_client/api/agent/batch_process_queries.py +1 -41
- robosystems_client/api/agent/execute_specific_agent.py +1 -41
- robosystems_client/api/agent/get_agent_metadata.py +5 -49
- robosystems_client/api/agent/list_agents.py +4 -42
- robosystems_client/api/agent/recommend_agent.py +5 -45
- robosystems_client/api/auth/generate_sso_token.py +0 -18
- robosystems_client/api/auth/get_current_auth_user.py +14 -74
- robosystems_client/api/auth/logout_user.py +14 -50
- robosystems_client/api/auth/refresh_auth_session.py +14 -50
- robosystems_client/api/auth/resend_verification_email.py +14 -74
- robosystems_client/api/backup/create_backup.py +5 -45
- robosystems_client/api/backup/get_backup_download_url.py +4 -42
- robosystems_client/api/backup/get_backup_stats.py +5 -49
- robosystems_client/api/backup/list_backups.py +4 -42
- robosystems_client/api/backup/restore_backup.py +5 -45
- robosystems_client/api/connections/create_connection.py +5 -45
- robosystems_client/api/connections/create_link_token.py +5 -45
- robosystems_client/api/connections/delete_connection.py +5 -49
- robosystems_client/api/connections/exchange_link_token.py +5 -45
- robosystems_client/api/connections/get_connection.py +5 -49
- robosystems_client/api/connections/get_connection_options.py +5 -49
- robosystems_client/api/connections/init_o_auth.py +5 -45
- robosystems_client/api/connections/list_connections.py +4 -42
- robosystems_client/api/connections/oauth_callback.py +5 -45
- robosystems_client/api/connections/sync_connection.py +5 -45
- robosystems_client/api/graph_analytics/get_graph_metrics.py +5 -49
- robosystems_client/api/graph_analytics/get_graph_usage_stats.py +4 -42
- robosystems_client/api/graph_billing/get_current_graph_bill.py +5 -49
- robosystems_client/api/graph_billing/get_graph_billing_history.py +4 -42
- robosystems_client/api/graph_billing/get_graph_monthly_bill.py +5 -49
- robosystems_client/api/graph_billing/get_graph_usage_details.py +4 -42
- robosystems_client/api/graph_credits/check_credit_balance.py +0 -38
- robosystems_client/api/graph_credits/check_storage_limits.py +1 -45
- robosystems_client/api/graph_credits/get_credit_summary.py +1 -45
- robosystems_client/api/graph_credits/get_storage_usage.py +0 -38
- robosystems_client/api/graph_credits/list_credit_transactions.py +4 -42
- robosystems_client/api/graph_health/get_database_health.py +5 -49
- robosystems_client/api/graph_info/get_database_info.py +5 -49
- robosystems_client/api/graph_limits/get_graph_limits.py +5 -49
- robosystems_client/api/graphs/create_graph.py +21 -57
- robosystems_client/api/graphs/get_available_extensions.py +131 -15
- robosystems_client/api/graphs/get_graphs.py +154 -79
- robosystems_client/api/graphs/select_graph.py +117 -49
- robosystems_client/api/mcp/call_mcp_tool.py +24 -47
- robosystems_client/api/mcp/list_mcp_tools.py +13 -61
- robosystems_client/api/operations/cancel_operation.py +1 -45
- robosystems_client/api/operations/get_operation_status.py +1 -45
- robosystems_client/api/query/execute_cypher_query.py +68 -57
- robosystems_client/api/schema/export_graph_schema.py +223 -65
- robosystems_client/api/schema/get_graph_schema.py +137 -79
- robosystems_client/api/schema/validate_schema.py +5 -45
- robosystems_client/api/subgraphs/create_subgraph.py +5 -45
- robosystems_client/api/subgraphs/delete_subgraph.py +5 -45
- robosystems_client/api/subgraphs/get_subgraph_info.py +5 -49
- robosystems_client/api/subgraphs/get_subgraph_quota.py +5 -49
- robosystems_client/api/subgraphs/list_subgraphs.py +5 -49
- robosystems_client/api/tables/delete_file.py +317 -0
- robosystems_client/api/tables/get_file_info.py +249 -0
- robosystems_client/api/tables/get_upload_url.py +352 -0
- robosystems_client/api/tables/ingest_tables.py +428 -0
- robosystems_client/api/tables/list_table_files.py +329 -0
- robosystems_client/api/tables/list_tables.py +316 -0
- robosystems_client/api/tables/query_tables.py +431 -0
- robosystems_client/api/tables/update_file_status.py +395 -0
- robosystems_client/api/user/create_user_api_key.py +1 -41
- robosystems_client/api/user/get_all_credit_summaries.py +14 -111
- robosystems_client/api/user/get_current_user.py +14 -75
- robosystems_client/api/user/list_user_api_keys.py +14 -75
- robosystems_client/api/user/revoke_user_api_key.py +1 -45
- robosystems_client/api/user/update_user.py +1 -41
- robosystems_client/api/user/update_user_api_key.py +1 -41
- robosystems_client/api/user/update_user_password.py +1 -41
- robosystems_client/api/user_analytics/get_detailed_user_analytics.py +0 -38
- robosystems_client/api/user_analytics/get_user_usage_overview.py +14 -75
- robosystems_client/api/user_limits/get_all_shared_repository_limits.py +14 -105
- robosystems_client/api/user_limits/get_shared_repository_limits.py +1 -45
- robosystems_client/api/user_limits/get_user_limits.py +14 -75
- robosystems_client/api/user_limits/get_user_usage.py +14 -75
- robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +1 -45
- robosystems_client/api/user_subscriptions/get_repository_credits.py +1 -45
- robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +14 -75
- robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +0 -38
- robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +1 -41
- robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +1 -41
- robosystems_client/extensions/__init__.py +8 -1
- robosystems_client/extensions/auth_integration.py +1 -2
- robosystems_client/extensions/graph_client.py +5 -0
- robosystems_client/extensions/query_client.py +3 -2
- robosystems_client/extensions/sse_client.py +1 -1
- robosystems_client/extensions/table_ingest_client.py +36 -40
- robosystems_client/extensions/utils.py +2 -2
- robosystems_client/models/__init__.py +17 -21
- robosystems_client/models/create_graph_request.py +15 -3
- robosystems_client/models/cypher_query_request.py +5 -22
- 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
- robosystems_client/models/file_info.py +169 -0
- robosystems_client/models/file_status_update.py +41 -0
- robosystems_client/models/get_file_info_response.py +205 -0
- robosystems_client/models/list_table_files_response.py +105 -0
- robosystems_client/models/schema_export_response.py +4 -2
- robosystems_client/models/schema_info_response.py +77 -0
- robosystems_client/models/{get_graph_schema_response_getgraphschema.py → schema_info_response_schema.py} +6 -6
- robosystems_client/models/schema_validation_response.py +7 -6
- robosystems_client/models/table_query_request.py +37 -2
- 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
- {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/METADATA +2 -4
- {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/RECORD +110 -108
- robosystems_client/api/tables/delete_file_v1_graphs_graph_id_tables_files_file_id_delete.py +0 -287
- robosystems_client/api/tables/get_file_info_v1_graphs_graph_id_tables_files_file_id_get.py +0 -283
- robosystems_client/api/tables/get_upload_url_v1_graphs_graph_id_tables_table_name_files_post.py +0 -260
- robosystems_client/api/tables/ingest_tables_v1_graphs_graph_id_tables_ingest_post.py +0 -251
- robosystems_client/api/tables/list_table_files_v1_graphs_graph_id_tables_table_name_files_get.py +0 -283
- robosystems_client/api/tables/list_tables_v1_graphs_graph_id_tables_get.py +0 -224
- robosystems_client/api/tables/query_tables_v1_graphs_graph_id_tables_query_post.py +0 -247
- robosystems_client/api/tables/update_file_v1_graphs_graph_id_tables_files_file_id_patch.py +0 -306
- robosystems_client/models/file_update_request.py +0 -62
- 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
- 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
- {robosystems_client-0.2.2.dist-info → robosystems_client-0.2.4.dist-info}/WHEEL +0 -0
- {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 ...
|
|
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
|
|
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
|
|
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
|
-
|
|
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
|
|
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
|
-
|
|
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
|
|
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
|
-
|
|
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
|
|
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
|
-
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
-
#
|
|
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
|
-
|
|
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")
|