blaxel 0.2.34__py3-none-any.whl → 0.2.35__py3-none-any.whl

blaxel/core/client/models/__init__.py

@@ -1,17 +1,15 @@
 """Contains all the data models used in inputs/outputs"""
 
-from .acl import ACL
 from .agent import Agent
+from .agent_runtime import AgentRuntime
+from .agent_runtime_generation import AgentRuntimeGeneration
 from .agent_spec import AgentSpec
 from .api_key import ApiKey
-from .billable_time_metric import BillableTimeMetric
 from .check_workspace_availability_body import CheckWorkspaceAvailabilityBody
 from .cleanup_images_response_200 import CleanupImagesResponse200
 from .configuration import Configuration
 from .continent import Continent
 from .core_event import CoreEvent
-from .core_spec import CoreSpec
-from .core_spec_configurations import CoreSpecConfigurations
 from .country import Country
 from .create_api_key_for_service_account_body import CreateApiKeyForServiceAccountBody
 from .create_job_execution_request import CreateJobExecutionRequest
@@ -23,25 +21,33 @@ from .create_workspace_service_account_response_200 import CreateWorkspaceServic
 from .custom_domain import CustomDomain
 from .custom_domain_metadata import CustomDomainMetadata
 from .custom_domain_spec import CustomDomainSpec
+from .custom_domain_spec_status import CustomDomainSpecStatus
 from .custom_domain_spec_txt_records import CustomDomainSpecTxtRecords
 from .delete_sandbox_preview_token_response_200 import DeleteSandboxPreviewTokenResponse200
 from .delete_volume_template_version_response_200 import DeleteVolumeTemplateVersionResponse200
 from .delete_workspace_service_account_response_200 import DeleteWorkspaceServiceAccountResponse200
 from .entrypoint import Entrypoint
+from .entrypoint_args_item import EntrypointArgsItem
 from .entrypoint_env import EntrypointEnv
+from .entrypoint_super_gateway_args_item import EntrypointSuperGatewayArgsItem
+from .env import Env
+from .error import Error
 from .expiration_policy import ExpirationPolicy
+from .expiration_policy_action import ExpirationPolicyAction
+from .expiration_policy_type import ExpirationPolicyType
 from .flavor import Flavor
+from .flavor_type import FlavorType
 from .form import Form
 from .form_config import FormConfig
-from .form_oauth import FormOauth
 from .form_secrets import FormSecrets
 from .function import Function
+from .function_runtime import FunctionRuntime
+from .function_runtime_generation import FunctionRuntimeGeneration
 from .function_spec import FunctionSpec
+from .function_spec_transport import FunctionSpecTransport
 from .get_workspace_service_accounts_response_200_item import (
     GetWorkspaceServiceAccountsResponse200Item,
 )
-from .histogram_bucket import HistogramBucket
-from .histogram_stats import HistogramStats
 from .image import Image
 from .image_metadata import ImageMetadata
 from .image_spec import ImageSpec
@@ -53,52 +59,40 @@ from .integration_connection_spec import IntegrationConnectionSpec
 from .integration_connection_spec_config import IntegrationConnectionSpecConfig
 from .integration_connection_spec_secret import IntegrationConnectionSpecSecret
 from .integration_endpoint import IntegrationEndpoint
+from .integration_endpoint_ignore_models_item import IntegrationEndpointIgnoreModelsItem
+from .integration_endpoint_models_item import IntegrationEndpointModelsItem
 from .integration_endpoint_token import IntegrationEndpointToken
 from .integration_endpoints import IntegrationEndpoints
 from .integration_headers import IntegrationHeaders
-from .integration_model import IntegrationModel
 from .integration_organization import IntegrationOrganization
 from .integration_query_params import IntegrationQueryParams
 from .integration_repository import IntegrationRepository
 from .invite_workspace_user_body import InviteWorkspaceUserBody
 from .job import Job
 from .job_execution import JobExecution
-from .job_execution_config import JobExecutionConfig
 from .job_execution_metadata import JobExecutionMetadata
 from .job_execution_spec import JobExecutionSpec
 from .job_execution_stats import JobExecutionStats
+from .job_execution_status import JobExecutionStatus
 from .job_execution_task import JobExecutionTask
 from .job_execution_task_condition import JobExecutionTaskCondition
 from .job_execution_task_metadata import JobExecutionTaskMetadata
 from .job_execution_task_spec import JobExecutionTaskSpec
-from .job_metrics import JobMetrics
-from .job_metrics_executions_total import JobMetricsExecutionsTotal
-from .job_metrics_tasks_total import JobMetricsTasksTotal
+from .job_execution_task_status import JobExecutionTaskStatus
+from .job_runtime import JobRuntime
+from .job_runtime_generation import JobRuntimeGeneration
 from .job_spec import JobSpec
-from .jobs_chart_value import JobsChartValue
-from .jobs_network_chart import JobsNetworkChart
-from .jobs_success_failed_chart import JobsSuccessFailedChart
-from .jobs_total import JobsTotal
-from .last_n_requests_metric import LastNRequestsMetric
-from .latency_metric import LatencyMetric
 from .location_response import LocationResponse
-from .logs_response import LogsResponse
-from .logs_response_data import LogsResponseData
 from .mcp_definition import MCPDefinition
-from .mcp_definition_entrypoint import MCPDefinitionEntrypoint
-from .mcp_definition_form import MCPDefinitionForm
-from .memory_allocation_by_name import MemoryAllocationByName
-from .memory_allocation_metric import MemoryAllocationMetric
+from .mcp_definition_categories_item import MCPDefinitionCategoriesItem
 from .metadata import Metadata
 from .metadata_labels import MetadataLabels
-from .metric import Metric
-from .metrics import Metrics
-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_runtime import ModelRuntime
+from .model_runtime_type import ModelRuntimeType
 from .model_spec import ModelSpec
 from .o_auth import OAuth
+from .o_auth_scope_item import OAuthScopeItem
 from .owner_fields import OwnerFields
 from .pending_invitation import PendingInvitation
 from .pending_invitation_accept import PendingInvitationAccept
@@ -106,12 +100,18 @@ 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_spec import PodTemplateSpec
+from .pending_invitation_workspace_details_emails_item import (
+    PendingInvitationWorkspaceDetailsEmailsItem,
+)
 from .policy import Policy
 from .policy_location import PolicyLocation
+from .policy_location_type import PolicyLocationType
 from .policy_max_tokens import PolicyMaxTokens
+from .policy_resource_type import PolicyResourceType
 from .policy_spec import PolicySpec
+from .policy_spec_type import PolicySpecType
 from .port import Port
+from .port_protocol import PortProtocol
 from .preview import Preview
 from .preview_metadata import PreviewMetadata
 from .preview_spec import PreviewSpec
@@ -125,61 +125,24 @@ from .public_ip import PublicIp
 from .public_ips import PublicIps
 from .region import Region
 from .repository import Repository
-from .request_duration_over_time_metric import RequestDurationOverTimeMetric
-from .request_duration_over_time_metrics import RequestDurationOverTimeMetrics
-from .request_total_by_origin_metric import RequestTotalByOriginMetric
-from .request_total_by_origin_metric_request_total_by_origin import (
-    RequestTotalByOriginMetricRequestTotalByOrigin,
-)
-from .request_total_by_origin_metric_request_total_by_origin_and_code import (
-    RequestTotalByOriginMetricRequestTotalByOriginAndCode,
-)
-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 .request_total_response_data import RequestTotalResponseData
-from .resource import Resource
-from .resource_log import ResourceLog
-from .resource_log_chart import ResourceLogChart
-from .resource_log_response import ResourceLogResponse
-from .resource_metrics import ResourceMetrics
-from .resource_metrics_request_total_per_code import ResourceMetricsRequestTotalPerCode
-from .resource_metrics_request_total_per_code_previous import (
-    ResourceMetricsRequestTotalPerCodePrevious,
-)
-from .resource_metrics_rps_per_code import ResourceMetricsRpsPerCode
-from .resource_metrics_rps_per_code_previous import ResourceMetricsRpsPerCodePrevious
-from .resource_trace import ResourceTrace
 from .revision_configuration import RevisionConfiguration
 from .revision_metadata import RevisionMetadata
-from .runtime import Runtime
-from .runtime_configuration import RuntimeConfiguration
-from .runtime_startup_probe import RuntimeStartupProbe
 from .sandbox import Sandbox
 from .sandbox_definition import SandboxDefinition
+from .sandbox_definition_categories_item import SandboxDefinitionCategoriesItem
+from .sandbox_error import SandboxError
+from .sandbox_error_details import SandboxErrorDetails
 from .sandbox_lifecycle import SandboxLifecycle
-from .sandbox_metrics import SandboxMetrics
+from .sandbox_runtime import SandboxRuntime
 from .sandbox_spec import SandboxSpec
-from .serverless_config import ServerlessConfig
-from .serverless_config_configuration import ServerlessConfigConfiguration
-from .spec_configuration import SpecConfiguration
-from .start_sandbox import StartSandbox
-from .stop_sandbox import StopSandbox
-from .store_agent import StoreAgent
-from .store_agent_labels import StoreAgentLabels
-from .store_configuration import StoreConfiguration
-from .store_configuration_option import StoreConfigurationOption
+from .status import Status
 from .template import Template
 from .template_variable import TemplateVariable
 from .time_fields import TimeFields
-from .time_to_first_token_over_time_metrics import TimeToFirstTokenOverTimeMetrics
-from .token_rate_metric import TokenRateMetric
-from .token_rate_metrics import TokenRateMetrics
-from .token_total_metric import TokenTotalMetric
-from .trace_ids_response import TraceIdsResponse
 from .trigger import Trigger
 from .trigger_configuration import TriggerConfiguration
 from .trigger_configuration_task import TriggerConfigurationTask
+from .trigger_type import TriggerType
 from .update_workspace_service_account_body import UpdateWorkspaceServiceAccountBody
 from .update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
 from .update_workspace_user_role_body import UpdateWorkspaceUserRoleBody
@@ -190,27 +153,25 @@ from .volume_state import VolumeState
 from .volume_template import VolumeTemplate
 from .volume_template_spec import VolumeTemplateSpec
 from .volume_template_state import VolumeTemplateState
+from .volume_template_state_status import VolumeTemplateStateStatus
 from .volume_template_version import VolumeTemplateVersion
-from .websocket_channel import WebsocketChannel
-from .websocket_message import WebsocketMessage
+from .volume_template_version_status import VolumeTemplateVersionStatus
 from .workspace import Workspace
-from .workspace_labels import WorkspaceLabels
 from .workspace_runtime import WorkspaceRuntime
+from .workspace_status import WorkspaceStatus
 from .workspace_user import WorkspaceUser
 
 __all__ = (
-    "ACL",
     "Agent",
+    "AgentRuntime",
+    "AgentRuntimeGeneration",
     "AgentSpec",
     "ApiKey",
-    "BillableTimeMetric",
     "CheckWorkspaceAvailabilityBody",
     "CleanupImagesResponse200",
     "Configuration",
     "Continent",
     "CoreEvent",
-    "CoreSpec",
-    "CoreSpecConfigurations",
     "Country",
     "CreateApiKeyForServiceAccountBody",
     "CreateJobExecutionRequest",
@@ -222,23 +183,31 @@ __all__ = (
     "CustomDomain",
     "CustomDomainMetadata",
     "CustomDomainSpec",
+    "CustomDomainSpecStatus",
     "CustomDomainSpecTxtRecords",
     "DeleteSandboxPreviewTokenResponse200",
     "DeleteVolumeTemplateVersionResponse200",
     "DeleteWorkspaceServiceAccountResponse200",
     "Entrypoint",
+    "EntrypointArgsItem",
     "EntrypointEnv",
+    "EntrypointSuperGatewayArgsItem",
+    "Env",
+    "Error",
     "ExpirationPolicy",
+    "ExpirationPolicyAction",
+    "ExpirationPolicyType",
     "Flavor",
+    "FlavorType",
     "Form",
     "FormConfig",
-    "FormOauth",
     "FormSecrets",
     "Function",
+    "FunctionRuntime",
+    "FunctionRuntimeGeneration",
     "FunctionSpec",
+    "FunctionSpecTransport",
     "GetWorkspaceServiceAccountsResponse200Item",
-    "HistogramBucket",
-    "HistogramStats",
     "Image",
     "ImageMetadata",
     "ImageSpec",
@@ -250,52 +219,40 @@ __all__ = (
     "IntegrationConnectionSpecConfig",
     "IntegrationConnectionSpecSecret",
     "IntegrationEndpoint",
+    "IntegrationEndpointIgnoreModelsItem",
+    "IntegrationEndpointModelsItem",
     "IntegrationEndpoints",
     "IntegrationEndpointToken",
     "IntegrationHeaders",
-    "IntegrationModel",
     "IntegrationOrganization",
     "IntegrationQueryParams",
     "IntegrationRepository",
     "InviteWorkspaceUserBody",
     "Job",
     "JobExecution",
-    "JobExecutionConfig",
     "JobExecutionMetadata",
     "JobExecutionSpec",
     "JobExecutionStats",
+    "JobExecutionStatus",
     "JobExecutionTask",
     "JobExecutionTaskCondition",
     "JobExecutionTaskMetadata",
     "JobExecutionTaskSpec",
-    "JobMetrics",
-    "JobMetricsExecutionsTotal",
-    "JobMetricsTasksTotal",
-    "JobsChartValue",
-    "JobsNetworkChart",
+    "JobExecutionTaskStatus",
+    "JobRuntime",
+    "JobRuntimeGeneration",
     "JobSpec",
-    "JobsSuccessFailedChart",
-    "JobsTotal",
-    "LastNRequestsMetric",
-    "LatencyMetric",
     "LocationResponse",
-    "LogsResponse",
-    "LogsResponseData",
     "MCPDefinition",
-    "MCPDefinitionEntrypoint",
-    "MCPDefinitionForm",
-    "MemoryAllocationByName",
-    "MemoryAllocationMetric",
+    "MCPDefinitionCategoriesItem",
     "Metadata",
     "MetadataLabels",
-    "Metric",
-    "Metrics",
-    "MetricsModels",
-    "MetricsRequestTotalPerCode",
-    "MetricsRpsPerCode",
     "Model",
+    "ModelRuntime",
+    "ModelRuntimeType",
     "ModelSpec",
     "OAuth",
+    "OAuthScopeItem",
     "OwnerFields",
     "PendingInvitation",
     "PendingInvitationAccept",
@@ -303,12 +260,16 @@ __all__ = (
     "PendingInvitationRenderInvitedBy",
     "PendingInvitationRenderWorkspace",
     "PendingInvitationWorkspaceDetails",
-    "PodTemplateSpec",
+    "PendingInvitationWorkspaceDetailsEmailsItem",
     "Policy",
     "PolicyLocation",
+    "PolicyLocationType",
     "PolicyMaxTokens",
+    "PolicyResourceType",
     "PolicySpec",
+    "PolicySpecType",
     "Port",
+    "PortProtocol",
     "Preview",
     "PreviewMetadata",
     "PreviewSpec",
@@ -322,55 +283,24 @@ __all__ = (
     "PublicIps",
     "Region",
     "Repository",
-    "RequestDurationOverTimeMetric",
-    "RequestDurationOverTimeMetrics",
-    "RequestTotalByOriginMetric",
-    "RequestTotalByOriginMetricRequestTotalByOrigin",
-    "RequestTotalByOriginMetricRequestTotalByOriginAndCode",
-    "RequestTotalMetric",
-    "RequestTotalMetricRequestTotalPerCode",
-    "RequestTotalMetricRpsPerCode",
-    "RequestTotalResponseData",
-    "Resource",
-    "ResourceLog",
-    "ResourceLogChart",
-    "ResourceLogResponse",
-    "ResourceMetrics",
-    "ResourceMetricsRequestTotalPerCode",
-    "ResourceMetricsRequestTotalPerCodePrevious",
-    "ResourceMetricsRpsPerCode",
-    "ResourceMetricsRpsPerCodePrevious",
-    "ResourceTrace",
     "RevisionConfiguration",
     "RevisionMetadata",
-    "Runtime",
-    "RuntimeConfiguration",
-    "RuntimeStartupProbe",
     "Sandbox",
     "SandboxDefinition",
+    "SandboxDefinitionCategoriesItem",
+    "SandboxError",
+    "SandboxErrorDetails",
     "SandboxLifecycle",
-    "SandboxMetrics",
+    "SandboxRuntime",
     "SandboxSpec",
-    "ServerlessConfig",
-    "ServerlessConfigConfiguration",
-    "SpecConfiguration",
-    "StartSandbox",
-    "StopSandbox",
-    "StoreAgent",
-    "StoreAgentLabels",
-    "StoreConfiguration",
-    "StoreConfigurationOption",
+    "Status",
     "Template",
     "TemplateVariable",
     "TimeFields",
-    "TimeToFirstTokenOverTimeMetrics",
-    "TokenRateMetric",
-    "TokenRateMetrics",
-    "TokenTotalMetric",
-    "TraceIdsResponse",
     "Trigger",
     "TriggerConfiguration",
     "TriggerConfigurationTask",
+    "TriggerType",
     "UpdateWorkspaceServiceAccountBody",
     "UpdateWorkspaceServiceAccountResponse200",
     "UpdateWorkspaceUserRoleBody",
@@ -381,11 +311,11 @@ __all__ = (
     "VolumeTemplate",
     "VolumeTemplateSpec",
     "VolumeTemplateState",
+    "VolumeTemplateStateStatus",
     "VolumeTemplateVersion",
-    "WebsocketChannel",
-    "WebsocketMessage",
+    "VolumeTemplateVersionStatus",
     "Workspace",
-    "WorkspaceLabels",
     "WorkspaceRuntime",
+    "WorkspaceStatus",
     "WorkspaceUser",
 )

blaxel/core/client/models/agent.py

@@ -3,6 +3,7 @@ from typing import TYPE_CHECKING, Any, TypeVar, Union
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.status import Status
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
@@ -16,22 +17,35 @@ T = TypeVar("T", bound="Agent")
 
 @_attrs_define
 class Agent:
-    """Agent
-
-    Attributes:
-        events (Union[Unset, list['CoreEvent']]): Core events
-        metadata (Union[Unset, Metadata]): Metadata
-        spec (Union[Unset, AgentSpec]): Agent specification
-        status (Union[Unset, str]): Agent status
+    """Serverless AI agent deployment that runs your custom agent code as an auto-scaling API endpoint. Agents are deployed
+    from your code repository and expose a global inference URL for querying.
+
+        Attributes:
+            metadata (Metadata): Common metadata fields shared by all Blaxel resources including name, labels, timestamps,
+                and ownership information
+            spec (AgentSpec): Configuration for an AI agent including runtime settings, repository source, and deployment
+                triggers
+            events (Union[Unset, list['CoreEvent']]): Events happening on a resource deployed on Blaxel
+            status (Union[Unset, Status]): Deployment status of a resource deployed on Blaxel
     """
 
+    metadata: "Metadata"
+    spec: "AgentSpec"
     events: Union[Unset, list["CoreEvent"]] = UNSET
-    metadata: Union[Unset, "Metadata"] = UNSET
-    spec: Union[Unset, "AgentSpec"] = UNSET
-    status: Union[Unset, str] = UNSET
+    status: Union[Unset, Status] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
+        if type(self.metadata) is dict:
+            metadata = self.metadata
+        else:
+            metadata = self.metadata.to_dict()
+
+        if type(self.spec) is dict:
+            spec = self.spec
+        else:
+            spec = self.spec.to_dict()
+
         events: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.events, Unset):
             events = []
@@ -44,33 +58,20 @@ class Agent:
                     )
                 events.append(componentsschemas_core_events_item)
 
-        metadata: Union[Unset, dict[str, Any]] = 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) 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
+        status: Union[Unset, str] = UNSET
+        if not isinstance(self.status, Unset):
+            status = self.status.value
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update({})
+        field_dict.update(
+            {
+                "metadata": metadata,
+                "spec": spec,
+            }
+        )
         if events is not UNSET:
             field_dict["events"] = events
-        if metadata is not UNSET:
-            field_dict["metadata"] = metadata
-        if spec is not UNSET:
-            field_dict["spec"] = spec
         if status is not UNSET:
             field_dict["status"] = status
 
@@ -85,6 +86,10 @@ class Agent:
         if not src_dict:
             return None
         d = src_dict.copy()
+        metadata = Metadata.from_dict(d.pop("metadata"))
+
+        spec = AgentSpec.from_dict(d.pop("spec"))
+
         events = []
         _events = d.pop("events", UNSET)
         for componentsschemas_core_events_item_data in _events or []:
@@ -94,26 +99,17 @@ class Agent:
 
             events.append(componentsschemas_core_events_item)
 
-        _metadata = d.pop("metadata", UNSET)
-        metadata: Union[Unset, Metadata]
-        if isinstance(_metadata, Unset):
-            metadata = UNSET
+        _status = d.pop("status", UNSET)
+        status: Union[Unset, Status]
+        if isinstance(_status, Unset):
+            status = UNSET
         else:
-            metadata = Metadata.from_dict(_metadata)
-
-        _spec = d.pop("spec", UNSET)
-        spec: Union[Unset, AgentSpec]
-        if isinstance(_spec, Unset):
-            spec = UNSET
-        else:
-            spec = AgentSpec.from_dict(_spec)
-
-        status = d.pop("status", UNSET)
+            status = Status(_status)
 
         agent = cls(
-            events=events,
             metadata=metadata,
             spec=spec,
+            events=events,
             status=status,
         )