athena-intelligence 0.1.77__tar.gz → 0.1.79__tar.gz

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/PKG-INFO

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

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/pyproject.toml

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

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/__init__.py

@@ -2,10 +2,16 @@
 
 from .types import (
     ConvertPdfToSheetOut,
+    DataFrameParsingError,
+    DataFrameRequestOut,
+    DataFrameRequestOutColumnsItem,
+    DataFrameRequestOutDataItemItem,
+    DataFrameRequestOutIndexItem,
     Dataset,
     Document,
     ExcecuteToolFirstWorkflowOut,
     FileDataResponse,
+    FileFetchError,
     FilterModel,
     FilterOperator,
     FirecrawlScrapeUrlDataReponseDto,
@@ -37,18 +43,25 @@ from .types import (
     ValidationErrorLocItem,
     WorkflowStatusOut,
 )
-from .errors import UnprocessableEntityError
+from .errors import InternalServerError, NotFoundError, UnprocessableEntityError
 from . import chain, dataset, message, query, report, search, snippet, tools, upload, workflow
 from .environment import AthenaEnvironment
+from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
     "AthenaEnvironment",
     "ConvertPdfToSheetOut",
+    "DataFrameParsingError",
+    "DataFrameRequestOut",
+    "DataFrameRequestOutColumnsItem",
+    "DataFrameRequestOutDataItemItem",
+    "DataFrameRequestOutIndexItem",
     "Dataset",
     "Document",
     "ExcecuteToolFirstWorkflowOut",
     "FileDataResponse",
+    "FileFetchError",
     "FilterModel",
     "FilterOperator",
     "FirecrawlScrapeUrlDataReponseDto",
@@ -57,12 +70,14 @@ __all__ = [
     "GetSnippetOut",
     "GetSnippetsResponse",
     "HttpValidationError",
+    "InternalServerError",
     "LangchainDocumentsRequestOut",
     "LlmModel",
     "MapReduceChainOut",
     "MessageOut",
     "MessageOutDto",
     "Model",
+    "NotFoundError",
     "PublishFormats",
     "QueryModel",
     "Report",
@@ -74,6 +89,7 @@ __all__ = [
     "StructuredParseResult",
     "TimeDimensionModel",
     "Tools",
+    "ToolsDataFrameRequestColumnsItem",
     "UnprocessableEntityError",
     "UploadDocumentsOut",
     "UrlResult",

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/chain/client.py

@@ -50,7 +50,7 @@ class ChainClient:
             api_key="YOUR_API_KEY",
         )
         client.chain.structured_parse(
-            text_input='Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows \n                by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot, \n                allowing you to hand over controls to her for autonomous execution with confidence." \n                \n                Give me all of the modes Athena provides.',
+            text_input='Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows\n                by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot,\n                allowing you to hand over controls to her for autonomous execution with confidence."\n\n                Give me all of the modes Athena provides.',
             custom_type_dict={"modes": {}},
             model=LlmModel.GPT_4_TURBO,
         )
@@ -219,7 +219,7 @@ class AsyncChainClient:
             api_key="YOUR_API_KEY",
         )
         await client.chain.structured_parse(
-            text_input='Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows \n                by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot, \n                allowing you to hand over controls to her for autonomous execution with confidence." \n                \n                Give me all of the modes Athena provides.',
+            text_input='Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows\n                by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot,\n                allowing you to hand over controls to her for autonomous execution with confidence."\n\n                Give me all of the modes Athena provides.',
             custom_type_dict={"modes": {}},
             model=LlmModel.GPT_4_TURBO,
         )

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/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.77",
+            "X-Fern-SDK-Version": "0.1.79",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena_intelligence-0.1.79/src/athena/errors/__init__.py

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

athena_intelligence-0.1.79/src/athena/errors/internal_server_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.data_frame_parsing_error import DataFrameParsingError
+
+
+class InternalServerError(ApiError):
+    def __init__(self, body: DataFrameParsingError):
+        super().__init__(status_code=500, body=body)

athena_intelligence-0.1.79/src/athena/errors/not_found_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.file_fetch_error import FileFetchError
+
+
+class NotFoundError(ApiError):
+    def __init__(self, body: FileFetchError):
+        super().__init__(status_code=404, body=body)

athena_intelligence-0.1.79/src/athena/tools/__init__.py

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

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/tools/client.py

@@ -10,9 +10,14 @@ from ..core.jsonable_encoder import jsonable_encoder
 from ..core.pydantic_utilities import pydantic_v1
 from ..core.remove_none_from_dict import remove_none_from_dict
 from ..core.request_options import RequestOptions
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.convert_pdf_to_sheet_out import ConvertPdfToSheetOut
+from ..types.data_frame_parsing_error import DataFrameParsingError
+from ..types.data_frame_request_out import DataFrameRequestOut
 from ..types.excecute_tool_first_workflow_out import ExcecuteToolFirstWorkflowOut
+from ..types.file_fetch_error import FileFetchError
 from ..types.firecrawl_scrape_url_data_reponse_dto import FirecrawlScrapeUrlDataReponseDto
 from ..types.http_validation_error import HttpValidationError
 from ..types.langchain_documents_request_out import LangchainDocumentsRequestOut
@@ -21,6 +26,7 @@ from ..types.publish_formats import PublishFormats
 from ..types.query_model import QueryModel
 from ..types.researcher_out import ResearcherOut
 from ..types.semantic_query_out import SemanticQueryOut
+from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -51,7 +57,7 @@ class ToolsClient:
             api_key="YOUR_API_KEY",
         )
         client.tools.scrape_url(
-            url="https://athenaintelligence.ai",
+            url="https://www.athenaintelligence.ai",
         )
         """
         _request: typing.Dict[str, typing.Any] = {"url": url}
@@ -169,6 +175,153 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def data_frame(
+        self,
+        *,
+        document_id: str,
+        row_limit: typing.Optional[int] = None,
+        index_column: typing.Optional[int] = None,
+        columns: typing.Optional[
+            typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
+        ] = None,
+        sheet_name: typing.Optional[str] = None,
+        separator: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DataFrameRequestOut:
+        """
+        Parameters:
+            - document_id: str.
+
+            - row_limit: typing.Optional[int].
+
+            - index_column: typing.Optional[int].
+
+            - columns: typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]. should be a list of strings or a list of integers
+
+            - sheet_name: typing.Optional[str]. only for excel files
+
+            - separator: typing.Optional[str]. only for csv files
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.data_frame(
+            document_id="document_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/data-frame"),
+            params=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        "document_id": document_id,
+                        "row_limit": row_limit,
+                        "index_column": index_column,
+                        "columns": columns,
+                        "sheet_name": sheet_name,
+                        "separator": separator,
+                        **(
+                            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 self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(
+                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
+            )
+        if _response.status_code == 500:
+            raise InternalServerError(pydantic_v1.parse_obj_as(DataFrameParsingError, _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 raw_data(self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Parameters:
+            - document_id: str.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.raw_data(
+            document_id="document_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/raw-data"),
+            params=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        "document_id": document_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 self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(
+                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
+            )
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def tool_first_workflow(
         self,
         *,
@@ -510,7 +663,7 @@ class AsyncToolsClient:
             api_key="YOUR_API_KEY",
         )
         await client.tools.scrape_url(
-            url="https://athenaintelligence.ai",
+            url="https://www.athenaintelligence.ai",
         )
         """
         _request: typing.Dict[str, typing.Any] = {"url": url}
@@ -628,6 +781,155 @@ class AsyncToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def data_frame(
+        self,
+        *,
+        document_id: str,
+        row_limit: typing.Optional[int] = None,
+        index_column: typing.Optional[int] = None,
+        columns: typing.Optional[
+            typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
+        ] = None,
+        sheet_name: typing.Optional[str] = None,
+        separator: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DataFrameRequestOut:
+        """
+        Parameters:
+            - document_id: str.
+
+            - row_limit: typing.Optional[int].
+
+            - index_column: typing.Optional[int].
+
+            - columns: typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]. should be a list of strings or a list of integers
+
+            - sheet_name: typing.Optional[str]. only for excel files
+
+            - separator: typing.Optional[str]. only for csv files
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.data_frame(
+            document_id="document_id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/data-frame"),
+            params=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        "document_id": document_id,
+                        "row_limit": row_limit,
+                        "index_column": index_column,
+                        "columns": columns,
+                        "sheet_name": sheet_name,
+                        "separator": separator,
+                        **(
+                            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 self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(
+                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
+            )
+        if _response.status_code == 500:
+            raise InternalServerError(pydantic_v1.parse_obj_as(DataFrameParsingError, _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 raw_data(
+        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Any:
+        """
+        Parameters:
+            - document_id: str.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.raw_data(
+            document_id="document_id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/raw-data"),
+            params=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        "document_id": document_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 self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(
+                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
+            )
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def tool_first_workflow(
         self,
         *,

athena_intelligence-0.1.79/src/athena/tools/types/__init__.py

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

athena_intelligence-0.1.79/src/athena/tools/types/tools_data_frame_request_columns_item.py

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

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/types/__init__.py

@@ -1,10 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .convert_pdf_to_sheet_out import ConvertPdfToSheetOut
+from .data_frame_parsing_error import DataFrameParsingError
+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
+from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
 from .dataset import Dataset
 from .document import Document
 from .excecute_tool_first_workflow_out import ExcecuteToolFirstWorkflowOut
 from .file_data_response import FileDataResponse
+from .file_fetch_error import FileFetchError
 from .filter_model import FilterModel
 from .filter_operator import FilterOperator
 from .firecrawl_scrape_url_data_reponse_dto import FirecrawlScrapeUrlDataReponseDto
@@ -38,10 +44,16 @@ from .workflow_status_out import WorkflowStatusOut
 
 __all__ = [
     "ConvertPdfToSheetOut",
+    "DataFrameParsingError",
+    "DataFrameRequestOut",
+    "DataFrameRequestOutColumnsItem",
+    "DataFrameRequestOutDataItemItem",
+    "DataFrameRequestOutIndexItem",
     "Dataset",
     "Document",
     "ExcecuteToolFirstWorkflowOut",
     "FileDataResponse",
+    "FileFetchError",
     "FilterModel",
     "FilterOperator",
     "FirecrawlScrapeUrlDataReponseDto",

athena_intelligence-0.1.79/src/athena/types/data_frame_parsing_error.py

@@ -0,0 +1,26 @@
+# 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 pydantic_v1
+
+
+class DataFrameParsingError(pydantic_v1.BaseModel):
+    message: str
+    document_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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.79/src/athena/types/data_frame_request_out.py

@@ -0,0 +1,30 @@
+# 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 pydantic_v1
+from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
+from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
+from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
+
+
+class DataFrameRequestOut(pydantic_v1.BaseModel):
+    columns: typing.List[DataFrameRequestOutColumnsItem]
+    index: typing.List[DataFrameRequestOutIndexItem]
+    data: typing.List[typing.List[DataFrameRequestOutDataItemItem]]
+
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.79/src/athena/types/data_frame_request_out_columns_item.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DataFrameRequestOutColumnsItem = typing.Union[str, int, float]

athena_intelligence-0.1.79/src/athena/types/data_frame_request_out_data_item_item.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DataFrameRequestOutDataItemItem = typing.Union[str, int, float]

athena_intelligence-0.1.79/src/athena/types/data_frame_request_out_index_item.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DataFrameRequestOutIndexItem = typing.Union[str, int, float]

athena_intelligence-0.1.79/src/athena/types/file_fetch_error.py

@@ -0,0 +1,26 @@
+# 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 pydantic_v1
+
+
+class FileFetchError(pydantic_v1.BaseModel):
+    message: str
+    status_code: typing.Optional[int] = 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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/types/llm_model.py

@@ -22,6 +22,7 @@ class LlmModel(str, enum.Enum):
     MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
     LLAMA_V_38_B_INSTRUCT = "llama-v3-8b-instruct"
     LLAMA_V_370_B_INSTRUCT = "llama-v3-70b-instruct"
+    LLAMA_V_3_P_1405_B_INSTRUCT = "llama-v3p1-405b-instruct"
     CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
     CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
     CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
@@ -30,6 +31,11 @@ class LlmModel(str, enum.Enum):
     GROQ_LLAMA_38_B_8192 = "groq-llama3-8b-8192"
     GROQ_LLAMA_370_B_8192 = "groq-llama3-70b-8192"
     GROQ_GEMMA_7_B_IT = "groq-gemma-7b-it"
+    GROQ_LLAMA_31405_B_REASONING = "groq-llama-3.1-405b-reasoning"
+    GROQ_LLAMA_3170_B_VERSATILE = "groq-llama-3.1-70b-versatile"
+    GROQ_LLAMA_318_B_INSTANT = "groq-llama-3.1-8b-instant"
+    GROQ_LLAMA_3_GROQ_70_B_8192_TOOL_USE_PREVIEW = "groq-llama3-groq-70b-8192-tool-use-preview"
+    GROQ_LLAMA_3_GROQ_8_B_8192_TOOL_USE_PREVIEW = "groq-llama3-groq-8b-8192-tool-use-preview"
     DATABRICKS_DBRX = "databricks-dbrx"
     GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
     GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
@@ -49,6 +55,7 @@ class LlmModel(str, enum.Enum):
         mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
         llama_v_38_b_instruct: typing.Callable[[], T_Result],
         llama_v_370_b_instruct: typing.Callable[[], T_Result],
+        llama_v_3_p_1405_b_instruct: typing.Callable[[], T_Result],
         claude_35_sonnet_20240620: typing.Callable[[], T_Result],
         claude_3_opus_20240229: typing.Callable[[], T_Result],
         claude_3_sonnet_20240229: typing.Callable[[], T_Result],
@@ -57,6 +64,11 @@ class LlmModel(str, enum.Enum):
         groq_llama_38_b_8192: typing.Callable[[], T_Result],
         groq_llama_370_b_8192: typing.Callable[[], T_Result],
         groq_gemma_7_b_it: typing.Callable[[], T_Result],
+        groq_llama_31405_b_reasoning: typing.Callable[[], T_Result],
+        groq_llama_3170_b_versatile: typing.Callable[[], T_Result],
+        groq_llama_318_b_instant: typing.Callable[[], T_Result],
+        groq_llama_3_groq_70_b_8192_tool_use_preview: typing.Callable[[], T_Result],
+        groq_llama_3_groq_8_b_8192_tool_use_preview: typing.Callable[[], T_Result],
         databricks_dbrx: typing.Callable[[], T_Result],
         google_gemini_10_pro_latest: typing.Callable[[], T_Result],
         google_gemini_15_pro_latest: typing.Callable[[], T_Result],
@@ -85,6 +97,8 @@ class LlmModel(str, enum.Enum):
             return llama_v_38_b_instruct()
         if self is LlmModel.LLAMA_V_370_B_INSTRUCT:
             return llama_v_370_b_instruct()
+        if self is LlmModel.LLAMA_V_3_P_1405_B_INSTRUCT:
+            return llama_v_3_p_1405_b_instruct()
         if self is LlmModel.CLAUDE_35_SONNET_20240620:
             return claude_35_sonnet_20240620()
         if self is LlmModel.CLAUDE_3_OPUS_20240229:
@@ -101,6 +115,16 @@ class LlmModel(str, enum.Enum):
             return groq_llama_370_b_8192()
         if self is LlmModel.GROQ_GEMMA_7_B_IT:
             return groq_gemma_7_b_it()
+        if self is LlmModel.GROQ_LLAMA_31405_B_REASONING:
+            return groq_llama_31405_b_reasoning()
+        if self is LlmModel.GROQ_LLAMA_3170_B_VERSATILE:
+            return groq_llama_3170_b_versatile()
+        if self is LlmModel.GROQ_LLAMA_318_B_INSTANT:
+            return groq_llama_318_b_instant()
+        if self is LlmModel.GROQ_LLAMA_3_GROQ_70_B_8192_TOOL_USE_PREVIEW:
+            return groq_llama_3_groq_70_b_8192_tool_use_preview()
+        if self is LlmModel.GROQ_LLAMA_3_GROQ_8_B_8192_TOOL_USE_PREVIEW:
+            return groq_llama_3_groq_8_b_8192_tool_use_preview()
         if self is LlmModel.DATABRICKS_DBRX:
             return databricks_dbrx()
         if self is LlmModel.GOOGLE_GEMINI_10_PRO_LATEST:

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/types/model.py

@@ -25,6 +25,7 @@ class Model(str, enum.Enum):
     CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
     CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
     CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
+    CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
     GOOGLE_GEMINI_10_PRO_LATEST = "google-gemini-1.0-pro-latest"
     DATABRICKS_DBRX = "databricks-dbrx"
 
@@ -44,6 +45,7 @@ class Model(str, enum.Enum):
         claude_3_opus_20240229: typing.Callable[[], T_Result],
         claude_3_sonnet_20240229: typing.Callable[[], T_Result],
         claude_3_haiku_20240307: typing.Callable[[], T_Result],
+        claude_35_sonnet_20240620: typing.Callable[[], T_Result],
         google_gemini_10_pro_latest: typing.Callable[[], T_Result],
         databricks_dbrx: typing.Callable[[], T_Result],
     ) -> T_Result:
@@ -75,6 +77,8 @@ class Model(str, enum.Enum):
             return claude_3_sonnet_20240229()
         if self is Model.CLAUDE_3_HAIKU_20240307:
             return claude_3_haiku_20240307()
+        if self is Model.CLAUDE_35_SONNET_20240620:
+            return claude_35_sonnet_20240620()
         if self is Model.GOOGLE_GEMINI_10_PRO_LATEST:
             return google_gemini_10_pro_latest()
         if self is Model.DATABRICKS_DBRX:

{athena_intelligence-0.1.77 → athena_intelligence-0.1.79}/src/athena/types/query_model.py

@@ -16,6 +16,7 @@ class QueryModel(pydantic_v1.BaseModel):
     )
     dimensions: typing.Optional[typing.List[str]] = None
     filters: typing.Optional[typing.List[FilterModel]] = None
+    limit: typing.Optional[int] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena_intelligence-0.1.77/src/athena/errors/__init__.py

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

athena_intelligence-0.1.77/src/athena/workflow/__init__.py

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