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

beamlit/api/generation/run_information_generation_agent.py

@@ -1,168 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.agent_information_request import AgentInformationRequest
-from ...models.agent_information_response import AgentInformationResponse
-from ...types import Response
-
-
-def _get_kwargs(
-    *,
-    body: AgentInformationRequest,
-) -> dict[str, Any]:
-    headers: dict[str, Any] = {}
-
-    _kwargs: dict[str, Any] = {
-        "method": "post",
-        "url": "/generation/agents/information",
-    }
-
-    _body = body.to_dict()
-
-    _kwargs["json"] = _body
-    headers["Content-Type"] = "application/json"
-
-    _kwargs["headers"] = headers
-    return _kwargs
-
-
-def _parse_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[AgentInformationResponse]:
-    if response.status_code == 200:
-        response_200 = AgentInformationResponse.from_dict(response.json())
-
-        return response_200
-    if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(response.status_code, response.content)
-    else:
-        return None
-
-
-def _build_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[AgentInformationResponse]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    *,
-    client: AuthenticatedClient,
-    body: AgentInformationRequest,
-) -> Response[AgentInformationResponse]:
-    """Run information generation agent
-
-     Run information generation agent
-
-    Args:
-        body (AgentInformationRequest): generation agent information request
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[AgentInformationResponse]
-    """
-
-    kwargs = _get_kwargs(
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-def sync(
-    *,
-    client: AuthenticatedClient,
-    body: AgentInformationRequest,
-) -> Optional[AgentInformationResponse]:
-    """Run information generation agent
-
-     Run information generation agent
-
-    Args:
-        body (AgentInformationRequest): generation agent information request
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        AgentInformationResponse
-    """
-
-    return sync_detailed(
-        client=client,
-        body=body,
-    ).parsed
-
-
-async def asyncio_detailed(
-    *,
-    client: AuthenticatedClient,
-    body: AgentInformationRequest,
-) -> Response[AgentInformationResponse]:
-    """Run information generation agent
-
-     Run information generation agent
-
-    Args:
-        body (AgentInformationRequest): generation agent information request
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[AgentInformationResponse]
-    """
-
-    kwargs = _get_kwargs(
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio(
-    *,
-    client: AuthenticatedClient,
-    body: AgentInformationRequest,
-) -> Optional[AgentInformationResponse]:
-    """Run information generation agent
-
-     Run information generation agent
-
-    Args:
-        body (AgentInformationRequest): generation agent information request
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        AgentInformationResponse
-    """
-
-    return (
-        await asyncio_detailed(
-            client=client,
-            body=body,
-        )
-    ).parsed

beamlit/api/history/get_agents_history.py

@@ -1,155 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.agent_history import AgentHistory
-from ...types import Response
-
-
-def _get_kwargs(
-    request_id: str,
-) -> dict[str, Any]:
-    _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": f"/history/agents/{request_id}",
-    }
-
-    return _kwargs
-
-
-def _parse_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[list["AgentHistory"]]:
-    if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = AgentHistory.from_dict(response_200_item_data)
-
-            response_200.append(response_200_item)
-
-        return response_200
-    if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(response.status_code, response.content)
-    else:
-        return None
-
-
-def _build_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[list["AgentHistory"]]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    request_id: str,
-    *,
-    client: AuthenticatedClient,
-) -> Response[list["AgentHistory"]]:
-    """Get all history for a specific request ID from all agents
-
-    Args:
-        request_id (str):
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[list['AgentHistory']]
-    """
-
-    kwargs = _get_kwargs(
-        request_id=request_id,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-def sync(
-    request_id: str,
-    *,
-    client: AuthenticatedClient,
-) -> Optional[list["AgentHistory"]]:
-    """Get all history for a specific request ID from all agents
-
-    Args:
-        request_id (str):
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        list['AgentHistory']
-    """
-
-    return sync_detailed(
-        request_id=request_id,
-        client=client,
-    ).parsed
-
-
-async def asyncio_detailed(
-    request_id: str,
-    *,
-    client: AuthenticatedClient,
-) -> Response[list["AgentHistory"]]:
-    """Get all history for a specific request ID from all agents
-
-    Args:
-        request_id (str):
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[list['AgentHistory']]
-    """
-
-    kwargs = _get_kwargs(
-        request_id=request_id,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio(
-    request_id: str,
-    *,
-    client: AuthenticatedClient,
-) -> Optional[list["AgentHistory"]]:
-    """Get all history for a specific request ID from all agents
-
-    Args:
-        request_id (str):
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        list['AgentHistory']
-    """
-
-    return (
-        await asyncio_detailed(
-            request_id=request_id,
-            client=client,
-        )
-    ).parsed

beamlit/api/history/list_agents_history.py

@@ -1,131 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.agent_history import AgentHistory
-from ...types import Response
-
-
-def _get_kwargs() -> dict[str, Any]:
-    _kwargs: dict[str, Any] = {
-        "method": "get",
-        "url": "/history/agents",
-    }
-
-    return _kwargs
-
-
-def _parse_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[list["AgentHistory"]]:
-    if response.status_code == 200:
-        response_200 = []
-        _response_200 = response.json()
-        for response_200_item_data in _response_200:
-            response_200_item = AgentHistory.from_dict(response_200_item_data)
-
-            response_200.append(response_200_item)
-
-        return response_200
-    if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(response.status_code, response.content)
-    else:
-        return None
-
-
-def _build_response(
-    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[list["AgentHistory"]]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    *,
-    client: AuthenticatedClient,
-) -> Response[list["AgentHistory"]]:
-    """Get all history for all agents
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[list['AgentHistory']]
-    """
-
-    kwargs = _get_kwargs()
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-def sync(
-    *,
-    client: AuthenticatedClient,
-) -> Optional[list["AgentHistory"]]:
-    """Get all history for all agents
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        list['AgentHistory']
-    """
-
-    return sync_detailed(
-        client=client,
-    ).parsed
-
-
-async def asyncio_detailed(
-    *,
-    client: AuthenticatedClient,
-) -> Response[list["AgentHistory"]]:
-    """Get all history for all agents
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        Response[list['AgentHistory']]
-    """
-
-    kwargs = _get_kwargs()
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio(
-    *,
-    client: AuthenticatedClient,
-) -> Optional[list["AgentHistory"]]:
-    """Get all history for all agents
-
-    Raises:
-        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-        httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-    Returns:
-        list['AgentHistory']
-    """
-
-    return (
-        await asyncio_detailed(
-            client=client,
-        )
-    ).parsed

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