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.

Files changed (99) hide show
  1. robosystems_client/__init__.py +15 -4
  2. robosystems_client/api/agent/auto_select_agent.py +26 -1
  3. robosystems_client/api/agent/batch_process_queries.py +26 -1
  4. robosystems_client/api/agent/execute_specific_agent.py +26 -1
  5. robosystems_client/api/agent/get_agent_metadata.py +26 -1
  6. robosystems_client/api/agent/list_agents.py +21 -1
  7. robosystems_client/api/agent/recommend_agent.py +26 -1
  8. robosystems_client/api/backup/create_backup.py +26 -1
  9. robosystems_client/api/backup/export_backup.py +26 -1
  10. robosystems_client/api/backup/get_backup_download_url.py +21 -1
  11. robosystems_client/api/backup/get_backup_stats.py +26 -1
  12. robosystems_client/api/backup/list_backups.py +21 -1
  13. robosystems_client/api/backup/restore_backup.py +26 -1
  14. robosystems_client/api/connections/create_connection.py +26 -1
  15. robosystems_client/api/connections/create_link_token.py +26 -1
  16. robosystems_client/api/connections/delete_connection.py +26 -1
  17. robosystems_client/api/connections/exchange_link_token.py +26 -1
  18. robosystems_client/api/connections/get_connection.py +26 -1
  19. robosystems_client/api/connections/get_connection_options.py +26 -1
  20. robosystems_client/api/connections/init_o_auth.py +26 -1
  21. robosystems_client/api/connections/list_connections.py +21 -1
  22. robosystems_client/api/connections/oauth_callback.py +26 -1
  23. robosystems_client/api/connections/sync_connection.py +26 -1
  24. robosystems_client/api/copy/copy_data_to_graph.py +26 -1
  25. robosystems_client/api/graph_analytics/get_graph_metrics.py +26 -1
  26. robosystems_client/api/graph_analytics/get_graph_usage_stats.py +21 -1
  27. robosystems_client/api/graph_billing/get_current_graph_bill.py +26 -1
  28. robosystems_client/api/graph_billing/get_graph_billing_history.py +21 -1
  29. robosystems_client/api/graph_billing/get_graph_monthly_bill.py +26 -1
  30. robosystems_client/api/graph_billing/get_graph_usage_details.py +21 -1
  31. robosystems_client/api/graph_credits/check_credit_balance.py +21 -1
  32. robosystems_client/api/graph_credits/check_storage_limits.py +26 -1
  33. robosystems_client/api/graph_credits/get_credit_summary.py +26 -1
  34. robosystems_client/api/graph_credits/get_storage_usage.py +21 -1
  35. robosystems_client/api/graph_credits/list_credit_transactions.py +21 -1
  36. robosystems_client/api/graph_health/get_database_health.py +26 -1
  37. robosystems_client/api/graph_info/get_database_info.py +26 -1
  38. robosystems_client/api/graph_limits/get_graph_limits.py +26 -1
  39. robosystems_client/api/{create → graphs}/create_graph.py +26 -1
  40. robosystems_client/api/{create → graphs}/get_available_extensions.py +1 -1
  41. robosystems_client/api/{user/get_user_graphs.py → graphs/get_graphs.py} +26 -1
  42. robosystems_client/api/{user/select_user_graph.py → graphs/select_graph.py} +30 -5
  43. robosystems_client/api/mcp/call_mcp_tool.py +21 -1
  44. robosystems_client/api/mcp/list_mcp_tools.py +26 -1
  45. robosystems_client/api/operations/cancel_operation.py +25 -0
  46. robosystems_client/api/operations/get_operation_status.py +25 -0
  47. robosystems_client/api/operations/stream_operation_events.py +20 -0
  48. robosystems_client/api/query/execute_cypher_query.py +21 -1
  49. robosystems_client/api/schema/export_graph_schema.py +21 -1
  50. robosystems_client/api/schema/get_graph_schema_info.py +26 -1
  51. robosystems_client/api/schema/list_schema_extensions.py +26 -1
  52. robosystems_client/api/schema/validate_schema.py +26 -1
  53. robosystems_client/api/subgraphs/create_subgraph.py +26 -1
  54. robosystems_client/api/subgraphs/delete_subgraph.py +26 -1
  55. robosystems_client/api/subgraphs/get_subgraph_info.py +26 -1
  56. robosystems_client/api/subgraphs/get_subgraph_quota.py +26 -1
  57. robosystems_client/api/subgraphs/list_subgraphs.py +26 -1
  58. robosystems_client/api/user/create_user_api_key.py +25 -0
  59. robosystems_client/api/user/get_all_credit_summaries.py +25 -0
  60. robosystems_client/api/user/get_current_user.py +25 -0
  61. robosystems_client/api/user/list_user_api_keys.py +25 -0
  62. robosystems_client/api/user/revoke_user_api_key.py +25 -0
  63. robosystems_client/api/user/update_user.py +25 -0
  64. robosystems_client/api/user/update_user_api_key.py +25 -0
  65. robosystems_client/api/user/update_user_password.py +25 -0
  66. robosystems_client/api/user_analytics/get_detailed_user_analytics.py +20 -0
  67. robosystems_client/api/user_analytics/get_user_usage_overview.py +25 -0
  68. robosystems_client/api/user_limits/get_all_shared_repository_limits.py +25 -0
  69. robosystems_client/api/user_limits/get_shared_repository_limits.py +25 -0
  70. robosystems_client/api/user_limits/get_user_limits.py +25 -0
  71. robosystems_client/api/user_limits/get_user_usage.py +25 -0
  72. robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +25 -0
  73. robosystems_client/api/user_subscriptions/get_repository_credits.py +25 -0
  74. robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +25 -0
  75. robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +20 -0
  76. robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +25 -0
  77. robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +25 -0
  78. robosystems_client/extensions/__init__.py +70 -0
  79. robosystems_client/extensions/auth_integration.py +14 -1
  80. robosystems_client/extensions/copy_client.py +32 -22
  81. robosystems_client/extensions/dataframe_utils.py +455 -0
  82. robosystems_client/extensions/extensions.py +16 -0
  83. robosystems_client/extensions/operation_client.py +43 -21
  84. robosystems_client/extensions/query_client.py +109 -12
  85. robosystems_client/extensions/tests/test_dataframe_utils.py +334 -0
  86. robosystems_client/extensions/tests/test_integration.py +1 -1
  87. robosystems_client/extensions/tests/test_token_utils.py +274 -0
  88. robosystems_client/extensions/token_utils.py +417 -0
  89. robosystems_client/extensions/utils.py +32 -2
  90. robosystems_client/models/__init__.py +0 -2
  91. robosystems_client/models/api_key_info.py +20 -0
  92. robosystems_client/models/create_api_key_request.py +20 -0
  93. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/METADATA +1 -1
  94. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/RECORD +97 -95
  95. robosystems_client/api/auth/sso_login.py +0 -177
  96. robosystems_client/models/sso_login_request.py +0 -60
  97. /robosystems_client/api/{create → graphs}/__init__.py +0 -0
  98. {robosystems_client-0.1.17.dist-info → robosystems_client-0.1.19.dist-info}/WHEEL +0 -0
  99. {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