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

asteroid_odyssey/api/generated/asteroid_agents_api_client/client.py

@@ -1,268 +0,0 @@
-import ssl
-from typing import Any, Dict, Optional, Union
-
-import httpx
-from attrs import define, evolve, field
-
-
-@define
-class Client:
-    """A class for keeping track of data related to the API
-
-    The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
-
-        ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
-
-        ``cookies``: A dictionary of cookies to be sent with every request
-
-        ``headers``: A dictionary of headers to be sent with every request
-
-        ``timeout``: The maximum amount of a time a request can take. API functions will raise
-        httpx.TimeoutException if this is exceeded.
-
-        ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
-        but can be set to False for testing purposes.
-
-        ``follow_redirects``: Whether or not to follow redirects. Default value is False.
-
-        ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
-
-
-    Attributes:
-        raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
-            status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
-            argument to the constructor.
-    """
-
-    raise_on_unexpected_status: bool = field(default=False, kw_only=True)
-    _base_url: str = field(alias="base_url")
-    _cookies: Dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
-    _headers: Dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
-    _timeout: Optional[httpx.Timeout] = field(default=None, kw_only=True, alias="timeout")
-    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
-    _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
-    _httpx_args: Dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
-    _client: Optional[httpx.Client] = field(default=None, init=False)
-    _async_client: Optional[httpx.AsyncClient] = field(default=None, init=False)
-
-    def with_headers(self, headers: Dict[str, str]) -> "Client":
-        """Get a new client matching this one with additional headers"""
-        if self._client is not None:
-            self._client.headers.update(headers)
-        if self._async_client is not None:
-            self._async_client.headers.update(headers)
-        return evolve(self, headers={**self._headers, **headers})
-
-    def with_cookies(self, cookies: Dict[str, str]) -> "Client":
-        """Get a new client matching this one with additional cookies"""
-        if self._client is not None:
-            self._client.cookies.update(cookies)
-        if self._async_client is not None:
-            self._async_client.cookies.update(cookies)
-        return evolve(self, cookies={**self._cookies, **cookies})
-
-    def with_timeout(self, timeout: httpx.Timeout) -> "Client":
-        """Get a new client matching this one with a new timeout (in seconds)"""
-        if self._client is not None:
-            self._client.timeout = timeout
-        if self._async_client is not None:
-            self._async_client.timeout = timeout
-        return evolve(self, timeout=timeout)
-
-    def set_httpx_client(self, client: httpx.Client) -> "Client":
-        """Manually set the underlying httpx.Client
-
-        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
-        """
-        self._client = client
-        return self
-
-    def get_httpx_client(self) -> httpx.Client:
-        """Get the underlying httpx.Client, constructing a new one if not previously set"""
-        if self._client is None:
-            self._client = httpx.Client(
-                base_url=self._base_url,
-                cookies=self._cookies,
-                headers=self._headers,
-                timeout=self._timeout,
-                verify=self._verify_ssl,
-                follow_redirects=self._follow_redirects,
-                **self._httpx_args,
-            )
-        return self._client
-
-    def __enter__(self) -> "Client":
-        """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
-        self.get_httpx_client().__enter__()
-        return self
-
-    def __exit__(self, *args: Any, **kwargs: Any) -> None:
-        """Exit a context manager for internal httpx.Client (see httpx docs)"""
-        self.get_httpx_client().__exit__(*args, **kwargs)
-
-    def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "Client":
-        """Manually the underlying httpx.AsyncClient
-
-        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
-        """
-        self._async_client = async_client
-        return self
-
-    def get_async_httpx_client(self) -> httpx.AsyncClient:
-        """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
-        if self._async_client is None:
-            self._async_client = httpx.AsyncClient(
-                base_url=self._base_url,
-                cookies=self._cookies,
-                headers=self._headers,
-                timeout=self._timeout,
-                verify=self._verify_ssl,
-                follow_redirects=self._follow_redirects,
-                **self._httpx_args,
-            )
-        return self._async_client
-
-    async def __aenter__(self) -> "Client":
-        """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
-        await self.get_async_httpx_client().__aenter__()
-        return self
-
-    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
-        """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
-        await self.get_async_httpx_client().__aexit__(*args, **kwargs)
-
-
-@define
-class AuthenticatedClient:
-    """A Client which has been authenticated for use on secured endpoints
-
-    The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
-
-        ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
-
-        ``cookies``: A dictionary of cookies to be sent with every request
-
-        ``headers``: A dictionary of headers to be sent with every request
-
-        ``timeout``: The maximum amount of a time a request can take. API functions will raise
-        httpx.TimeoutException if this is exceeded.
-
-        ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
-        but can be set to False for testing purposes.
-
-        ``follow_redirects``: Whether or not to follow redirects. Default value is False.
-
-        ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
-
-
-    Attributes:
-        raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
-            status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
-            argument to the constructor.
-        token: The token to use for authentication
-        prefix: The prefix to use for the Authorization header
-        auth_header_name: The name of the Authorization header
-    """
-
-    raise_on_unexpected_status: bool = field(default=False, kw_only=True)
-    _base_url: str = field(alias="base_url")
-    _cookies: Dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
-    _headers: Dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
-    _timeout: Optional[httpx.Timeout] = field(default=None, kw_only=True, alias="timeout")
-    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
-    _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
-    _httpx_args: Dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
-    _client: Optional[httpx.Client] = field(default=None, init=False)
-    _async_client: Optional[httpx.AsyncClient] = field(default=None, init=False)
-
-    token: str
-    prefix: str = "Bearer"
-    auth_header_name: str = "Authorization"
-
-    def with_headers(self, headers: Dict[str, str]) -> "AuthenticatedClient":
-        """Get a new client matching this one with additional headers"""
-        if self._client is not None:
-            self._client.headers.update(headers)
-        if self._async_client is not None:
-            self._async_client.headers.update(headers)
-        return evolve(self, headers={**self._headers, **headers})
-
-    def with_cookies(self, cookies: Dict[str, str]) -> "AuthenticatedClient":
-        """Get a new client matching this one with additional cookies"""
-        if self._client is not None:
-            self._client.cookies.update(cookies)
-        if self._async_client is not None:
-            self._async_client.cookies.update(cookies)
-        return evolve(self, cookies={**self._cookies, **cookies})
-
-    def with_timeout(self, timeout: httpx.Timeout) -> "AuthenticatedClient":
-        """Get a new client matching this one with a new timeout (in seconds)"""
-        if self._client is not None:
-            self._client.timeout = timeout
-        if self._async_client is not None:
-            self._async_client.timeout = timeout
-        return evolve(self, timeout=timeout)
-
-    def set_httpx_client(self, client: httpx.Client) -> "AuthenticatedClient":
-        """Manually set the underlying httpx.Client
-
-        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
-        """
-        self._client = client
-        return self
-
-    def get_httpx_client(self) -> httpx.Client:
-        """Get the underlying httpx.Client, constructing a new one if not previously set"""
-        if self._client is None:
-            self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
-            self._client = httpx.Client(
-                base_url=self._base_url,
-                cookies=self._cookies,
-                headers=self._headers,
-                timeout=self._timeout,
-                verify=self._verify_ssl,
-                follow_redirects=self._follow_redirects,
-                **self._httpx_args,
-            )
-        return self._client
-
-    def __enter__(self) -> "AuthenticatedClient":
-        """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
-        self.get_httpx_client().__enter__()
-        return self
-
-    def __exit__(self, *args: Any, **kwargs: Any) -> None:
-        """Exit a context manager for internal httpx.Client (see httpx docs)"""
-        self.get_httpx_client().__exit__(*args, **kwargs)
-
-    def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "AuthenticatedClient":
-        """Manually the underlying httpx.AsyncClient
-
-        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
-        """
-        self._async_client = async_client
-        return self
-
-    def get_async_httpx_client(self) -> httpx.AsyncClient:
-        """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
-        if self._async_client is None:
-            self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
-            self._async_client = httpx.AsyncClient(
-                base_url=self._base_url,
-                cookies=self._cookies,
-                headers=self._headers,
-                timeout=self._timeout,
-                verify=self._verify_ssl,
-                follow_redirects=self._follow_redirects,
-                **self._httpx_args,
-            )
-        return self._async_client
-
-    async def __aenter__(self) -> "AuthenticatedClient":
-        """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
-        await self.get_async_httpx_client().__aenter__()
-        return self
-
-    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
-        """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
-        await self.get_async_httpx_client().__aexit__(*args, **kwargs)

asteroid_odyssey/api/generated/asteroid_agents_api_client/errors.py

@@ -1,16 +0,0 @@
-"""Contains shared errors types that can be raised from API functions"""
-
-
-class UnexpectedStatus(Exception):
-    """Raised by api functions when the response status an undocumented status and Client.raise_on_unexpected_status is True"""
-
-    def __init__(self, status_code: int, content: bytes):
-        self.status_code = status_code
-        self.content = content
-
-        super().__init__(
-            f"Unexpected status code: {status_code}\n\nResponse content:\n{content.decode(errors='ignore')}"
-        )
-
-
-__all__ = ["UnexpectedStatus"]

asteroid_odyssey/api/generated/asteroid_agents_api_client/models/__init__.py

@@ -1,61 +0,0 @@
-"""Contains all the data models used in inputs/outputs"""
-
-from .agent import Agent
-from .browser_session import BrowserSession
-from .create_workflow_request import CreateWorkflowRequest
-from .create_workflow_request_fields import CreateWorkflowRequestFields
-from .create_workflow_request_provider import CreateWorkflowRequestProvider
-from .credential import Credential
-from .credentials_request import CredentialsRequest
-from .credentials_response import CredentialsResponse
-from .delete_execution_response_200 import DeleteExecutionResponse200
-from .delete_execution_response_404 import DeleteExecutionResponse404
-from .delete_workflow_response_200 import DeleteWorkflowResponse200
-from .delete_workflow_response_404 import DeleteWorkflowResponse404
-from .execution import Execution
-from .execution_dynamic_data import ExecutionDynamicData
-from .execution_result import ExecutionResult
-from .execution_status import ExecutionStatus
-from .file import File
-from .health_check_response_200 import HealthCheckResponse200
-from .health_check_response_500 import HealthCheckResponse500
-from .optimisation_request import OptimisationRequest
-from .progress_update import ProgressUpdate
-from .result_schema import ResultSchema
-from .slack_channel_request import SlackChannelRequest
-from .status import Status
-from .workflow import Workflow
-from .workflow_execution import WorkflowExecution
-from .workflow_execution_request import WorkflowExecutionRequest
-from .workflow_fields import WorkflowFields
-
-__all__ = (
-    "Agent",
-    "BrowserSession",
-    "CreateWorkflowRequest",
-    "CreateWorkflowRequestFields",
-    "CreateWorkflowRequestProvider",
-    "Credential",
-    "CredentialsRequest",
-    "CredentialsResponse",
-    "DeleteExecutionResponse200",
-    "DeleteExecutionResponse404",
-    "DeleteWorkflowResponse200",
-    "DeleteWorkflowResponse404",
-    "Execution",
-    "ExecutionDynamicData",
-    "ExecutionResult",
-    "ExecutionStatus",
-    "File",
-    "HealthCheckResponse200",
-    "HealthCheckResponse500",
-    "OptimisationRequest",
-    "ProgressUpdate",
-    "ResultSchema",
-    "SlackChannelRequest",
-    "Status",
-    "Workflow",
-    "WorkflowExecution",
-    "WorkflowExecutionRequest",
-    "WorkflowFields",
-)

asteroid_odyssey/api/generated/asteroid_agents_api_client/models/agent.py

@@ -1,90 +0,0 @@
-from typing import Any, Dict, List, Type, TypeVar, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="Agent")
-
-
-@_attrs_define
-class Agent:
-    """
-    Attributes:
-        name (str): The name of the agent Example: my_agent.
-        description (str): The description of the agent Example: This agent is used to queue jobs.
-        visibility (str): The visibility of the agent Example: public.
-        required_fields (List[str]): The required fields for the agent Example: ['system_prompt_template'].
-        required_prompts (List[str]): The prompts for the agent Example: ['Hello {{.name}}, your model is {{.model}}'].
-    """
-
-    name: str
-    description: str
-    visibility: str
-    required_fields: List[str]
-    required_prompts: List[str]
-    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> Dict[str, Any]:
-        name = self.name
-
-        description = self.description
-
-        visibility = self.visibility
-
-        required_fields = self.required_fields
-
-        required_prompts = self.required_prompts
-
-        field_dict: Dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "name": name,
-                "description": description,
-                "visibility": visibility,
-                "required_fields": required_fields,
-                "required_prompts": required_prompts,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
-        d = src_dict.copy()
-        name = d.pop("name")
-
-        description = d.pop("description")
-
-        visibility = d.pop("visibility")
-
-        required_fields = cast(List[str], d.pop("required_fields"))
-
-        required_prompts = cast(List[str], d.pop("required_prompts"))
-
-        agent = cls(
-            name=name,
-            description=description,
-            visibility=visibility,
-            required_fields=required_fields,
-            required_prompts=required_prompts,
-        )
-
-        agent.additional_properties = d
-        return agent
-
-    @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

asteroid_odyssey/api/generated/asteroid_agents_api_client/models/browser_session.py

@@ -1,136 +0,0 @@
-from typing import Any, Dict, List, Type, TypeVar, Union
-from uuid import UUID
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="BrowserSession")
-
-
-@_attrs_define
-class BrowserSession:
-    """
-    Attributes:
-        id (Union[Unset, UUID]): Browser session identifier.
-        browser_name (Union[Unset, str]): Browser name (Anchor, Browserbase, etc.)
-        execution_id (Union[Unset, UUID]): Execution ID.
-        cdp_url (Union[Unset, str]): CDP URL.
-        debugger_url (Union[Unset, str]): Debugger URL.
-        session_id (Union[Unset, str]): Session ID.
-        session_url (Union[Unset, str]): Session URL.
-        recording_url (Union[Unset, str]): Recording URL.
-    """
-
-    id: Union[Unset, UUID] = UNSET
-    browser_name: Union[Unset, str] = UNSET
-    execution_id: Union[Unset, UUID] = UNSET
-    cdp_url: Union[Unset, str] = UNSET
-    debugger_url: Union[Unset, str] = UNSET
-    session_id: Union[Unset, str] = UNSET
-    session_url: Union[Unset, str] = UNSET
-    recording_url: Union[Unset, str] = UNSET
-    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> Dict[str, Any]:
-        id: Union[Unset, str] = UNSET
-        if not isinstance(self.id, Unset):
-            id = str(self.id)
-
-        browser_name = self.browser_name
-
-        execution_id: Union[Unset, str] = UNSET
-        if not isinstance(self.execution_id, Unset):
-            execution_id = str(self.execution_id)
-
-        cdp_url = self.cdp_url
-
-        debugger_url = self.debugger_url
-
-        session_id = self.session_id
-
-        session_url = self.session_url
-
-        recording_url = self.recording_url
-
-        field_dict: Dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if id is not UNSET:
-            field_dict["id"] = id
-        if browser_name is not UNSET:
-            field_dict["browser_name"] = browser_name
-        if execution_id is not UNSET:
-            field_dict["execution_id"] = execution_id
-        if cdp_url is not UNSET:
-            field_dict["cdp_url"] = cdp_url
-        if debugger_url is not UNSET:
-            field_dict["debugger_url"] = debugger_url
-        if session_id is not UNSET:
-            field_dict["session_id"] = session_id
-        if session_url is not UNSET:
-            field_dict["session_url"] = session_url
-        if recording_url is not UNSET:
-            field_dict["recording_url"] = recording_url
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
-        d = src_dict.copy()
-        _id = d.pop("id", UNSET)
-        id: Union[Unset, UUID]
-        if isinstance(_id, Unset):
-            id = UNSET
-        else:
-            id = UUID(_id)
-
-        browser_name = d.pop("browser_name", UNSET)
-
-        _execution_id = d.pop("execution_id", UNSET)
-        execution_id: Union[Unset, UUID]
-        if isinstance(_execution_id, Unset):
-            execution_id = UNSET
-        else:
-            execution_id = UUID(_execution_id)
-
-        cdp_url = d.pop("cdp_url", UNSET)
-
-        debugger_url = d.pop("debugger_url", UNSET)
-
-        session_id = d.pop("session_id", UNSET)
-
-        session_url = d.pop("session_url", UNSET)
-
-        recording_url = d.pop("recording_url", UNSET)
-
-        browser_session = cls(
-            id=id,
-            browser_name=browser_name,
-            execution_id=execution_id,
-            cdp_url=cdp_url,
-            debugger_url=debugger_url,
-            session_id=session_id,
-            session_url=session_url,
-            recording_url=recording_url,
-        )
-
-        browser_session.additional_properties = d
-        return browser_session
-
-    @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