athena-intelligence 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

athena/__init__.py

@@ -1,6 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .resources import Message, message
+from .types import HttpValidationError, MessageOut, MessageOutDto, Model, Tools, ValidationError, ValidationErrorLocItem
+from .errors import UnprocessableEntityError
 from .environment import AthenaEnvironment
 
-__all__ = ["AthenaEnvironment", "Message", "message"]
+__all__ = [
+    "AthenaEnvironment",
+    "HttpValidationError",
+    "MessageOut",
+    "MessageOutDto",
+    "Model",
+    "Tools",
+    "UnprocessableEntityError",
+    "ValidationError",
+    "ValidationErrorLocItem",
+]

athena/client.py

@@ -1,14 +1,31 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import os
 import typing
+import urllib.parse
+from json.decoder import JSONDecodeError
 
 import httpx
 
 from .core.api_error import ApiError
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .core.jsonable_encoder import jsonable_encoder
+from .core.remove_none_from_dict import remove_none_from_dict
+from .core.request_options import RequestOptions
 from .environment import AthenaEnvironment
-from .resources.message.client import AsyncMessageClient, MessageClient
+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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
 
 
 class Athena:
@@ -20,9 +37,7 @@ class Athena:
 
         - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
 
-                                          Defaults to AthenaEnvironment.PRODUCTION
-
-        - api_key: typing.Optional[typing.Union[str, typing.Callable[[], str]]].
+                                          Defaults to AthenaEnvironment.DEFAULT
 
         - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
 
@@ -30,30 +45,177 @@ class Athena:
     ---
     from athena.client import Athena
 
-    client = Athena(
-        api_key="YOUR_API_KEY",
-    )
+    client = Athena()
     """
 
     def __init__(
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
-        api_key: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = os.getenv("ATHENA_API_KEY"),
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
         timeout: typing.Optional[float] = 60,
-        httpx_client: typing.Optional[httpx.Client] = None
+        httpx_client: typing.Optional[httpx.Client] = None,
     ):
-        if api_key is None:
-            raise ApiError(
-                body="The client must be instantiated be either passing in api_key or setting ATHENA_API_KEY"
-            )
         self._client_wrapper = SyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
-            api_key=api_key,
             httpx_client=httpx.Client(timeout=timeout) if httpx_client is None else httpx_client,
         )
-        self.message = MessageClient(client_wrapper=self._client_wrapper)
+
+    def health_check(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Checks the health of a project.
+
+        It returns 200 if the project is healthy.
+
+        Parameters:
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import Athena
+
+        client = Athena()
+        client.health_check()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/health"),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else 60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _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 athena_get_message(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> MessageOutDto:
+        """
+        Parameters:
+            - id: str.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import Athena
+
+        client = Athena()
+        client.athena_get_message(
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v0/message/{jsonable_encoder(id)}"),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else 60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(MessageOutDto, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.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 post_message(
+        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.
+        ---
+        from athena.client import Athena
+
+        client = Athena()
+        client.post_message(
+            content="content",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"content": content}
+        if model is not OMIT:
+            _request["model"] = model.value if model is not None else None
+        if tools is not OMIT:
+            _request["tools"] = tools
+        if conversation_id is not OMIT:
+            _request["conversation_id"] = conversation_id
+        if conversation_name is not OMIT:
+            _request["conversation_name"] = conversation_name
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/message"),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            json=jsonable_encoder(_request)
+            if request_options is None or request_options.get("additional_body_parameters") is None
+            else {
+                **jsonable_encoder(_request),
+                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
+            },
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else 60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(MessageOut, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.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 AsyncAthena:
@@ -65,9 +227,7 @@ class AsyncAthena:
 
         - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
 
-                                          Defaults to AthenaEnvironment.PRODUCTION
-
-        - api_key: typing.Optional[typing.Union[str, typing.Callable[[], str]]].
+                                          Defaults to AthenaEnvironment.DEFAULT
 
         - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
 
@@ -75,30 +235,179 @@ class AsyncAthena:
     ---
     from athena.client import AsyncAthena
 
-    client = AsyncAthena(
-        api_key="YOUR_API_KEY",
-    )
+    client = AsyncAthena()
     """
 
     def __init__(
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
-        api_key: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = os.getenv("ATHENA_API_KEY"),
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
         timeout: typing.Optional[float] = 60,
-        httpx_client: typing.Optional[httpx.AsyncClient] = None
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
     ):
-        if api_key is None:
-            raise ApiError(
-                body="The client must be instantiated be either passing in api_key or setting ATHENA_API_KEY"
-            )
         self._client_wrapper = AsyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
-            api_key=api_key,
             httpx_client=httpx.AsyncClient(timeout=timeout) if httpx_client is None else httpx_client,
         )
-        self.message = AsyncMessageClient(client_wrapper=self._client_wrapper)
+
+    async def health_check(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Checks the health of a project.
+
+        It returns 200 if the project is healthy.
+
+        Parameters:
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena()
+        await client.health_check()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/health"),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else 60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _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 athena_get_message(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> MessageOutDto:
+        """
+        Parameters:
+            - id: str.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena()
+        await client.athena_get_message(
+            id="id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v0/message/{jsonable_encoder(id)}"),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else 60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(MessageOutDto, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.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 post_message(
+        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.
+        ---
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena()
+        await client.post_message(
+            content="content",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"content": content}
+        if model is not OMIT:
+            _request["model"] = model.value if model is not None else None
+        if tools is not OMIT:
+            _request["tools"] = tools
+        if conversation_id is not OMIT:
+            _request["conversation_id"] = conversation_id
+        if conversation_name is not OMIT:
+            _request["conversation_name"] = conversation_name
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/message"),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            json=jsonable_encoder(_request)
+            if request_options is None or request_options.get("additional_body_parameters") is None
+            else {
+                **jsonable_encoder(_request),
+                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
+            },
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else 60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(MessageOut, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.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_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:

athena/core/client_wrapper.py

@@ -6,40 +6,28 @@ import httpx
 
 
 class BaseClientWrapper:
-    def __init__(self, *, api_key: typing.Union[str, typing.Callable[[], str]], base_url: str):
-        self._api_key = api_key
+    def __init__(self, *, base_url: str):
         self._base_url = base_url
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.3",
+            "X-Fern-SDK-Version": "0.1.5",
         }
-        headers["Authorization"] = f"Bearer {self._get_api_key()}"
         return headers
 
-    def _get_api_key(self) -> str:
-        if isinstance(self._api_key, str):
-            return self._api_key
-        else:
-            return self._api_key()
-
     def get_base_url(self) -> str:
         return self._base_url
 
 
 class SyncClientWrapper(BaseClientWrapper):
-    def __init__(
-        self, *, api_key: typing.Union[str, typing.Callable[[], str]], base_url: str, httpx_client: httpx.Client
-    ):
-        super().__init__(api_key=api_key, base_url=base_url)
+    def __init__(self, *, base_url: str, httpx_client: httpx.Client):
+        super().__init__(base_url=base_url)
         self.httpx_client = httpx_client
 
 
 class AsyncClientWrapper(BaseClientWrapper):
-    def __init__(
-        self, *, api_key: typing.Union[str, typing.Callable[[], str]], base_url: str, httpx_client: httpx.AsyncClient
-    ):
-        super().__init__(api_key=api_key, base_url=base_url)
+    def __init__(self, *, base_url: str, httpx_client: httpx.AsyncClient):
+        super().__init__(base_url=base_url)
         self.httpx_client = httpx_client

athena/environment.py

@@ -4,4 +4,4 @@ import enum
 
 
 class AthenaEnvironment(enum.Enum):
-    PRODUCTION = "https://app.athenaintelligence.ai"
+    DEFAULT = "https://api.athenaintelligence.ai"

athena/errors/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .unprocessable_entity_error import UnprocessableEntityError
+
+__all__ = ["UnprocessableEntityError"]

athena/errors/unprocessable_entity_error.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.api_error import ApiError
+from ..types.http_validation_error import HttpValidationError
+
+
+class UnprocessableEntityError(ApiError):
+    def __init__(self, body: HttpValidationError):
+        super().__init__(status_code=422, body=body)

athena/types/__init__.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .http_validation_error import HttpValidationError
+from .message_out import MessageOut
+from .message_out_dto import MessageOutDto
+from .model import Model
+from .tools import Tools
+from .validation_error import ValidationError
+from .validation_error_loc_item import ValidationErrorLocItem
+
+__all__ = [
+    "HttpValidationError",
+    "MessageOut",
+    "MessageOutDto",
+    "Model",
+    "Tools",
+    "ValidationError",
+    "ValidationErrorLocItem",
+]

athena/types/http_validation_error.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .validation_error import ValidationError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class HttpValidationError(pydantic.BaseModel):
+    detail: typing.Optional[typing.List[ValidationError]] = None
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

athena/types/message_out.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class MessageOut(pydantic.BaseModel):
+    id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

athena/{resources/message/types/message.py → types/message_out_dto.py}

@@ -3,7 +3,7 @@
 import datetime as dt
 import typing
 
-from ....core.datetime_utils import serialize_datetime
+from ..core.datetime_utils import serialize_datetime
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -11,15 +11,13 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class Message(pydantic.BaseModel):
-    message_id: str
+class MessageOutDto(pydantic.BaseModel):
+    id: str
     conversation_id: str
-    sender: str
-    content: str
+    logs: typing.Optional[str] = None
+    content: typing.Optional[str] = None
+    created_at: str
     status: str
-    created_at: dt.datetime
-    config: typing.Dict[str, typing.Any]
-    final_output: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena/types/model.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class Model(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    GPT_4 = "gpt_4"
+    GPT_4_TURBO_PREVIEW = "gpt_4_turbo_preview"
+    GPT_3_5_TURBO = "gpt_3_5_turbo"
+
+    def visit(
+        self,
+        gpt_4: typing.Callable[[], T_Result],
+        gpt_4_turbo_preview: typing.Callable[[], T_Result],
+        gpt_3_5_turbo: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is Model.GPT_4:
+            return gpt_4()
+        if self is Model.GPT_4_TURBO_PREVIEW:
+            return gpt_4_turbo_preview()
+        if self is Model.GPT_3_5_TURBO:
+            return gpt_3_5_turbo()

athena/types/tools.py

@@ -0,0 +1,45 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class Tools(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    SEARCH = "search"
+    BROWSE = "browse"
+    DATABASE_METADATA = "database_metadata"
+    QUERY = "query"
+    CHART = "chart"
+    ENRICH_PERSON = "enrich_person"
+    ENRICH_COMPANY = "enrich_company"
+
+    def visit(
+        self,
+        search: typing.Callable[[], T_Result],
+        browse: typing.Callable[[], T_Result],
+        database_metadata: typing.Callable[[], T_Result],
+        query: typing.Callable[[], T_Result],
+        chart: typing.Callable[[], T_Result],
+        enrich_person: typing.Callable[[], T_Result],
+        enrich_company: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is Tools.SEARCH:
+            return search()
+        if self is Tools.BROWSE:
+            return browse()
+        if self is Tools.DATABASE_METADATA:
+            return database_metadata()
+        if self is Tools.QUERY:
+            return query()
+        if self is Tools.CHART:
+            return chart()
+        if self is Tools.ENRICH_PERSON:
+            return enrich_person()
+        if self is Tools.ENRICH_COMPANY:
+            return enrich_company()

athena/types/validation_error.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .validation_error_loc_item import ValidationErrorLocItem
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ValidationError(pydantic.BaseModel):
+    loc: typing.List[ValidationErrorLocItem]
+    msg: str
+    type: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

athena/{resources/message/__init__.py → types/validation_error_loc_item.py}

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import Message
+import typing
 
-__all__ = ["Message"]
+ValidationErrorLocItem = typing.Union[str, int]

{athena_intelligence-0.1.3.dist-info → athena_intelligence-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.3
+Version: 0.1.5
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3

athena_intelligence-0.1.5.dist-info/RECORD

@@ -0,0 +1,25 @@
+athena/__init__.py,sha256=S4TY2NjQ19rPwYCWsw8-Cxrr51PjtSN2pma_ybgaUtQ,491
+athena/client.py,sha256=diL-uLcWo2lhyxOXdZRdUn8DrhjII4yLYgdStelnkh8,17713
+athena/core/__init__.py,sha256=95onSWXymaL0iV-nBsuKNgjh1LbsymAkRkx7POn1nc0,696
+athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
+athena/core/client_wrapper.py,sha256=sDd5yzNnq_9nIwN3MDMpVLv6dWB0hauTPxulICxMRt8,937
+athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
+athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
+athena/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
+athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
+athena/core/request_options.py,sha256=9n3ZHWZkw60MGp7GbdjGEuGo73Pa-6JeusRGhS533aA,1297
+athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
+athena/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
+athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
+athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+athena/types/__init__.py,sha256=sG74gyPktXegm01dMPmvxhq9um_3GdxMtF8kbSR9L4s,517
+athena/types/http_validation_error.py,sha256=Fcv_CTMMrLvCeTHjF0n5xf5tskMDgt-J6H9gp654eQw,973
+athena/types/message_out.py,sha256=uvZY_Podv2XccEk8CICug9I_S2hFJTSzCBwcHiauW7A,865
+athena/types/message_out_dto.py,sha256=GZ30Lk4PS0opAJS24cC_VfpPVZ87lFW171YH82_dEaQ,1008
+athena/types/model.py,sha256=SXpMQM_Dk-AIEbisBd4OGwLiqI1ePaDNEqXrs9ijuEY,732
+athena/types/tools.py,sha256=Vfb-D9CYcPZH6MBATNODgfXjFyBpCs4qbkqpCMl7eBM,1277
+athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
+athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
+athena_intelligence-0.1.5.dist-info/METADATA,sha256=XXNAl8v11mjjc24OVqFUYEngibxbnUsjdsG2BmQUSc4,3741
+athena_intelligence-0.1.5.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.5.dist-info/RECORD,,

athena/resources/__init__.py

@@ -1,6 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from . import message
-from .message import Message
-
-__all__ = ["Message", "message"]

athena/resources/message/client.py

@@ -1,292 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-import urllib.parse
-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.remove_none_from_dict import remove_none_from_dict
-from ...core.request_options import RequestOptions
-from .types.message import Message
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-# 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 get(
-        self, message_id: typing.Optional[str], *, request_options: typing.Optional[RequestOptions] = None
-    ) -> Message:
-        """
-        Get an Athena message by ID.
-        Returns an Athena message final output.
-
-        Parameters:
-            - message_id: typing.Optional[str].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"api/agent/{jsonable_encoder(message_id)}"
-            ),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else 60,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(Message, _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 send(
-        self,
-        *,
-        content: str,
-        conversation_id: typing.Optional[str] = OMIT,
-        user_id: typing.Optional[str] = OMIT,
-        workspace_id: typing.Optional[str] = OMIT,
-        sdataset_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        tools: typing.Optional[typing.Sequence[str]] = OMIT,
-        model: typing.Optional[str] = OMIT,
-        documents: typing.Optional[typing.Sequence[str]] = OMIT,
-        config: typing.Dict[str, typing.Any],
-        additional_context: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> Message:
-        """
-        Send a message to Athena.
-        Returns an Athena message with ID, content and final output.
-
-        Parameters:
-            - content: str.
-
-            - conversation_id: typing.Optional[str].
-
-            - user_id: typing.Optional[str].
-
-            - workspace_id: typing.Optional[str].
-
-            - sdataset_ids: typing.Optional[typing.Sequence[str]].
-
-            - tools: typing.Optional[typing.Sequence[str]].
-
-            - model: typing.Optional[str].
-
-            - documents: typing.Optional[typing.Sequence[str]].
-
-            - config: typing.Dict[str, typing.Any].
-
-            - additional_context: typing.Optional[str].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        """
-        _request: typing.Dict[str, typing.Any] = {"content": content, "config": config}
-        if conversation_id is not OMIT:
-            _request["conversation_id"] = conversation_id
-        if user_id is not OMIT:
-            _request["user_id"] = user_id
-        if workspace_id is not OMIT:
-            _request["workspace_id"] = workspace_id
-        if sdataset_ids is not OMIT:
-            _request["sdataset_ids"] = sdataset_ids
-        if tools is not OMIT:
-            _request["tools"] = tools
-        if model is not OMIT:
-            _request["model"] = model
-        if documents is not OMIT:
-            _request["documents"] = documents
-        if additional_context is not OMIT:
-            _request["additional_context"] = additional_context
-        _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/agent"),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else 60,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(Message, _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 get(
-        self, message_id: typing.Optional[str], *, request_options: typing.Optional[RequestOptions] = None
-    ) -> Message:
-        """
-        Get an Athena message by ID.
-        Returns an Athena message final output.
-
-        Parameters:
-            - message_id: typing.Optional[str].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"api/agent/{jsonable_encoder(message_id)}"
-            ),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else 60,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(Message, _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 send(
-        self,
-        *,
-        content: str,
-        conversation_id: typing.Optional[str] = OMIT,
-        user_id: typing.Optional[str] = OMIT,
-        workspace_id: typing.Optional[str] = OMIT,
-        sdataset_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        tools: typing.Optional[typing.Sequence[str]] = OMIT,
-        model: typing.Optional[str] = OMIT,
-        documents: typing.Optional[typing.Sequence[str]] = OMIT,
-        config: typing.Dict[str, typing.Any],
-        additional_context: typing.Optional[str] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> Message:
-        """
-        Send a message to Athena.
-        Returns an Athena message with ID, content and final output.
-
-        Parameters:
-            - content: str.
-
-            - conversation_id: typing.Optional[str].
-
-            - user_id: typing.Optional[str].
-
-            - workspace_id: typing.Optional[str].
-
-            - sdataset_ids: typing.Optional[typing.Sequence[str]].
-
-            - tools: typing.Optional[typing.Sequence[str]].
-
-            - model: typing.Optional[str].
-
-            - documents: typing.Optional[typing.Sequence[str]].
-
-            - config: typing.Dict[str, typing.Any].
-
-            - additional_context: typing.Optional[str].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        """
-        _request: typing.Dict[str, typing.Any] = {"content": content, "config": config}
-        if conversation_id is not OMIT:
-            _request["conversation_id"] = conversation_id
-        if user_id is not OMIT:
-            _request["user_id"] = user_id
-        if workspace_id is not OMIT:
-            _request["workspace_id"] = workspace_id
-        if sdataset_ids is not OMIT:
-            _request["sdataset_ids"] = sdataset_ids
-        if tools is not OMIT:
-            _request["tools"] = tools
-        if model is not OMIT:
-            _request["model"] = model
-        if documents is not OMIT:
-            _request["documents"] = documents
-        if additional_context is not OMIT:
-            _request["additional_context"] = additional_context
-        _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/agent"),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else 60,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(Message, _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/resources/message/types/__init__.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from .message import Message
-
-__all__ = ["Message"]

athena_intelligence-0.1.3.dist-info/RECORD

@@ -1,20 +0,0 @@
-athena/__init__.py,sha256=AGJSNJ3GjJguxtGkkfaiypZ3yFgm-JFQFWa47SDcbx8,203
-athena/client.py,sha256=0KtesyhdKJAMEvbVsYYntHh3pV-6137rFyrmSDq6DBk,4676
-athena/core/__init__.py,sha256=95onSWXymaL0iV-nBsuKNgjh1LbsymAkRkx7POn1nc0,696
-athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=Jn4izMkQlUJcK9QTyxFSx58dAHkplqU92Yko3ggQCc8,1421
-athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
-athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
-athena/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
-athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
-athena/core/request_options.py,sha256=9n3ZHWZkw60MGp7GbdjGEuGo73Pa-6JeusRGhS533aA,1297
-athena/environment.py,sha256=JjPAuJkoJ7KTRtol_3WGiZt7_azy6JygWnew89L36C4,168
-athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-athena/resources/__init__.py,sha256=JM7cJIx5wLNRx4rmJyQ7ZbiuAfVOp1LT-x2W5IVURUU,150
-athena/resources/message/__init__.py,sha256=E3xv7UtdbvniqZDUXBOov2pICgBfdeHKZ7RRDseNXKU,115
-athena/resources/message/client.py,sha256=0GVx6AmrXY4tvwfUBLQb9f04kNT1FOYYCw3E84asd8M,11893
-athena/resources/message/types/__init__.py,sha256=vMTc7TB6uvZ64bGGxdYAJx_Txj4naeghaVLRasHPKEc,117
-athena/resources/message/types/message.py,sha256=2FDVoG84ckvC4_v_uGol_354fmSyUBXTdCYelDiO5r4,1037
-athena_intelligence-0.1.3.dist-info/METADATA,sha256=xb1xEKg7T_IIimogxPV24mlnZnzVGW3gKuQmP9rDl0s,3741
-athena_intelligence-0.1.3.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.3.dist-info/RECORD,,