athena-intelligence 0.1.94__py3-none-any.whl → 0.1.95__py3-none-any.whl

athena/tools/types/tools_data_frame_request_columns_item.py

@@ -2,4 +2,4 @@
 
 import typing
 
-ToolsDataFrameRequestColumnsItem = typing.Union[int, str]
+ToolsDataFrameRequestColumnsItem = typing.Union[str, int]

athena/types/__init__.py

@@ -1,91 +1,25 @@
 # 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
 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 .data_frame_unknown_format_error import DataFrameUnknownFormatError
-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
-from .firecrawl_scrape_url_metadata import FirecrawlScrapeUrlMetadata
-from .get_datasets_response import GetDatasetsResponse
-from .get_snippet_out import GetSnippetOut
-from .get_snippets_response import GetSnippetsResponse
 from .http_validation_error import HttpValidationError
-from .langchain_documents_request_out import LangchainDocumentsRequestOut
-from .llm_model import LlmModel
-from .map_reduce_chain_out import MapReduceChainOut
-from .message_out import MessageOut
-from .message_out_dto import MessageOutDto
-from .model import Model
-from .publish_formats import PublishFormats
-from .query_model import QueryModel
-from .report import Report
-from .researcher_out import ResearcherOut
-from .semantic_query_out import SemanticQueryOut
-from .snippet import Snippet
-from .sql_results import SqlResults
-from .status_enum import StatusEnum
-from .structured_parse_result import StructuredParseResult
-from .time_dimension_model import TimeDimensionModel
-from .tools import Tools
-from .upload_documents_out import UploadDocumentsOut
-from .url_result import UrlResult
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
-from .workflow_status_out import WorkflowStatusOut
 
 __all__ = [
-    "AthenaDocumentV2Out",
-    "ConvertPdfToSheetOut",
     "DataFrameParsingError",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
-    "Dataset",
-    "Document",
-    "ExcecuteToolFirstWorkflowOut",
-    "FileDataResponse",
     "FileFetchError",
-    "FilterModel",
-    "FilterOperator",
-    "FirecrawlScrapeUrlDataReponseDto",
-    "FirecrawlScrapeUrlMetadata",
-    "GetDatasetsResponse",
-    "GetSnippetOut",
-    "GetSnippetsResponse",
     "HttpValidationError",
-    "LangchainDocumentsRequestOut",
-    "LlmModel",
-    "MapReduceChainOut",
-    "MessageOut",
-    "MessageOutDto",
-    "Model",
-    "PublishFormats",
-    "QueryModel",
-    "Report",
-    "ResearcherOut",
-    "SemanticQueryOut",
-    "Snippet",
-    "SqlResults",
-    "StatusEnum",
-    "StructuredParseResult",
-    "TimeDimensionModel",
-    "Tools",
-    "UploadDocumentsOut",
-    "UrlResult",
     "ValidationError",
     "ValidationErrorLocItem",
-    "WorkflowStatusOut",
 ]

athena/types/data_frame_parsing_error.py

@@ -8,8 +8,8 @@ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
 class DataFrameParsingError(pydantic_v1.BaseModel):
-    message: str
     document_id: str
+    message: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena/types/data_frame_request_out.py

@@ -11,9 +11,13 @@ from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
 
 
 class DataFrameRequestOut(pydantic_v1.BaseModel):
+    """
+    Response model with JSON dataframe representation.
+    """
+
     columns: typing.List[DataFrameRequestOutColumnsItem]
-    index: typing.List[DataFrameRequestOutIndexItem]
     data: typing.List[typing.List[DataFrameRequestOutDataItemItem]]
+    index: typing.List[DataFrameRequestOutIndexItem]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena/types/data_frame_unknown_format_error.py

@@ -8,9 +8,9 @@ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
 class DataFrameUnknownFormatError(pydantic_v1.BaseModel):
-    message: str
     document_id: str
     media_type: str
+    message: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{athena_intelligence-0.1.94.dist-info → athena_intelligence-0.1.95.dist-info}/METADATA

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

athena_intelligence-0.1.95.dist-info/RECORD

@@ -0,0 +1,40 @@
+athena/__init__.py,sha256=Puh6njWEP0yMV514GrJSg2UrY6ojVOboWmoM38a-jeo,1161
+athena/base_client.py,sha256=KS-PX4GHtkn89PP3IHPcSX54TfyBsFeOSuDn05p9YLc,5238
+athena/client.py,sha256=SKZjcEFf_-NThIm8WuPUCsBVSlmWPVnVsXEsvJfWK_I,9415
+athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
+athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
+athena/core/client_wrapper.py,sha256=UPff8A8yuhnh05jPpibiw4L3_v4Fw6MOLAHAKxiH5Ss,1805
+athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
+athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
+athena/core/http_client.py,sha256=Z4NuAsJD-51yqmoME17O5sxwx5orSp1wsnd6bPyKcgA,17768
+athena/core/jsonable_encoder.py,sha256=L6G68Py6gEo8n87rXwlkLPUyzHvXftEBjJZNb2tCuOA,3742
+athena/core/pydantic_utilities.py,sha256=hI3vcpSG47sVlafyPol2T2ICt8HNMIu_rM9amc2zf7w,748
+athena/core/query_encoder.py,sha256=sI6XiwFby-WRIk4MVN_5wlAlm30aCHE73Uvm09fb9qQ,1266
+athena/core/remove_none_from_dict.py,sha256=EU9SGgYidWq7SexuJbNs4-PZ-5Bl3Vppd864mS6vQZw,342
+athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
+athena/environment.py,sha256=qTOfpg8B5EGqwcJFB4ANkLFn2Y6rsPWGuhU8GRPlFl0,162
+athena/errors/__init__.py,sha256=0LicWKR1fNQwwL8_ohKhBkxXTD8StFlBYDFG1jPgqOk,405
+athena/errors/internal_server_error.py,sha256=WEwVqzsfpBTaqZipvse-kvKbW-3NbpXVvuHXs_64U0M,315
+athena/errors/not_found_error.py,sha256=uBK3JapPPgTkMoSCX9j22C6MDtSuTuSPmjT0lDRigpA,287
+athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
+athena/errors/unsupported_media_type_error.py,sha256=fQ7TYQ3QYcT_YzFzO4f8_bLger7UQfZmuF22I9jYxFA,340
+athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+athena/tools/__init__.py,sha256=3n7oOoMebo06MAQqYRE2CX9Q0fTNnKBYE0cTlh1MPkM,165
+athena/tools/client.py,sha256=2sBdhdNv-BZqLBgCtlwLFs9uFaM_bhGAGfhW8XWshkY,11028
+athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
+athena/tools/types/tools_data_frame_request_columns_item.py,sha256=4rve98HoxzGpDhfEcNwqkH1mUQQXL8vAnX4T04PTTLk,138
+athena/types/__init__.py,sha256=ZGBn5o7_zVgGVt67H1Ukh8wDknD5M0J0afgZUwggv5s,1022
+athena/types/data_frame_parsing_error.py,sha256=lECOObzDIziXpsr8fokECZmYYRG_GBtAwA6bwChMwrE,1145
+athena/types/data_frame_request_out.py,sha256=cm3QgumqJ2bx2sJ2IVedkXPQ97qTu2835MoK0m1UKTg,1594
+athena/types/data_frame_request_out_columns_item.py,sha256=9cjzciFv6C8n8Griytt_q_8ovkzHViS5tvUcMDfkfKE,143
+athena/types/data_frame_request_out_data_item_item.py,sha256=KMTJRr-1bdKDNMbUericCliwRoPHLGRV-n2bJtxdRW0,144
+athena/types/data_frame_request_out_index_item.py,sha256=bW7oe912trpkYKodj-I_AiTXXy61yWzliekcsUZkZE0,141
+athena/types/data_frame_unknown_format_error.py,sha256=qeWdaMN5cRTS7JGzp0BGCZFo3IWCHtOQ-Fr0CwAO3z0,1171
+athena/types/file_fetch_error.py,sha256=iR7IXjj2C4lJFcWV3aS2gfhmT3CE-_hhBDSWFaCktHU,1162
+athena/types/http_validation_error.py,sha256=ZuVHc-T6RFEo7_qBO8DWHa2jGJ6O--tUIZynmLFezsI,1216
+athena/types/validation_error.py,sha256=zub9I6bl192KHCE8RA0pEkdaEODR3ktTFnZ7E1B7Eco,1235
+athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
+athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
+athena_intelligence-0.1.95.dist-info/METADATA,sha256=TuP2Tv8YY434Zc-KVJAMdSKCe4iPAAFDQ0L2Ylyx1yI,5273
+athena_intelligence-0.1.95.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.95.dist-info/RECORD,,

athena/chain/__init__.py

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

athena/chain/client.py

@@ -1,302 +0,0 @@
-# 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.document import Document
-from ..types.http_validation_error import HttpValidationError
-from ..types.llm_model import LlmModel
-from ..types.map_reduce_chain_out import MapReduceChainOut
-from ..types.structured_parse_result import StructuredParseResult
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class ChainClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def structured_parse(
-        self,
-        *,
-        text_input: str,
-        custom_type_dict: typing.Dict[str, typing.Any],
-        model: LlmModel,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> StructuredParseResult:
-        """
-        Parameters
-        ----------
-        text_input : str
-            The text input to be parsed.
-
-        custom_type_dict : typing.Dict[str, typing.Any]
-            A dictionary of field names and their default values.
-
-        model : LlmModel
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        StructuredParseResult
-            Successful Response
-
-        Examples
-        --------
-        from athena import LlmModel
-        from athena.client import Athena
-
-        client = Athena(
-            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.',
-            custom_type_dict={},
-            model=LlmModel.GPT_4_TURBO,
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/structured-parse",
-            method="POST",
-            json={"text_input": text_input, "custom_type_dict": custom_type_dict, "model": model},
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(StructuredParseResult, _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 map_reduce_chain(
-        self,
-        *,
-        documents: typing.Sequence[Document],
-        model: LlmModel,
-        operator_prompt: str,
-        reducer_prompt: str,
-        input: str,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> MapReduceChainOut:
-        """
-        Parameters
-        ----------
-        documents : typing.Sequence[Document]
-
-        model : LlmModel
-
-        operator_prompt : str
-
-        reducer_prompt : str
-
-        input : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        MapReduceChainOut
-            Successful Response
-
-        Examples
-        --------
-        from athena import Document, LlmModel
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.chain.map_reduce_chain(
-            documents=[
-                Document(
-                    metadata={"key": "value"},
-                    page_content="Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot, allowing you to hand over controls to her for autonomous execution with confidence.",
-                )
-            ],
-            model=LlmModel.MISTRAL_LARGE_0224,
-            operator_prompt="summarize the content",
-            reducer_prompt="Combine these summaries",
-            input="return a summary in a single sentence",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/tools/map-reduce",
-            method="POST",
-            json={
-                "documents": documents,
-                "model": model,
-                "operator_prompt": operator_prompt,
-                "reducer_prompt": reducer_prompt,
-                "input": input,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(MapReduceChainOut, _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 AsyncChainClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def structured_parse(
-        self,
-        *,
-        text_input: str,
-        custom_type_dict: typing.Dict[str, typing.Any],
-        model: LlmModel,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> StructuredParseResult:
-        """
-        Parameters
-        ----------
-        text_input : str
-            The text input to be parsed.
-
-        custom_type_dict : typing.Dict[str, typing.Any]
-            A dictionary of field names and their default values.
-
-        model : LlmModel
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        StructuredParseResult
-            Successful Response
-
-        Examples
-        --------
-        from athena import LlmModel
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            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.',
-            custom_type_dict={},
-            model=LlmModel.GPT_4_TURBO,
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/structured-parse",
-            method="POST",
-            json={"text_input": text_input, "custom_type_dict": custom_type_dict, "model": model},
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(StructuredParseResult, _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 map_reduce_chain(
-        self,
-        *,
-        documents: typing.Sequence[Document],
-        model: LlmModel,
-        operator_prompt: str,
-        reducer_prompt: str,
-        input: str,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> MapReduceChainOut:
-        """
-        Parameters
-        ----------
-        documents : typing.Sequence[Document]
-
-        model : LlmModel
-
-        operator_prompt : str
-
-        reducer_prompt : str
-
-        input : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        MapReduceChainOut
-            Successful Response
-
-        Examples
-        --------
-        from athena import Document, LlmModel
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.chain.map_reduce_chain(
-            documents=[
-                Document(
-                    metadata={"key": "value"},
-                    page_content="Athena is an AI-native analytics platform and artificial employee built to accelerate analytics workflows by offering enterprise teams co-pilot and auto-pilot modes. Athena learns your workflow as a co-pilot, allowing you to hand over controls to her for autonomous execution with confidence.",
-                )
-            ],
-            model=LlmModel.MISTRAL_LARGE_0224,
-            operator_prompt="summarize the content",
-            reducer_prompt="Combine these summaries",
-            input="return a summary in a single sentence",
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/tools/map-reduce",
-            method="POST",
-            json={
-                "documents": documents,
-                "model": model,
-                "operator_prompt": operator_prompt,
-                "reducer_prompt": reducer_prompt,
-                "input": input,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(MapReduceChainOut, _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/dataset/__init__.py

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

athena/dataset/client.py

@@ -1,124 +0,0 @@
-# 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.get_datasets_response import GetDatasetsResponse
-from ..types.http_validation_error import HttpValidationError
-
-
-class DatasetClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def get(
-        self,
-        *,
-        page: typing.Optional[int] = None,
-        page_size: typing.Optional[int] = None,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> GetDatasetsResponse:
-        """
-        Parameters
-        ----------
-        page : typing.Optional[int]
-            Page number starting from 1
-
-        page_size : typing.Optional[int]
-            Number of items per page
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        GetDatasetsResponse
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.dataset.get()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "api/v0/datasets",
-            method="GET",
-            params={"page": page, "page_size": page_size},
-            request_options=request_options,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(GetDatasetsResponse, _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 AsyncDatasetClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def get(
-        self,
-        *,
-        page: typing.Optional[int] = None,
-        page_size: typing.Optional[int] = None,
-        request_options: typing.Optional[RequestOptions] = None
-    ) -> GetDatasetsResponse:
-        """
-        Parameters
-        ----------
-        page : typing.Optional[int]
-            Page number starting from 1
-
-        page_size : typing.Optional[int]
-            Number of items per page
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        GetDatasetsResponse
-            Successful Response
-
-        Examples
-        --------
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.dataset.get()
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "api/v0/datasets",
-            method="GET",
-            params={"page": page, "page_size": page_size},
-            request_options=request_options,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic_v1.parse_obj_as(GetDatasetsResponse, _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/message/__init__.py

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