asteroid-odyssey 0.1.21__py3-none-any.whl → 1.0.0__py3-none-any.whl

asteroid_odyssey/api/generated/asteroid_agents_api_client/api/execution/update_execution_status.py

@@ -1,115 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Dict, Optional, Union
-from uuid import UUID
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.execution_status import ExecutionStatus
-from ...types import Response
-
-
-def _get_kwargs(
-    id: UUID,
-    *,
-    body: ExecutionStatus,
-) -> Dict[str, Any]:
-    headers: Dict[str, Any] = {}
-
-    _kwargs: Dict[str, Any] = {
-        "method": "post",
-        "url": f"/execution/{id}/status",
-    }
-
-    _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[Any]:
-    if response.status_code == 200:
-        return None
-    if response.status_code == 400:
-        return None
-    if response.status_code == 404:
-        return None
-    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[Any]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    id: UUID,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: ExecutionStatus,
-) -> Response[Any]:
-    """Update the status of an execution
-
-    Args:
-        id (UUID):
-        body (ExecutionStatus):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        id=id,
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio_detailed(
-    id: UUID,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: ExecutionStatus,
-) -> Response[Any]:
-    """Update the status of an execution
-
-    Args:
-        id (UUID):
-        body (ExecutionStatus):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        id=id,
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)

asteroid_odyssey/api/generated/asteroid_agents_api_client/api/notifications/set_slack_channel.py

@@ -1,107 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Dict, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.slack_channel_request import SlackChannelRequest
-from ...types import Response
-
-
-def _get_kwargs(
-    *,
-    body: SlackChannelRequest,
-) -> Dict[str, Any]:
-    headers: Dict[str, Any] = {}
-
-    _kwargs: Dict[str, Any] = {
-        "method": "post",
-        "url": "/slack_channel",
-    }
-
-    _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[Any]:
-    if response.status_code == 200:
-        return None
-    if response.status_code == 400:
-        return None
-    if response.status_code == 404:
-        return None
-    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[Any]:
-    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: Union[AuthenticatedClient, Client],
-    body: SlackChannelRequest,
-) -> Response[Any]:
-    """Set your slack notification channel
-
-    Args:
-        body (SlackChannelRequest):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio_detailed(
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: SlackChannelRequest,
-) -> Response[Any]:
-    """Set your slack notification channel
-
-    Args:
-        body (SlackChannelRequest):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)

asteroid_odyssey/api/generated/asteroid_agents_api_client/api/optimiser/queue_optimisation_job.py

@@ -1,103 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Dict, Optional, Union
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.optimisation_request import OptimisationRequest
-from ...types import Response
-
-
-def _get_kwargs(
-    *,
-    body: OptimisationRequest,
-) -> Dict[str, Any]:
-    headers: Dict[str, Any] = {}
-
-    _kwargs: Dict[str, Any] = {
-        "method": "post",
-        "url": "/optimiser",
-    }
-
-    _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[Any]:
-    if response.status_code == 202:
-        return None
-    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[Any]:
-    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: Union[AuthenticatedClient, Client],
-    body: OptimisationRequest,
-) -> Response[Any]:
-    """Queue an optimisation job
-
-    Args:
-        body (OptimisationRequest):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio_detailed(
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: OptimisationRequest,
-) -> Response[Any]:
-    """Queue an optimisation job
-
-    Args:
-        body (OptimisationRequest):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)

asteroid_odyssey/api/generated/asteroid_agents_api_client/api/workflow/add_workflow_credential.py

@@ -1,111 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Dict, Optional, Union
-from uuid import UUID
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.credentials_request import CredentialsRequest
-from ...types import Response
-
-
-def _get_kwargs(
-    workflow_id: UUID,
-    *,
-    body: CredentialsRequest,
-) -> Dict[str, Any]:
-    headers: Dict[str, Any] = {}
-
-    _kwargs: Dict[str, Any] = {
-        "method": "patch",
-        "url": f"/workflow/{workflow_id}/credentials",
-    }
-
-    _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[Any]:
-    if response.status_code == 200:
-        return None
-    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[Any]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    workflow_id: UUID,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: CredentialsRequest,
-) -> Response[Any]:
-    """Add a credential for a workflow
-
-    Args:
-        workflow_id (UUID):
-        body (CredentialsRequest):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        workflow_id=workflow_id,
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio_detailed(
-    workflow_id: UUID,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: CredentialsRequest,
-) -> Response[Any]:
-    """Add a credential for a workflow
-
-    Args:
-        workflow_id (UUID):
-        body (CredentialsRequest):
-
-    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[Any]
-    """
-
-    kwargs = _get_kwargs(
-        workflow_id=workflow_id,
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)

asteroid_odyssey/api/generated/asteroid_agents_api_client/api/workflow/create_workflow.py

@@ -1,174 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Dict, Optional, Union, cast
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.create_workflow_request import CreateWorkflowRequest
-from ...types import Response
-
-
-def _get_kwargs(
-    agent_name: str,
-    *,
-    body: CreateWorkflowRequest,
-) -> Dict[str, Any]:
-    headers: Dict[str, Any] = {}
-
-    _kwargs: Dict[str, Any] = {
-        "method": "post",
-        "url": f"/workflow/{agent_name}/create",
-    }
-
-    _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[Union[Any, str]]:
-    if response.status_code == 201:
-        response_201 = cast(str, response.json())
-        return response_201
-    if response.status_code == 400:
-        response_400 = cast(Any, None)
-        return response_400
-    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[Union[Any, str]]:
-    return Response(
-        status_code=HTTPStatus(response.status_code),
-        content=response.content,
-        headers=response.headers,
-        parsed=_parse_response(client=client, response=response),
-    )
-
-
-def sync_detailed(
-    agent_name: str,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: CreateWorkflowRequest,
-) -> Response[Union[Any, str]]:
-    """Create a new workflow for an agent
-
-    Args:
-        agent_name (str):
-        body (CreateWorkflowRequest):
-
-    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[Union[Any, str]]
-    """
-
-    kwargs = _get_kwargs(
-        agent_name=agent_name,
-        body=body,
-    )
-
-    response = client.get_httpx_client().request(
-        **kwargs,
-    )
-
-    return _build_response(client=client, response=response)
-
-
-def sync(
-    agent_name: str,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: CreateWorkflowRequest,
-) -> Optional[Union[Any, str]]:
-    """Create a new workflow for an agent
-
-    Args:
-        agent_name (str):
-        body (CreateWorkflowRequest):
-
-    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:
-        Union[Any, str]
-    """
-
-    return sync_detailed(
-        agent_name=agent_name,
-        client=client,
-        body=body,
-    ).parsed
-
-
-async def asyncio_detailed(
-    agent_name: str,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: CreateWorkflowRequest,
-) -> Response[Union[Any, str]]:
-    """Create a new workflow for an agent
-
-    Args:
-        agent_name (str):
-        body (CreateWorkflowRequest):
-
-    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[Union[Any, str]]
-    """
-
-    kwargs = _get_kwargs(
-        agent_name=agent_name,
-        body=body,
-    )
-
-    response = await client.get_async_httpx_client().request(**kwargs)
-
-    return _build_response(client=client, response=response)
-
-
-async def asyncio(
-    agent_name: str,
-    *,
-    client: Union[AuthenticatedClient, Client],
-    body: CreateWorkflowRequest,
-) -> Optional[Union[Any, str]]:
-    """Create a new workflow for an agent
-
-    Args:
-        agent_name (str):
-        body (CreateWorkflowRequest):
-
-    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:
-        Union[Any, str]
-    """
-
-    return (
-        await asyncio_detailed(
-            agent_name=agent_name,
-            client=client,
-            body=body,
-        )
-    ).parsed