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

beamlit/api/environments/update_environment.py

@@ -1,180 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.environment import Environment
-from ...types import Response
-
-
-def _get_kwargs(
-    environment_name: str,
-    *,
-    body: Environment,
-) -> dict[str, Any]:
-    headers: dict[str, Any] = {}
-
-    _kwargs: dict[str, Any] = {
-        "method": "put",
-        "url": f"/environments/{environment_name}",
-    }
-
-    _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[Environment]:
-    if response.status_code == 200:
-        response_200 = Environment.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[Environment]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    environment_name: str,
-    *,
-    client: AuthenticatedClient,
-    body: Environment,
-) -> Response[Environment]:
-    """Update environment
-
-     Updates an environment.
-
-    Args:
-        environment_name (str):
-        body (Environment): Environment on which deployments will be made (e.g. development,
-            production), enforcing multiple policies at once.
-
-    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[Environment]
-    """
-
-    kwargs = _get_kwargs(
-        environment_name=environment_name,
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-def sync(
-    environment_name: str,
-    *,
-    client: AuthenticatedClient,
-    body: Environment,
-) -> Optional[Environment]:
-    """Update environment
-
-     Updates an environment.
-
-    Args:
-        environment_name (str):
-        body (Environment): Environment on which deployments will be made (e.g. development,
-            production), enforcing multiple policies at once.
-
-    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:
-        Environment
-    """
-
-    return sync_detailed(
-        environment_name=environment_name,
-        client=client,
-        body=body,
-    ).parsed
-
-
-async def asyncio_detailed(
-    environment_name: str,
-    *,
-    client: AuthenticatedClient,
-    body: Environment,
-) -> Response[Environment]:
-    """Update environment
-
-     Updates an environment.
-
-    Args:
-        environment_name (str):
-        body (Environment): Environment on which deployments will be made (e.g. development,
-            production), enforcing multiple policies at once.
-
-    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[Environment]
-    """
-
-    kwargs = _get_kwargs(
-        environment_name=environment_name,
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio(
-    environment_name: str,
-    *,
-    client: AuthenticatedClient,
-    body: Environment,
-) -> Optional[Environment]:
-    """Update environment
-
-     Updates an environment.
-
-    Args:
-        environment_name (str):
-        body (Environment): Environment on which deployments will be made (e.g. development,
-            production), enforcing multiple policies at once.
-
-    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:
-        Environment
-    """
-
-    return (
-        await asyncio_detailed(
-            environment_name=environment_name,
-            client=client,
-            body=body,
-        )
-    ).parsed

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