beamlit 0.0.45rc91__py3-none-any.whl → 0.0.47__py3-none-any.whl

beamlit/api/agents/get_agent_metrics.py

@@ -5,7 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.resource_metrics import ResourceMetrics
+from ...models.resource_environment_metrics import ResourceEnvironmentMetrics
 from ...types import UNSET, Response, Unset
 
 
@@ -31,9 +31,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     if response.status_code == 200:
-        response_200 = ResourceMetrics.from_dict(response.json())
+        response_200 = ResourceEnvironmentMetrics.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -44,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,7 +58,7 @@ def sync_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -70,7 +70,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -90,7 +90,7 @@ def sync(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -102,7 +102,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return sync_detailed(
@@ -117,7 +117,7 @@ async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -129,7 +129,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -147,7 +147,7 @@ async def asyncio(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -159,7 +159,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return (

beamlit/api/functions/get_function_metrics.py

@@ -5,7 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.resource_metrics import ResourceMetrics
+from ...models.resource_environment_metrics import ResourceEnvironmentMetrics
 from ...types import UNSET, Response, Unset
 
 
@@ -31,9 +31,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     if response.status_code == 200:
-        response_200 = ResourceMetrics.from_dict(response.json())
+        response_200 = ResourceEnvironmentMetrics.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -44,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,7 +58,7 @@ def sync_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -70,7 +70,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -90,7 +90,7 @@ def sync(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -102,7 +102,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return sync_detailed(
@@ -117,7 +117,7 @@ async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -129,7 +129,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -147,7 +147,7 @@ async def asyncio(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -159,7 +159,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return (

beamlit/api/models/get_model_metrics.py

@@ -5,7 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.resource_metrics import ResourceMetrics
+from ...models.resource_environment_metrics import ResourceEnvironmentMetrics
 from ...types import UNSET, Response, Unset
 
 
@@ -31,9 +31,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     if response.status_code == 200:
-        response_200 = ResourceMetrics.from_dict(response.json())
+        response_200 = ResourceEnvironmentMetrics.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -44,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,7 +58,7 @@ def sync_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -72,7 +72,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -92,7 +92,7 @@ def sync(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -106,7 +106,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return sync_detailed(
@@ -121,7 +121,7 @@ async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -135,7 +135,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -153,7 +153,7 @@ async def asyncio(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -167,7 +167,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return (

beamlit/functions/common.py

@@ -4,7 +4,6 @@ import asyncio
 import importlib.util
 import os
 import traceback
-
 from logging import getLogger
 from typing import Union
 

beamlit/functions/mcp/mcp.py

@@ -6,13 +6,14 @@ import pydantic
 import pydantic_core
 import requests
 import typing_extensions as t
-from beamlit.authentication.authentication import AuthenticatedClient
-from beamlit.common.settings import get_settings
 from langchain_core.tools.base import BaseTool, BaseToolkit, ToolException
 from mcp.types import CallToolResult, ListToolsResult
 from pydantic.json_schema import JsonSchemaValue
 from pydantic_core import core_schema as cs
 
+from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.common.settings import get_settings
+
 settings = get_settings()
 
 TYPE_MAP = {

beamlit/models/__init__.py

@@ -5,7 +5,6 @@ from .agent import Agent
 from .agent_chain import AgentChain
 from .agent_history import AgentHistory
 from .agent_history_event import AgentHistoryEvent
-from .agent_metadata import AgentMetadata
 from .agent_release import AgentRelease
 from .agent_spec import AgentSpec
 from .api_key import ApiKey
@@ -37,12 +36,10 @@ from .get_workspace_service_accounts_response_200_item import (
 )
 from .histogram_bucket import HistogramBucket
 from .histogram_stats import HistogramStats
-from .increase_and_rate_metric import IncreaseAndRateMetric
-from .integration_config import IntegrationConfig
 from .integration_connection import IntegrationConnection
-from .integration_connection_config import IntegrationConnectionConfig
-from .integration_connection_secret import IntegrationConnectionSecret
 from .integration_connection_spec import IntegrationConnectionSpec
+from .integration_connection_spec_config import IntegrationConnectionSpecConfig
+from .integration_connection_spec_secret import IntegrationConnectionSpecSecret
 from .integration_model import IntegrationModel
 from .integration_repository import IntegrationRepository
 from .invite_workspace_user_body import InviteWorkspaceUserBody
@@ -57,7 +54,6 @@ from .metrics_models import MetricsModels
 from .metrics_request_total_per_code import MetricsRequestTotalPerCode
 from .metrics_rps_per_code import MetricsRpsPerCode
 from .model import Model
-from .model_metadata import ModelMetadata
 from .model_private_cluster import ModelPrivateCluster
 from .model_release import ModelRelease
 from .model_spec import ModelSpec
@@ -68,14 +64,14 @@ from .pending_invitation_render import PendingInvitationRender
 from .pending_invitation_render_invited_by import PendingInvitationRenderInvitedBy
 from .pending_invitation_render_workspace import PendingInvitationRenderWorkspace
 from .pending_invitation_workspace_details import PendingInvitationWorkspaceDetails
+from .pod_template import PodTemplate
 from .pod_template_spec import PodTemplateSpec
 from .policy import Policy
 from .policy_location import PolicyLocation
+from .policy_max_tokens import PolicyMaxTokens
 from .policy_spec import PolicySpec
 from .private_cluster import PrivateCluster
 from .private_location import PrivateLocation
-from .provider_config import ProviderConfig
-from .qps import QPS
 from .repository import Repository
 from .request_duration_over_time_metric import RequestDurationOverTimeMetric
 from .request_duration_over_time_metrics import RequestDurationOverTimeMetrics
@@ -89,20 +85,12 @@ from .request_total_by_origin_metric_request_total_by_origin_and_code import (
 from .request_total_metric import RequestTotalMetric
 from .request_total_metric_request_total_per_code import RequestTotalMetricRequestTotalPerCode
 from .request_total_metric_rps_per_code import RequestTotalMetricRpsPerCode
-from .resource_deployment_metrics import ResourceDeploymentMetrics
-from .resource_deployment_metrics_inference_per_second_per_region import (
-    ResourceDeploymentMetricsInferencePerSecondPerRegion,
-)
-from .resource_deployment_metrics_query_per_second_per_region_per_code import (
-    ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode,
-)
 from .resource_environment_metrics import ResourceEnvironmentMetrics
 from .resource_environment_metrics_request_total_per_code import (
     ResourceEnvironmentMetricsRequestTotalPerCode,
 )
 from .resource_environment_metrics_rps_per_code import ResourceEnvironmentMetricsRpsPerCode
 from .resource_log import ResourceLog
-from .resource_metrics import ResourceMetrics
 from .runtime import Runtime
 from .runtime_readiness_probe import RuntimeReadinessProbe
 from .runtime_resources import RuntimeResources
@@ -135,7 +123,6 @@ __all__ = (
     "AgentChain",
     "AgentHistory",
     "AgentHistoryEvent",
-    "AgentMetadata",
     "AgentRelease",
     "AgentSpec",
     "ApiKey",
@@ -165,12 +152,10 @@ __all__ = (
     "GetWorkspaceServiceAccountsResponse200Item",
     "HistogramBucket",
     "HistogramStats",
-    "IncreaseAndRateMetric",
-    "IntegrationConfig",
     "IntegrationConnection",
-    "IntegrationConnectionConfig",
-    "IntegrationConnectionSecret",
     "IntegrationConnectionSpec",
+    "IntegrationConnectionSpecConfig",
+    "IntegrationConnectionSpecSecret",
     "IntegrationModel",
     "IntegrationRepository",
     "InviteWorkspaceUserBody",
@@ -185,7 +170,6 @@ __all__ = (
     "MetricsRequestTotalPerCode",
     "MetricsRpsPerCode",
     "Model",
-    "ModelMetadata",
     "ModelPrivateCluster",
     "ModelRelease",
     "ModelSpec",
@@ -196,14 +180,14 @@ __all__ = (
     "PendingInvitationRenderInvitedBy",
     "PendingInvitationRenderWorkspace",
     "PendingInvitationWorkspaceDetails",
+    "PodTemplate",
     "PodTemplateSpec",
     "Policy",
     "PolicyLocation",
+    "PolicyMaxTokens",
     "PolicySpec",
     "PrivateCluster",
     "PrivateLocation",
-    "ProviderConfig",
-    "QPS",
     "Repository",
     "RequestDurationOverTimeMetric",
     "RequestDurationOverTimeMetrics",
@@ -213,14 +197,10 @@ __all__ = (
     "RequestTotalMetric",
     "RequestTotalMetricRequestTotalPerCode",
     "RequestTotalMetricRpsPerCode",
-    "ResourceDeploymentMetrics",
-    "ResourceDeploymentMetricsInferencePerSecondPerRegion",
-    "ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode",
     "ResourceEnvironmentMetrics",
     "ResourceEnvironmentMetricsRequestTotalPerCode",
     "ResourceEnvironmentMetricsRpsPerCode",
     "ResourceLog",
-    "ResourceMetrics",
     "Runtime",
     "RuntimeReadinessProbe",
     "RuntimeResources",

beamlit/models/agent.py

@@ -40,12 +40,16 @@ class Agent:
                 events.append(componentsschemas_core_events_item)
 
         metadata: Union[Unset, dict[str, Any]] = UNSET
-        if self.metadata and not isinstance(self.metadata, Unset):
+        if self.metadata and not isinstance(self.metadata, Unset) and not isinstance(self.metadata, dict):
             metadata = self.metadata.to_dict()
+        elif self.metadata and isinstance(self.metadata, dict):
+            metadata = self.metadata
 
         spec: Union[Unset, dict[str, Any]] = UNSET
-        if self.spec and not isinstance(self.spec, Unset):
+        if self.spec and not isinstance(self.spec, Unset) and not isinstance(self.spec, dict):
             spec = self.spec.to_dict()
+        elif self.spec and isinstance(self.spec, dict):
+            spec = self.spec
 
         status = self.status
 

beamlit/models/agent_spec.py

@@ -62,8 +62,14 @@ class AgentSpec:
 
     def to_dict(self) -> dict[str, Any]:
         configurations: Union[Unset, dict[str, Any]] = UNSET
-        if self.configurations and not isinstance(self.configurations, Unset):
+        if (
+            self.configurations
+            and not isinstance(self.configurations, Unset)
+            and not isinstance(self.configurations, dict)
+        ):
             configurations = self.configurations.to_dict()
+        elif self.configurations and isinstance(self.configurations, dict):
+            configurations = self.configurations
 
         enabled = self.enabled
 
@@ -79,26 +85,42 @@ class AgentSpec:
             integration_connections = self.integration_connections
 
         pod_template: Union[Unset, dict[str, Any]] = UNSET
-        if self.pod_template and not isinstance(self.pod_template, Unset):
+        if self.pod_template and not isinstance(self.pod_template, Unset) and not isinstance(self.pod_template, dict):
             pod_template = self.pod_template.to_dict()
+        elif self.pod_template and isinstance(self.pod_template, dict):
+            pod_template = self.pod_template
 
         policies: Union[Unset, list[str]] = UNSET
         if not isinstance(self.policies, Unset):
             policies = self.policies
 
         private_clusters: Union[Unset, dict[str, Any]] = UNSET
-        if self.private_clusters and not isinstance(self.private_clusters, Unset):
+        if (
+            self.private_clusters
+            and not isinstance(self.private_clusters, Unset)
+            and not isinstance(self.private_clusters, dict)
+        ):
             private_clusters = self.private_clusters.to_dict()
+        elif self.private_clusters and isinstance(self.private_clusters, dict):
+            private_clusters = self.private_clusters
 
         runtime: Union[Unset, dict[str, Any]] = UNSET
-        if self.runtime and not isinstance(self.runtime, Unset):
+        if self.runtime and not isinstance(self.runtime, Unset) and not isinstance(self.runtime, dict):
             runtime = self.runtime.to_dict()
+        elif self.runtime and isinstance(self.runtime, dict):
+            runtime = self.runtime
 
         sandbox = self.sandbox
 
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
-        if self.serverless_config and not isinstance(self.serverless_config, Unset):
+        if (
+            self.serverless_config
+            and not isinstance(self.serverless_config, Unset)
+            and not isinstance(self.serverless_config, dict)
+        ):
             serverless_config = self.serverless_config.to_dict()
+        elif self.serverless_config and isinstance(self.serverless_config, dict):
+            serverless_config = self.serverless_config
 
         agent_chain: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.agent_chain, Unset):
@@ -116,8 +138,10 @@ class AgentSpec:
         model = self.model
 
         repository: Union[Unset, dict[str, Any]] = UNSET
-        if self.repository and not isinstance(self.repository, Unset):
+        if self.repository and not isinstance(self.repository, Unset) and not isinstance(self.repository, dict):
             repository = self.repository.to_dict()
+        elif self.repository and isinstance(self.repository, dict):
+            repository = self.repository
 
         store_id = self.store_id
 

beamlit/models/core_spec.py

@@ -48,8 +48,14 @@ class CoreSpec:
 
     def to_dict(self) -> dict[str, Any]:
         configurations: Union[Unset, dict[str, Any]] = UNSET
-        if self.configurations and not isinstance(self.configurations, Unset):
+        if (
+            self.configurations
+            and not isinstance(self.configurations, Unset)
+            and not isinstance(self.configurations, dict)
+        ):
             configurations = self.configurations.to_dict()
+        elif self.configurations and isinstance(self.configurations, dict):
+            configurations = self.configurations
 
         enabled = self.enabled
 
@@ -65,26 +71,42 @@ class CoreSpec:
             integration_connections = self.integration_connections
 
         pod_template: Union[Unset, dict[str, Any]] = UNSET
-        if self.pod_template and not isinstance(self.pod_template, Unset):
+        if self.pod_template and not isinstance(self.pod_template, Unset) and not isinstance(self.pod_template, dict):
             pod_template = self.pod_template.to_dict()
+        elif self.pod_template and isinstance(self.pod_template, dict):
+            pod_template = self.pod_template
 
         policies: Union[Unset, list[str]] = UNSET
         if not isinstance(self.policies, Unset):
             policies = self.policies
 
         private_clusters: Union[Unset, dict[str, Any]] = UNSET
-        if self.private_clusters and not isinstance(self.private_clusters, Unset):
+        if (
+            self.private_clusters
+            and not isinstance(self.private_clusters, Unset)
+            and not isinstance(self.private_clusters, dict)
+        ):
             private_clusters = self.private_clusters.to_dict()
+        elif self.private_clusters and isinstance(self.private_clusters, dict):
+            private_clusters = self.private_clusters
 
         runtime: Union[Unset, dict[str, Any]] = UNSET
-        if self.runtime and not isinstance(self.runtime, Unset):
+        if self.runtime and not isinstance(self.runtime, Unset) and not isinstance(self.runtime, dict):
             runtime = self.runtime.to_dict()
+        elif self.runtime and isinstance(self.runtime, dict):
+            runtime = self.runtime
 
         sandbox = self.sandbox
 
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
-        if self.serverless_config and not isinstance(self.serverless_config, Unset):
+        if (
+            self.serverless_config
+            and not isinstance(self.serverless_config, Unset)
+            and not isinstance(self.serverless_config, dict)
+        ):
             serverless_config = self.serverless_config.to_dict()
+        elif self.serverless_config and isinstance(self.serverless_config, dict):
+            serverless_config = self.serverless_config
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/core_spec_configurations.py

@@ -26,8 +26,10 @@ class CoreSpecConfigurations:
 
     def to_dict(self) -> dict[str, Any]:
         key: Union[Unset, dict[str, Any]] = UNSET
-        if self.key and not isinstance(self.key, Unset):
+        if self.key and not isinstance(self.key, Unset) and not isinstance(self.key, dict):
             key = self.key.to_dict()
+        elif self.key and isinstance(self.key, dict):
+            key = self.key
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/environment.py

@@ -28,12 +28,16 @@ class Environment:
 
     def to_dict(self) -> dict[str, Any]:
         metadata: Union[Unset, dict[str, Any]] = UNSET
-        if self.metadata and not isinstance(self.metadata, Unset):
+        if self.metadata and not isinstance(self.metadata, Unset) and not isinstance(self.metadata, dict):
             metadata = self.metadata.to_dict()
+        elif self.metadata and isinstance(self.metadata, dict):
+            metadata = self.metadata
 
         spec: Union[Unset, dict[str, Any]] = UNSET
-        if self.spec and not isinstance(self.spec, Unset):
+        if self.spec and not isinstance(self.spec, Unset) and not isinstance(self.spec, dict):
             spec = self.spec.to_dict()
+        elif self.spec and isinstance(self.spec, dict):
+            spec = self.spec
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)

beamlit/models/environment_metadata.py

@@ -51,8 +51,10 @@ class EnvironmentMetadata:
         display_name = self.display_name
 
         labels: Union[Unset, dict[str, Any]] = UNSET
-        if self.labels and not isinstance(self.labels, Unset):
+        if self.labels and not isinstance(self.labels, Unset) and not isinstance(self.labels, dict):
             labels = self.labels.to_dict()
+        elif self.labels and isinstance(self.labels, dict):
+            labels = self.labels
 
         name = self.name
 

beamlit/models/function.py

@@ -40,12 +40,16 @@ class Function:
                 events.append(componentsschemas_core_events_item)
 
         metadata: Union[Unset, dict[str, Any]] = UNSET
-        if self.metadata and not isinstance(self.metadata, Unset):
+        if self.metadata and not isinstance(self.metadata, Unset) and not isinstance(self.metadata, dict):
             metadata = self.metadata.to_dict()
+        elif self.metadata and isinstance(self.metadata, dict):
+            metadata = self.metadata
 
         spec: Union[Unset, dict[str, Any]] = UNSET
-        if self.spec and not isinstance(self.spec, Unset):
+        if self.spec and not isinstance(self.spec, Unset) and not isinstance(self.spec, dict):
             spec = self.spec.to_dict()
+        elif self.spec and isinstance(self.spec, dict):
+            spec = self.spec
 
         status = self.status