athena-intelligence 0.1.99__py3-none-any.whl → 0.1.101__py3-none-any.whl

athena/__init__.py

@@ -1,40 +1,44 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
-    DataFrameParsingError,
     DataFrameRequestOut,
     DataFrameRequestOutColumnsItem,
     DataFrameRequestOutDataItemItem,
     DataFrameRequestOutIndexItem,
     DataFrameUnknownFormatError,
     FileFetchError,
-    HttpValidationError,
-    ValidationError,
-    ValidationErrorLocItem,
+    SqlUnauthorizedError,
 )
-from .errors import InternalServerError, NotFoundError, UnprocessableEntityError, UnsupportedMediaTypeError
-from . import tools
+from .errors import (
+    InternalServerError,
+    NotFoundError,
+    UnauthorizedError,
+    UnprocessableEntityError,
+    UnsupportedMediaTypeError,
+)
+from . import query, tools
 from .environment import AthenaEnvironment
+from .query import QueryExecuteRequestDatabaseAssetIds
 from .tools import ToolsDataFrameRequestColumnsItem
 from .version import __version__
 
 __all__ = [
     "AthenaEnvironment",
-    "DataFrameParsingError",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
     "FileFetchError",
-    "HttpValidationError",
     "InternalServerError",
     "NotFoundError",
+    "QueryExecuteRequestDatabaseAssetIds",
+    "SqlUnauthorizedError",
     "ToolsDataFrameRequestColumnsItem",
+    "UnauthorizedError",
     "UnprocessableEntityError",
     "UnsupportedMediaTypeError",
-    "ValidationError",
-    "ValidationErrorLocItem",
     "__version__",
+    "query",
     "tools",
 ]

athena/base_client.py

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

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

athena/errors/__init__.py

@@ -2,7 +2,14 @@
 
 from .internal_server_error import InternalServerError
 from .not_found_error import NotFoundError
+from .unauthorized_error import UnauthorizedError
 from .unprocessable_entity_error import UnprocessableEntityError
 from .unsupported_media_type_error import UnsupportedMediaTypeError
 
-__all__ = ["InternalServerError", "NotFoundError", "UnprocessableEntityError", "UnsupportedMediaTypeError"]
+__all__ = [
+    "InternalServerError",
+    "NotFoundError",
+    "UnauthorizedError",
+    "UnprocessableEntityError",
+    "UnsupportedMediaTypeError",
+]

athena/errors/internal_server_error.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import typing
+
 from ..core.api_error import ApiError
-from ..types.data_frame_parsing_error import DataFrameParsingError
 
 
 class InternalServerError(ApiError):
-    def __init__(self, body: DataFrameParsingError):
+    def __init__(self, body: typing.Any):
         super().__init__(status_code=500, body=body)

athena/errors/unauthorized_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.sql_unauthorized_error import SqlUnauthorizedError
+
+
+class UnauthorizedError(ApiError):
+    def __init__(self, body: SqlUnauthorizedError):
+        super().__init__(status_code=401, body=body)

athena/errors/unprocessable_entity_error.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import typing
+
 from ..core.api_error import ApiError
-from ..types.http_validation_error import HttpValidationError
 
 
 class UnprocessableEntityError(ApiError):
-    def __init__(self, body: HttpValidationError):
+    def __init__(self, body: typing.Any):
         super().__init__(status_code=422, body=body)

athena/query/__init__.py

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

athena/query/client.py

@@ -0,0 +1,240 @@
+# 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.jsonable_encoder import jsonable_encoder
+from ..core.pydantic_utilities import pydantic_v1
+from ..core.request_options import RequestOptions
+from ..errors.internal_server_error import InternalServerError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.data_frame_request_out import DataFrameRequestOut
+from ..types.sql_unauthorized_error import SqlUnauthorizedError
+from .types.query_execute_request_database_asset_ids import QueryExecuteRequestDatabaseAssetIds
+
+
+class QueryClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def execute(
+        self,
+        *,
+        sql_command: str,
+        database_asset_ids: QueryExecuteRequestDatabaseAssetIds,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> DataFrameRequestOut:
+        """
+        Get the result of an SQL query over given assets.
+
+        Parameters
+        ----------
+        sql_command : str
+            SQL query string
+
+        database_asset_ids : QueryExecuteRequestDatabaseAssetIds
+            Single ID or list of asset IDs
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DataFrameRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.query.execute(
+            sql_command="sql_command",
+            database_asset_ids="database_asset_ids",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/query/sql/code/execute",
+            method="GET",
+            params={"sql_command": sql_command, "database_asset_ids": jsonable_encoder(database_asset_ids)},
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 500:
+            raise InternalServerError(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 execute_snippet(
+        self, *, snippet_asset_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DataFrameRequestOut:
+        """
+        Get the result of an SQL query over given assets.
+
+        Parameters
+        ----------
+        snippet_asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DataFrameRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.query.execute_snippet(
+            snippet_asset_id="snippet_asset_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/query/sql/snippet/execute",
+            method="GET",
+            params={"snippet_asset_id": snippet_asset_id},
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 500:
+            raise InternalServerError(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 AsyncQueryClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def execute(
+        self,
+        *,
+        sql_command: str,
+        database_asset_ids: QueryExecuteRequestDatabaseAssetIds,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> DataFrameRequestOut:
+        """
+        Get the result of an SQL query over given assets.
+
+        Parameters
+        ----------
+        sql_command : str
+            SQL query string
+
+        database_asset_ids : QueryExecuteRequestDatabaseAssetIds
+            Single ID or list of asset IDs
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DataFrameRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.query.execute(
+            sql_command="sql_command",
+            database_asset_ids="database_asset_ids",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/query/sql/code/execute",
+            method="GET",
+            params={"sql_command": sql_command, "database_asset_ids": jsonable_encoder(database_asset_ids)},
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 500:
+            raise InternalServerError(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 execute_snippet(
+        self, *, snippet_asset_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DataFrameRequestOut:
+        """
+        Get the result of an SQL query over given assets.
+
+        Parameters
+        ----------
+        snippet_asset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DataFrameRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.query.execute_snippet(
+            snippet_asset_id="snippet_asset_id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/query/sql/snippet/execute",
+            method="GET",
+            params={"snippet_asset_id": snippet_asset_id},
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 401:
+            raise UnauthorizedError(pydantic_v1.parse_obj_as(SqlUnauthorizedError, _response.json()))  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 500:
+            raise InternalServerError(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/query/types/__init__.py

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

athena/{types/validation_error_loc_item.py → query/types/query_execute_request_database_asset_ids.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-ValidationErrorLocItem = typing.Union[str, int]
+QueryExecuteRequestDatabaseAssetIds = typing.Union[str, typing.List[str]]

athena/tools/client.py

@@ -11,11 +11,9 @@ 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.data_frame_parsing_error import DataFrameParsingError
 from ..types.data_frame_request_out import DataFrameRequestOut
 from ..types.data_frame_unknown_format_error import DataFrameUnknownFormatError
 from ..types.file_fetch_error import FileFetchError
-from ..types.http_validation_error import HttpValidationError
 from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
 
 
@@ -95,11 +93,9 @@ class ToolsClient:
                 pydantic_v1.parse_obj_as(DataFrameUnknownFormatError, _response.json())  # type: ignore
             )
         if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
-            raise InternalServerError(pydantic_v1.parse_obj_as(DataFrameParsingError, _response.json()))  # type: ignore
+            raise InternalServerError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -146,9 +142,7 @@ class ToolsClient:
             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
-                )
+                raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             try:
                 _response_json = _response.json()
             except JSONDecodeError:
@@ -232,11 +226,9 @@ class AsyncToolsClient:
                 pydantic_v1.parse_obj_as(DataFrameUnknownFormatError, _response.json())  # type: ignore
             )
         if _response.status_code == 422:
-            raise UnprocessableEntityError(
-                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
-            )
+            raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
-            raise InternalServerError(pydantic_v1.parse_obj_as(DataFrameParsingError, _response.json()))  # type: ignore
+            raise InternalServerError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -283,9 +275,7 @@ class AsyncToolsClient:
             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
-                )
+                raise UnprocessableEntityError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             try:
                 _response_json = _response.json()
             except JSONDecodeError:

athena/types/__init__.py

@@ -1,25 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-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 .file_fetch_error import FileFetchError
-from .http_validation_error import HttpValidationError
-from .validation_error import ValidationError
-from .validation_error_loc_item import ValidationErrorLocItem
+from .sql_unauthorized_error import SqlUnauthorizedError
 
 __all__ = [
-    "DataFrameParsingError",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
     "FileFetchError",
-    "HttpValidationError",
-    "ValidationError",
-    "ValidationErrorLocItem",
+    "SqlUnauthorizedError",
 ]

athena/types/data_frame_request_out.py

@@ -17,7 +17,7 @@ class DataFrameRequestOut(pydantic_v1.BaseModel):
 
     columns: typing.List[typing.Optional[DataFrameRequestOutColumnsItem]]
     data: typing.List[typing.List[typing.Optional[DataFrameRequestOutDataItemItem]]]
-    index: typing.List[typing.Optional[DataFrameRequestOutIndexItem]]
+    index: typing.Optional[typing.List[typing.Optional[DataFrameRequestOutIndexItem]]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena/types/{data_frame_parsing_error.py → sql_unauthorized_error.py}

@@ -7,10 +7,7 @@ from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
-class DataFrameParsingError(pydantic_v1.BaseModel):
-    asset_id: str
-    message: str
-
+class SqlUnauthorizedError(pydantic_v1.BaseModel):
     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)

{athena_intelligence-0.1.99.dist-info → athena_intelligence-0.1.101.dist-info}/METADATA

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

{athena_intelligence-0.1.99.dist-info → athena_intelligence-0.1.101.dist-info}/RECORD

@@ -1,9 +1,9 @@
-athena/__init__.py,sha256=Puh6njWEP0yMV514GrJSg2UrY6ojVOboWmoM38a-jeo,1161
-athena/base_client.py,sha256=KS-PX4GHtkn89PP3IHPcSX54TfyBsFeOSuDn05p9YLc,5238
+athena/__init__.py,sha256=lGo6ASUYoueZCu_oWrD7ruCsa0nLRNo9qZ_-2RV4WxY,1192
+athena/base_client.py,sha256=TlP599mOBvj7Tk8IpFe5dgrDM98GJu3lEQh_zwl4vtA,5439
 athena/client.py,sha256=2jxJ8mZWLa9LOH9XCxbxwLV6qVPXw53gi7GiYUTcpZg,9346
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=Stqk_c6EsVwY9O4XvChh7ubnzLVgtLFLm29x03MQNWo,1805
+athena/core/client_wrapper.py,sha256=OWm1dmWnrVS5aatw4sJNN9fa54tiO28qSndRzyy8ksw,1806
 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
@@ -13,28 +13,30 @@ athena/core/query_encoder.py,sha256=sI6XiwFby-WRIk4MVN_5wlAlm30aCHE73Uvm09fb9qQ,
 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=_e7YwByXALEk7fNJNJvjF81caYTbsx4c7wwftCJHQ7g,214
-athena/errors/__init__.py,sha256=0LicWKR1fNQwwL8_ohKhBkxXTD8StFlBYDFG1jPgqOk,405
-athena/errors/internal_server_error.py,sha256=WEwVqzsfpBTaqZipvse-kvKbW-3NbpXVvuHXs_64U0M,315
+athena/errors/__init__.py,sha256=3hOHgz6h6rQZ_8URX6GpKLhKIhMVxYHANI-X8jTO1t0,499
+athena/errors/internal_server_error.py,sha256=E0rgqJC0-LcetLi1HmSi92KpvNkGSRCIdBeEqT_ln1s,252
 athena/errors/not_found_error.py,sha256=uBK3JapPPgTkMoSCX9j22C6MDtSuTuSPmjT0lDRigpA,287
-athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
+athena/errors/unauthorized_error.py,sha256=6qGByG08vA3iFHmK5ReHUK2AiGeCsK3Vf50C5LQNILM,309
+athena/errors/unprocessable_entity_error.py,sha256=OztAzRntOJVxbZtXwoR4epahhG6chKOB5chVLUC8D9Q,257
 athena/errors/unsupported_media_type_error.py,sha256=fQ7TYQ3QYcT_YzFzO4f8_bLger7UQfZmuF22I9jYxFA,340
 athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+athena/query/__init__.py,sha256=EIeVs3aDVNAqLyWetSrQldzc2yUdJkf_cJXyrLdMDj0,171
+athena/query/client.py,sha256=3i_yiGD7JM204TsTwY7R1CvGRs0odCh6QkYeNggHbc8,9147
+athena/query/types/__init__.py,sha256=WX-Or2h5NY2sv93ojrZsHcmiFHGYdqd0yxNo-5iGHR4,206
+athena/query/types/query_execute_request_database_asset_ids.py,sha256=aoVl5Xb34Q27hYGuVTnByGIxtHkL67wAwzXh7eJctew,154
 athena/tools/__init__.py,sha256=3n7oOoMebo06MAQqYRE2CX9Q0fTNnKBYE0cTlh1MPkM,165
-athena/tools/client.py,sha256=i_1TD-Cp5ztcCc45h4XxEDmk2xPLfyUtZLueHID2G6g,10888
+athena/tools/client.py,sha256=sMV_--PzUReXXILfKJeS8g9GQzYxxPDZI59wlos_rSo,10565
 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=CbSu1YI-dpp9MpGWCFW5EvrkubPYPWhAWeW9ByksVfY,1142
-athena/types/data_frame_request_out.py,sha256=Od205PAKQL_Eq14GChRx-DIEPe00aIJBOuO89TwaQbQ,1645
+athena/types/__init__.py,sha256=9pojMYMfD3Bk4w4V8FXZ_Apwf1DFDCDsbKtqsMR6EK0,775
+athena/types/data_frame_request_out.py,sha256=1CEBe-baDQi0uz_EgMw0TKGYXGj6KV44cL3ViRTZLKM,1669
 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=lbgAUArEgIYZt3P5Ji4Clp-xyXnKJR7-v5VzXwKu0J8,1168
 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/types/sql_unauthorized_error.py,sha256=c3LR7lDgg66km9J0BYDCRu0hdDYaPqpnfcE3YXXKkVM,1105
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
-athena_intelligence-0.1.99.dist-info/METADATA,sha256=-mIyox4gwnnHaN5DD8gy-TOkUM2APAuOvcVI1H4CZ0c,5273
-athena_intelligence-0.1.99.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.99.dist-info/RECORD,,
+athena_intelligence-0.1.101.dist-info/METADATA,sha256=vYeSC3K17uL8bbSXmpJ1Vz0mn80WW3LiR5zhJouF0Vw,5274
+athena_intelligence-0.1.101.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.101.dist-info/RECORD,,

athena/types/http_validation_error.py

@@ -1,30 +0,0 @@
-# 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 .validation_error import ValidationError
-
-
-class HttpValidationError(pydantic_v1.BaseModel):
-    detail: typing.Optional[typing.List[ValidationError]] = 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/types/validation_error.py

@@ -1,32 +0,0 @@
-# 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 .validation_error_loc_item import ValidationErrorLocItem
-
-
-class ValidationError(pydantic_v1.BaseModel):
-    loc: typing.List[ValidationErrorLocItem]
-    msg: str
-    type: 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_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}