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

athena/__init__.py

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

athena/client.py

@@ -1,12 +1,31 @@
 # 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 .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:
@@ -18,9 +37,7 @@ class Athena:
 
         - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
 
-                                          Defaults to AthenaEnvironment.PRODUCTION
-
-        - token: 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.
 
@@ -28,26 +45,177 @@ class Athena:
     ---
     from athena.client import Athena
 
-    client = Athena(
-        token="YOUR_TOKEN",
-    )
+    client = Athena()
     """
 
     def __init__(
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
-        token: typing.Union[str, typing.Callable[[], str]],
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
         timeout: typing.Optional[float] = 60,
-        httpx_client: typing.Optional[httpx.Client] = None
+        httpx_client: typing.Optional[httpx.Client] = None,
     ):
         self._client_wrapper = SyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
-            token=token,
             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:
@@ -59,9 +227,7 @@ class AsyncAthena:
 
         - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
 
-                                          Defaults to AthenaEnvironment.PRODUCTION
-
-        - token: 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.
 
@@ -69,26 +235,179 @@ class AsyncAthena:
     ---
     from athena.client import AsyncAthena
 
-    client = AsyncAthena(
-        token="YOUR_TOKEN",
-    )
+    client = AsyncAthena()
     """
 
     def __init__(
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
-        token: typing.Union[str, typing.Callable[[], str]],
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
         timeout: typing.Optional[float] = 60,
-        httpx_client: typing.Optional[httpx.AsyncClient] = None
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
     ):
         self._client_wrapper = AsyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
-            token=token,
             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, *, token: typing.Union[str, typing.Callable[[], str]], base_url: str):
-        self._token = token
+    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.4",
+            "X-Fern-SDK-Version": "0.1.5",
         }
-        headers["Authorization"] = f"Bearer {self._get_token()}"
         return headers
 
-    def _get_token(self) -> str:
-        if isinstance(self._token, str):
-            return self._token
-        else:
-            return self._token()
-
     def get_base_url(self) -> str:
         return self._base_url
 
 
 class SyncClientWrapper(BaseClientWrapper):
-    def __init__(
-        self, *, token: typing.Union[str, typing.Callable[[], str]], base_url: str, httpx_client: httpx.Client
-    ):
-        super().__init__(token=token, 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, *, token: typing.Union[str, typing.Callable[[], str]], base_url: str, httpx_client: httpx.AsyncClient
-    ):
-        super().__init__(token=token, 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://api.athenaintelligence.ai"
+    DEFAULT = "https://api.athenaintelligence.ai"

athena/types/__init__.py

@@ -1,6 +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", "ValidationError"]
+__all__ = [
+    "HttpValidationError",
+    "MessageOut",
+    "MessageOutDto",
+    "Model",
+    "Tools",
+    "ValidationError",
+    "ValidationErrorLocItem",
+]

athena/types/http_validation_error.py

@@ -13,7 +13,7 @@ except ImportError:
 
 
 class HttpValidationError(pydantic.BaseModel):
-    detail: typing.List[ValidationError]
+    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}

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/types/message_out_dto.py

@@ -0,0 +1,33 @@
+# 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 MessageOutDto(pydantic.BaseModel):
+    id: str
+    conversation_id: str
+    logs: typing.Optional[str] = None
+    content: typing.Optional[str] = None
+    created_at: str
+    status: 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/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

@@ -4,6 +4,7 @@ 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
@@ -12,9 +13,9 @@ except ImportError:
 
 
 class ValidationError(pydantic.BaseModel):
-    loc: typing.Optional[typing.List[str]] = None
-    msg: typing.Optional[str] = None
-    type: typing.Optional[str] = None
+    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}

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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.4
+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/message/__init__.py

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

athena/resources/message/client.py

@@ -1,329 +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 ...errors.unprocessable_entity_error import UnprocessableEntityError
-from ...types.http_validation_error import HttpValidationError
-
-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: str, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
-        """
-        Parameters:
-            - message_id: str. The ID of the message to retrieve
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena.client import Athena
-
-        client = Athena(
-            token="YOUR_TOKEN",
-        )
-        client.message.get(
-            message_id="message_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/auth/athena-get-message"),
-            params=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        "message_id": message_id,
-                        **(
-                            request_options.get("additional_query_parameters", {})
-                            if request_options is not None
-                            else {}
-                        ),
-                    }
-                )
-            ),
-            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
-        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 submit(
-        self,
-        *,
-        additional_context: typing.Optional[str] = OMIT,
-        config: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
-        content: str,
-        conversation_id: typing.Optional[str] = OMIT,
-        conversation_name: typing.Optional[str] = OMIT,
-        documents: typing.Optional[typing.Sequence[str]] = OMIT,
-        model: typing.Optional[str] = OMIT,
-        sdataset_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        tools: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Any:
-        """
-        Parameters:
-            - additional_context: typing.Optional[str].
-
-            - config: typing.Optional[typing.Dict[str, typing.Any]].
-
-            - content: str.
-
-            - conversation_id: typing.Optional[str].
-
-            - conversation_name: typing.Optional[str].
-
-            - documents: typing.Optional[typing.Sequence[str]].
-
-            - model: typing.Optional[str].
-
-            - sdataset_ids: typing.Optional[typing.Sequence[str]].
-
-            - tools: typing.Optional[typing.Sequence[str]].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena.client import Athena
-
-        client = Athena(
-            token="YOUR_TOKEN",
-        )
-        client.message.submit(
-            content="content",
-        )
-        """
-        _request: typing.Dict[str, typing.Any] = {"content": content}
-        if additional_context is not OMIT:
-            _request["additional_context"] = additional_context
-        if config is not OMIT:
-            _request["config"] = config
-        if conversation_id is not OMIT:
-            _request["conversation_id"] = conversation_id
-        if conversation_name is not OMIT:
-            _request["conversation_name"] = conversation_name
-        if documents is not OMIT:
-            _request["documents"] = documents
-        if model is not OMIT:
-            _request["model"] = model
-        if sdataset_ids is not OMIT:
-            _request["sdataset_ids"] = sdataset_ids
-        if tools is not OMIT:
-            _request["tools"] = tools
-        _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/auth/athena-submit-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(typing.Any, _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 AsyncMessageClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def get(self, *, message_id: str, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
-        """
-        Parameters:
-            - message_id: str. The ID of the message to retrieve
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            token="YOUR_TOKEN",
-        )
-        await client.message.get(
-            message_id="message_id",
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/auth/athena-get-message"),
-            params=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        "message_id": message_id,
-                        **(
-                            request_options.get("additional_query_parameters", {})
-                            if request_options is not None
-                            else {}
-                        ),
-                    }
-                )
-            ),
-            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
-        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 submit(
-        self,
-        *,
-        additional_context: typing.Optional[str] = OMIT,
-        config: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
-        content: str,
-        conversation_id: typing.Optional[str] = OMIT,
-        conversation_name: typing.Optional[str] = OMIT,
-        documents: typing.Optional[typing.Sequence[str]] = OMIT,
-        model: typing.Optional[str] = OMIT,
-        sdataset_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        tools: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Any:
-        """
-        Parameters:
-            - additional_context: typing.Optional[str].
-
-            - config: typing.Optional[typing.Dict[str, typing.Any]].
-
-            - content: str.
-
-            - conversation_id: typing.Optional[str].
-
-            - conversation_name: typing.Optional[str].
-
-            - documents: typing.Optional[typing.Sequence[str]].
-
-            - model: typing.Optional[str].
-
-            - sdataset_ids: typing.Optional[typing.Sequence[str]].
-
-            - tools: typing.Optional[typing.Sequence[str]].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            token="YOUR_TOKEN",
-        )
-        await client.message.submit(
-            content="content",
-        )
-        """
-        _request: typing.Dict[str, typing.Any] = {"content": content}
-        if additional_context is not OMIT:
-            _request["additional_context"] = additional_context
-        if config is not OMIT:
-            _request["config"] = config
-        if conversation_id is not OMIT:
-            _request["conversation_id"] = conversation_id
-        if conversation_name is not OMIT:
-            _request["conversation_name"] = conversation_name
-        if documents is not OMIT:
-            _request["documents"] = documents
-        if model is not OMIT:
-            _request["model"] = model
-        if sdataset_ids is not OMIT:
-            _request["sdataset_ids"] = sdataset_ids
-        if tools is not OMIT:
-            _request["tools"] = tools
-        _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/auth/athena-submit-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(typing.Any, _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)

athena_intelligence-0.1.4.dist-info/RECORD

@@ -1,23 +0,0 @@
-athena/__init__.py,sha256=mM2ORuc8lE5dQBWDvOVv4pYwGC9-v5BKks3LGTssVHY,354
-athena/client.py,sha256=yZ1H80Vp8fSf1-0poxJCoK0cnjsQGIHM0C1ucKuQCPM,4117
-athena/core/__init__.py,sha256=95onSWXymaL0iV-nBsuKNgjh1LbsymAkRkx7POn1nc0,696
-athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=tQkKCpqMT4JcxxkLHSeLlt3NW2HRgMGcrJJv5iSqXW8,1393
-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=5-uIjSP-twha9NxudzbrBcoevnUcQmsG_JCe9pLAji4,168
-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/resources/__init__.py,sha256=H2Sg5QOlqjAbAl_ADBNYPqFArb-9plPdXDPVdeIkXDk,110
-athena/resources/message/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/resources/message/client.py,sha256=pZwHlwLyhOyYi2XLyBnf3z02YspFYAHnRx9ojqcwvAE,13485
-athena/types/__init__.py,sha256=h85tlX9zfEuZsJZVvtfTV801zX-hNyahVmKx2rpKf28,220
-athena/types/http_validation_error.py,sha256=jyYYic0XAUCV7cw-E8cqCysIbJ8citJ48Jk6UAR3Z18,949
-athena/types/validation_error.py,sha256=U1eCkenyCs3oULwpUi3T2DGaXO7py0hyuY99H1Bh36Q,983
-athena_intelligence-0.1.4.dist-info/METADATA,sha256=c4RgWOrKuVFFSs-Jr0gz5jxe46bonTsVmw0QZJr-OzE,3741
-athena_intelligence-0.1.4.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.4.dist-info/RECORD,,