beamlit 0.0.24rc20__py3-none-any.whl → 0.0.24rc21__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (170) hide show
  1. beamlit/agents/chat.py +37 -11
  2. beamlit/agents/decorator.py +97 -65
  3. beamlit/api/agents/create_agent.py +9 -14
  4. beamlit/api/agents/delete_agent.py +22 -1
  5. beamlit/api/agents/delete_agent_history.py +2 -6
  6. beamlit/api/agents/get_agent.py +22 -1
  7. beamlit/api/agents/get_agent_environment_logs.py +11 -11
  8. beamlit/api/agents/get_agent_history.py +2 -6
  9. beamlit/api/agents/get_agent_metrics.py +22 -1
  10. beamlit/api/agents/list_agent_history.py +11 -11
  11. beamlit/api/agents/list_agents.py +36 -4
  12. beamlit/api/agents/put_agent_history.py +2 -6
  13. beamlit/api/agents/update_agent.py +9 -14
  14. beamlit/api/functions/create_function.py +9 -14
  15. beamlit/api/functions/delete_function.py +22 -1
  16. beamlit/api/functions/get_function.py +22 -1
  17. beamlit/api/functions/get_function_environment_logs.py +11 -11
  18. beamlit/api/functions/get_function_metrics.py +22 -1
  19. beamlit/api/functions/list_functions.py +36 -4
  20. beamlit/api/functions/update_function.py +9 -14
  21. beamlit/api/history/get_agents_history.py +12 -12
  22. beamlit/api/history/list_agents_history.py +12 -12
  23. beamlit/api/integrations/get_integration_connection.py +16 -12
  24. beamlit/api/integrations/get_integration_connection_model.py +8 -1
  25. beamlit/api/integrations/get_integration_model.py +8 -1
  26. beamlit/api/locations/list_locations.py +12 -12
  27. beamlit/api/models/create_model.py +13 -14
  28. beamlit/api/models/delete_model.py +22 -1
  29. beamlit/api/models/get_model.py +22 -1
  30. beamlit/api/models/get_model_environment_logs.py +11 -11
  31. beamlit/api/models/get_model_metrics.py +38 -13
  32. beamlit/api/models/list_models.py +36 -4
  33. beamlit/api/models/update_model.py +13 -14
  34. beamlit/api/privateclusters/__init__.py +0 -0
  35. beamlit/api/{agents/get_agent_deployment_metrics.py → privateclusters/create_private_cluster.py} +28 -55
  36. beamlit/api/{functions/list_function_deployments.py → privateclusters/delete_private_cluster.py} +38 -37
  37. beamlit/api/{agents/get_agent_deployment.py → privateclusters/get_private_cluster.py} +40 -44
  38. beamlit/api/privateclusters/get_private_cluster_health.py +97 -0
  39. beamlit/api/{agents/list_agent_deployments.py → privateclusters/list_private_clusters.py} +30 -45
  40. beamlit/api/{models/list_model_deployments.py → privateclusters/update_private_cluster.py} +38 -45
  41. beamlit/api/privateclusters/update_private_cluster_health.py +97 -0
  42. beamlit/authentication/device_mode.py +3 -9
  43. beamlit/common/generate.py +47 -34
  44. beamlit/common/settings.py +29 -27
  45. beamlit/deploy/deploy.py +101 -342
  46. beamlit/deploy/format.py +70 -0
  47. beamlit/deploy/parser.py +175 -0
  48. beamlit/functions/decorator.py +8 -11
  49. beamlit/models/__init__.py +64 -78
  50. beamlit/models/acl.py +4 -22
  51. beamlit/models/agent.py +34 -95
  52. beamlit/models/agent_history.py +7 -5
  53. beamlit/models/agent_history_event.py +11 -9
  54. beamlit/models/agent_metadata.py +5 -3
  55. beamlit/models/agent_spec.py +54 -45
  56. beamlit/models/api_key.py +14 -14
  57. beamlit/models/configuration.py +11 -0
  58. beamlit/models/core_spec.py +45 -32
  59. beamlit/models/{model_deployment_metrics_query_per_second_per_region_per_code.py → core_spec_configurations.py} +22 -22
  60. beamlit/models/environment.py +33 -105
  61. beamlit/models/environment_metadata.py +146 -0
  62. beamlit/models/environment_spec.py +8 -6
  63. beamlit/models/function.py +34 -95
  64. beamlit/models/function_metadata.py +5 -3
  65. beamlit/models/function_spec.py +53 -40
  66. beamlit/models/integration_connection.py +31 -137
  67. beamlit/models/integration_connection_spec.py +6 -4
  68. beamlit/models/metadata.py +5 -3
  69. beamlit/models/metadata_labels.py +4 -2
  70. beamlit/models/model.py +33 -94
  71. beamlit/models/model_metadata.py +5 -3
  72. beamlit/models/{function_provider_ref.py → model_private_cluster.py} +23 -14
  73. beamlit/models/model_provider.py +27 -43
  74. beamlit/models/model_spec.py +45 -32
  75. beamlit/models/owner_fields.py +4 -2
  76. beamlit/models/pending_invitation.py +14 -14
  77. beamlit/models/pod_template_spec.py +4 -2
  78. beamlit/models/policy.py +33 -159
  79. beamlit/models/policy_spec.py +24 -28
  80. beamlit/models/private_cluster.py +183 -0
  81. beamlit/models/{model_provider_ref.py → private_location.py} +7 -16
  82. beamlit/models/resource_deployment_metrics.py +0 -108
  83. beamlit/models/resource_environment_metrics.py +143 -71
  84. beamlit/models/{resource_deployment_metrics_inference_per_region.py → resource_environment_metrics_inference_per_region.py} +5 -5
  85. beamlit/models/resource_environment_metrics_inference_per_second_per_region.py +9 -11
  86. beamlit/models/{resource_deployment_metrics_query_per_region_per_code.py → resource_environment_metrics_query_per_region_per_code.py} +5 -5
  87. beamlit/models/resource_environment_metrics_query_per_second_per_region_per_code.py +5 -3
  88. beamlit/models/resource_log.py +4 -2
  89. beamlit/models/runtime.py +20 -2
  90. beamlit/models/serverless_config.py +28 -21
  91. beamlit/models/spec_configuration.py +7 -5
  92. beamlit/models/store_agent.py +14 -14
  93. beamlit/models/store_function.py +14 -14
  94. beamlit/models/time_fields.py +4 -2
  95. beamlit/models/websocket_channel.py +4 -2
  96. beamlit/models/workspace.py +23 -14
  97. beamlit/run.py +0 -1
  98. beamlit/serve/app.py +1 -0
  99. beamlit/serve/middlewares/accesslog.py +3 -1
  100. {beamlit-0.0.24rc20.dist-info → beamlit-0.0.24rc21.dist-info}/METADATA +1 -1
  101. beamlit-0.0.24rc21.dist-info/RECORD +243 -0
  102. beamlit/api/agents/delete_agent_deployment.py +0 -163
  103. beamlit/api/agents/delete_agent_deployment_history.py +0 -172
  104. beamlit/api/agents/get_agent_deployment_history.py +0 -172
  105. beamlit/api/agents/get_agent_deployment_logs.py +0 -164
  106. beamlit/api/agents/list_agent_deployment_history.py +0 -164
  107. beamlit/api/agents/put_agent_deployment.py +0 -185
  108. beamlit/api/agents/put_agent_deployment_history.py +0 -198
  109. beamlit/api/functions/delete_function_deployment.py +0 -163
  110. beamlit/api/functions/get_function_deployment.py +0 -163
  111. beamlit/api/functions/get_function_deployment_logs.py +0 -164
  112. beamlit/api/functions/get_function_deployment_metrics.py +0 -159
  113. beamlit/api/functions/put_function_deployment.py +0 -185
  114. beamlit/api/models/delete_model_deployment.py +0 -171
  115. beamlit/api/models/get_model_deployment.py +0 -171
  116. beamlit/api/models/get_model_deployment_logs.py +0 -168
  117. beamlit/api/models/get_model_deployment_metrics.py +0 -163
  118. beamlit/api/models/put_model_deployment.py +0 -193
  119. beamlit/models/agent_configuration.py +0 -70
  120. beamlit/models/agent_deployment.py +0 -340
  121. beamlit/models/agent_deployment_configuration.py +0 -45
  122. beamlit/models/agent_deployment_configuration_type_0.py +0 -43
  123. beamlit/models/agent_deployment_history.py +0 -185
  124. beamlit/models/agent_deployment_history_event.py +0 -133
  125. beamlit/models/agent_deployment_pod_template.py +0 -45
  126. beamlit/models/agent_deployment_pod_template_type_0.py +0 -43
  127. beamlit/models/agent_with_deployments.py +0 -176
  128. beamlit/models/authentication_provider_model.py +0 -144
  129. beamlit/models/authentication_provider_organization.py +0 -88
  130. beamlit/models/deployment_configuration.py +0 -70
  131. beamlit/models/deployment_configurations.py +0 -58
  132. beamlit/models/deployment_serverless_config.py +0 -131
  133. beamlit/models/deployment_serverless_config_type_0.py +0 -220
  134. beamlit/models/function_configuration.py +0 -70
  135. beamlit/models/function_deployment.py +0 -340
  136. beamlit/models/function_deployment_configuration.py +0 -45
  137. beamlit/models/function_deployment_configuration_type_0.py +0 -43
  138. beamlit/models/function_deployment_pod_template.py +0 -45
  139. beamlit/models/function_deployment_pod_template_type_0.py +0 -43
  140. beamlit/models/function_with_deployments.py +0 -176
  141. beamlit/models/integration.py +0 -198
  142. beamlit/models/integration_config.py +0 -45
  143. beamlit/models/integration_secret.py +0 -61
  144. beamlit/models/labels_type_0.py +0 -45
  145. beamlit/models/location.py +0 -122
  146. beamlit/models/model_deployment.py +0 -296
  147. beamlit/models/model_deployment_log.py +0 -70
  148. beamlit/models/model_deployment_metrics.py +0 -172
  149. beamlit/models/model_deployment_metrics_inference_per_second_per_region.py +0 -77
  150. beamlit/models/model_deployment_pod_template.py +0 -45
  151. beamlit/models/model_deployment_pod_template_type_0.py +0 -43
  152. beamlit/models/model_metrics.py +0 -96
  153. beamlit/models/model_with_deployments.py +0 -176
  154. beamlit/models/resource_deployment_log.py +0 -70
  155. beamlit/models/resource_deployment_metrics_inference_per_region_type_0.py +0 -79
  156. beamlit/models/resource_deployment_metrics_inference_per_second_per_region_type_0.py +0 -79
  157. beamlit/models/resource_deployment_metrics_query_per_region_per_code_type_0.py +0 -73
  158. beamlit/models/resource_deployment_metrics_query_per_second_per_region_per_code_type_0.py +0 -73
  159. beamlit/models/runtime_readiness_probe_type_0.py +0 -43
  160. beamlit/models/runtime_type_0.py +0 -111
  161. beamlit/models/runtime_type_0_readiness_probe.py +0 -43
  162. beamlit/models/runtime_type_0_readiness_probe_type_0.py +0 -43
  163. beamlit/models/runtime_type_0_resources.py +0 -59
  164. beamlit/models/standard_fields_dynamo_db.py +0 -88
  165. beamlit/models/store_agent_configuration.py +0 -97
  166. beamlit/models/store_agent_labels_type_0.py +0 -43
  167. beamlit/models/store_function_configuration.py +0 -97
  168. beamlit/models/store_function_labels_type_0.py +0 -43
  169. beamlit-0.0.24rc20.dist-info/RECORD +0 -303
  170. {beamlit-0.0.24rc20.dist-info → beamlit-0.0.24rc21.dist-info}/WHEEL +0 -0
@@ -5,7 +5,7 @@ import httpx
5
5
 
6
6
  from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
- from ...models.agent_deployment_history import AgentDeploymentHistory
8
+ from ...models.agent_history import AgentHistory
9
9
  from ...types import Response
10
10
 
11
11
 
@@ -22,12 +22,12 @@ def _get_kwargs(
22
22
 
23
23
  def _parse_response(
24
24
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
25
- ) -> Optional[list["AgentDeploymentHistory"]]:
25
+ ) -> Optional[list["AgentHistory"]]:
26
26
  if response.status_code == 200:
27
27
  response_200 = []
28
28
  _response_200 = response.json()
29
29
  for response_200_item_data in _response_200:
30
- response_200_item = AgentDeploymentHistory.from_dict(response_200_item_data)
30
+ response_200_item = AgentHistory.from_dict(response_200_item_data)
31
31
 
32
32
  response_200.append(response_200_item)
33
33
 
@@ -40,7 +40,7 @@ def _parse_response(
40
40
 
41
41
  def _build_response(
42
42
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
43
- ) -> Response[list["AgentDeploymentHistory"]]:
43
+ ) -> Response[list["AgentHistory"]]:
44
44
  return Response(
45
45
  status_code=HTTPStatus(response.status_code),
46
46
  content=response.content,
@@ -53,7 +53,7 @@ def sync_detailed(
53
53
  request_id: str,
54
54
  *,
55
55
  client: AuthenticatedClient,
56
- ) -> Response[list["AgentDeploymentHistory"]]:
56
+ ) -> Response[list["AgentHistory"]]:
57
57
  """Get all history for a specific request ID from all agents
58
58
 
59
59
  Args:
@@ -64,7 +64,7 @@ def sync_detailed(
64
64
  httpx.TimeoutException: If the request takes longer than Client.timeout.
65
65
 
66
66
  Returns:
67
- Response[list['AgentDeploymentHistory']]
67
+ Response[list['AgentHistory']]
68
68
  """
69
69
 
70
70
  kwargs = _get_kwargs(
@@ -82,7 +82,7 @@ def sync(
82
82
  request_id: str,
83
83
  *,
84
84
  client: AuthenticatedClient,
85
- ) -> Optional[list["AgentDeploymentHistory"]]:
85
+ ) -> Optional[list["AgentHistory"]]:
86
86
  """Get all history for a specific request ID from all agents
87
87
 
88
88
  Args:
@@ -93,7 +93,7 @@ def sync(
93
93
  httpx.TimeoutException: If the request takes longer than Client.timeout.
94
94
 
95
95
  Returns:
96
- list['AgentDeploymentHistory']
96
+ list['AgentHistory']
97
97
  """
98
98
 
99
99
  return sync_detailed(
@@ -106,7 +106,7 @@ async def asyncio_detailed(
106
106
  request_id: str,
107
107
  *,
108
108
  client: AuthenticatedClient,
109
- ) -> Response[list["AgentDeploymentHistory"]]:
109
+ ) -> Response[list["AgentHistory"]]:
110
110
  """Get all history for a specific request ID from all agents
111
111
 
112
112
  Args:
@@ -117,7 +117,7 @@ async def asyncio_detailed(
117
117
  httpx.TimeoutException: If the request takes longer than Client.timeout.
118
118
 
119
119
  Returns:
120
- Response[list['AgentDeploymentHistory']]
120
+ Response[list['AgentHistory']]
121
121
  """
122
122
 
123
123
  kwargs = _get_kwargs(
@@ -133,7 +133,7 @@ async def asyncio(
133
133
  request_id: str,
134
134
  *,
135
135
  client: AuthenticatedClient,
136
- ) -> Optional[list["AgentDeploymentHistory"]]:
136
+ ) -> Optional[list["AgentHistory"]]:
137
137
  """Get all history for a specific request ID from all agents
138
138
 
139
139
  Args:
@@ -144,7 +144,7 @@ async def asyncio(
144
144
  httpx.TimeoutException: If the request takes longer than Client.timeout.
145
145
 
146
146
  Returns:
147
- list['AgentDeploymentHistory']
147
+ list['AgentHistory']
148
148
  """
149
149
 
150
150
  return (
@@ -5,7 +5,7 @@ import httpx
5
5
 
6
6
  from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
- from ...models.agent_deployment_history import AgentDeploymentHistory
8
+ from ...models.agent_history import AgentHistory
9
9
  from ...types import Response
10
10
 
11
11
 
@@ -20,12 +20,12 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[list["AgentDeploymentHistory"]]:
23
+ ) -> Optional[list["AgentHistory"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
27
27
  for response_200_item_data in _response_200:
28
- response_200_item = AgentDeploymentHistory.from_dict(response_200_item_data)
28
+ response_200_item = AgentHistory.from_dict(response_200_item_data)
29
29
 
30
30
  response_200.append(response_200_item)
31
31
 
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[list["AgentDeploymentHistory"]]:
41
+ ) -> Response[list["AgentHistory"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: AuthenticatedClient,
53
- ) -> Response[list["AgentDeploymentHistory"]]:
53
+ ) -> Response[list["AgentHistory"]]:
54
54
  """Get all history for all agents
55
55
 
56
56
  Raises:
@@ -58,7 +58,7 @@ def sync_detailed(
58
58
  httpx.TimeoutException: If the request takes longer than Client.timeout.
59
59
 
60
60
  Returns:
61
- Response[list['AgentDeploymentHistory']]
61
+ Response[list['AgentHistory']]
62
62
  """
63
63
 
64
64
  kwargs = _get_kwargs()
@@ -73,7 +73,7 @@ def sync_detailed(
73
73
  def sync(
74
74
  *,
75
75
  client: AuthenticatedClient,
76
- ) -> Optional[list["AgentDeploymentHistory"]]:
76
+ ) -> Optional[list["AgentHistory"]]:
77
77
  """Get all history for all agents
78
78
 
79
79
  Raises:
@@ -81,7 +81,7 @@ def sync(
81
81
  httpx.TimeoutException: If the request takes longer than Client.timeout.
82
82
 
83
83
  Returns:
84
- list['AgentDeploymentHistory']
84
+ list['AgentHistory']
85
85
  """
86
86
 
87
87
  return sync_detailed(
@@ -92,7 +92,7 @@ def sync(
92
92
  async def asyncio_detailed(
93
93
  *,
94
94
  client: AuthenticatedClient,
95
- ) -> Response[list["AgentDeploymentHistory"]]:
95
+ ) -> Response[list["AgentHistory"]]:
96
96
  """Get all history for all agents
97
97
 
98
98
  Raises:
@@ -100,7 +100,7 @@ async def asyncio_detailed(
100
100
  httpx.TimeoutException: If the request takes longer than Client.timeout.
101
101
 
102
102
  Returns:
103
- Response[list['AgentDeploymentHistory']]
103
+ Response[list['AgentHistory']]
104
104
  """
105
105
 
106
106
  kwargs = _get_kwargs()
@@ -113,7 +113,7 @@ async def asyncio_detailed(
113
113
  async def asyncio(
114
114
  *,
115
115
  client: AuthenticatedClient,
116
- ) -> Optional[list["AgentDeploymentHistory"]]:
116
+ ) -> Optional[list["AgentHistory"]]:
117
117
  """Get all history for all agents
118
118
 
119
119
  Raises:
@@ -121,7 +121,7 @@ async def asyncio(
121
121
  httpx.TimeoutException: If the request takes longer than Client.timeout.
122
122
 
123
123
  Returns:
124
- list['AgentDeploymentHistory']
124
+ list['AgentHistory']
125
125
  """
126
126
 
127
127
  return (
@@ -5,7 +5,7 @@ import httpx
5
5
 
6
6
  from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
- from ...models.integration import Integration
8
+ from ...models.integration_connection import IntegrationConnection
9
9
  from ...types import Response
10
10
 
11
11
 
@@ -20,9 +20,11 @@ def _get_kwargs(
20
20
  return _kwargs
21
21
 
22
22
 
23
- def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Integration]:
23
+ def _parse_response(
24
+ *, client: Union[AuthenticatedClient, Client], response: httpx.Response
25
+ ) -> Optional[IntegrationConnection]:
24
26
  if response.status_code == 200:
25
- response_200 = Integration.from_dict(response.json())
27
+ response_200 = IntegrationConnection.from_dict(response.json())
26
28
 
27
29
  return response_200
28
30
  if client.raise_on_unexpected_status:
@@ -31,7 +33,9 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
31
33
  return None
32
34
 
33
35
 
34
- def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Integration]:
36
+ def _build_response(
37
+ *, client: Union[AuthenticatedClient, Client], response: httpx.Response
38
+ ) -> Response[IntegrationConnection]:
35
39
  return Response(
36
40
  status_code=HTTPStatus(response.status_code),
37
41
  content=response.content,
@@ -44,7 +48,7 @@ def sync_detailed(
44
48
  connection_name: str,
45
49
  *,
46
50
  client: Union[AuthenticatedClient, Client],
47
- ) -> Response[Integration]:
51
+ ) -> Response[IntegrationConnection]:
48
52
  """Get integration
49
53
 
50
54
  Returns an integration connection by integration name and connection name.
@@ -57,7 +61,7 @@ def sync_detailed(
57
61
  httpx.TimeoutException: If the request takes longer than Client.timeout.
58
62
 
59
63
  Returns:
60
- Response[Integration]
64
+ Response[IntegrationConnection]
61
65
  """
62
66
 
63
67
  kwargs = _get_kwargs(
@@ -75,7 +79,7 @@ def sync(
75
79
  connection_name: str,
76
80
  *,
77
81
  client: Union[AuthenticatedClient, Client],
78
- ) -> Optional[Integration]:
82
+ ) -> Optional[IntegrationConnection]:
79
83
  """Get integration
80
84
 
81
85
  Returns an integration connection by integration name and connection name.
@@ -88,7 +92,7 @@ def sync(
88
92
  httpx.TimeoutException: If the request takes longer than Client.timeout.
89
93
 
90
94
  Returns:
91
- Integration
95
+ IntegrationConnection
92
96
  """
93
97
 
94
98
  return sync_detailed(
@@ -101,7 +105,7 @@ async def asyncio_detailed(
101
105
  connection_name: str,
102
106
  *,
103
107
  client: Union[AuthenticatedClient, Client],
104
- ) -> Response[Integration]:
108
+ ) -> Response[IntegrationConnection]:
105
109
  """Get integration
106
110
 
107
111
  Returns an integration connection by integration name and connection name.
@@ -114,7 +118,7 @@ async def asyncio_detailed(
114
118
  httpx.TimeoutException: If the request takes longer than Client.timeout.
115
119
 
116
120
  Returns:
117
- Response[Integration]
121
+ Response[IntegrationConnection]
118
122
  """
119
123
 
120
124
  kwargs = _get_kwargs(
@@ -130,7 +134,7 @@ async def asyncio(
130
134
  connection_name: str,
131
135
  *,
132
136
  client: Union[AuthenticatedClient, Client],
133
- ) -> Optional[Integration]:
137
+ ) -> Optional[IntegrationConnection]:
134
138
  """Get integration
135
139
 
136
140
  Returns an integration connection by integration name and connection name.
@@ -143,7 +147,7 @@ async def asyncio(
143
147
  httpx.TimeoutException: If the request takes longer than Client.timeout.
144
148
 
145
149
  Returns:
146
- Integration
150
+ IntegrationConnection
147
151
  """
148
152
 
149
153
  return (
@@ -10,10 +10,11 @@ from ...types import Response
10
10
 
11
11
  def _get_kwargs(
12
12
  connection_name: str,
13
+ model_id: str,
13
14
  ) -> dict[str, Any]:
14
15
  _kwargs: dict[str, Any] = {
15
16
  "method": "get",
16
- "url": f"/integrations/connections/{connection_name}/models/*modelId",
17
+ "url": f"/integrations/connections/{connection_name}/models/{model_id}",
17
18
  }
18
19
 
19
20
  return _kwargs
@@ -39,6 +40,7 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
39
40
 
40
41
  def sync_detailed(
41
42
  connection_name: str,
43
+ model_id: str,
42
44
  *,
43
45
  client: AuthenticatedClient,
44
46
  ) -> Response[Any]:
@@ -48,6 +50,7 @@ def sync_detailed(
48
50
 
49
51
  Args:
50
52
  connection_name (str):
53
+ model_id (str):
51
54
 
52
55
  Raises:
53
56
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -59,6 +62,7 @@ def sync_detailed(
59
62
 
60
63
  kwargs = _get_kwargs(
61
64
  connection_name=connection_name,
65
+ model_id=model_id,
62
66
  )
63
67
 
64
68
  response = client.get_httpx_client().request(
@@ -70,6 +74,7 @@ def sync_detailed(
70
74
 
71
75
  async def asyncio_detailed(
72
76
  connection_name: str,
77
+ model_id: str,
73
78
  *,
74
79
  client: AuthenticatedClient,
75
80
  ) -> Response[Any]:
@@ -79,6 +84,7 @@ async def asyncio_detailed(
79
84
 
80
85
  Args:
81
86
  connection_name (str):
87
+ model_id (str):
82
88
 
83
89
  Raises:
84
90
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -90,6 +96,7 @@ async def asyncio_detailed(
90
96
 
91
97
  kwargs = _get_kwargs(
92
98
  connection_name=connection_name,
99
+ model_id=model_id,
93
100
  )
94
101
 
95
102
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -10,10 +10,11 @@ from ...types import Response
10
10
 
11
11
  def _get_kwargs(
12
12
  integration_name: str,
13
+ model_id: str,
13
14
  ) -> dict[str, Any]:
14
15
  _kwargs: dict[str, Any] = {
15
16
  "method": "get",
16
- "url": f"/integrations/{integration_name}/models/*modelId",
17
+ "url": f"/integrations/{integration_name}/models/{model_id}",
17
18
  }
18
19
 
19
20
  return _kwargs
@@ -39,6 +40,7 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
39
40
 
40
41
  def sync_detailed(
41
42
  integration_name: str,
43
+ model_id: str,
42
44
  *,
43
45
  client: AuthenticatedClient,
44
46
  ) -> Response[Any]:
@@ -48,6 +50,7 @@ def sync_detailed(
48
50
 
49
51
  Args:
50
52
  integration_name (str):
53
+ model_id (str):
51
54
 
52
55
  Raises:
53
56
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -59,6 +62,7 @@ def sync_detailed(
59
62
 
60
63
  kwargs = _get_kwargs(
61
64
  integration_name=integration_name,
65
+ model_id=model_id,
62
66
  )
63
67
 
64
68
  response = client.get_httpx_client().request(
@@ -70,6 +74,7 @@ def sync_detailed(
70
74
 
71
75
  async def asyncio_detailed(
72
76
  integration_name: str,
77
+ model_id: str,
73
78
  *,
74
79
  client: AuthenticatedClient,
75
80
  ) -> Response[Any]:
@@ -79,6 +84,7 @@ async def asyncio_detailed(
79
84
 
80
85
  Args:
81
86
  integration_name (str):
87
+ model_id (str):
82
88
 
83
89
  Raises:
84
90
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -90,6 +96,7 @@ async def asyncio_detailed(
90
96
 
91
97
  kwargs = _get_kwargs(
92
98
  integration_name=integration_name,
99
+ model_id=model_id,
93
100
  )
94
101
 
95
102
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -5,7 +5,7 @@ import httpx
5
5
 
6
6
  from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
- from ...models.location import Location
8
+ from ...models.location_response import LocationResponse
9
9
  from ...types import Response
10
10
 
11
11
 
@@ -20,12 +20,12 @@ def _get_kwargs() -> dict[str, Any]:
20
20
 
21
21
  def _parse_response(
22
22
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
23
- ) -> Optional[list["Location"]]:
23
+ ) -> Optional[list["LocationResponse"]]:
24
24
  if response.status_code == 200:
25
25
  response_200 = []
26
26
  _response_200 = response.json()
27
27
  for response_200_item_data in _response_200:
28
- response_200_item = Location.from_dict(response_200_item_data)
28
+ response_200_item = LocationResponse.from_dict(response_200_item_data)
29
29
 
30
30
  response_200.append(response_200_item)
31
31
 
@@ -38,7 +38,7 @@ def _parse_response(
38
38
 
39
39
  def _build_response(
40
40
  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
41
- ) -> Response[list["Location"]]:
41
+ ) -> Response[list["LocationResponse"]]:
42
42
  return Response(
43
43
  status_code=HTTPStatus(response.status_code),
44
44
  content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
50
50
  def sync_detailed(
51
51
  *,
52
52
  client: AuthenticatedClient,
53
- ) -> Response[list["Location"]]:
53
+ ) -> Response[list["LocationResponse"]]:
54
54
  """List locations
55
55
 
56
56
  Returns a list of all locations available with status.
@@ -60,7 +60,7 @@ def sync_detailed(
60
60
  httpx.TimeoutException: If the request takes longer than Client.timeout.
61
61
 
62
62
  Returns:
63
- Response[list['Location']]
63
+ Response[list['LocationResponse']]
64
64
  """
65
65
 
66
66
  kwargs = _get_kwargs()
@@ -75,7 +75,7 @@ def sync_detailed(
75
75
  def sync(
76
76
  *,
77
77
  client: AuthenticatedClient,
78
- ) -> Optional[list["Location"]]:
78
+ ) -> Optional[list["LocationResponse"]]:
79
79
  """List locations
80
80
 
81
81
  Returns a list of all locations available with status.
@@ -85,7 +85,7 @@ def sync(
85
85
  httpx.TimeoutException: If the request takes longer than Client.timeout.
86
86
 
87
87
  Returns:
88
- list['Location']
88
+ list['LocationResponse']
89
89
  """
90
90
 
91
91
  return sync_detailed(
@@ -96,7 +96,7 @@ def sync(
96
96
  async def asyncio_detailed(
97
97
  *,
98
98
  client: AuthenticatedClient,
99
- ) -> Response[list["Location"]]:
99
+ ) -> Response[list["LocationResponse"]]:
100
100
  """List locations
101
101
 
102
102
  Returns a list of all locations available with status.
@@ -106,7 +106,7 @@ async def asyncio_detailed(
106
106
  httpx.TimeoutException: If the request takes longer than Client.timeout.
107
107
 
108
108
  Returns:
109
- Response[list['Location']]
109
+ Response[list['LocationResponse']]
110
110
  """
111
111
 
112
112
  kwargs = _get_kwargs()
@@ -119,7 +119,7 @@ async def asyncio_detailed(
119
119
  async def asyncio(
120
120
  *,
121
121
  client: AuthenticatedClient,
122
- ) -> Optional[list["Location"]]:
122
+ ) -> Optional[list["LocationResponse"]]:
123
123
  """List locations
124
124
 
125
125
  Returns a list of all locations available with status.
@@ -129,7 +129,7 @@ async def asyncio(
129
129
  httpx.TimeoutException: If the request takes longer than Client.timeout.
130
130
 
131
131
  Returns:
132
- list['Location']
132
+ list['LocationResponse']
133
133
  """
134
134
 
135
135
  return (
@@ -6,13 +6,12 @@ import httpx
6
6
  from ... import errors
7
7
  from ...client import AuthenticatedClient, Client
8
8
  from ...models.model import Model
9
- from ...models.model_with_deployments import ModelWithDeployments
10
9
  from ...types import Response
11
10
 
12
11
 
13
12
  def _get_kwargs(
14
13
  *,
15
- body: ModelWithDeployments,
14
+ body: Model,
16
15
  ) -> dict[str, Any]:
17
16
  headers: dict[str, Any] = {}
18
17
 
@@ -53,15 +52,15 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
53
52
  def sync_detailed(
54
53
  *,
55
54
  client: AuthenticatedClient,
56
- body: ModelWithDeployments,
55
+ body: Model,
57
56
  ) -> Response[Model]:
58
57
  """Create model
59
58
 
60
59
  Creates a model.
61
60
 
62
61
  Args:
63
- body (ModelWithDeployments): Logical object representing a model but with deployment
64
- definition inside
62
+ body (Model): Logical object representing a model, that can be instantiated in multiple
63
+ environments as model deployments
65
64
 
66
65
  Raises:
67
66
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -85,15 +84,15 @@ def sync_detailed(
85
84
  def sync(
86
85
  *,
87
86
  client: AuthenticatedClient,
88
- body: ModelWithDeployments,
87
+ body: Model,
89
88
  ) -> Optional[Model]:
90
89
  """Create model
91
90
 
92
91
  Creates a model.
93
92
 
94
93
  Args:
95
- body (ModelWithDeployments): Logical object representing a model but with deployment
96
- definition inside
94
+ body (Model): Logical object representing a model, that can be instantiated in multiple
95
+ environments as model deployments
97
96
 
98
97
  Raises:
99
98
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -112,15 +111,15 @@ def sync(
112
111
  async def asyncio_detailed(
113
112
  *,
114
113
  client: AuthenticatedClient,
115
- body: ModelWithDeployments,
114
+ body: Model,
116
115
  ) -> Response[Model]:
117
116
  """Create model
118
117
 
119
118
  Creates a model.
120
119
 
121
120
  Args:
122
- body (ModelWithDeployments): Logical object representing a model but with deployment
123
- definition inside
121
+ body (Model): Logical object representing a model, that can be instantiated in multiple
122
+ environments as model deployments
124
123
 
125
124
  Raises:
126
125
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -142,15 +141,15 @@ async def asyncio_detailed(
142
141
  async def asyncio(
143
142
  *,
144
143
  client: AuthenticatedClient,
145
- body: ModelWithDeployments,
144
+ body: Model,
146
145
  ) -> Optional[Model]:
147
146
  """Create model
148
147
 
149
148
  Creates a model.
150
149
 
151
150
  Args:
152
- body (ModelWithDeployments): Logical object representing a model but with deployment
153
- definition inside
151
+ body (Model): Logical object representing a model, that can be instantiated in multiple
152
+ environments as model deployments
154
153
 
155
154
  Raises:
156
155
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.