athena-intelligence 0.1.84__tar.gz → 0.1.86__tar.gz

{athena_intelligence-0.1.84 → athena_intelligence-0.1.86}/PKG-INFO

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

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

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

{athena_intelligence-0.1.84 → athena_intelligence-0.1.86}/src/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_intelligence-0.1.84 → athena_intelligence-0.1.86}/src/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_intelligence-0.1.86/src/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_intelligence-0.1.84 → athena_intelligence-0.1.86}/src/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_intelligence-0.1.84 → athena_intelligence-0.1.86}/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
+    order: typing.Optional[typing.Dict[str, typing.Any]] = None
     limit: typing.Optional[int] = None
 
     def json(self, **kwargs: typing.Any) -> str: