beamlit 0.0.57rc108__py3-none-any.whl → 0.0.57rc110__py3-none-any.whl

beamlit/models/agent_history.py

@@ -1,167 +0,0 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-if TYPE_CHECKING:
-    from ..models.agent_history_event import AgentHistoryEvent
-
-
-T = TypeVar("T", bound="AgentHistory")
-
-
-@_attrs_define
-class AgentHistory:
-    """Agent deployment history
-
-    Attributes:
-        created_at (Union[Unset, str]): The date and time when the resource was created
-        updated_at (Union[Unset, str]): The date and time when the resource was updated
-        agent (Union[Unset, str]): Agent name
-        end (Union[Unset, str]): End time
-        environment (Union[Unset, str]): Environment name
-        events (Union[Unset, list['AgentHistoryEvent']]): Events
-        request_id (Union[Unset, str]): Request ID
-        start (Union[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
-        workspace (Union[Unset, str]): The workspace the agent deployment belongs to
-    """
-
-    created_at: Union[Unset, str] = UNSET
-    updated_at: Union[Unset, str] = UNSET
-    agent: Union[Unset, str] = UNSET
-    end: Union[Unset, str] = UNSET
-    environment: Union[Unset, str] = UNSET
-    events: Union[Unset, list["AgentHistoryEvent"]] = UNSET
-    request_id: Union[Unset, str] = UNSET
-    start: Union[Unset, str] = UNSET
-    status: Union[Unset, str] = UNSET
-    took: Union[Unset, int] = UNSET
-    workspace: Union[Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        created_at = self.created_at
-
-        updated_at = self.updated_at
-
-        agent = self.agent
-
-        end = self.end
-
-        environment = self.environment
-
-        events: Union[Unset, list[dict[str, Any]]] = UNSET
-        if not isinstance(self.events, Unset):
-            events = []
-            for events_item_data in self.events:
-                events_item = events_item_data.to_dict()
-                events.append(events_item)
-
-        request_id = self.request_id
-
-        start = self.start
-
-        status = self.status
-
-        took = self.took
-
-        workspace = self.workspace
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if created_at is not UNSET:
-            field_dict["createdAt"] = created_at
-        if updated_at is not UNSET:
-            field_dict["updatedAt"] = updated_at
-        if agent is not UNSET:
-            field_dict["agent"] = agent
-        if end is not UNSET:
-            field_dict["end"] = end
-        if environment is not UNSET:
-            field_dict["environment"] = environment
-        if events is not UNSET:
-            field_dict["events"] = events
-        if request_id is not UNSET:
-            field_dict["request_id"] = request_id
-        if start is not UNSET:
-            field_dict["start"] = start
-        if status is not UNSET:
-            field_dict["status"] = status
-        if took is not UNSET:
-            field_dict["took"] = took
-        if workspace is not UNSET:
-            field_dict["workspace"] = workspace
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.agent_history_event import AgentHistoryEvent
-
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        created_at = d.pop("createdAt", UNSET)
-
-        updated_at = d.pop("updatedAt", UNSET)
-
-        agent = d.pop("agent", UNSET)
-
-        end = d.pop("end", UNSET)
-
-        environment = d.pop("environment", UNSET)
-
-        events = []
-        _events = d.pop("events", UNSET)
-        for events_item_data in _events or []:
-            events_item = AgentHistoryEvent.from_dict(events_item_data)
-
-            events.append(events_item)
-
-        request_id = d.pop("request_id", UNSET)
-
-        start = d.pop("start", UNSET)
-
-        status = d.pop("status", UNSET)
-
-        took = d.pop("took", UNSET)
-
-        workspace = d.pop("workspace", UNSET)
-
-        agent_history = cls(
-            created_at=created_at,
-            updated_at=updated_at,
-            agent=agent,
-            end=end,
-            environment=environment,
-            events=events,
-            request_id=request_id,
-            start=start,
-            status=status,
-            took=took,
-            workspace=workspace,
-        )
-
-        agent_history.additional_properties = d
-        return agent_history
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

beamlit/models/agent_history_event.py

@@ -1,133 +0,0 @@
-from typing import Any, TypeVar, Union
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="AgentHistoryEvent")
-
-
-@_attrs_define
-class AgentHistoryEvent:
-    """Agent deployment history event
-
-    Attributes:
-        end (Union[Unset, str]): End time
-        error (Union[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
-        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
-        type_ (Union[Unset, str]): Type, one of function or agent
-    """
-
-    end: Union[Unset, str] = UNSET
-    error: Union[Unset, str] = UNSET
-    name: Union[Unset, str] = UNSET
-    parameters: Union[Unset, str] = UNSET
-    start: Union[Unset, str] = UNSET
-    status: Union[Unset, str] = UNSET
-    sub_function: Union[Unset, str] = UNSET
-    took: Union[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
-
-        name = self.name
-
-        parameters = self.parameters
-
-        start = self.start
-
-        status = self.status
-
-        sub_function = self.sub_function
-
-        took = self.took
-
-        type_ = self.type_
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if end is not UNSET:
-            field_dict["end"] = end
-        if error is not UNSET:
-            field_dict["error"] = error
-        if name is not UNSET:
-            field_dict["name"] = name
-        if parameters is not UNSET:
-            field_dict["parameters"] = parameters
-        if start is not UNSET:
-            field_dict["start"] = start
-        if status is not UNSET:
-            field_dict["status"] = status
-        if sub_function is not UNSET:
-            field_dict["subFunction"] = sub_function
-        if took is not UNSET:
-            field_dict["took"] = took
-        if type_ is not UNSET:
-            field_dict["type"] = type_
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        end = d.pop("end", UNSET)
-
-        error = d.pop("error", UNSET)
-
-        name = d.pop("name", UNSET)
-
-        parameters = d.pop("parameters", UNSET)
-
-        start = d.pop("start", UNSET)
-
-        status = d.pop("status", UNSET)
-
-        sub_function = d.pop("subFunction", UNSET)
-
-        took = d.pop("took", UNSET)
-
-        type_ = d.pop("type", UNSET)
-
-        agent_history_event = cls(
-            end=end,
-            error=error,
-            name=name,
-            parameters=parameters,
-            start=start,
-            status=status,
-            sub_function=sub_function,
-            took=took,
-            type_=type_,
-        )
-
-        agent_history_event.additional_properties = d
-        return agent_history_event
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

beamlit/models/agent_information_request.py

@@ -1,63 +0,0 @@
-from typing import Any, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="AgentInformationRequest")
-
-
-@_attrs_define
-class AgentInformationRequest:
-    """generation agent information request
-
-    Attributes:
-        functions (Union[Unset, list[Any]]): Functions to generate information for
-    """
-
-    functions: Union[Unset, list[Any]] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        functions: Union[Unset, list[Any]] = UNSET
-        if not isinstance(self.functions, Unset):
-            functions = self.functions
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if functions is not UNSET:
-            field_dict["functions"] = functions
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        functions = cast(list[Any], d.pop("functions", UNSET))
-
-        agent_information_request = cls(
-            functions=functions,
-        )
-
-        agent_information_request.additional_properties = d
-        return agent_information_request
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

beamlit/models/agent_information_response.py

@@ -1,79 +0,0 @@
-from typing import Any, TypeVar, Union
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="AgentInformationResponse")
-
-
-@_attrs_define
-class AgentInformationResponse:
-    """generation agent information response
-
-    Attributes:
-        description (Union[Unset, str]): Description of the agent
-        display_name (Union[Unset, str]): Display name of the agent
-        prompt (Union[Unset, str]): Prompt of the agent
-    """
-
-    description: Union[Unset, str] = UNSET
-    display_name: Union[Unset, str] = UNSET
-    prompt: Union[Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        description = self.description
-
-        display_name = self.display_name
-
-        prompt = self.prompt
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if description is not UNSET:
-            field_dict["description"] = description
-        if display_name is not UNSET:
-            field_dict["displayName"] = display_name
-        if prompt is not UNSET:
-            field_dict["prompt"] = prompt
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        description = d.pop("description", UNSET)
-
-        display_name = d.pop("displayName", UNSET)
-
-        prompt = d.pop("prompt", UNSET)
-
-        agent_information_response = cls(
-            description=description,
-            display_name=display_name,
-            prompt=prompt,
-        )
-
-        agent_information_response.additional_properties = d
-        return agent_information_response
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

beamlit/models/agent_release.py

@@ -1,70 +0,0 @@
-from typing import Any, TypeVar, Union
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="AgentRelease")
-
-
-@_attrs_define
-class AgentRelease:
-    """Agent release, used to deploy a agent from one environment to another
-
-    Attributes:
-        from_ (Union[Unset, str]): Origin environment from which the agent is released
-        to (Union[Unset, str]): Destination environment to which the agent is released
-    """
-
-    from_: Union[Unset, str] = UNSET
-    to: Union[Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        from_ = self.from_
-
-        to = self.to
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if from_ is not UNSET:
-            field_dict["from"] = from_
-        if to is not UNSET:
-            field_dict["to"] = to
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        from_ = d.pop("from", UNSET)
-
-        to = d.pop("to", UNSET)
-
-        agent_release = cls(
-            from_=from_,
-            to=to,
-        )
-
-        agent_release.additional_properties = d
-        return agent_release
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

beamlit/models/environment.py

@@ -1,96 +0,0 @@
-from typing import TYPE_CHECKING, Any, TypeVar, Union
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-if TYPE_CHECKING:
-    from ..models.environment_spec import EnvironmentSpec
-    from ..models.metadata import Metadata
-
-
-T = TypeVar("T", bound="Environment")
-
-
-@_attrs_define
-class Environment:
-    """Environment on which deployments will be made (e.g. development, production), enforcing multiple policies at once.
-
-    Attributes:
-        metadata (Union[Unset, Metadata]): Metadata
-        spec (Union[Unset, EnvironmentSpec]): Environment specification
-    """
-
-    metadata: Union[Unset, "Metadata"] = UNSET
-    spec: Union[Unset, "EnvironmentSpec"] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        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
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if metadata is not UNSET:
-            field_dict["metadata"] = metadata
-        if spec is not UNSET:
-            field_dict["spec"] = spec
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.environment_spec import EnvironmentSpec
-        from ..models.metadata import Metadata
-
-        if not src_dict:
-            return None
-        d = src_dict.copy()
-        _metadata = d.pop("metadata", UNSET)
-        metadata: Union[Unset, Metadata]
-        if isinstance(_metadata, Unset):
-            metadata = UNSET
-        else:
-            metadata = Metadata.from_dict(_metadata)
-
-        _spec = d.pop("spec", UNSET)
-        spec: Union[Unset, EnvironmentSpec]
-        if isinstance(_spec, Unset):
-            spec = UNSET
-        else:
-            spec = EnvironmentSpec.from_dict(_spec)
-
-        environment = cls(
-            metadata=metadata,
-            spec=spec,
-        )
-
-        environment.additional_properties = d
-        return environment
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties