athena-intelligence 0.1.118__tar.gz → 0.1.119__tar.gz

{athena_intelligence-0.1.118 → athena_intelligence-0.1.119}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.118
+Version: 0.1.119
 Summary: Athena Intelligence Python Library
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{athena_intelligence-0.1.118 → athena_intelligence-0.1.119}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.118"
+version = "0.1.119"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.118 → athena_intelligence-0.1.119}/src/athena/__init__.py

@@ -2,6 +2,10 @@
 
 from .types import (
     AssetNotFoundError,
+    AthenaAssistantConfig,
+    AthenaAssistantConfigEnabledToolsItem,
+    AthenaAssistantReponse,
+    AthenaAssistantRequest,
     DataFrameRequestOut,
     DataFrameRequestOutColumnsItem,
     DataFrameRequestOutDataItemItem,
@@ -12,6 +16,8 @@ from .types import (
     FileTooLargeError,
     ParentFolderError,
     SaveAssetRequestOut,
+    StructuredDataExtractorReponse,
+    Tool,
 )
 from .errors import (
     BadRequestError,
@@ -22,7 +28,7 @@ from .errors import (
     UnprocessableEntityError,
     UnsupportedMediaTypeError,
 )
-from . import query, tools
+from . import agents, query, tools
 from .environment import AthenaEnvironment
 from .query import QueryExecuteRequestDatabaseAssetIds
 from .tools import ToolsDataFrameRequestColumnsItem
@@ -30,6 +36,10 @@ from .version import __version__
 
 __all__ = [
     "AssetNotFoundError",
+    "AthenaAssistantConfig",
+    "AthenaAssistantConfigEnabledToolsItem",
+    "AthenaAssistantReponse",
+    "AthenaAssistantRequest",
     "AthenaEnvironment",
     "BadRequestError",
     "ContentTooLargeError",
@@ -46,11 +56,14 @@ __all__ = [
     "ParentFolderError",
     "QueryExecuteRequestDatabaseAssetIds",
     "SaveAssetRequestOut",
+    "StructuredDataExtractorReponse",
+    "Tool",
     "ToolsDataFrameRequestColumnsItem",
     "UnauthorizedError",
     "UnprocessableEntityError",
     "UnsupportedMediaTypeError",
     "__version__",
+    "agents",
     "query",
     "tools",
 ]

athena_intelligence-0.1.119/src/athena/agents/__init__.py

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

athena_intelligence-0.1.119/src/athena/agents/athena_assistant/__init__.py

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

athena_intelligence-0.1.119/src/athena/agents/athena_assistant/client.py

@@ -0,0 +1,353 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.athena_assistant_reponse import AthenaAssistantReponse
+from ...types.athena_assistant_request import AthenaAssistantRequest
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class AthenaAssistantClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def batch(
+        self,
+        *,
+        request: typing.Sequence[AthenaAssistantRequest],
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[AthenaAssistantReponse]:
+        """
+        Call the Athena assistant with batched requests and return the results.
+
+        Parameters
+        ----------
+        request : typing.Sequence[AthenaAssistantRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[AthenaAssistantReponse]
+            Successful Response
+
+        Examples
+        --------
+        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.athena_assistant.batch(
+            request=[
+                AthenaAssistantRequest(
+                    config=AthenaAssistantConfig(
+                        enabled_tools=[Tool.SEARCH],
+                    ),
+                    messages=[],
+                )
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/athena-assistant/batch",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.List[AthenaAssistantReponse], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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 invoke(
+        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> AthenaAssistantReponse:
+        """
+        Call the Athena Assistant agent synchronously.
+
+        Call the agent with the messages list, wait for the agent to complete,
+        and return the result.
+
+        Parameters
+        ----------
+        request : AthenaAssistantRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AthenaAssistantReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.athena_assistant.invoke(
+            request=AthenaAssistantRequest(
+                config=AthenaAssistantConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/athena-assistant/invoke",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(AthenaAssistantReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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 stream_events(
+        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> AthenaAssistantReponse:
+        """
+        Call the Athena assistant and stream events.
+
+        Parameters
+        ----------
+        request : AthenaAssistantRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AthenaAssistantReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.athena_assistant.stream_events(
+            request=AthenaAssistantRequest(
+                config=AthenaAssistantConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/athena-assistant/stream_events",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(AthenaAssistantReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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)
+
+
+class AsyncAthenaAssistantClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def batch(
+        self,
+        *,
+        request: typing.Sequence[AthenaAssistantRequest],
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[AthenaAssistantReponse]:
+        """
+        Call the Athena assistant with batched requests and return the results.
+
+        Parameters
+        ----------
+        request : typing.Sequence[AthenaAssistantRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[AthenaAssistantReponse]
+            Successful Response
+
+        Examples
+        --------
+        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.athena_assistant.batch(
+            request=[
+                AthenaAssistantRequest(
+                    config=AthenaAssistantConfig(
+                        enabled_tools=[Tool.SEARCH],
+                    ),
+                    messages=[],
+                )
+            ],
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/athena-assistant/batch",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.List[AthenaAssistantReponse], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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 invoke(
+        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> AthenaAssistantReponse:
+        """
+        Call the Athena Assistant agent synchronously.
+
+        Call the agent with the messages list, wait for the agent to complete,
+        and return the result.
+
+        Parameters
+        ----------
+        request : AthenaAssistantRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AthenaAssistantReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.athena_assistant.invoke(
+            request=AthenaAssistantRequest(
+                config=AthenaAssistantConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[],
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/athena-assistant/invoke",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(AthenaAssistantReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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 stream_events(
+        self, *, request: AthenaAssistantRequest, request_options: typing.Optional[RequestOptions] = None
+    ) -> AthenaAssistantReponse:
+        """
+        Call the Athena assistant and stream events.
+
+        Parameters
+        ----------
+        request : AthenaAssistantRequest
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AthenaAssistantReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import AthenaAssistantConfig, AthenaAssistantRequest, Tool
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.athena_assistant.stream_events(
+            request=AthenaAssistantRequest(
+                config=AthenaAssistantConfig(
+                    enabled_tools=[Tool.SEARCH],
+                ),
+                messages=[],
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/athena-assistant/stream_events",
+            method="POST",
+            json=request,
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(AthenaAssistantReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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)

athena_intelligence-0.1.119/src/athena/agents/client.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .athena_assistant.client import AsyncAthenaAssistantClient, AthenaAssistantClient
+from .structured_data_extractor.client import AsyncStructuredDataExtractorClient, StructuredDataExtractorClient
+
+
+class AgentsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+        self.athena_assistant = AthenaAssistantClient(client_wrapper=self._client_wrapper)
+        self.structured_data_extractor = StructuredDataExtractorClient(client_wrapper=self._client_wrapper)
+
+
+class AsyncAgentsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+        self.athena_assistant = AsyncAthenaAssistantClient(client_wrapper=self._client_wrapper)
+        self.structured_data_extractor = AsyncStructuredDataExtractorClient(client_wrapper=self._client_wrapper)

athena_intelligence-0.1.119/src/athena/agents/structured_data_extractor/__init__.py

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

athena_intelligence-0.1.119/src/athena/agents/structured_data_extractor/client.py

@@ -0,0 +1,150 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.structured_data_extractor_reponse import StructuredDataExtractorReponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class StructuredDataExtractorClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def invoke(
+        self,
+        *,
+        asset_ids: typing.Sequence[str],
+        json_schema: typing.Dict[str, typing.Any],
+        map_: typing.Optional[bool] = OMIT,
+        reduce: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> StructuredDataExtractorReponse:
+        """
+        Coming soon! Extract structured data from assets of arbitrary length.
+
+        Parameters
+        ----------
+        asset_ids : typing.Sequence[str]
+            The IDs of the assets from which to extract structured data matching `json_schema`.
+
+        json_schema : typing.Dict[str, typing.Any]
+            The JSON schema to use for validation.
+
+        map_ : typing.Optional[bool]
+            Whether to split the asset into chunks and attempt to extract the schema from each chunk.  Set to false if you know the asset is small.
+
+        reduce : typing.Optional[bool]
+            If `map`, whether to reduce the chunks to a single structured object (true) or return the full list (false).  Use True unless you want to preserve duplicates from each page or expect the object to overflow the output context.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        StructuredDataExtractorReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.agents.structured_data_extractor.invoke(
+            asset_ids=["asset_ids"],
+            json_schema={"key": "value"},
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/agents/structured-data-extractor/invoke",
+            method="POST",
+            json={"asset_ids": asset_ids, "json_schema": json_schema, "map": map_, "reduce": reduce},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(StructuredDataExtractorReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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)
+
+
+class AsyncStructuredDataExtractorClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def invoke(
+        self,
+        *,
+        asset_ids: typing.Sequence[str],
+        json_schema: typing.Dict[str, typing.Any],
+        map_: typing.Optional[bool] = OMIT,
+        reduce: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> StructuredDataExtractorReponse:
+        """
+        Coming soon! Extract structured data from assets of arbitrary length.
+
+        Parameters
+        ----------
+        asset_ids : typing.Sequence[str]
+            The IDs of the assets from which to extract structured data matching `json_schema`.
+
+        json_schema : typing.Dict[str, typing.Any]
+            The JSON schema to use for validation.
+
+        map_ : typing.Optional[bool]
+            Whether to split the asset into chunks and attempt to extract the schema from each chunk.  Set to false if you know the asset is small.
+
+        reduce : typing.Optional[bool]
+            If `map`, whether to reduce the chunks to a single structured object (true) or return the full list (false).  Use True unless you want to preserve duplicates from each page or expect the object to overflow the output context.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        StructuredDataExtractorReponse
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.agents.structured_data_extractor.invoke(
+            asset_ids=["asset_ids"],
+            json_schema={"key": "value"},
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/agents/structured-data-extractor/invoke",
+            method="POST",
+            json={"asset_ids": asset_ids, "json_schema": json_schema, "map": map_, "reduce": reduce},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(StructuredDataExtractorReponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.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)

{athena_intelligence-0.1.118 → athena_intelligence-0.1.119}/src/athena/base_client.py

@@ -4,6 +4,7 @@ import typing
 
 import httpx
 
+from .agents.client import AgentsClient, AsyncAgentsClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import AthenaEnvironment
 from .query.client import AsyncQueryClient, QueryClient
@@ -68,6 +69,7 @@ class BaseAthena:
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.agents = AgentsClient(client_wrapper=self._client_wrapper)
         self.query = QueryClient(client_wrapper=self._client_wrapper)
         self.tools = ToolsClient(client_wrapper=self._client_wrapper)
 
@@ -130,6 +132,7 @@ class AsyncBaseAthena:
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
         self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
         self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
 

{athena_intelligence-0.1.118 → athena_intelligence-0.1.119}/src/athena/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.118",
+            "X-Fern-SDK-Version": "0.1.119",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.118 → athena_intelligence-0.1.119}/src/athena/types/__init__.py

@@ -1,6 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .asset_not_found_error import AssetNotFoundError
+from .athena_assistant_config import AthenaAssistantConfig
+from .athena_assistant_config_enabled_tools_item import AthenaAssistantConfigEnabledToolsItem
+from .athena_assistant_reponse import AthenaAssistantReponse
+from .athena_assistant_request import AthenaAssistantRequest
 from .data_frame_request_out import DataFrameRequestOut
 from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
 from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
@@ -11,9 +15,15 @@ from .file_chunk_request_out import FileChunkRequestOut
 from .file_too_large_error import FileTooLargeError
 from .parent_folder_error import ParentFolderError
 from .save_asset_request_out import SaveAssetRequestOut
+from .structured_data_extractor_reponse import StructuredDataExtractorReponse
+from .tool import Tool
 
 __all__ = [
     "AssetNotFoundError",
+    "AthenaAssistantConfig",
+    "AthenaAssistantConfigEnabledToolsItem",
+    "AthenaAssistantReponse",
+    "AthenaAssistantRequest",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
@@ -24,4 +34,6 @@ __all__ = [
     "FileTooLargeError",
     "ParentFolderError",
     "SaveAssetRequestOut",
+    "StructuredDataExtractorReponse",
+    "Tool",
 ]

athena_intelligence-0.1.119/src/athena/types/athena_assistant_config.py

@@ -0,0 +1,36 @@
+# 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 ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .athena_assistant_config_enabled_tools_item import AthenaAssistantConfigEnabledToolsItem
+
+
+class AthenaAssistantConfig(pydantic_v1.BaseModel):
+    """
+    Configurable fields for the general agent.
+    """
+
+    enabled_tools: typing.Optional[typing.List[AthenaAssistantConfigEnabledToolsItem]] = None
+    knowledge_base_asset_ids: typing.Optional[typing.List[str]] = None
+    system_prompt: typing.Optional[str] = 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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.119/src/athena/types/athena_assistant_config_enabled_tools_item.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .tool import Tool
+
+AthenaAssistantConfigEnabledToolsItem = typing.Union[Tool, str]

athena_intelligence-0.1.119/src/athena/types/athena_assistant_reponse.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
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class AthenaAssistantReponse(pydantic_v1.BaseModel):
+    """
+    The response from the assistant.
+    """
+
+    messages: typing.List[typing.Dict[str, typing.Any]]
+
+    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.119/src/athena/types/athena_assistant_request.py

@@ -0,0 +1,38 @@
+# 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 ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+from .athena_assistant_config import AthenaAssistantConfig
+
+
+class AthenaAssistantRequest(pydantic_v1.BaseModel):
+    """
+    A chat request for the Athena SDK.
+    """
+
+    config: AthenaAssistantConfig
+    messages: typing.List[typing.Any] = pydantic_v1.Field()
+    """
+    The messages to send to the assistant. Each message should be a string (for text inputs) or a list of multimodal content parts.
+    """
+
+    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.119/src/athena/types/structured_data_extractor_reponse.py

@@ -0,0 +1,36 @@
+# 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 ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class StructuredDataExtractorReponse(pydantic_v1.BaseModel):
+    """
+    The agent's response.
+    """
+
+    data: typing.Dict[str, typing.Any] = pydantic_v1.Field()
+    """
+    The extracted structured data. Guaranteed to match `json_schema`.
+    """
+
+    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.119/src/athena/types/tool.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class Tool(str, enum.Enum):
+    SEARCH = "search"
+    BROWSE = "browse"
+
+    def visit(self, search: typing.Callable[[], T_Result], browse: typing.Callable[[], T_Result]) -> T_Result:
+        if self is Tool.SEARCH:
+            return search()
+        if self is Tool.BROWSE:
+            return browse()