athena-intelligence 0.1.3__tar.gz → 0.1.5__tar.gz

{athena_intelligence-0.1.3 → athena_intelligence-0.1.5}/PKG-INFO

@@ -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.3 → athena_intelligence-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.3"
+version = "0.1.5"
 description = ""
 readme = "README.md"
 authors = []

athena_intelligence-0.1.5/src/athena/__init__.py

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

athena_intelligence-0.1.5/src/athena/client.py

@@ -0,0 +1,419 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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 .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:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
+
+    Parameters:
+        - base_url: typing.Optional[str]. The base url to use for requests from the client.
+
+        - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
+
+                                          Defaults to AthenaEnvironment.DEFAULT
+
+        - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
+
+        - httpx_client: typing.Optional[httpx.Client]. The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+    ---
+    from athena.client import Athena
+
+    client = Athena()
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        timeout: typing.Optional[float] = 60,
+        httpx_client: typing.Optional[httpx.Client] = None,
+    ):
+        self._client_wrapper = SyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            httpx_client=httpx.Client(timeout=timeout) if httpx_client is None else httpx_client,
+        )
+
+    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:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
+
+    Parameters:
+        - base_url: typing.Optional[str]. The base url to use for requests from the client.
+
+        - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
+
+                                          Defaults to AthenaEnvironment.DEFAULT
+
+        - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
+
+        - httpx_client: typing.Optional[httpx.AsyncClient]. The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+    ---
+    from athena.client import AsyncAthena
+
+    client = AsyncAthena()
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        timeout: typing.Optional[float] = 60,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+    ):
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            httpx_client=httpx.AsyncClient(timeout=timeout) if httpx_client is None else httpx_client,
+        )
+
+    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:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

athena_intelligence-0.1.5/src/athena/core/client_wrapper.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import httpx
+
+
+class BaseClientWrapper:
+    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.5",
+        }
+        return headers
+
+    def get_base_url(self) -> str:
+        return self._base_url
+
+
+class SyncClientWrapper(BaseClientWrapper):
+    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, *, base_url: str, httpx_client: httpx.AsyncClient):
+        super().__init__(base_url=base_url)
+        self.httpx_client = httpx_client

{athena_intelligence-0.1.3 → athena_intelligence-0.1.5}/src/athena/environment.py

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

athena_intelligence-0.1.5/src/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_intelligence-0.1.5/src/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_intelligence-0.1.5/src/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_intelligence-0.1.5/src/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_intelligence-0.1.5/src/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_intelligence-0.1.3/src/athena/resources/message/types/message.py → athena_intelligence-0.1.5/src/athena/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_intelligence-0.1.5/src/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_intelligence-0.1.5/src/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_intelligence-0.1.5/src/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_intelligence-0.1.3/src/athena/resources/message/__init__.py → athena_intelligence-0.1.5/src/athena/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/src/athena/__init__.py

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

athena_intelligence-0.1.3/src/athena/client.py

@@ -1,110 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import os
-import typing
-
-import httpx
-
-from .core.api_error import ApiError
-from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from .environment import AthenaEnvironment
-from .resources.message.client import AsyncMessageClient, MessageClient
-
-
-class Athena:
-    """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
-
-    Parameters:
-        - base_url: typing.Optional[str]. The base url to use for requests from the client.
-
-        - 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]]].
-
-        - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
-
-        - httpx_client: typing.Optional[httpx.Client]. The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
-    ---
-    from athena.client import Athena
-
-    client = Athena(
-        api_key="YOUR_API_KEY",
-    )
-    """
-
-    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"),
-        timeout: typing.Optional[float] = 60,
-        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)
-
-
-class AsyncAthena:
-    """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
-
-    Parameters:
-        - base_url: typing.Optional[str]. The base url to use for requests from the client.
-
-        - 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]]].
-
-        - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
-
-        - httpx_client: typing.Optional[httpx.AsyncClient]. The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
-    ---
-    from athena.client import AsyncAthena
-
-    client = AsyncAthena(
-        api_key="YOUR_API_KEY",
-    )
-    """
-
-    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"),
-        timeout: typing.Optional[float] = 60,
-        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)
-
-
-def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:
-    if base_url is not None:
-        return base_url
-    elif environment is not None:
-        return environment.value
-    else:
-        raise Exception("Please pass in either base_url or environment to construct the client")

athena_intelligence-0.1.3/src/athena/core/client_wrapper.py

@@ -1,45 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-import httpx
-
-
-class BaseClientWrapper:
-    def __init__(self, *, api_key: typing.Union[str, typing.Callable[[], str]], base_url: str):
-        self._api_key = api_key
-        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",
-        }
-        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)
-        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)
-        self.httpx_client = httpx_client

athena_intelligence-0.1.3/src/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_intelligence-0.1.3/src/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_intelligence-0.1.3/src/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"]