robosystems-client 0.1.17__py3-none-any.whl → 0.1.19__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/__init__.py +15 -4
- robosystems_client/api/agent/auto_select_agent.py +26 -1
- robosystems_client/api/agent/batch_process_queries.py +26 -1
- robosystems_client/api/agent/execute_specific_agent.py +26 -1
- robosystems_client/api/agent/get_agent_metadata.py +26 -1
- robosystems_client/api/agent/list_agents.py +21 -1
- robosystems_client/api/agent/recommend_agent.py +26 -1
- robosystems_client/api/backup/create_backup.py +26 -1
- robosystems_client/api/backup/export_backup.py +26 -1
- robosystems_client/api/backup/get_backup_download_url.py +21 -1
- robosystems_client/api/backup/get_backup_stats.py +26 -1
- robosystems_client/api/backup/list_backups.py +21 -1
- robosystems_client/api/backup/restore_backup.py +26 -1
- robosystems_client/api/connections/create_connection.py +26 -1
- robosystems_client/api/connections/create_link_token.py +26 -1
- robosystems_client/api/connections/delete_connection.py +26 -1
- robosystems_client/api/connections/exchange_link_token.py +26 -1
- robosystems_client/api/connections/get_connection.py +26 -1
- robosystems_client/api/connections/get_connection_options.py +26 -1
- robosystems_client/api/connections/init_o_auth.py +26 -1
- robosystems_client/api/connections/list_connections.py +21 -1
- robosystems_client/api/connections/oauth_callback.py +26 -1
- robosystems_client/api/connections/sync_connection.py +26 -1
- robosystems_client/api/copy/copy_data_to_graph.py +26 -1
- robosystems_client/api/graph_analytics/get_graph_metrics.py +26 -1
- robosystems_client/api/graph_analytics/get_graph_usage_stats.py +21 -1
- robosystems_client/api/graph_billing/get_current_graph_bill.py +26 -1
- robosystems_client/api/graph_billing/get_graph_billing_history.py +21 -1
- robosystems_client/api/graph_billing/get_graph_monthly_bill.py +26 -1
- robosystems_client/api/graph_billing/get_graph_usage_details.py +21 -1
- robosystems_client/api/graph_credits/check_credit_balance.py +21 -1
- robosystems_client/api/graph_credits/check_storage_limits.py +26 -1
- robosystems_client/api/graph_credits/get_credit_summary.py +26 -1
- robosystems_client/api/graph_credits/get_storage_usage.py +21 -1
- robosystems_client/api/graph_credits/list_credit_transactions.py +21 -1
- robosystems_client/api/graph_health/get_database_health.py +26 -1
- robosystems_client/api/graph_info/get_database_info.py +26 -1
- robosystems_client/api/graph_limits/get_graph_limits.py +26 -1
- robosystems_client/api/{create → graphs}/create_graph.py +26 -1
- robosystems_client/api/{create → graphs}/get_available_extensions.py +1 -1
- robosystems_client/api/{user/get_user_graphs.py → graphs/get_graphs.py} +26 -1
- robosystems_client/api/{user/select_user_graph.py → graphs/select_graph.py} +30 -5
- robosystems_client/api/mcp/call_mcp_tool.py +21 -1
- robosystems_client/api/mcp/list_mcp_tools.py +26 -1
- robosystems_client/api/operations/cancel_operation.py +25 -0
- robosystems_client/api/operations/get_operation_status.py +25 -0
- robosystems_client/api/operations/stream_operation_events.py +20 -0
- robosystems_client/api/query/execute_cypher_query.py +21 -1
- robosystems_client/api/schema/export_graph_schema.py +21 -1
- robosystems_client/api/schema/get_graph_schema_info.py +26 -1
- robosystems_client/api/schema/list_schema_extensions.py +26 -1
- robosystems_client/api/schema/validate_schema.py +26 -1
- robosystems_client/api/subgraphs/create_subgraph.py +26 -1
- robosystems_client/api/subgraphs/delete_subgraph.py +26 -1
- robosystems_client/api/subgraphs/get_subgraph_info.py +26 -1
- robosystems_client/api/subgraphs/get_subgraph_quota.py +26 -1
- robosystems_client/api/subgraphs/list_subgraphs.py +26 -1
- robosystems_client/api/user/create_user_api_key.py +25 -0
- robosystems_client/api/user/get_all_credit_summaries.py +25 -0
- robosystems_client/api/user/get_current_user.py +25 -0
- robosystems_client/api/user/list_user_api_keys.py +25 -0
- robosystems_client/api/user/revoke_user_api_key.py +25 -0
- robosystems_client/api/user/update_user.py +25 -0
- robosystems_client/api/user/update_user_api_key.py +25 -0
- robosystems_client/api/user/update_user_password.py +25 -0
- robosystems_client/api/user_analytics/get_detailed_user_analytics.py +20 -0
- robosystems_client/api/user_analytics/get_user_usage_overview.py +25 -0
- robosystems_client/api/user_limits/get_all_shared_repository_limits.py +25 -0
- robosystems_client/api/user_limits/get_shared_repository_limits.py +25 -0
- robosystems_client/api/user_limits/get_user_limits.py +25 -0
- robosystems_client/api/user_limits/get_user_usage.py +25 -0
- robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +25 -0
- robosystems_client/api/user_subscriptions/get_repository_credits.py +25 -0
- robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +25 -0
- robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +20 -0
- robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +25 -0
- robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +25 -0
- robosystems_client/extensions/__init__.py +70 -0
- robosystems_client/extensions/auth_integration.py +14 -1
- robosystems_client/extensions/copy_client.py +32 -22
- robosystems_client/extensions/dataframe_utils.py +455 -0
- robosystems_client/extensions/extensions.py +16 -0
- robosystems_client/extensions/operation_client.py +43 -21
- robosystems_client/extensions/query_client.py +109 -12
- robosystems_client/extensions/tests/test_dataframe_utils.py +334 -0
- robosystems_client/extensions/tests/test_integration.py +1 -1
- robosystems_client/extensions/tests/test_token_utils.py +274 -0
- robosystems_client/extensions/token_utils.py +417 -0
- robosystems_client/extensions/utils.py +32 -2
- robosystems_client/models/__init__.py +0 -2
- robosystems_client/models/api_key_info.py +20 -0
- robosystems_client/models/create_api_key_request.py +20 -0
- {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/METADATA +1 -1
- {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/RECORD +97 -95
- robosystems_client/api/auth/sso_login.py +0 -177
- robosystems_client/models/sso_login_request.py +0 -60
- /robosystems_client/api/{create → graphs}/__init__.py +0 -0
- {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/WHEEL +0 -0
- {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/licenses/LICENSE +0 -0
|
@@ -1,177 +0,0 @@
|
|
|
1
|
-
from http import HTTPStatus
|
|
2
|
-
from typing import Any, Optional, Union
|
|
3
|
-
|
|
4
|
-
import httpx
|
|
5
|
-
|
|
6
|
-
from ... import errors
|
|
7
|
-
from ...client import AuthenticatedClient, Client
|
|
8
|
-
from ...models.auth_response import AuthResponse
|
|
9
|
-
from ...models.error_response import ErrorResponse
|
|
10
|
-
from ...models.http_validation_error import HTTPValidationError
|
|
11
|
-
from ...models.sso_login_request import SSOLoginRequest
|
|
12
|
-
from ...types import Response
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
def _get_kwargs(
|
|
16
|
-
*,
|
|
17
|
-
body: SSOLoginRequest,
|
|
18
|
-
) -> dict[str, Any]:
|
|
19
|
-
headers: dict[str, Any] = {}
|
|
20
|
-
|
|
21
|
-
_kwargs: dict[str, Any] = {
|
|
22
|
-
"method": "post",
|
|
23
|
-
"url": "/v1/auth/sso-login",
|
|
24
|
-
}
|
|
25
|
-
|
|
26
|
-
_kwargs["json"] = body.to_dict()
|
|
27
|
-
|
|
28
|
-
headers["Content-Type"] = "application/json"
|
|
29
|
-
|
|
30
|
-
_kwargs["headers"] = headers
|
|
31
|
-
return _kwargs
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
def _parse_response(
|
|
35
|
-
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
|
36
|
-
) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
|
|
37
|
-
if response.status_code == 200:
|
|
38
|
-
response_200 = AuthResponse.from_dict(response.json())
|
|
39
|
-
|
|
40
|
-
return response_200
|
|
41
|
-
if response.status_code == 401:
|
|
42
|
-
response_401 = ErrorResponse.from_dict(response.json())
|
|
43
|
-
|
|
44
|
-
return response_401
|
|
45
|
-
if response.status_code == 422:
|
|
46
|
-
response_422 = HTTPValidationError.from_dict(response.json())
|
|
47
|
-
|
|
48
|
-
return response_422
|
|
49
|
-
if client.raise_on_unexpected_status:
|
|
50
|
-
raise errors.UnexpectedStatus(response.status_code, response.content)
|
|
51
|
-
else:
|
|
52
|
-
return None
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
def _build_response(
|
|
56
|
-
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
|
57
|
-
) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
|
|
58
|
-
return Response(
|
|
59
|
-
status_code=HTTPStatus(response.status_code),
|
|
60
|
-
content=response.content,
|
|
61
|
-
headers=response.headers,
|
|
62
|
-
parsed=_parse_response(client=client, response=response),
|
|
63
|
-
)
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
def sync_detailed(
|
|
67
|
-
*,
|
|
68
|
-
client: Union[AuthenticatedClient, Client],
|
|
69
|
-
body: SSOLoginRequest,
|
|
70
|
-
) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
|
|
71
|
-
"""SSO Login
|
|
72
|
-
|
|
73
|
-
Authenticate user with SSO token and establish session.
|
|
74
|
-
|
|
75
|
-
Args:
|
|
76
|
-
body (SSOLoginRequest): SSO login request model.
|
|
77
|
-
|
|
78
|
-
Raises:
|
|
79
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
80
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
81
|
-
|
|
82
|
-
Returns:
|
|
83
|
-
Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]
|
|
84
|
-
"""
|
|
85
|
-
|
|
86
|
-
kwargs = _get_kwargs(
|
|
87
|
-
body=body,
|
|
88
|
-
)
|
|
89
|
-
|
|
90
|
-
response = client.get_httpx_client().request(
|
|
91
|
-
**kwargs,
|
|
92
|
-
)
|
|
93
|
-
|
|
94
|
-
return _build_response(client=client, response=response)
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
def sync(
|
|
98
|
-
*,
|
|
99
|
-
client: Union[AuthenticatedClient, Client],
|
|
100
|
-
body: SSOLoginRequest,
|
|
101
|
-
) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
|
|
102
|
-
"""SSO Login
|
|
103
|
-
|
|
104
|
-
Authenticate user with SSO token and establish session.
|
|
105
|
-
|
|
106
|
-
Args:
|
|
107
|
-
body (SSOLoginRequest): SSO login request model.
|
|
108
|
-
|
|
109
|
-
Raises:
|
|
110
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
111
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
112
|
-
|
|
113
|
-
Returns:
|
|
114
|
-
Union[AuthResponse, ErrorResponse, HTTPValidationError]
|
|
115
|
-
"""
|
|
116
|
-
|
|
117
|
-
return sync_detailed(
|
|
118
|
-
client=client,
|
|
119
|
-
body=body,
|
|
120
|
-
).parsed
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
async def asyncio_detailed(
|
|
124
|
-
*,
|
|
125
|
-
client: Union[AuthenticatedClient, Client],
|
|
126
|
-
body: SSOLoginRequest,
|
|
127
|
-
) -> Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
|
|
128
|
-
"""SSO Login
|
|
129
|
-
|
|
130
|
-
Authenticate user with SSO token and establish session.
|
|
131
|
-
|
|
132
|
-
Args:
|
|
133
|
-
body (SSOLoginRequest): SSO login request model.
|
|
134
|
-
|
|
135
|
-
Raises:
|
|
136
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
137
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
138
|
-
|
|
139
|
-
Returns:
|
|
140
|
-
Response[Union[AuthResponse, ErrorResponse, HTTPValidationError]]
|
|
141
|
-
"""
|
|
142
|
-
|
|
143
|
-
kwargs = _get_kwargs(
|
|
144
|
-
body=body,
|
|
145
|
-
)
|
|
146
|
-
|
|
147
|
-
response = await client.get_async_httpx_client().request(**kwargs)
|
|
148
|
-
|
|
149
|
-
return _build_response(client=client, response=response)
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
async def asyncio(
|
|
153
|
-
*,
|
|
154
|
-
client: Union[AuthenticatedClient, Client],
|
|
155
|
-
body: SSOLoginRequest,
|
|
156
|
-
) -> Optional[Union[AuthResponse, ErrorResponse, HTTPValidationError]]:
|
|
157
|
-
"""SSO Login
|
|
158
|
-
|
|
159
|
-
Authenticate user with SSO token and establish session.
|
|
160
|
-
|
|
161
|
-
Args:
|
|
162
|
-
body (SSOLoginRequest): SSO login request model.
|
|
163
|
-
|
|
164
|
-
Raises:
|
|
165
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
166
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
167
|
-
|
|
168
|
-
Returns:
|
|
169
|
-
Union[AuthResponse, ErrorResponse, HTTPValidationError]
|
|
170
|
-
"""
|
|
171
|
-
|
|
172
|
-
return (
|
|
173
|
-
await asyncio_detailed(
|
|
174
|
-
client=client,
|
|
175
|
-
body=body,
|
|
176
|
-
)
|
|
177
|
-
).parsed
|
|
@@ -1,60 +0,0 @@
|
|
|
1
|
-
from collections.abc import Mapping
|
|
2
|
-
from typing import Any, TypeVar
|
|
3
|
-
|
|
4
|
-
from attrs import define as _attrs_define
|
|
5
|
-
from attrs import field as _attrs_field
|
|
6
|
-
|
|
7
|
-
T = TypeVar("T", bound="SSOLoginRequest")
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
@_attrs_define
|
|
11
|
-
class SSOLoginRequest:
|
|
12
|
-
"""SSO login request model.
|
|
13
|
-
|
|
14
|
-
Attributes:
|
|
15
|
-
token (str): Temporary SSO token
|
|
16
|
-
"""
|
|
17
|
-
|
|
18
|
-
token: str
|
|
19
|
-
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
20
|
-
|
|
21
|
-
def to_dict(self) -> dict[str, Any]:
|
|
22
|
-
token = self.token
|
|
23
|
-
|
|
24
|
-
field_dict: dict[str, Any] = {}
|
|
25
|
-
field_dict.update(self.additional_properties)
|
|
26
|
-
field_dict.update(
|
|
27
|
-
{
|
|
28
|
-
"token": token,
|
|
29
|
-
}
|
|
30
|
-
)
|
|
31
|
-
|
|
32
|
-
return field_dict
|
|
33
|
-
|
|
34
|
-
@classmethod
|
|
35
|
-
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
|
|
36
|
-
d = dict(src_dict)
|
|
37
|
-
token = d.pop("token")
|
|
38
|
-
|
|
39
|
-
sso_login_request = cls(
|
|
40
|
-
token=token,
|
|
41
|
-
)
|
|
42
|
-
|
|
43
|
-
sso_login_request.additional_properties = d
|
|
44
|
-
return sso_login_request
|
|
45
|
-
|
|
46
|
-
@property
|
|
47
|
-
def additional_keys(self) -> list[str]:
|
|
48
|
-
return list(self.additional_properties.keys())
|
|
49
|
-
|
|
50
|
-
def __getitem__(self, key: str) -> Any:
|
|
51
|
-
return self.additional_properties[key]
|
|
52
|
-
|
|
53
|
-
def __setitem__(self, key: str, value: Any) -> None:
|
|
54
|
-
self.additional_properties[key] = value
|
|
55
|
-
|
|
56
|
-
def __delitem__(self, key: str) -> None:
|
|
57
|
-
del self.additional_properties[key]
|
|
58
|
-
|
|
59
|
-
def __contains__(self, key: str) -> bool:
|
|
60
|
-
return key in self.additional_properties
|
|
File without changes
|
|
File without changes
|
{robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/licenses/LICENSE
RENAMED
|
File without changes
|