athena-intelligence 0.1.84__py3-none-any.whl → 0.1.86__py3-none-any.whl

athena/__init__.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AthenaDocumentV2Out,
     ConvertPdfToSheetOut,
     DataFrameParsingError,
     DataFrameRequestOut,
@@ -51,6 +52,7 @@ from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
+    "AthenaDocumentV2Out",
     "AthenaEnvironment",
     "ConvertPdfToSheetOut",
     "DataFrameParsingError",

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.84",
+            "X-Fern-SDK-Version": "0.1.86",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/tools/client.py

@@ -15,6 +15,7 @@ from ..errors.internal_server_error import InternalServerError
 from ..errors.not_found_error import NotFoundError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..errors.unsupported_media_type_error import UnsupportedMediaTypeError
+from ..types.athena_document_v_2_out import AthenaDocumentV2Out
 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
@@ -210,7 +211,7 @@ class ToolsClient:
         columns: typing.Optional[
             typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
         ] = None,
-        sheet_name: typing.Optional[str] = None,
+        sheet_name: typing.Optional[int] = None,
         separator: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DataFrameRequestOut:
@@ -226,7 +227,7 @@ class ToolsClient:
         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]
+        sheet_name : typing.Optional[int]
             only for excel files
 
         separator : typing.Optional[str]
@@ -741,6 +742,73 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def athena_documents(
+        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AthenaDocumentV2Out:
+        """
+        Parameters
+        ----------
+        document_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AthenaDocumentV2Out
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.athena_documents(
+            document_id="doc_9249292-d118-42d3-95b4-00eccfe0754f",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            method="POST",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/athena-document-v2"),
+            params=encode_query(
+                jsonable_encoder(
+                    request_options.get("additional_query_parameters") if request_options is not None else None
+                )
+            ),
+            json=jsonable_encoder({"document_id": document_id})
+            if request_options is None or request_options.get("additional_body_parameters") is None
+            else {
+                **jsonable_encoder({"document_id": document_id}),
+                **(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 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(AthenaDocumentV2Out, _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)
+
 
 class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -917,7 +985,7 @@ class AsyncToolsClient:
         columns: typing.Optional[
             typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
         ] = None,
-        sheet_name: typing.Optional[str] = None,
+        sheet_name: typing.Optional[int] = None,
         separator: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DataFrameRequestOut:
@@ -933,7 +1001,7 @@ class AsyncToolsClient:
         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]
+        sheet_name : typing.Optional[int]
             only for excel files
 
         separator : typing.Optional[str]
@@ -1447,3 +1515,70 @@ class AsyncToolsClient:
         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_documents(
+        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AthenaDocumentV2Out:
+        """
+        Parameters
+        ----------
+        document_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AthenaDocumentV2Out
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.athena_documents(
+            document_id="doc_9249292-d118-42d3-95b4-00eccfe0754f",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            method="POST",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/athena-document-v2"),
+            params=encode_query(
+                jsonable_encoder(
+                    request_options.get("additional_query_parameters") if request_options is not None else None
+                )
+            ),
+            json=jsonable_encoder({"document_id": document_id})
+            if request_options is None or request_options.get("additional_body_parameters") is None
+            else {
+                **jsonable_encoder({"document_id": document_id}),
+                **(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 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(AthenaDocumentV2Out, _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)

athena/types/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .athena_document_v_2_out import AthenaDocumentV2Out
 from .convert_pdf_to_sheet_out import ConvertPdfToSheetOut
 from .data_frame_parsing_error import DataFrameParsingError
 from .data_frame_request_out import DataFrameRequestOut
@@ -44,6 +45,7 @@ from .validation_error_loc_item import ValidationErrorLocItem
 from .workflow_status_out import WorkflowStatusOut
 
 __all__ = [
+    "AthenaDocumentV2Out",
     "ConvertPdfToSheetOut",
     "DataFrameParsingError",
     "DataFrameRequestOut",

athena/types/athena_document_v_2_out.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class AthenaDocumentV2Out(pydantic_v1.BaseModel):
+    output: 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/types/model.py

@@ -27,6 +27,7 @@ class Model(str, enum.Enum):
     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"
+    GOOGLE_GEMINI_15_PRO_LATEST = "google-gemini-1.5-pro-latest"
     DATABRICKS_DBRX = "databricks-dbrx"
 
     def visit(
@@ -47,6 +48,7 @@ class Model(str, enum.Enum):
         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],
+        google_gemini_15_pro_latest: typing.Callable[[], T_Result],
         databricks_dbrx: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is Model.GPT_35_TURBO:
@@ -81,5 +83,7 @@ class Model(str, enum.Enum):
             return claude_35_sonnet_20240620()
         if self is Model.GOOGLE_GEMINI_10_PRO_LATEST:
             return google_gemini_10_pro_latest()
+        if self is Model.GOOGLE_GEMINI_15_PRO_LATEST:
+            return google_gemini_15_pro_latest()
         if self is Model.DATABRICKS_DBRX:
             return databricks_dbrx()

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
+    order: typing.Optional[typing.Dict[str, typing.Any]] = None
     limit: typing.Optional[int] = None
 
     def json(self, **kwargs: typing.Any) -> str:

{athena_intelligence-0.1.84.dist-info → athena_intelligence-0.1.86.dist-info}/METADATA

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

{athena_intelligence-0.1.84.dist-info → athena_intelligence-0.1.86.dist-info}/RECORD

@@ -1,11 +1,11 @@
-athena/__init__.py,sha256=S-yzA6G5fLjtyXBy2EPQ7eBzd_czab5tJGAJUcv_LgA,2749
+athena/__init__.py,sha256=vS1F7G3hdiHGM-q24LBoKf77EHpaDBrb3ZzM_sgXoGY,2801
 athena/base_client.py,sha256=4HD21xS-Kl2oyy0OQGPVSH8X3aRpKdefKKOGXkAa-1M,7101
 athena/chain/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/chain/client.py,sha256=3O5tTl9skrhNS--6Jt-Za7CmmBz1MQLiqeNK0Ewtup0,16977
 athena/client.py,sha256=gVUvs9Wui80KKYfXOzWQjfcar98ijg8NzNaPLkULFFA,6757
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=Ias36Y3yOhjFDZRabC6_FlVgWN0p0pf2MSZg2SMh-GI,1495
+athena/core/client_wrapper.py,sha256=ujw-SZMlg7IKRB5rwW50cLXxVAMNLoMM7e6vQ-2Pcig,1495
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=5ok6hqgZDJhg57EHvMnr0BBaHdG50QxFPKaCZ9aVWTc,5059
@@ -35,10 +35,11 @@ athena/search/client.py,sha256=1XD9KW-crpB4U2bGOrYngOxdomjoUInthNe_H9tXdgA,7978
 athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/snippet/client.py,sha256=0_lhr8_IHNah2qdDT8OpwOSHN_SCqE0F7KLx4YHp-8I,12217
 athena/tools/__init__.py,sha256=3n7oOoMebo06MAQqYRE2CX9Q0fTNnKBYE0cTlh1MPkM,165
-athena/tools/client.py,sha256=C55orGhXtURrn8T8-XHJjCAHEUABz9KeCxA13t2dkKc,58155
+athena/tools/client.py,sha256=ur1PDs-ftZDWIB_5-FUKLlHNzvGgN2DgJea_B09lidM,63809
 athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
 athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
-athena/types/__init__.py,sha256=mOox5XE1eT51OIcJmBmHt9y3FKe412J-ydKk7WLkFrg,3214
+athena/types/__init__.py,sha256=UKGjYkgCipVr4_CBwGkCL6PdYGC4XsS7qYdRmCGpr8Q,3298
+athena/types/athena_document_v_2_out.py,sha256=_wUNU1ygb0FNcf3ls9Xf0_3ePHkwXjRhh7YAKJcfirw,1146
 athena/types/convert_pdf_to_sheet_out.py,sha256=ce27tpf6OzV7eVTpMV3oeEqvqVzOwUBFRxxHnS6Y60M,1152
 athena/types/data_frame_parsing_error.py,sha256=gXJXZQsZKgzRoF7iLKvy1JFZAPI-M-ajY7ybLW-XYvA,1145
 athena/types/data_frame_request_out.py,sha256=u2dVlNwkzIStseKbXHGwt8kvjre64TNrulDxE7Gi7E0,1522
@@ -64,9 +65,9 @@ athena/types/llm_model.py,sha256=kGCC7WEAe-Lq7r6kiXhy5UCn6vdSdb7ydbq5gIjeye0,666
 athena/types/map_reduce_chain_out.py,sha256=p9rN21wMZaki_rpd1Z_0DM4tvlDdYG8gHE2Xg4TEAvE,1213
 athena/types/message_out.py,sha256=Jli5sJBAtwOgkZjpx28kVHfBRoBLuINyNTm_9AXPuys,1108
 athena/types/message_out_dto.py,sha256=iELbR1DZBrlSshzIdfpZ3ZR2V1fzAzVw4aprxcfzg38,1294
-athena/types/model.py,sha256=qUqa8Tcmb3o0Cxul_1UGiNlJGPlbr4OGqYNLtxf8oEQ,3510
+athena/types/model.py,sha256=2dpNftSHkrgeUl-00Zk3UZ1tSkeygOfimHFDu5mc3ek,3746
 athena/types/publish_formats.py,sha256=1j2fgJo0MMaHzEEYxd9uCFq_Y7bFoQox0JgKCfcPXLo,1148
-athena/types/query_model.py,sha256=IIOOn36s6JWuDNxy4-OvXhSh2hcL0ZLDo9pnPvlbB28,1620
+athena/types/query_model.py,sha256=F-HSvgcTys4hA3e7sYas_71APMZmAu_LOKqWVZ0-Jx4,1684
 athena/types/report.py,sha256=Yr_Ph7BS-2QgMJ3az5GP_fwlRMl6i4-1MGWYcLPSPhY,1189
 athena/types/researcher_out.py,sha256=XC8f90Q1G18zSSEzJ3pM5hrMnBDt0rrBIri0pJVFWQg,1142
 athena/types/semantic_query_out.py,sha256=sdsVToZ5o5AUTKVSeCeyhb8TwINMX5RZgs5XbZgkiaQ,1143
@@ -86,6 +87,6 @@ athena/upload/client.py,sha256=L66H0qFpFH1RYIfKJLQgPDZ3HK5ZagRwYQX5BIen1rc,6825
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
 athena/workflow/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/workflow/client.py,sha256=3LCFYbAoOWcvzwME677syefoJ55ukdTUxiQ71jXw7so,6663
-athena_intelligence-0.1.84.dist-info/METADATA,sha256=CVZngWwieSVyUaAXXYdR1IK9ujhUk8NKckYXMVBJWwk,5273
-athena_intelligence-0.1.84.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.84.dist-info/RECORD,,
+athena_intelligence-0.1.86.dist-info/METADATA,sha256=oiAY6HrOZNxyXsjDtH8XB6QZkv5mgS77ouasBMLWlfQ,5273
+athena_intelligence-0.1.86.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.86.dist-info/RECORD,,