beamlit 0.0.18__py3-none-any.whl → 0.0.20rc1__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.
Files changed (73) hide show
  1. beamlit/api/agents/create_agent.py +14 -9
  2. beamlit/api/agents/get_agent_deployment.py +22 -1
  3. beamlit/api/agents/update_agent.py +14 -9
  4. beamlit/api/functions/create_function.py +14 -9
  5. beamlit/api/functions/update_function.py +14 -9
  6. beamlit/api/{authentication_providers/list_organizations_for_authentication_provider.py → integrations/create_integration_connection.py} +45 -41
  7. beamlit/api/integrations/delete_integration_connection.py +158 -0
  8. beamlit/api/integrations/get_integration.py +97 -0
  9. beamlit/api/integrations/get_integration_connection.py +154 -0
  10. beamlit/api/integrations/get_integration_connection_model.py +97 -0
  11. beamlit/api/integrations/get_integration_model.py +97 -0
  12. beamlit/api/integrations/list_integration_connection_models.py +97 -0
  13. beamlit/api/{authentication_providers/list_models_for_authentication_provider.py → integrations/list_integration_connections.py} +24 -48
  14. beamlit/api/integrations/list_integration_models.py +97 -0
  15. beamlit/api/integrations/update_integration_connection.py +180 -0
  16. beamlit/authentication/__init__.py +2 -1
  17. beamlit/authentication/authentication.py +3 -0
  18. beamlit/authentication/clientcredentials.py +99 -0
  19. beamlit/authentication/credentials.py +7 -1
  20. beamlit/authentication/device_mode.py +0 -2
  21. beamlit/common/generate.py +166 -0
  22. beamlit/common/logger.py +30 -0
  23. beamlit/common/settings.py +112 -0
  24. beamlit/common/utils.py +13 -0
  25. beamlit/models/__init__.py +50 -24
  26. beamlit/models/agent_deployment.py +117 -49
  27. beamlit/models/{agent_deployment_configuration.py → agent_deployment_configuration_type_0.py} +5 -5
  28. beamlit/models/agent_deployment_history.py +46 -13
  29. beamlit/models/agent_deployment_history_event.py +78 -22
  30. beamlit/models/{function_deployment_pod_template.py → agent_deployment_pod_template_type_0.py} +5 -5
  31. beamlit/models/agent_with_deployments.py +174 -0
  32. beamlit/models/deployment_configurations.py +19 -6
  33. beamlit/models/deployment_serverless_config_type_0.py +218 -0
  34. beamlit/models/environment_metrics.py +19 -0
  35. beamlit/models/function_deployment.py +110 -42
  36. beamlit/models/{function_deployment_configuration.py → function_deployment_configuration_type_0.py} +5 -5
  37. beamlit/models/{agent_deployment_pod_template.py → function_deployment_pod_template_type_0.py} +5 -5
  38. beamlit/models/function_with_deployments.py +174 -0
  39. beamlit/models/increase_and_rate_metric.py +102 -0
  40. beamlit/models/integration.py +196 -0
  41. beamlit/models/integration_config.py +43 -0
  42. beamlit/models/integration_connection.py +196 -0
  43. beamlit/models/integration_connection_config.py +43 -0
  44. beamlit/models/integration_connection_secret.py +59 -0
  45. beamlit/models/integration_model.py +142 -0
  46. beamlit/models/integration_secret.py +59 -0
  47. beamlit/models/metrics.py +61 -20
  48. beamlit/models/model_deployment.py +96 -64
  49. beamlit/models/{model_deployment_pod_template.py → model_deployment_pod_template_type_0.py} +5 -5
  50. beamlit/models/policy.py +8 -34
  51. beamlit/models/provider_config.py +0 -9
  52. beamlit/models/resource_deployment_metrics.py +231 -36
  53. beamlit/models/resource_deployment_metrics_inference_per_region_type_0.py +75 -0
  54. beamlit/models/{resource_deployment_metrics_inference_per_second_per_region.py → resource_deployment_metrics_inference_per_second_per_region_type_0.py} +5 -5
  55. beamlit/models/resource_deployment_metrics_query_per_region_per_code_type_0.py +73 -0
  56. beamlit/models/{resource_deployment_metrics_query_per_second_per_region_per_code.py → resource_deployment_metrics_query_per_second_per_region_per_code_type_0.py} +5 -5
  57. beamlit/models/resource_metrics.py +36 -0
  58. beamlit/models/runtime.py +29 -12
  59. beamlit/models/{runtime_readiness_probe.py → runtime_readiness_probe_type_0.py} +5 -5
  60. beamlit/models/store_agent.py +29 -13
  61. beamlit/models/{store_agent_labels.py → store_agent_labels_type_0.py} +5 -5
  62. beamlit/models/store_configuration.py +15 -4
  63. beamlit/models/store_configuration_option.py +18 -5
  64. beamlit/models/store_function.py +29 -13
  65. beamlit/models/{store_function_labels.py → store_function_labels_type_0.py} +5 -5
  66. beamlit/run.py +49 -0
  67. {beamlit-0.0.18.dist-info → beamlit-0.0.20rc1.dist-info}/METADATA +5 -2
  68. {beamlit-0.0.18.dist-info → beamlit-0.0.20rc1.dist-info}/RECORD +70 -46
  69. beamlit/api/authentication_providers/get_model_with_repo_for_authentication_provider.py +0 -184
  70. beamlit/models/deployment_serverless_config.py +0 -129
  71. beamlit/models/labels.py +0 -43
  72. /beamlit/api/{authentication_providers → integrations}/__init__.py +0 -0
  73. {beamlit-0.0.18.dist-info → beamlit-0.0.20rc1.dist-info}/WHEEL +0 -0
@@ -0,0 +1,166 @@
1
+ from typing import Dict, Tuple
2
+
3
+ from beamlit.common.settings import Settings, get_settings
4
+ from beamlit.models.agent_deployment import AgentDeployment
5
+ from beamlit.models.function_deployment import FunctionDeployment
6
+ from beamlit.models.function_kit import FunctionKit
7
+
8
+
9
+ def get_titles_name(name: str) -> str:
10
+ return name.title().replace("-", "").replace("_", "")
11
+
12
+ def generate_kit_function_code(settings: Settings, function: FunctionDeployment, kit: FunctionKit) -> Tuple[str, str]:
13
+ export_code = ""
14
+ code = ""
15
+ for kit in kit:
16
+ body = {"function": kit.name, "workspace": settings.workspace, **kit.to_dict()}
17
+ new_code, export = generate_function_code(body, force_name_in_endpoint=function.function, kit=True)
18
+ code += new_code
19
+ export_code += export
20
+ return code, export_code
21
+
22
+ def generate_function_code(settings: Settings, function: FunctionDeployment, force_name_in_endpoint: str = "", kit: bool = False) -> Tuple[str, str]:
23
+ name = get_titles_name(function.function)
24
+ if function.parameters and len(function.parameters) > 0:
25
+ args_list = ", ".join(f"{param.name}: str" for param in function.parameters)
26
+ args_list += ", "
27
+ else:
28
+ args_list = ""
29
+ args_schema = ""
30
+ if function.parameters:
31
+ for param in function.parameters:
32
+ args_schema += f'{param.name}: str = Field(description="""{param.description}""")\n '
33
+ if len(args_schema) == 0:
34
+ args_schema = "pass"
35
+
36
+ # TODO: add return direct in function configuration
37
+ return_direct = False
38
+ endpoint_name = force_name_in_endpoint or function.function
39
+ body = "{}"
40
+ if function.parameters:
41
+ body = f'{", ".join(f'"{param.name}": {param.name}' for param in function.parameters)}'
42
+ if kit is True:
43
+ has_name = False
44
+ if function.parameters:
45
+ for param in function.parameters:
46
+ if param.name == "name":
47
+ has_name = True
48
+ break
49
+ if not has_name:
50
+ if len(body) > 0:
51
+ body += ", "
52
+ body += f'"name": "{function.function}"'
53
+ return f'''
54
+
55
+ class Beamlit{name}Input(BaseModel):
56
+ {args_schema}
57
+
58
+ class Beamlit{name}(BaseTool):
59
+ name: str = "beamlit_{function.function.replace("-", "_")}"
60
+ description: str = """{function.description}"""
61
+ args_schema: Type[BaseModel] = Beamlit{name}Input
62
+
63
+ response_format: Literal["content_and_artifact"] = "content_and_artifact"
64
+ return_direct: bool = {return_direct}
65
+
66
+ def _run(self, {args_list} run_manager: Optional[CallbackManagerForToolRun] = None) -> Tuple[Union[List[Dict[str, str]], str], Dict]:
67
+ try:
68
+ params = self.metadata.get("params", {{}})
69
+ response = run_client.run("function", "{endpoint_name}", settings.environment, "POST", json={{{body}}})
70
+ if response.status_code >= 400:
71
+ logger.error(f"Failed to run function {name}, {{response.status_code}}::{{response.text}}")
72
+ raise Exception(f"Failed to run function {name}, {{response.status_code}}::{{response.text}}")
73
+ return response.json(), {{}}
74
+ except Exception as e:
75
+ return repr(e), {{}}
76
+ ''', f'Beamlit{get_titles_name(function.function)},'
77
+
78
+ def generate_chain_code(settings: Settings, agent: AgentDeployment) -> Tuple[str, str]:
79
+ name = get_titles_name(agent.agent)
80
+ # TODO: add return direct in agent configuration
81
+ return_direct = False
82
+ return f'''
83
+ class BeamlitChain{name}Input(BaseModel):
84
+ input: str = Field(description='{agent.description}')
85
+
86
+ class BeamlitChain{name}(BaseTool):
87
+ name: str = "beamlit_chain_{agent.agent.replace("-", "_")}"
88
+ description: str = """{agent.description}"""
89
+ args_schema: Type[BaseModel] = BeamlitChain{name}Input
90
+
91
+ response_format: Literal["content_and_artifact"] = "content_and_artifact"
92
+ return_direct: bool = {return_direct}
93
+
94
+ def _run(
95
+ self,
96
+ input: str,
97
+ run_manager: Optional[CallbackManagerForToolRun] = None,
98
+ ) -> Tuple[Union[List[Dict[str, str]], str], Dict]:
99
+ try:
100
+ params = self.metadata.get("params", {{}})
101
+ response = run_client.run("agent", "{agent.agent}", settings.environment, "POST", json={{"input": input}})
102
+ if response.status_code >= 400:
103
+ logger.error(f"Failed to run tool {agent.agent}, {{response.status_code}}::{{response.text}}")
104
+ raise Exception(f"Failed to run tool {agent.agent}, {{response.status_code}}::{{response.text}}")
105
+ if response.headers.get("Content-Type") == "application/json":
106
+ return response.json(), {{}}
107
+ else:
108
+ return response.text, {{}}
109
+ except Exception as e:
110
+ return repr(e), {{}}
111
+ ''', f'BeamlitChain{name},'
112
+
113
+ def generate(destination: str, dry_run: bool = False):
114
+ imports = '''from logging import getLogger
115
+ from typing import Dict, List, Literal, Optional, Tuple, Type, Union
116
+
117
+ from langchain_core.callbacks import CallbackManagerForToolRun
118
+ from langchain_core.tools import BaseTool
119
+ from pydantic import BaseModel, Field
120
+ from beamlit.authentication import (RunClientWithCredentials,
121
+ load_credentials_from_config,
122
+ new_client_with_credentials)
123
+ from beamlit.common.settings import get_settings
124
+ from beamlit.run import RunClient
125
+
126
+ logger = getLogger(__name__)
127
+ settings = get_settings()
128
+ credentials = load_credentials_from_config(settings)
129
+
130
+ client_config = RunClientWithCredentials(
131
+ credentials=credentials,
132
+ workspace=settings.workspace,
133
+ )
134
+ client = new_client_with_credentials(client_config)
135
+ run_client = RunClient(client=client)
136
+ '''
137
+ settings = get_settings()
138
+ export_code = '\n\nfunctions = ['
139
+ export_chain = '\n\nchains = ['
140
+ code = imports
141
+ if settings.agent_functions and len(settings.agent_functions) > 0:
142
+ for function_config in settings.agent_functions:
143
+ if function_config.kit and len(function_config.kit) > 0:
144
+ new_code, export = generate_kit_function_code(settings, function_config)
145
+ code += new_code
146
+ export_code += export
147
+ else:
148
+ new_code, export = generate_function_code(settings, function_config)
149
+ code += new_code
150
+ export_code += export
151
+ if settings.agent_chain and len(settings.agent_chain) > 0:
152
+ for agent in settings.agent_chain:
153
+ new_code, export = generate_chain_code(settings, agent)
154
+ code += new_code
155
+ export_chain += export
156
+ if settings.agent_functions and len(settings.agent_functions) > 0:
157
+ export_code = export_code[:-1]
158
+ export_code += ']'
159
+ if settings.agent_chain and len(settings.agent_chain) > 0:
160
+ export_chain = export_chain[:-1]
161
+ export_chain += ']'
162
+ content = code + export_code + export_chain
163
+ if not dry_run:
164
+ with open(destination, "w") as f:
165
+ f.write(content)
166
+ return content
@@ -0,0 +1,30 @@
1
+ import logging
2
+ import os
3
+
4
+
5
+ class ColoredFormatter(logging.Formatter):
6
+ COLORS = {
7
+ 'DEBUG': '\033[1;36m', # Cyan
8
+ 'INFO': '\033[1;32m', # Green
9
+ 'WARNING': '\033[1;33m', # Yellow
10
+ 'ERROR': '\033[1;31m', # Red
11
+ 'CRITICAL': '\033[1;41m' # Red background
12
+ }
13
+
14
+ def format(self, record):
15
+ color = self.COLORS.get(record.levelname, '\033[0m')
16
+ record.levelname = f"{color}{record.levelname}\033[0m"
17
+ return super().format(record)
18
+
19
+ def init(log_level: str):
20
+ logging.getLogger("uvicorn.access").handlers.clear()
21
+ logging.getLogger("uvicorn.access").propagate = False
22
+ logging.getLogger("httpx").handlers.clear()
23
+ logging.getLogger("httpx").propagate = False
24
+
25
+ handler = logging.StreamHandler()
26
+ handler.setFormatter(ColoredFormatter('%(levelname)s:\t %(name)s - %(message)s'))
27
+ logging.basicConfig(
28
+ level=log_level,
29
+ handlers=[handler]
30
+ )
@@ -0,0 +1,112 @@
1
+ import os
2
+ from logging import getLogger
3
+ from typing import Dict, List, Tuple, Type, Union
4
+
5
+ from beamlit.api.functions import get_function_deployment
6
+ from beamlit.api.models import get_model_deployment
7
+ from beamlit.client import AuthenticatedClient
8
+ from beamlit.common.logger import init as init_logger
9
+ from beamlit.models.agent_chain import AgentChain
10
+ from beamlit.models.agent_deployment import AgentDeployment
11
+ from beamlit.models.function_deployment import FunctionDeployment
12
+ from beamlit.models.model_deployment import ModelDeployment
13
+ from beamlit.types import UNSET, Unset
14
+ from pydantic import Field
15
+ from pydantic_settings import (BaseSettings, PydanticBaseSettingsSource,
16
+ SettingsConfigDict, YamlConfigSettingsSource)
17
+
18
+ global SETTINGS
19
+ SETTINGS = None
20
+
21
+ def get_settings():
22
+ return SETTINGS
23
+
24
+ class Settings(BaseSettings):
25
+ model_config = SettingsConfigDict(
26
+ cli_parse_args=True,
27
+ yaml_file="beamlit.yaml",
28
+ env_prefix="bl_",
29
+ )
30
+
31
+ workspace: str
32
+ type: str
33
+ name: str
34
+ environment: str = Field(default="production")
35
+ base_url: str = Field(default="https://api.beamlit.dev/v0")
36
+ run_url: str = Field(default="https://run.beamlit.dev")
37
+ port: int = Field(default=80)
38
+ host: str = Field(default="0.0.0.0")
39
+ log_level: str = Field(default="INFO")
40
+ api_key: Union[None, str] = None
41
+ jwt: Union[None, str] = None
42
+ client_credentials: Union[None, str] = None
43
+ agent_chain: Union[Unset, List[AgentDeployment]] = UNSET
44
+ agent_functions: Union[Unset, List[FunctionDeployment]] = UNSET
45
+ agent_model: Union[Unset, ModelDeployment] = UNSET
46
+
47
+ @classmethod
48
+ def settings_customise_sources(
49
+ cls,
50
+ settings_cls: Type[BaseSettings],
51
+ init_settings: PydanticBaseSettingsSource,
52
+ env_settings: PydanticBaseSettingsSource,
53
+ dotenv_settings: PydanticBaseSettingsSource,
54
+ file_secret_settings: PydanticBaseSettingsSource,
55
+ ) -> Tuple[PydanticBaseSettingsSource, ...]:
56
+ return (env_settings, dotenv_settings, file_secret_settings, YamlConfigSettingsSource(settings_cls), init_settings, )
57
+
58
+ def init_agent(client: AuthenticatedClient):
59
+ from beamlit.api.agents import get_agent_deployment
60
+ from beamlit.common.generate import generate
61
+
62
+ logger = getLogger(__name__)
63
+
64
+ # Init configuration from environment variables
65
+ if SETTINGS.agent_functions or SETTINGS.agent_chain:
66
+ return
67
+
68
+ # Init configuration from beamlit control plane
69
+ name = SETTINGS.name
70
+ env = SETTINGS.environment
71
+
72
+ agent_deployment = get_agent_deployment.sync(name, env, client=client)
73
+ function_deployments = []
74
+ agent_chain_deployments = []
75
+ if agent_deployment.functions:
76
+ for function in agent_deployment.functions:
77
+ function_deployment = get_function_deployment.sync(function, env, client=client)
78
+ function_deployments.append(function_deployment)
79
+ SETTINGS.agent_functions = function_deployments
80
+
81
+ if agent_deployment.agent_chain:
82
+ for chain in agent_deployment.agent_chain:
83
+ if chain.enabled:
84
+ agent_deployment = get_agent_deployment.sync(chain.name, env, client=client)
85
+ if chain.description:
86
+ agent_deployment.description = chain.description
87
+ agent_chain_deployments.append(agent_deployment)
88
+ SETTINGS.agent_chain = agent_chain_deployments
89
+ if agent_deployment.model:
90
+ model_deployment = get_model_deployment.sync(agent_deployment.model, env , client=client)
91
+ SETTINGS.agent_model = model_deployment
92
+
93
+
94
+ destination = f"{os.getcwd()}/src/beamlit_generated.py"
95
+
96
+
97
+ content_generate = generate(destination, dry_run=True)
98
+ compared_content = None
99
+ if os.path.exists(destination):
100
+ compared_content = open(destination, "r").read()
101
+
102
+ if not os.path.exists(destination) or (compared_content and content_generate != compared_content):
103
+ logger.info("Generating agent code")
104
+ generate(destination)
105
+
106
+ def init() -> Settings:
107
+ """Parse the beamlit.yaml file to get configurations."""
108
+ global SETTINGS
109
+
110
+ SETTINGS = Settings()
111
+ init_logger(SETTINGS.log_level)
112
+ return SETTINGS
@@ -0,0 +1,13 @@
1
+ import filecmp
2
+ import os
3
+ import shutil
4
+
5
+
6
+ def copy_folder(source_folder: str, destination_folder: str):
7
+ for file in os.listdir(source_folder):
8
+ if os.path.isdir(f"{source_folder}/{file}"):
9
+ if not os.path.exists(f"{destination_folder}/{file}"):
10
+ os.makedirs(f"{destination_folder}/{file}")
11
+ copy_folder(f"{source_folder}/{file}", f"{destination_folder}/{file}")
12
+ elif not os.path.exists(f"{destination_folder}/{file}") or not filecmp.cmp(f"{source_folder}/{file}", f"{destination_folder}/{file}"):
13
+ shutil.copy(f"{source_folder}/{file}", f"{destination_folder}/{file}")
@@ -5,11 +5,12 @@ from .agent import Agent
5
5
  from .agent_chain import AgentChain
6
6
  from .agent_configuration import AgentConfiguration
7
7
  from .agent_deployment import AgentDeployment
8
- from .agent_deployment_configuration import AgentDeploymentConfiguration
8
+ from .agent_deployment_configuration_type_0 import AgentDeploymentConfigurationType0
9
9
  from .agent_deployment_history import AgentDeploymentHistory
10
10
  from .agent_deployment_history_event import AgentDeploymentHistoryEvent
11
- from .agent_deployment_pod_template import AgentDeploymentPodTemplate
11
+ from .agent_deployment_pod_template_type_0 import AgentDeploymentPodTemplateType0
12
12
  from .agent_release import AgentRelease
13
+ from .agent_with_deployments import AgentWithDeployments
13
14
  from .api_key import ApiKey
14
15
  from .authentication_provider_model import AuthenticationProviderModel
15
16
  from .authentication_provider_organization import AuthenticationProviderOrganization
@@ -22,19 +23,28 @@ from .create_workspace_service_account_response_200 import CreateWorkspaceServic
22
23
  from .delete_workspace_service_account_response_200 import DeleteWorkspaceServiceAccountResponse200
23
24
  from .deployment_configuration import DeploymentConfiguration
24
25
  from .deployment_configurations import DeploymentConfigurations
25
- from .deployment_serverless_config import DeploymentServerlessConfig
26
+ from .deployment_serverless_config_type_0 import DeploymentServerlessConfigType0
26
27
  from .environment import Environment
27
28
  from .environment_metrics import EnvironmentMetrics
28
29
  from .flavor import Flavor
29
30
  from .function import Function
30
31
  from .function_configuration import FunctionConfiguration
31
32
  from .function_deployment import FunctionDeployment
32
- from .function_deployment_configuration import FunctionDeploymentConfiguration
33
- from .function_deployment_pod_template import FunctionDeploymentPodTemplate
33
+ from .function_deployment_configuration_type_0 import FunctionDeploymentConfigurationType0
34
+ from .function_deployment_pod_template_type_0 import FunctionDeploymentPodTemplateType0
34
35
  from .function_kit import FunctionKit
35
36
  from .function_provider_ref import FunctionProviderRef
36
37
  from .function_release import FunctionRelease
38
+ from .function_with_deployments import FunctionWithDeployments
37
39
  from .get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
40
+ from .increase_and_rate_metric import IncreaseAndRateMetric
41
+ from .integration import Integration
42
+ from .integration_config import IntegrationConfig
43
+ from .integration_connection import IntegrationConnection
44
+ from .integration_connection_config import IntegrationConnectionConfig
45
+ from .integration_connection_secret import IntegrationConnectionSecret
46
+ from .integration_model import IntegrationModel
47
+ from .integration_secret import IntegrationSecret
38
48
  from .invite_workspace_user_body import InviteWorkspaceUserBody
39
49
  from .labels_type_0 import LabelsType0
40
50
  from .location import Location
@@ -49,7 +59,7 @@ from .model_deployment_metrics_inference_per_second_per_region import ModelDeplo
49
59
  from .model_deployment_metrics_query_per_second_per_region_per_code import (
50
60
  ModelDeploymentMetricsQueryPerSecondPerRegionPerCode,
51
61
  )
52
- from .model_deployment_pod_template import ModelDeploymentPodTemplate
62
+ from .model_deployment_pod_template_type_0 import ModelDeploymentPodTemplateType0
53
63
  from .model_metrics import ModelMetrics
54
64
  from .model_provider import ModelProvider
55
65
  from .model_provider_ref import ModelProviderRef
@@ -67,27 +77,31 @@ from .provider_config import ProviderConfig
67
77
  from .qps import QPS
68
78
  from .resource_deployment_log import ResourceDeploymentLog
69
79
  from .resource_deployment_metrics import ResourceDeploymentMetrics
70
- from .resource_deployment_metrics_inference_per_second_per_region import (
71
- ResourceDeploymentMetricsInferencePerSecondPerRegion,
80
+ from .resource_deployment_metrics_inference_per_region_type_0 import ResourceDeploymentMetricsInferencePerRegionType0
81
+ from .resource_deployment_metrics_inference_per_second_per_region_type_0 import (
82
+ ResourceDeploymentMetricsInferencePerSecondPerRegionType0,
72
83
  )
73
- from .resource_deployment_metrics_query_per_second_per_region_per_code import (
74
- ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode,
84
+ from .resource_deployment_metrics_query_per_region_per_code_type_0 import (
85
+ ResourceDeploymentMetricsQueryPerRegionPerCodeType0,
86
+ )
87
+ from .resource_deployment_metrics_query_per_second_per_region_per_code_type_0 import (
88
+ ResourceDeploymentMetricsQueryPerSecondPerRegionPerCodeType0,
75
89
  )
76
90
  from .resource_metrics import ResourceMetrics
77
91
  from .runtime import Runtime
78
- from .runtime_readiness_probe import RuntimeReadinessProbe
92
+ from .runtime_readiness_probe_type_0 import RuntimeReadinessProbeType0
79
93
  from .runtime_resources import RuntimeResources
80
94
  from .serverless_config import ServerlessConfig
81
95
  from .standard_fields_dynamo_db import StandardFieldsDynamoDb
82
96
  from .store_agent import StoreAgent
83
97
  from .store_agent_configuration import StoreAgentConfiguration
84
- from .store_agent_labels import StoreAgentLabels
98
+ from .store_agent_labels_type_0 import StoreAgentLabelsType0
85
99
  from .store_configuration import StoreConfiguration
86
100
  from .store_configuration_option import StoreConfigurationOption
87
101
  from .store_function import StoreFunction
88
102
  from .store_function_configuration import StoreFunctionConfiguration
89
103
  from .store_function_kit import StoreFunctionKit
90
- from .store_function_labels import StoreFunctionLabels
104
+ from .store_function_labels_type_0 import StoreFunctionLabelsType0
91
105
  from .store_function_parameter import StoreFunctionParameter
92
106
  from .update_workspace_service_account_body import UpdateWorkspaceServiceAccountBody
93
107
  from .update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
@@ -102,11 +116,12 @@ __all__ = (
102
116
  "AgentChain",
103
117
  "AgentConfiguration",
104
118
  "AgentDeployment",
105
- "AgentDeploymentConfiguration",
119
+ "AgentDeploymentConfigurationType0",
106
120
  "AgentDeploymentHistory",
107
121
  "AgentDeploymentHistoryEvent",
108
- "AgentDeploymentPodTemplate",
122
+ "AgentDeploymentPodTemplateType0",
109
123
  "AgentRelease",
124
+ "AgentWithDeployments",
110
125
  "ApiKey",
111
126
  "AuthenticationProviderModel",
112
127
  "AuthenticationProviderOrganization",
@@ -119,19 +134,28 @@ __all__ = (
119
134
  "DeleteWorkspaceServiceAccountResponse200",
120
135
  "DeploymentConfiguration",
121
136
  "DeploymentConfigurations",
122
- "DeploymentServerlessConfig",
137
+ "DeploymentServerlessConfigType0",
123
138
  "Environment",
124
139
  "EnvironmentMetrics",
125
140
  "Flavor",
126
141
  "Function",
127
142
  "FunctionConfiguration",
128
143
  "FunctionDeployment",
129
- "FunctionDeploymentConfiguration",
130
- "FunctionDeploymentPodTemplate",
144
+ "FunctionDeploymentConfigurationType0",
145
+ "FunctionDeploymentPodTemplateType0",
131
146
  "FunctionKit",
132
147
  "FunctionProviderRef",
133
148
  "FunctionRelease",
149
+ "FunctionWithDeployments",
134
150
  "GetWorkspaceServiceAccountsResponse200Item",
151
+ "IncreaseAndRateMetric",
152
+ "Integration",
153
+ "IntegrationConfig",
154
+ "IntegrationConnection",
155
+ "IntegrationConnectionConfig",
156
+ "IntegrationConnectionSecret",
157
+ "IntegrationModel",
158
+ "IntegrationSecret",
135
159
  "InviteWorkspaceUserBody",
136
160
  "LabelsType0",
137
161
  "Location",
@@ -144,7 +168,7 @@ __all__ = (
144
168
  "ModelDeploymentMetrics",
145
169
  "ModelDeploymentMetricsInferencePerSecondPerRegion",
146
170
  "ModelDeploymentMetricsQueryPerSecondPerRegionPerCode",
147
- "ModelDeploymentPodTemplate",
171
+ "ModelDeploymentPodTemplateType0",
148
172
  "ModelMetrics",
149
173
  "ModelProvider",
150
174
  "ModelProviderRef",
@@ -162,23 +186,25 @@ __all__ = (
162
186
  "QPS",
163
187
  "ResourceDeploymentLog",
164
188
  "ResourceDeploymentMetrics",
165
- "ResourceDeploymentMetricsInferencePerSecondPerRegion",
166
- "ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode",
189
+ "ResourceDeploymentMetricsInferencePerRegionType0",
190
+ "ResourceDeploymentMetricsInferencePerSecondPerRegionType0",
191
+ "ResourceDeploymentMetricsQueryPerRegionPerCodeType0",
192
+ "ResourceDeploymentMetricsQueryPerSecondPerRegionPerCodeType0",
167
193
  "ResourceMetrics",
168
194
  "Runtime",
169
- "RuntimeReadinessProbe",
195
+ "RuntimeReadinessProbeType0",
170
196
  "RuntimeResources",
171
197
  "ServerlessConfig",
172
198
  "StandardFieldsDynamoDb",
173
199
  "StoreAgent",
174
200
  "StoreAgentConfiguration",
175
- "StoreAgentLabels",
201
+ "StoreAgentLabelsType0",
176
202
  "StoreConfiguration",
177
203
  "StoreConfigurationOption",
178
204
  "StoreFunction",
179
205
  "StoreFunctionConfiguration",
180
206
  "StoreFunctionKit",
181
- "StoreFunctionLabels",
207
+ "StoreFunctionLabelsType0",
182
208
  "StoreFunctionParameter",
183
209
  "UpdateWorkspaceServiceAccountBody",
184
210
  "UpdateWorkspaceServiceAccountResponse200",