beamlit 0.1.0__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
Potentially problematic release.
This version of beamlit might be problematic. Click here for more details.
- beamlit/__init__.py +8 -0
- beamlit/api/__init__.py +1 -0
- beamlit/api/agents/__init__.py +0 -0
- beamlit/api/agents/create_agent.py +155 -0
- beamlit/api/agents/create_agent_release.py +146 -0
- beamlit/api/agents/delete_agent.py +146 -0
- beamlit/api/agents/delete_agent_deployment.py +163 -0
- beamlit/api/agents/delete_agent_deployment_history.py +172 -0
- beamlit/api/agents/get_agent.py +146 -0
- beamlit/api/agents/get_agent_deployment.py +163 -0
- beamlit/api/agents/get_agent_deployment_history.py +172 -0
- beamlit/api/agents/get_agent_deployment_logs.py +164 -0
- beamlit/api/agents/get_agent_deployment_metrics.py +159 -0
- beamlit/api/agents/get_agent_metrics.py +150 -0
- beamlit/api/agents/list_agent_deployment_history.py +164 -0
- beamlit/api/agents/list_agent_deployments.py +155 -0
- beamlit/api/agents/list_agents.py +127 -0
- beamlit/api/agents/put_agent_deployment.py +185 -0
- beamlit/api/agents/put_agent_deployment_history.py +198 -0
- beamlit/api/agents/update_agent.py +168 -0
- beamlit/api/authentication_providers/__init__.py +0 -0
- beamlit/api/authentication_providers/get_model_with_repo_for_authentication_provider.py +184 -0
- beamlit/api/authentication_providers/list_models_for_authentication_provider.py +163 -0
- beamlit/api/authentication_providers/list_organizations_for_authentication_provider.py +163 -0
- beamlit/api/configurations/__init__.py +0 -0
- beamlit/api/configurations/get_configuration.py +122 -0
- beamlit/api/environments/__init__.py +0 -0
- beamlit/api/environments/create_environment.py +167 -0
- beamlit/api/environments/delete_environment.py +154 -0
- beamlit/api/environments/get_environment.py +154 -0
- beamlit/api/environments/get_environment_metrics.py +158 -0
- beamlit/api/environments/list_environments.py +139 -0
- beamlit/api/environments/update_environment.py +180 -0
- beamlit/api/functions/__init__.py +0 -0
- beamlit/api/functions/create_function.py +155 -0
- beamlit/api/functions/create_function_release.py +150 -0
- beamlit/api/functions/delete_function.py +146 -0
- beamlit/api/functions/delete_function_deployment.py +163 -0
- beamlit/api/functions/get_function.py +146 -0
- beamlit/api/functions/get_function_deployment.py +163 -0
- beamlit/api/functions/get_function_deployment_logs.py +164 -0
- beamlit/api/functions/get_function_deployment_metrics.py +159 -0
- beamlit/api/functions/get_function_metrics.py +150 -0
- beamlit/api/functions/list_function_deployments.py +155 -0
- beamlit/api/functions/list_functions.py +131 -0
- beamlit/api/functions/put_function_deployment.py +185 -0
- beamlit/api/functions/update_function.py +168 -0
- beamlit/api/history/__init__.py +0 -0
- beamlit/api/history/get_agents_history.py +155 -0
- beamlit/api/history/list_agents_history.py +131 -0
- beamlit/api/invitations/__init__.py +0 -0
- beamlit/api/invitations/list_all_pending_invitations.py +142 -0
- beamlit/api/locations/__init__.py +0 -0
- beamlit/api/locations/list_locations.py +139 -0
- beamlit/api/metrics/__init__.py +0 -0
- beamlit/api/metrics/get_metrics.py +130 -0
- beamlit/api/model_providers/__init__.py +0 -0
- beamlit/api/model_providers/create_model_provider.py +163 -0
- beamlit/api/model_providers/delete_model_provider.py +154 -0
- beamlit/api/model_providers/get_model_provider.py +154 -0
- beamlit/api/model_providers/list_model_providers.py +139 -0
- beamlit/api/model_providers/update_model_provider.py +176 -0
- beamlit/api/models/__init__.py +0 -0
- beamlit/api/models/create_model.py +168 -0
- beamlit/api/models/delete_model.py +154 -0
- beamlit/api/models/delete_model_deployment.py +171 -0
- beamlit/api/models/get_model.py +154 -0
- beamlit/api/models/get_model_deployment.py +171 -0
- beamlit/api/models/get_model_deployment_logs.py +168 -0
- beamlit/api/models/get_model_deployment_metrics.py +163 -0
- beamlit/api/models/get_model_metrics.py +154 -0
- beamlit/api/models/list_model_deployments.py +163 -0
- beamlit/api/models/list_models.py +135 -0
- beamlit/api/models/put_model_deployment.py +193 -0
- beamlit/api/models/release_model.py +154 -0
- beamlit/api/models/update_model.py +181 -0
- beamlit/api/policies/__init__.py +0 -0
- beamlit/api/policies/create_policy.py +167 -0
- beamlit/api/policies/delete_policy.py +154 -0
- beamlit/api/policies/get_policy.py +154 -0
- beamlit/api/policies/list_policies.py +139 -0
- beamlit/api/policies/update_policy.py +180 -0
- beamlit/api/service_accounts/__init__.py +0 -0
- beamlit/api/service_accounts/create_api_key_for_service_account.py +177 -0
- beamlit/api/service_accounts/create_workspace_service_account.py +168 -0
- beamlit/api/service_accounts/delete_api_key_for_service_account.py +104 -0
- beamlit/api/service_accounts/delete_workspace_service_account.py +158 -0
- beamlit/api/service_accounts/get_workspace_service_accounts.py +139 -0
- beamlit/api/service_accounts/list_api_keys_for_service_account.py +163 -0
- beamlit/api/service_accounts/update_workspace_service_account.py +181 -0
- beamlit/api/store/__init__.py +0 -0
- beamlit/api/store/get_store_agent.py +146 -0
- beamlit/api/store/get_store_function.py +146 -0
- beamlit/api/store/list_store_agents.py +131 -0
- beamlit/api/store/list_store_functions.py +131 -0
- beamlit/api/workspaces/__init__.py +0 -0
- beamlit/api/workspaces/accept_workspace_invitation.py +161 -0
- beamlit/api/workspaces/create_worspace.py +163 -0
- beamlit/api/workspaces/decline_workspace_invitation.py +158 -0
- beamlit/api/workspaces/delete_workspace.py +154 -0
- beamlit/api/workspaces/get_workspace.py +154 -0
- beamlit/api/workspaces/invite_workspace_user.py +174 -0
- beamlit/api/workspaces/leave_workspace.py +161 -0
- beamlit/api/workspaces/list_workspace_users.py +139 -0
- beamlit/api/workspaces/list_workspaces.py +139 -0
- beamlit/api/workspaces/remove_workspace_user.py +101 -0
- beamlit/api/workspaces/update_workspace.py +176 -0
- beamlit/api/workspaces/update_workspace_user_role.py +187 -0
- beamlit/authentication/__init__.py +24 -0
- beamlit/authentication/apikey.py +14 -0
- beamlit/authentication/authentication.py +35 -0
- beamlit/authentication/credentials.py +165 -0
- beamlit/authentication/device_mode.py +115 -0
- beamlit/client.py +270 -0
- beamlit/errors.py +16 -0
- beamlit/models/__init__.py +189 -0
- beamlit/models/acl.py +149 -0
- beamlit/models/agent.py +151 -0
- beamlit/models/agent_chain.py +77 -0
- beamlit/models/agent_configuration.py +68 -0
- beamlit/models/agent_deployment.py +327 -0
- beamlit/models/agent_deployment_configuration.py +43 -0
- beamlit/models/agent_deployment_history.py +183 -0
- beamlit/models/agent_deployment_history_event.py +131 -0
- beamlit/models/agent_deployment_pod_template.py +43 -0
- beamlit/models/agent_release.py +68 -0
- beamlit/models/api_key.py +140 -0
- beamlit/models/authentication_provider_model.py +142 -0
- beamlit/models/authentication_provider_organization.py +86 -0
- beamlit/models/configuration.py +72 -0
- beamlit/models/continent.py +68 -0
- beamlit/models/country.py +68 -0
- beamlit/models/create_api_key_for_service_account_body.py +67 -0
- beamlit/models/create_workspace_service_account_body.py +69 -0
- beamlit/models/create_workspace_service_account_response_200.py +103 -0
- beamlit/models/delete_workspace_service_account_response_200.py +94 -0
- beamlit/models/deployment_configuration.py +68 -0
- beamlit/models/deployment_configurations.py +43 -0
- beamlit/models/deployment_serverless_config.py +129 -0
- beamlit/models/environment.py +162 -0
- beamlit/models/environment_metrics.py +75 -0
- beamlit/models/flavor.py +68 -0
- beamlit/models/function.py +151 -0
- beamlit/models/function_configuration.py +68 -0
- beamlit/models/function_deployment.py +327 -0
- beamlit/models/function_deployment_configuration.py +43 -0
- beamlit/models/function_deployment_pod_template.py +43 -0
- beamlit/models/function_kit.py +95 -0
- beamlit/models/function_provider_ref.py +68 -0
- beamlit/models/function_release.py +68 -0
- beamlit/models/get_workspace_service_accounts_response_200_item.py +94 -0
- beamlit/models/invite_workspace_user_body.py +58 -0
- beamlit/models/labels.py +43 -0
- beamlit/models/labels_type_0.py +43 -0
- beamlit/models/location.py +120 -0
- beamlit/models/location_response.py +111 -0
- beamlit/models/metric.py +68 -0
- beamlit/models/metrics.py +111 -0
- beamlit/models/model.py +151 -0
- beamlit/models/model_deployment.py +283 -0
- beamlit/models/model_deployment_log.py +68 -0
- beamlit/models/model_deployment_metrics.py +170 -0
- beamlit/models/model_deployment_metrics_inference_per_second_per_region.py +75 -0
- beamlit/models/model_deployment_metrics_query_per_second_per_region_per_code.py +73 -0
- beamlit/models/model_deployment_pod_template.py +43 -0
- beamlit/models/model_metrics.py +94 -0
- beamlit/models/model_provider.py +187 -0
- beamlit/models/model_provider_ref.py +68 -0
- beamlit/models/model_release.py +68 -0
- beamlit/models/model_with_deployments.py +174 -0
- beamlit/models/pending_invitation.py +122 -0
- beamlit/models/pending_invitation_accept.py +81 -0
- beamlit/models/pending_invitation_render.py +135 -0
- beamlit/models/pending_invitation_render_invited_by.py +86 -0
- beamlit/models/pending_invitation_render_workspace.py +68 -0
- beamlit/models/pending_invitation_workspace_details.py +70 -0
- beamlit/models/policy.py +216 -0
- beamlit/models/policy_location.py +68 -0
- beamlit/models/provider_config.py +101 -0
- beamlit/models/qps.py +59 -0
- beamlit/models/resource_deployment_log.py +68 -0
- beamlit/models/resource_deployment_metrics.py +172 -0
- beamlit/models/resource_deployment_metrics_inference_per_second_per_region.py +75 -0
- beamlit/models/resource_deployment_metrics_query_per_second_per_region_per_code.py +73 -0
- beamlit/models/resource_metrics.py +94 -0
- beamlit/models/runtime.py +152 -0
- beamlit/models/runtime_readiness_probe.py +43 -0
- beamlit/models/runtime_resources.py +43 -0
- beamlit/models/serverless_config.py +122 -0
- beamlit/models/standard_fields_dynamo_db.py +86 -0
- beamlit/models/store_agent.py +165 -0
- beamlit/models/store_agent_configuration.py +95 -0
- beamlit/models/store_agent_labels.py +43 -0
- beamlit/models/store_configuration.py +149 -0
- beamlit/models/store_configuration_option.py +77 -0
- beamlit/models/store_function.py +207 -0
- beamlit/models/store_function_configuration.py +95 -0
- beamlit/models/store_function_kit.py +95 -0
- beamlit/models/store_function_labels.py +43 -0
- beamlit/models/store_function_parameter.py +86 -0
- beamlit/models/update_workspace_service_account_body.py +67 -0
- beamlit/models/update_workspace_service_account_response_200.py +94 -0
- beamlit/models/update_workspace_user_role_body.py +58 -0
- beamlit/models/workspace.py +126 -0
- beamlit/models/workspace_labels.py +43 -0
- beamlit/models/workspace_user.py +113 -0
- beamlit/py.typed +1 -0
- beamlit/types.py +46 -0
- beamlit-0.1.0.dist-info/METADATA +59 -0
- beamlit-0.1.0.dist-info/RECORD +211 -0
- beamlit-0.1.0.dist-info/WHEEL +4 -0
@@ -0,0 +1,172 @@
|
|
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.agent_deployment_history import AgentDeploymentHistory
|
9
|
+
from ...types import Response
|
10
|
+
|
11
|
+
|
12
|
+
def _get_kwargs(
|
13
|
+
agent_name: str,
|
14
|
+
environment_name: str,
|
15
|
+
request_id: str,
|
16
|
+
) -> dict[str, Any]:
|
17
|
+
_kwargs: dict[str, Any] = {
|
18
|
+
"method": "delete",
|
19
|
+
"url": f"/agents/{agent_name}/deployments/{environment_name}/history/{request_id}",
|
20
|
+
}
|
21
|
+
|
22
|
+
return _kwargs
|
23
|
+
|
24
|
+
|
25
|
+
def _parse_response(
|
26
|
+
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
27
|
+
) -> Optional[AgentDeploymentHistory]:
|
28
|
+
if response.status_code == 200:
|
29
|
+
response_200 = AgentDeploymentHistory.from_dict(response.json())
|
30
|
+
|
31
|
+
return response_200
|
32
|
+
if client.raise_on_unexpected_status:
|
33
|
+
raise errors.UnexpectedStatus(response.status_code, response.content)
|
34
|
+
else:
|
35
|
+
return None
|
36
|
+
|
37
|
+
|
38
|
+
def _build_response(
|
39
|
+
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
40
|
+
) -> Response[AgentDeploymentHistory]:
|
41
|
+
return Response(
|
42
|
+
status_code=HTTPStatus(response.status_code),
|
43
|
+
content=response.content,
|
44
|
+
headers=response.headers,
|
45
|
+
parsed=_parse_response(client=client, response=response),
|
46
|
+
)
|
47
|
+
|
48
|
+
|
49
|
+
def sync_detailed(
|
50
|
+
agent_name: str,
|
51
|
+
environment_name: str,
|
52
|
+
request_id: str,
|
53
|
+
*,
|
54
|
+
client: AuthenticatedClient,
|
55
|
+
) -> Response[AgentDeploymentHistory]:
|
56
|
+
"""
|
57
|
+
Args:
|
58
|
+
agent_name (str):
|
59
|
+
environment_name (str):
|
60
|
+
request_id (str):
|
61
|
+
|
62
|
+
Raises:
|
63
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
64
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
65
|
+
|
66
|
+
Returns:
|
67
|
+
Response[AgentDeploymentHistory]
|
68
|
+
"""
|
69
|
+
|
70
|
+
kwargs = _get_kwargs(
|
71
|
+
agent_name=agent_name,
|
72
|
+
environment_name=environment_name,
|
73
|
+
request_id=request_id,
|
74
|
+
)
|
75
|
+
|
76
|
+
response = client.get_httpx_client().request(
|
77
|
+
**kwargs,
|
78
|
+
)
|
79
|
+
|
80
|
+
return _build_response(client=client, response=response)
|
81
|
+
|
82
|
+
|
83
|
+
def sync(
|
84
|
+
agent_name: str,
|
85
|
+
environment_name: str,
|
86
|
+
request_id: str,
|
87
|
+
*,
|
88
|
+
client: AuthenticatedClient,
|
89
|
+
) -> Optional[AgentDeploymentHistory]:
|
90
|
+
"""
|
91
|
+
Args:
|
92
|
+
agent_name (str):
|
93
|
+
environment_name (str):
|
94
|
+
request_id (str):
|
95
|
+
|
96
|
+
Raises:
|
97
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
98
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
99
|
+
|
100
|
+
Returns:
|
101
|
+
AgentDeploymentHistory
|
102
|
+
"""
|
103
|
+
|
104
|
+
return sync_detailed(
|
105
|
+
agent_name=agent_name,
|
106
|
+
environment_name=environment_name,
|
107
|
+
request_id=request_id,
|
108
|
+
client=client,
|
109
|
+
).parsed
|
110
|
+
|
111
|
+
|
112
|
+
async def asyncio_detailed(
|
113
|
+
agent_name: str,
|
114
|
+
environment_name: str,
|
115
|
+
request_id: str,
|
116
|
+
*,
|
117
|
+
client: AuthenticatedClient,
|
118
|
+
) -> Response[AgentDeploymentHistory]:
|
119
|
+
"""
|
120
|
+
Args:
|
121
|
+
agent_name (str):
|
122
|
+
environment_name (str):
|
123
|
+
request_id (str):
|
124
|
+
|
125
|
+
Raises:
|
126
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
127
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
128
|
+
|
129
|
+
Returns:
|
130
|
+
Response[AgentDeploymentHistory]
|
131
|
+
"""
|
132
|
+
|
133
|
+
kwargs = _get_kwargs(
|
134
|
+
agent_name=agent_name,
|
135
|
+
environment_name=environment_name,
|
136
|
+
request_id=request_id,
|
137
|
+
)
|
138
|
+
|
139
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
140
|
+
|
141
|
+
return _build_response(client=client, response=response)
|
142
|
+
|
143
|
+
|
144
|
+
async def asyncio(
|
145
|
+
agent_name: str,
|
146
|
+
environment_name: str,
|
147
|
+
request_id: str,
|
148
|
+
*,
|
149
|
+
client: AuthenticatedClient,
|
150
|
+
) -> Optional[AgentDeploymentHistory]:
|
151
|
+
"""
|
152
|
+
Args:
|
153
|
+
agent_name (str):
|
154
|
+
environment_name (str):
|
155
|
+
request_id (str):
|
156
|
+
|
157
|
+
Raises:
|
158
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
159
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
160
|
+
|
161
|
+
Returns:
|
162
|
+
AgentDeploymentHistory
|
163
|
+
"""
|
164
|
+
|
165
|
+
return (
|
166
|
+
await asyncio_detailed(
|
167
|
+
agent_name=agent_name,
|
168
|
+
environment_name=environment_name,
|
169
|
+
request_id=request_id,
|
170
|
+
client=client,
|
171
|
+
)
|
172
|
+
).parsed
|
@@ -0,0 +1,146 @@
|
|
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.agent import Agent
|
9
|
+
from ...types import Response
|
10
|
+
|
11
|
+
|
12
|
+
def _get_kwargs(
|
13
|
+
agent_name: str,
|
14
|
+
) -> dict[str, Any]:
|
15
|
+
_kwargs: dict[str, Any] = {
|
16
|
+
"method": "get",
|
17
|
+
"url": f"/agents/{agent_name}",
|
18
|
+
}
|
19
|
+
|
20
|
+
return _kwargs
|
21
|
+
|
22
|
+
|
23
|
+
def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Agent]:
|
24
|
+
if response.status_code == 200:
|
25
|
+
response_200 = Agent.from_dict(response.json())
|
26
|
+
|
27
|
+
return response_200
|
28
|
+
if client.raise_on_unexpected_status:
|
29
|
+
raise errors.UnexpectedStatus(response.status_code, response.content)
|
30
|
+
else:
|
31
|
+
return None
|
32
|
+
|
33
|
+
|
34
|
+
def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Agent]:
|
35
|
+
return Response(
|
36
|
+
status_code=HTTPStatus(response.status_code),
|
37
|
+
content=response.content,
|
38
|
+
headers=response.headers,
|
39
|
+
parsed=_parse_response(client=client, response=response),
|
40
|
+
)
|
41
|
+
|
42
|
+
|
43
|
+
def sync_detailed(
|
44
|
+
agent_name: str,
|
45
|
+
*,
|
46
|
+
client: AuthenticatedClient,
|
47
|
+
) -> Response[Agent]:
|
48
|
+
"""Get agent by name
|
49
|
+
|
50
|
+
Args:
|
51
|
+
agent_name (str):
|
52
|
+
|
53
|
+
Raises:
|
54
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
55
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
56
|
+
|
57
|
+
Returns:
|
58
|
+
Response[Agent]
|
59
|
+
"""
|
60
|
+
|
61
|
+
kwargs = _get_kwargs(
|
62
|
+
agent_name=agent_name,
|
63
|
+
)
|
64
|
+
|
65
|
+
response = client.get_httpx_client().request(
|
66
|
+
**kwargs,
|
67
|
+
)
|
68
|
+
|
69
|
+
return _build_response(client=client, response=response)
|
70
|
+
|
71
|
+
|
72
|
+
def sync(
|
73
|
+
agent_name: str,
|
74
|
+
*,
|
75
|
+
client: AuthenticatedClient,
|
76
|
+
) -> Optional[Agent]:
|
77
|
+
"""Get agent by name
|
78
|
+
|
79
|
+
Args:
|
80
|
+
agent_name (str):
|
81
|
+
|
82
|
+
Raises:
|
83
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
84
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
85
|
+
|
86
|
+
Returns:
|
87
|
+
Agent
|
88
|
+
"""
|
89
|
+
|
90
|
+
return sync_detailed(
|
91
|
+
agent_name=agent_name,
|
92
|
+
client=client,
|
93
|
+
).parsed
|
94
|
+
|
95
|
+
|
96
|
+
async def asyncio_detailed(
|
97
|
+
agent_name: str,
|
98
|
+
*,
|
99
|
+
client: AuthenticatedClient,
|
100
|
+
) -> Response[Agent]:
|
101
|
+
"""Get agent by name
|
102
|
+
|
103
|
+
Args:
|
104
|
+
agent_name (str):
|
105
|
+
|
106
|
+
Raises:
|
107
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
108
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
109
|
+
|
110
|
+
Returns:
|
111
|
+
Response[Agent]
|
112
|
+
"""
|
113
|
+
|
114
|
+
kwargs = _get_kwargs(
|
115
|
+
agent_name=agent_name,
|
116
|
+
)
|
117
|
+
|
118
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
119
|
+
|
120
|
+
return _build_response(client=client, response=response)
|
121
|
+
|
122
|
+
|
123
|
+
async def asyncio(
|
124
|
+
agent_name: str,
|
125
|
+
*,
|
126
|
+
client: AuthenticatedClient,
|
127
|
+
) -> Optional[Agent]:
|
128
|
+
"""Get agent by name
|
129
|
+
|
130
|
+
Args:
|
131
|
+
agent_name (str):
|
132
|
+
|
133
|
+
Raises:
|
134
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
135
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
136
|
+
|
137
|
+
Returns:
|
138
|
+
Agent
|
139
|
+
"""
|
140
|
+
|
141
|
+
return (
|
142
|
+
await asyncio_detailed(
|
143
|
+
agent_name=agent_name,
|
144
|
+
client=client,
|
145
|
+
)
|
146
|
+
).parsed
|
@@ -0,0 +1,163 @@
|
|
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.agent_deployment import AgentDeployment
|
9
|
+
from ...types import Response
|
10
|
+
|
11
|
+
|
12
|
+
def _get_kwargs(
|
13
|
+
agent_name: str,
|
14
|
+
environment_name: str,
|
15
|
+
) -> dict[str, Any]:
|
16
|
+
_kwargs: dict[str, Any] = {
|
17
|
+
"method": "get",
|
18
|
+
"url": f"/agents/{agent_name}/deployments/{environment_name}",
|
19
|
+
}
|
20
|
+
|
21
|
+
return _kwargs
|
22
|
+
|
23
|
+
|
24
|
+
def _parse_response(
|
25
|
+
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
26
|
+
) -> Optional[AgentDeployment]:
|
27
|
+
if response.status_code == 200:
|
28
|
+
response_200 = AgentDeployment.from_dict(response.json())
|
29
|
+
|
30
|
+
return response_200
|
31
|
+
if client.raise_on_unexpected_status:
|
32
|
+
raise errors.UnexpectedStatus(response.status_code, response.content)
|
33
|
+
else:
|
34
|
+
return None
|
35
|
+
|
36
|
+
|
37
|
+
def _build_response(
|
38
|
+
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
39
|
+
) -> Response[AgentDeployment]:
|
40
|
+
return Response(
|
41
|
+
status_code=HTTPStatus(response.status_code),
|
42
|
+
content=response.content,
|
43
|
+
headers=response.headers,
|
44
|
+
parsed=_parse_response(client=client, response=response),
|
45
|
+
)
|
46
|
+
|
47
|
+
|
48
|
+
def sync_detailed(
|
49
|
+
agent_name: str,
|
50
|
+
environment_name: str,
|
51
|
+
*,
|
52
|
+
client: AuthenticatedClient,
|
53
|
+
) -> Response[AgentDeployment]:
|
54
|
+
"""Get agent deployment by environment name
|
55
|
+
|
56
|
+
Args:
|
57
|
+
agent_name (str):
|
58
|
+
environment_name (str):
|
59
|
+
|
60
|
+
Raises:
|
61
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
62
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
63
|
+
|
64
|
+
Returns:
|
65
|
+
Response[AgentDeployment]
|
66
|
+
"""
|
67
|
+
|
68
|
+
kwargs = _get_kwargs(
|
69
|
+
agent_name=agent_name,
|
70
|
+
environment_name=environment_name,
|
71
|
+
)
|
72
|
+
|
73
|
+
response = client.get_httpx_client().request(
|
74
|
+
**kwargs,
|
75
|
+
)
|
76
|
+
|
77
|
+
return _build_response(client=client, response=response)
|
78
|
+
|
79
|
+
|
80
|
+
def sync(
|
81
|
+
agent_name: str,
|
82
|
+
environment_name: str,
|
83
|
+
*,
|
84
|
+
client: AuthenticatedClient,
|
85
|
+
) -> Optional[AgentDeployment]:
|
86
|
+
"""Get agent deployment by environment name
|
87
|
+
|
88
|
+
Args:
|
89
|
+
agent_name (str):
|
90
|
+
environment_name (str):
|
91
|
+
|
92
|
+
Raises:
|
93
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
94
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
95
|
+
|
96
|
+
Returns:
|
97
|
+
AgentDeployment
|
98
|
+
"""
|
99
|
+
|
100
|
+
return sync_detailed(
|
101
|
+
agent_name=agent_name,
|
102
|
+
environment_name=environment_name,
|
103
|
+
client=client,
|
104
|
+
).parsed
|
105
|
+
|
106
|
+
|
107
|
+
async def asyncio_detailed(
|
108
|
+
agent_name: str,
|
109
|
+
environment_name: str,
|
110
|
+
*,
|
111
|
+
client: AuthenticatedClient,
|
112
|
+
) -> Response[AgentDeployment]:
|
113
|
+
"""Get agent deployment by environment name
|
114
|
+
|
115
|
+
Args:
|
116
|
+
agent_name (str):
|
117
|
+
environment_name (str):
|
118
|
+
|
119
|
+
Raises:
|
120
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
121
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
122
|
+
|
123
|
+
Returns:
|
124
|
+
Response[AgentDeployment]
|
125
|
+
"""
|
126
|
+
|
127
|
+
kwargs = _get_kwargs(
|
128
|
+
agent_name=agent_name,
|
129
|
+
environment_name=environment_name,
|
130
|
+
)
|
131
|
+
|
132
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
133
|
+
|
134
|
+
return _build_response(client=client, response=response)
|
135
|
+
|
136
|
+
|
137
|
+
async def asyncio(
|
138
|
+
agent_name: str,
|
139
|
+
environment_name: str,
|
140
|
+
*,
|
141
|
+
client: AuthenticatedClient,
|
142
|
+
) -> Optional[AgentDeployment]:
|
143
|
+
"""Get agent deployment by environment name
|
144
|
+
|
145
|
+
Args:
|
146
|
+
agent_name (str):
|
147
|
+
environment_name (str):
|
148
|
+
|
149
|
+
Raises:
|
150
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
151
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
152
|
+
|
153
|
+
Returns:
|
154
|
+
AgentDeployment
|
155
|
+
"""
|
156
|
+
|
157
|
+
return (
|
158
|
+
await asyncio_detailed(
|
159
|
+
agent_name=agent_name,
|
160
|
+
environment_name=environment_name,
|
161
|
+
client=client,
|
162
|
+
)
|
163
|
+
).parsed
|
@@ -0,0 +1,172 @@
|
|
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.agent_deployment_history import AgentDeploymentHistory
|
9
|
+
from ...types import Response
|
10
|
+
|
11
|
+
|
12
|
+
def _get_kwargs(
|
13
|
+
agent_name: str,
|
14
|
+
environment_name: str,
|
15
|
+
request_id: str,
|
16
|
+
) -> dict[str, Any]:
|
17
|
+
_kwargs: dict[str, Any] = {
|
18
|
+
"method": "get",
|
19
|
+
"url": f"/agents/{agent_name}/deployments/{environment_name}/history/{request_id}",
|
20
|
+
}
|
21
|
+
|
22
|
+
return _kwargs
|
23
|
+
|
24
|
+
|
25
|
+
def _parse_response(
|
26
|
+
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
27
|
+
) -> Optional[AgentDeploymentHistory]:
|
28
|
+
if response.status_code == 200:
|
29
|
+
response_200 = AgentDeploymentHistory.from_dict(response.json())
|
30
|
+
|
31
|
+
return response_200
|
32
|
+
if client.raise_on_unexpected_status:
|
33
|
+
raise errors.UnexpectedStatus(response.status_code, response.content)
|
34
|
+
else:
|
35
|
+
return None
|
36
|
+
|
37
|
+
|
38
|
+
def _build_response(
|
39
|
+
*, client: Union[AuthenticatedClient, Client], response: httpx.Response
|
40
|
+
) -> Response[AgentDeploymentHistory]:
|
41
|
+
return Response(
|
42
|
+
status_code=HTTPStatus(response.status_code),
|
43
|
+
content=response.content,
|
44
|
+
headers=response.headers,
|
45
|
+
parsed=_parse_response(client=client, response=response),
|
46
|
+
)
|
47
|
+
|
48
|
+
|
49
|
+
def sync_detailed(
|
50
|
+
agent_name: str,
|
51
|
+
environment_name: str,
|
52
|
+
request_id: str,
|
53
|
+
*,
|
54
|
+
client: AuthenticatedClient,
|
55
|
+
) -> Response[AgentDeploymentHistory]:
|
56
|
+
"""
|
57
|
+
Args:
|
58
|
+
agent_name (str):
|
59
|
+
environment_name (str):
|
60
|
+
request_id (str):
|
61
|
+
|
62
|
+
Raises:
|
63
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
64
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
65
|
+
|
66
|
+
Returns:
|
67
|
+
Response[AgentDeploymentHistory]
|
68
|
+
"""
|
69
|
+
|
70
|
+
kwargs = _get_kwargs(
|
71
|
+
agent_name=agent_name,
|
72
|
+
environment_name=environment_name,
|
73
|
+
request_id=request_id,
|
74
|
+
)
|
75
|
+
|
76
|
+
response = client.get_httpx_client().request(
|
77
|
+
**kwargs,
|
78
|
+
)
|
79
|
+
|
80
|
+
return _build_response(client=client, response=response)
|
81
|
+
|
82
|
+
|
83
|
+
def sync(
|
84
|
+
agent_name: str,
|
85
|
+
environment_name: str,
|
86
|
+
request_id: str,
|
87
|
+
*,
|
88
|
+
client: AuthenticatedClient,
|
89
|
+
) -> Optional[AgentDeploymentHistory]:
|
90
|
+
"""
|
91
|
+
Args:
|
92
|
+
agent_name (str):
|
93
|
+
environment_name (str):
|
94
|
+
request_id (str):
|
95
|
+
|
96
|
+
Raises:
|
97
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
98
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
99
|
+
|
100
|
+
Returns:
|
101
|
+
AgentDeploymentHistory
|
102
|
+
"""
|
103
|
+
|
104
|
+
return sync_detailed(
|
105
|
+
agent_name=agent_name,
|
106
|
+
environment_name=environment_name,
|
107
|
+
request_id=request_id,
|
108
|
+
client=client,
|
109
|
+
).parsed
|
110
|
+
|
111
|
+
|
112
|
+
async def asyncio_detailed(
|
113
|
+
agent_name: str,
|
114
|
+
environment_name: str,
|
115
|
+
request_id: str,
|
116
|
+
*,
|
117
|
+
client: AuthenticatedClient,
|
118
|
+
) -> Response[AgentDeploymentHistory]:
|
119
|
+
"""
|
120
|
+
Args:
|
121
|
+
agent_name (str):
|
122
|
+
environment_name (str):
|
123
|
+
request_id (str):
|
124
|
+
|
125
|
+
Raises:
|
126
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
127
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
128
|
+
|
129
|
+
Returns:
|
130
|
+
Response[AgentDeploymentHistory]
|
131
|
+
"""
|
132
|
+
|
133
|
+
kwargs = _get_kwargs(
|
134
|
+
agent_name=agent_name,
|
135
|
+
environment_name=environment_name,
|
136
|
+
request_id=request_id,
|
137
|
+
)
|
138
|
+
|
139
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
140
|
+
|
141
|
+
return _build_response(client=client, response=response)
|
142
|
+
|
143
|
+
|
144
|
+
async def asyncio(
|
145
|
+
agent_name: str,
|
146
|
+
environment_name: str,
|
147
|
+
request_id: str,
|
148
|
+
*,
|
149
|
+
client: AuthenticatedClient,
|
150
|
+
) -> Optional[AgentDeploymentHistory]:
|
151
|
+
"""
|
152
|
+
Args:
|
153
|
+
agent_name (str):
|
154
|
+
environment_name (str):
|
155
|
+
request_id (str):
|
156
|
+
|
157
|
+
Raises:
|
158
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
159
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
160
|
+
|
161
|
+
Returns:
|
162
|
+
AgentDeploymentHistory
|
163
|
+
"""
|
164
|
+
|
165
|
+
return (
|
166
|
+
await asyncio_detailed(
|
167
|
+
agent_name=agent_name,
|
168
|
+
environment_name=environment_name,
|
169
|
+
request_id=request_id,
|
170
|
+
client=client,
|
171
|
+
)
|
172
|
+
).parsed
|