beamlit 0.0.18__py3-none-any.whl → 0.0.20rc1__py3-none-any.whl

beamlit/models/agent_deployment.py

@@ -7,9 +7,9 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.agent_chain import AgentChain
-    from ..models.agent_deployment_configuration import AgentDeploymentConfiguration
-    from ..models.agent_deployment_pod_template import AgentDeploymentPodTemplate
-    from ..models.deployment_serverless_config import DeploymentServerlessConfig
+    from ..models.agent_deployment_configuration_type_0 import AgentDeploymentConfigurationType0
+    from ..models.agent_deployment_pod_template_type_0 import AgentDeploymentPodTemplateType0
+    from ..models.deployment_serverless_config_type_0 import DeploymentServerlessConfigType0
     from ..models.flavor import Flavor
     from ..models.labels_type_0 import LabelsType0
     from ..models.runtime import Runtime
@@ -29,22 +29,24 @@ class AgentDeployment:
         updated_by (Union[Unset, str]): The user or service account who updated the resource
         agent (Union[Unset, str]): The name of the agent
         agent_chain (Union[Unset, List['AgentChain']]): Agent chaining configuration
-        configuration (Union[Unset, AgentDeploymentConfiguration]): Agent configuration, this is a key value storage. In
-            your agent you can retrieve the value with config[key]
+        configuration (Union['AgentDeploymentConfigurationType0', None, Unset]): Agent configuration, this is a key
+            value storage. In your agent you can retrieve the value with config[key]
         description (Union[Unset, str]): Agent description, very important to have a clear description for your agent if
             you want to make it work with agent chaining
-        enabled (Union[Unset, bool]): Whether the agent deployment is enabled
+        enabled (Union[None, Unset, bool]): Whether the agent deployment is enabled
         environment (Union[Unset, str]): The name of the environment
         flavors (Union[Unset, List['Flavor']]): Types of hardware available for deployments
         functions (Union[Unset, List[str]]): Functions used by the agent, those functions needs to be created before
             setting it here
+        integration_connections (Union[Unset, List[str]]):
         labels (Union['LabelsType0', None, Unset]): Labels
-        llm (Union[Unset, str]): LLM to use for agent, it should be compatible with function calling
-        pod_template (Union[Unset, AgentDeploymentPodTemplate]): The pod template, should be a valid Kubernetes pod
-            template
+        model (Union[Unset, str]): Model beamlit to use for agent, it should be compatible with function calling
+        pod_template (Union['AgentDeploymentPodTemplateType0', None, Unset]): The pod template, should be a valid
+            Kubernetes pod template
         policies (Union[Unset, List[str]]):
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
-        serverless_config (Union[Unset, DeploymentServerlessConfig]): Configuration for a serverless deployment
+        serverless_config (Union['DeploymentServerlessConfigType0', None, Unset]): Configuration for a serverless
+            deployment
         store_id (Union[Unset, str]): Create from a store registered function
         workspace (Union[Unset, str]): The workspace the agent deployment belongs to
     """
@@ -55,23 +57,27 @@ class AgentDeployment:
     updated_by: Union[Unset, str] = UNSET
     agent: Union[Unset, str] = UNSET
     agent_chain: Union[Unset, List["AgentChain"]] = UNSET
-    configuration: Union[Unset, "AgentDeploymentConfiguration"] = UNSET
+    configuration: Union["AgentDeploymentConfigurationType0", None, Unset] = UNSET
     description: Union[Unset, str] = UNSET
-    enabled: Union[Unset, bool] = UNSET
+    enabled: Union[None, Unset, bool] = UNSET
     environment: Union[Unset, str] = UNSET
     flavors: Union[Unset, List["Flavor"]] = UNSET
     functions: Union[Unset, List[str]] = UNSET
+    integration_connections: Union[Unset, List[str]] = UNSET
     labels: Union["LabelsType0", None, Unset] = UNSET
-    llm: Union[Unset, str] = UNSET
-    pod_template: Union[Unset, "AgentDeploymentPodTemplate"] = UNSET
+    model: Union[Unset, str] = UNSET
+    pod_template: Union["AgentDeploymentPodTemplateType0", None, Unset] = UNSET
     policies: Union[Unset, List[str]] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
-    serverless_config: Union[Unset, "DeploymentServerlessConfig"] = UNSET
+    serverless_config: Union["DeploymentServerlessConfigType0", None, Unset] = UNSET
     store_id: Union[Unset, str] = UNSET
     workspace: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        from ..models.agent_deployment_configuration_type_0 import AgentDeploymentConfigurationType0
+        from ..models.agent_deployment_pod_template_type_0 import AgentDeploymentPodTemplateType0
+        from ..models.deployment_serverless_config_type_0 import DeploymentServerlessConfigType0
         from ..models.labels_type_0 import LabelsType0
 
         created_at = self.created_at
@@ -91,13 +97,21 @@ class AgentDeployment:
                 agent_chain_item = agent_chain_item_data.to_dict()
                 agent_chain.append(agent_chain_item)
 
-        configuration: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.configuration, Unset):
+        configuration: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.configuration, Unset):
+            configuration = UNSET
+        elif isinstance(self.configuration, AgentDeploymentConfigurationType0):
             configuration = self.configuration.to_dict()
+        else:
+            configuration = self.configuration
 
         description = self.description
 
-        enabled = self.enabled
+        enabled: Union[None, Unset, bool]
+        if isinstance(self.enabled, Unset):
+            enabled = UNSET
+        else:
+            enabled = self.enabled
 
         environment = self.environment
 
@@ -112,6 +126,10 @@ class AgentDeployment:
         if not isinstance(self.functions, Unset):
             functions = self.functions
 
+        integration_connections: Union[Unset, List[str]] = UNSET
+        if not isinstance(self.integration_connections, Unset):
+            integration_connections = self.integration_connections
+
         labels: Union[Dict[str, Any], None, Unset]
         if isinstance(self.labels, Unset):
             labels = UNSET
@@ -120,11 +138,15 @@ class AgentDeployment:
         else:
             labels = self.labels
 
-        llm = self.llm
+        model = self.model
 
-        pod_template: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.pod_template, Unset):
+        pod_template: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.pod_template, Unset):
+            pod_template = UNSET
+        elif isinstance(self.pod_template, AgentDeploymentPodTemplateType0):
             pod_template = self.pod_template.to_dict()
+        else:
+            pod_template = self.pod_template
 
         policies: Union[Unset, List[str]] = UNSET
         if not isinstance(self.policies, Unset):
@@ -134,9 +156,13 @@ class AgentDeployment:
         if not isinstance(self.runtime, Unset):
             runtime = self.runtime.to_dict()
 
-        serverless_config: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.serverless_config, Unset):
+        serverless_config: Union[Dict[str, Any], None, Unset]
+        if isinstance(self.serverless_config, Unset):
+            serverless_config = UNSET
+        elif isinstance(self.serverless_config, DeploymentServerlessConfigType0):
             serverless_config = self.serverless_config.to_dict()
+        else:
+            serverless_config = self.serverless_config
 
         store_id = self.store_id
 
@@ -169,10 +195,12 @@ class AgentDeployment:
             field_dict["flavors"] = flavors
         if functions is not UNSET:
             field_dict["functions"] = functions
+        if integration_connections is not UNSET:
+            field_dict["integration_connections"] = integration_connections
         if labels is not UNSET:
             field_dict["labels"] = labels
-        if llm is not UNSET:
-            field_dict["llm"] = llm
+        if model is not UNSET:
+            field_dict["model"] = model
         if pod_template is not UNSET:
             field_dict["pod_template"] = pod_template
         if policies is not UNSET:
@@ -191,9 +219,9 @@ class AgentDeployment:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         from ..models.agent_chain import AgentChain
-        from ..models.agent_deployment_configuration import AgentDeploymentConfiguration
-        from ..models.agent_deployment_pod_template import AgentDeploymentPodTemplate
-        from ..models.deployment_serverless_config import DeploymentServerlessConfig
+        from ..models.agent_deployment_configuration_type_0 import AgentDeploymentConfigurationType0
+        from ..models.agent_deployment_pod_template_type_0 import AgentDeploymentPodTemplateType0
+        from ..models.deployment_serverless_config_type_0 import DeploymentServerlessConfigType0
         from ..models.flavor import Flavor
         from ..models.labels_type_0 import LabelsType0
         from ..models.runtime import Runtime
@@ -216,16 +244,33 @@ class AgentDeployment:
 
             agent_chain.append(agent_chain_item)
 
-        _configuration = d.pop("configuration", UNSET)
-        configuration: Union[Unset, AgentDeploymentConfiguration]
-        if isinstance(_configuration, Unset):
-            configuration = UNSET
-        else:
-            configuration = AgentDeploymentConfiguration.from_dict(_configuration)
+        def _parse_configuration(data: object) -> Union["AgentDeploymentConfigurationType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                configuration_type_0 = AgentDeploymentConfigurationType0.from_dict(data)
+
+                return configuration_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["AgentDeploymentConfigurationType0", None, Unset], data)
+
+        configuration = _parse_configuration(d.pop("configuration", UNSET))
 
         description = d.pop("description", UNSET)
 
-        enabled = d.pop("enabled", UNSET)
+        def _parse_enabled(data: object) -> Union[None, Unset, bool]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, bool], data)
+
+        enabled = _parse_enabled(d.pop("enabled", UNSET))
 
         environment = d.pop("environment", UNSET)
 
@@ -238,6 +283,8 @@ class AgentDeployment:
 
         functions = cast(List[str], d.pop("functions", UNSET))
 
+        integration_connections = cast(List[str], d.pop("integration_connections", UNSET))
+
         def _parse_labels(data: object) -> Union["LabelsType0", None, Unset]:
             if data is None:
                 return data
@@ -255,14 +302,24 @@ class AgentDeployment:
 
         labels = _parse_labels(d.pop("labels", UNSET))
 
-        llm = d.pop("llm", UNSET)
+        model = d.pop("model", UNSET)
 
-        _pod_template = d.pop("pod_template", UNSET)
-        pod_template: Union[Unset, AgentDeploymentPodTemplate]
-        if isinstance(_pod_template, Unset):
-            pod_template = UNSET
-        else:
-            pod_template = AgentDeploymentPodTemplate.from_dict(_pod_template)
+        def _parse_pod_template(data: object) -> Union["AgentDeploymentPodTemplateType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                pod_template_type_0 = AgentDeploymentPodTemplateType0.from_dict(data)
+
+                return pod_template_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["AgentDeploymentPodTemplateType0", None, Unset], data)
+
+        pod_template = _parse_pod_template(d.pop("pod_template", UNSET))
 
         policies = cast(List[str], d.pop("policies", UNSET))
 
@@ -273,12 +330,22 @@ class AgentDeployment:
         else:
             runtime = Runtime.from_dict(_runtime)
 
-        _serverless_config = d.pop("serverless_config", UNSET)
-        serverless_config: Union[Unset, DeploymentServerlessConfig]
-        if isinstance(_serverless_config, Unset):
-            serverless_config = UNSET
-        else:
-            serverless_config = DeploymentServerlessConfig.from_dict(_serverless_config)
+        def _parse_serverless_config(data: object) -> Union["DeploymentServerlessConfigType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                componentsschemas_deployment_serverless_config_type_0 = DeploymentServerlessConfigType0.from_dict(data)
+
+                return componentsschemas_deployment_serverless_config_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["DeploymentServerlessConfigType0", None, Unset], data)
+
+        serverless_config = _parse_serverless_config(d.pop("serverless_config", UNSET))
 
         store_id = d.pop("store_id", UNSET)
 
@@ -297,8 +364,9 @@ class AgentDeployment:
             environment=environment,
             flavors=flavors,
             functions=functions,
+            integration_connections=integration_connections,
             labels=labels,
-            llm=llm,
+            model=model,
             pod_template=pod_template,
             policies=policies,
             runtime=runtime,

beamlit/models/{agent_deployment_configuration.py → agent_deployment_configuration_type_0.py}

@@ -3,11 +3,11 @@ from typing import Any, Type, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="AgentDeploymentConfiguration")
+T = TypeVar("T", bound="AgentDeploymentConfigurationType0")
 
 
 @_attrs_define
-class AgentDeploymentConfiguration:
+class AgentDeploymentConfigurationType0:
     """Agent configuration, this is a key value storage. In your agent you can retrieve the value with config[key]"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -21,10 +21,10 @@ class AgentDeploymentConfiguration:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        agent_deployment_configuration = cls()
+        agent_deployment_configuration_type_0 = cls()
 
-        agent_deployment_configuration.additional_properties = d
-        return agent_deployment_configuration
+        agent_deployment_configuration_type_0.additional_properties = d
+        return agent_deployment_configuration_type_0
 
     @property
     def additional_keys(self) -> list[str]:

beamlit/models/agent_deployment_history.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -22,13 +22,13 @@ class AgentDeploymentHistory:
         updated_at (Union[Unset, str]): The date and time when the resource was updated
         updated_by (Union[Unset, str]): The user or service account who updated the resource
         agent (Union[Unset, str]): Agent name
-        end (Union[Unset, str]): End time
+        end (Union[None, Unset, str]): End time
         environment (Union[Unset, str]): Environment name
         events (Union[Unset, List['AgentDeploymentHistoryEvent']]): Events
         request_id (Union[Unset, str]): Request ID
-        start (Union[Unset, str]): Start time
+        start (Union[None, Unset, str]): Start time
         status (Union[Unset, str]): Status, eg: running, success, failed
-        took (Union[Unset, int]): Number of milliseconds it took to complete the event
+        took (Union[None, Unset, int]): Number of milliseconds it took to complete the event
         workspace (Union[Unset, str]): The workspace the agent deployment belongs to
     """
 
@@ -37,13 +37,13 @@ class AgentDeploymentHistory:
     updated_at: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
     agent: Union[Unset, str] = UNSET
-    end: Union[Unset, str] = UNSET
+    end: Union[None, Unset, str] = UNSET
     environment: Union[Unset, str] = UNSET
     events: Union[Unset, List["AgentDeploymentHistoryEvent"]] = UNSET
     request_id: Union[Unset, str] = UNSET
-    start: Union[Unset, str] = UNSET
+    start: Union[None, Unset, str] = UNSET
     status: Union[Unset, str] = UNSET
-    took: Union[Unset, int] = UNSET
+    took: Union[None, Unset, int] = UNSET
     workspace: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -58,7 +58,11 @@ class AgentDeploymentHistory:
 
         agent = self.agent
 
-        end = self.end
+        end: Union[None, Unset, str]
+        if isinstance(self.end, Unset):
+            end = UNSET
+        else:
+            end = self.end
 
         environment = self.environment
 
@@ -71,11 +75,19 @@ class AgentDeploymentHistory:
 
         request_id = self.request_id
 
-        start = self.start
+        start: Union[None, Unset, str]
+        if isinstance(self.start, Unset):
+            start = UNSET
+        else:
+            start = self.start
 
         status = self.status
 
-        took = self.took
+        took: Union[None, Unset, int]
+        if isinstance(self.took, Unset):
+            took = UNSET
+        else:
+            took = self.took
 
         workspace = self.workspace
 
@@ -126,7 +138,14 @@ class AgentDeploymentHistory:
 
         agent = d.pop("agent", UNSET)
 
-        end = d.pop("end", UNSET)
+        def _parse_end(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        end = _parse_end(d.pop("end", UNSET))
 
         environment = d.pop("environment", UNSET)
 
@@ -139,11 +158,25 @@ class AgentDeploymentHistory:
 
         request_id = d.pop("request_id", UNSET)
 
-        start = d.pop("start", UNSET)
+        def _parse_start(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        start = _parse_start(d.pop("start", UNSET))
 
         status = d.pop("status", UNSET)
 
-        took = d.pop("took", UNSET)
+        def _parse_took(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
+
+        took = _parse_took(d.pop("took", UNSET))
 
         workspace = d.pop("workspace", UNSET)
 

beamlit/models/agent_deployment_history_event.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, Type, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -13,44 +13,64 @@ class AgentDeploymentHistoryEvent:
     """Agent deployment history event
 
     Attributes:
-        end (Union[Unset, str]): End time
-        error (Union[Unset, str]): Error message
+        end (Union[None, Unset, str]): End time
+        error (Union[None, Unset, str]): Error message
         name (Union[Unset, str]): Name of the function or agent
         parameters (Union[Unset, str]): Parameters
-        start (Union[Unset, str]): Start time
+        start (Union[None, Unset, str]): Start time
         status (Union[Unset, str]): Status, eg: running, success, failed
-        sub_function (Union[Unset, str]): Function used in kit if a kit was used
-        took (Union[Unset, int]): Number of milliseconds it took to complete the event
+        sub_function (Union[None, Unset, str]): Function used in kit if a kit was used
+        took (Union[None, Unset, int]): Number of milliseconds it took to complete the event
         type (Union[Unset, str]): Type, one of function or agent
     """
 
-    end: Union[Unset, str] = UNSET
-    error: Union[Unset, str] = UNSET
+    end: Union[None, Unset, str] = UNSET
+    error: Union[None, Unset, str] = UNSET
     name: Union[Unset, str] = UNSET
     parameters: Union[Unset, str] = UNSET
-    start: Union[Unset, str] = UNSET
+    start: Union[None, Unset, str] = UNSET
     status: Union[Unset, str] = UNSET
-    sub_function: Union[Unset, str] = UNSET
-    took: Union[Unset, int] = UNSET
+    sub_function: Union[None, Unset, str] = UNSET
+    took: Union[None, Unset, int] = UNSET
     type: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        end = self.end
-
-        error = self.error
+        end: Union[None, Unset, str]
+        if isinstance(self.end, Unset):
+            end = UNSET
+        else:
+            end = self.end
+
+        error: Union[None, Unset, str]
+        if isinstance(self.error, Unset):
+            error = UNSET
+        else:
+            error = self.error
 
         name = self.name
 
         parameters = self.parameters
 
-        start = self.start
+        start: Union[None, Unset, str]
+        if isinstance(self.start, Unset):
+            start = UNSET
+        else:
+            start = self.start
 
         status = self.status
 
-        sub_function = self.sub_function
+        sub_function: Union[None, Unset, str]
+        if isinstance(self.sub_function, Unset):
+            sub_function = UNSET
+        else:
+            sub_function = self.sub_function
 
-        took = self.took
+        took: Union[None, Unset, int]
+        if isinstance(self.took, Unset):
+            took = UNSET
+        else:
+            took = self.took
 
         type = self.type
 
@@ -81,21 +101,57 @@ class AgentDeploymentHistoryEvent:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        end = d.pop("end", UNSET)
 
-        error = d.pop("error", UNSET)
+        def _parse_end(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        end = _parse_end(d.pop("end", UNSET))
+
+        def _parse_error(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        error = _parse_error(d.pop("error", UNSET))
 
         name = d.pop("name", UNSET)
 
         parameters = d.pop("parameters", UNSET)
 
-        start = d.pop("start", UNSET)
+        def _parse_start(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        start = _parse_start(d.pop("start", UNSET))
 
         status = d.pop("status", UNSET)
 
-        sub_function = d.pop("sub_function", UNSET)
+        def _parse_sub_function(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        sub_function = _parse_sub_function(d.pop("sub_function", UNSET))
+
+        def _parse_took(data: object) -> Union[None, Unset, int]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, int], data)
 
-        took = d.pop("took", UNSET)
+        took = _parse_took(d.pop("took", UNSET))
 
         type = d.pop("type", UNSET)
 

beamlit/models/{function_deployment_pod_template.py → agent_deployment_pod_template_type_0.py}

@@ -3,11 +3,11 @@ from typing import Any, Type, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="FunctionDeploymentPodTemplate")
+T = TypeVar("T", bound="AgentDeploymentPodTemplateType0")
 
 
 @_attrs_define
-class FunctionDeploymentPodTemplate:
+class AgentDeploymentPodTemplateType0:
     """The pod template, should be a valid Kubernetes pod template"""
 
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -21,10 +21,10 @@ class FunctionDeploymentPodTemplate:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        function_deployment_pod_template = cls()
+        agent_deployment_pod_template_type_0 = cls()
 
-        function_deployment_pod_template.additional_properties = d
-        return function_deployment_pod_template
+        agent_deployment_pod_template_type_0.additional_properties = d
+        return agent_deployment_pod_template_type_0
 
     @property
     def additional_keys(self) -> list[str]: