athena-intelligence 0.1.94__py3-none-any.whl → 0.1.95__py3-none-any.whl

athena/message/client.py

@@ -1,249 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.jsonable_encoder import jsonable_encoder
-from ..core.pydantic_utilities import pydantic_v1
-from ..core.request_options import RequestOptions
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
-from ..types.message_out import MessageOut
-from ..types.message_out_dto import MessageOutDto
-from ..types.model import Model
-from ..types.tools import Tools
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class MessageClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def submit(
-        self,
-        *,
-        content: str,
-        model: typing.Optional[Model] = OMIT,
-        tools: typing.Optional[typing.Sequence[Tools]] = OMIT,
-        conversation_id: typing.Optional[str] = OMIT,
-        conversation_name: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> MessageOut:
-        """
-        Parameters
-        ----------
-        content : str
-
-        model : typing.Optional[Model]
-
-        tools : typing.Optional[typing.Sequence[Tools]]
-
-        conversation_id : typing.Optional[str]
-
-        conversation_name : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        MessageOut
-            Successful Response
-
-        Examples
-        --------
-        from athena import Model, Tools
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.message.submit(
-            content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
-            model=Model.GPT_35_TURBO,
-            tools=[Tools.SEARCH, Tools.BROWSE],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/message",
-            method="POST",
-            json={
-                "content": content,
-                "model": model,
-                "tools": tools,
-                "conversation_id": conversation_id,
-                "conversation_name": conversation_name,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(MessageOut, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def get(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> MessageOutDto:
-        """
-        Parameters
-        ----------
-        id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        MessageOutDto
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.message.get(
-            id="id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"api/v0/message/{jsonable_encoder(id)}", method="GET", request_options=request_options
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(MessageOutDto, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-
-class AsyncMessageClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def submit(
-        self,
-        *,
-        content: str,
-        model: typing.Optional[Model] = OMIT,
-        tools: typing.Optional[typing.Sequence[Tools]] = OMIT,
-        conversation_id: typing.Optional[str] = OMIT,
-        conversation_name: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> MessageOut:
-        """
-        Parameters
-        ----------
-        content : str
-
-        model : typing.Optional[Model]
-
-        tools : typing.Optional[typing.Sequence[Tools]]
-
-        conversation_id : typing.Optional[str]
-
-        conversation_name : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        MessageOut
-            Successful Response
-
-        Examples
-        --------
-        from athena import Model, Tools
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.message.submit(
-            content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
-            model=Model.GPT_35_TURBO,
-            tools=[Tools.SEARCH, Tools.BROWSE],
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/message",
-            method="POST",
-            json={
-                "content": content,
-                "model": model,
-                "tools": tools,
-                "conversation_id": conversation_id,
-                "conversation_name": conversation_name,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(MessageOut, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    async def get(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> MessageOutDto:
-        """
-        Parameters
-        ----------
-        id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        MessageOutDto
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.message.get(
-            id="id",
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"api/v0/message/{jsonable_encoder(id)}", method="GET", request_options=request_options
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(MessageOutDto, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/polling_message_client.py

@@ -1,125 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import time
-import typing
-
-from athena.types.message_out_dto import MessageOutDto
-
-from .core import RequestOptions
-from .message.client import AsyncMessageClient, MessageClient
-from .types.model import Model
-from .types.tools import Tools
-from .types.status_enum import StatusEnum
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class MessagePollingClient(MessageClient):
-
-    def submit_and_poll(
-        self,
-        *,
-        content: str,
-        model: typing.Optional[Model] = OMIT,
-        tools: typing.Optional[typing.Sequence[Tools]] = OMIT,
-        conversation_id: typing.Optional[str] = OMIT,
-        conversation_name: typing.Optional[str] = OMIT,
-        poll_interval: int = 2,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> MessageOutDto:
-        """
-        Parameters:
-            - content: str.
-
-            - model: typing.Optional[Model].
-
-            - tools: typing.Optional[typing.Sequence[Tools]].
-
-            - conversation_id: typing.Optional[str].
-
-            - conversation_name: typing.Optional[str].
-
-            - poll_interval: int. The polling interval in seconds.
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena import Model, Tools
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.message.submit(
-            content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
-            model=Model.GPT_4_TURBO_PREVIEW,
-            tools=[Tools.SEARCH, Tools.BROWSE],
-        )
-        """
-        response = super().submit(
-          content=content,
-          model=model,
-          tools=tools,
-          conversation_id=conversation_id, 
-          conversation_name=conversation_name,
-          request_options=request_options)
-        message = self.get(id=response.id)
-        while message.status == StatusEnum.PENDING:
-            time.sleep(poll_interval)
-            message = self.get(id=message.id)
-        return message
-
-
-class AsyncMessagePollingClient(AsyncMessageClient):
-
-    async def submit_and_poll(
-        self,
-        *,
-        content: str,
-        model: typing.Optional[Model] = OMIT,
-        tools: typing.Optional[typing.Sequence[Tools]] = OMIT,
-        conversation_id: typing.Optional[str] = OMIT,
-        conversation_name: typing.Optional[str] = OMIT,
-        poll_interval: int = 2,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> MessageOutDto: 
-        """
-        Parameters:
-            - content: str.
-
-            - model: typing.Optional[Model].
-
-            - tools: typing.Optional[typing.Sequence[Tools]].
-
-            - conversation_id: typing.Optional[str].
-
-            - conversation_name: typing.Optional[str].
-
-            - poll_interval: int. The polling interval in seconds.
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena import Model, Tools
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY"
-        )
-        client.message.submit(
-            content="visit www.athenaintelligence.ai and summarize the website in one paragraph",
-            model=Model.GPT_4_TURBO_PREVIEW,
-            tools=[Tools.SEARCH, Tools.BROWSE],
-        )
-        """
-        response = await super().submit(
-          content=content,
-          model=model,
-          tools=tools,
-          conversation_id=conversation_id, 
-          conversation_name=conversation_name,
-          request_options=request_options)
-        message = await self.get(id=response.id)
-        while message.status == StatusEnum.PENDING:
-            time.sleep(poll_interval)
-            message = await self.get(id=message.id)
-        return message

athena/query/__init__.py

@@ -1,2 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-

athena/query/client.py

@@ -1,123 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.pydantic_utilities import pydantic_v1
-from ..core.request_options import RequestOptions
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
-from ..types.sql_results import SqlResults
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class QueryClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def execute(
-        self, *, sql_command: str, database_id: int, request_options: typing.Optional[RequestOptions] = None
-    ) -> SqlResults:
-        """
-        Parameters
-        ----------
-        sql_command : str
-
-        database_id : int
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        SqlResults
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.query.execute(
-            sql_command="sql_command",
-            database_id=1,
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/execute-sql",
-            method="POST",
-            json={"sql_command": sql_command, "database_id": database_id},
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(SqlResults, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-
-class AsyncQueryClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def execute(
-        self, *, sql_command: str, database_id: int, request_options: typing.Optional[RequestOptions] = None
-    ) -> SqlResults:
-        """
-        Parameters
-        ----------
-        sql_command : str
-
-        database_id : int
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        SqlResults
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.query.execute(
-            sql_command="sql_command",
-            database_id=1,
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/execute-sql",
-            method="POST",
-            json={"sql_command": sql_command, "database_id": database_id},
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(SqlResults, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/report/__init__.py

@@ -1,2 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-

athena/report/client.py

@@ -1,125 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.pydantic_utilities import pydantic_v1
-from ..core.request_options import RequestOptions
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
-from ..types.report import Report
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class ReportClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def create(
-        self,
-        *,
-        name: typing.Optional[str] = OMIT,
-        workspace_access: typing.Optional[bool] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> Report:
-        """
-        Parameters
-        ----------
-        name : typing.Optional[str]
-
-        workspace_access : typing.Optional[bool]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Report
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.report.create()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/report",
-            method="POST",
-            json={"name": name, "workspace_access": workspace_access},
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Report, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-
-class AsyncReportClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def create(
-        self,
-        *,
-        name: typing.Optional[str] = OMIT,
-        workspace_access: typing.Optional[bool] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> Report:
-        """
-        Parameters
-        ----------
-        name : typing.Optional[str]
-
-        workspace_access : typing.Optional[bool]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        Report
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.report.create()
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/report",
-            method="POST",
-            json={"name": name, "workspace_access": workspace_access},
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(Report, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/search/__init__.py

@@ -1,2 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-