athena-intelligence 0.1.24__py3-none-any.whl → 0.1.26__py3-none-any.whl

athena/base_client.py

@@ -0,0 +1,112 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import httpx
+
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .dataset.client import AsyncDatasetClient, DatasetClient
+from .environment import AthenaEnvironment
+from .message.client import AsyncMessageClient, MessageClient
+from .report.client import AsyncReportClient, ReportClient
+from .snippet.client import AsyncSnippetClient, SnippetClient
+from .sql.client import AsyncSqlClient, SqlClient
+
+
+class BaseAthena:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
+
+    Parameters:
+        - base_url: typing.Optional[str]. The base url to use for requests from the client.
+
+        - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
+
+                                          Defaults to AthenaEnvironment.DEFAULT
+
+        - api_key: str.
+
+        - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
+
+        - httpx_client: typing.Optional[httpx.Client]. The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+    ---
+    from athena.client import Athena
+
+    client = Athena(
+        api_key="YOUR_API_KEY",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        api_key: str,
+        timeout: typing.Optional[float] = 60,
+        httpx_client: typing.Optional[httpx.Client] = None
+    ):
+        self._client_wrapper = SyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            api_key=api_key,
+            httpx_client=httpx.Client(timeout=timeout) if httpx_client is None else httpx_client,
+        )
+        self.message = MessageClient(client_wrapper=self._client_wrapper)
+        self.dataset = DatasetClient(client_wrapper=self._client_wrapper)
+        self.snippet = SnippetClient(client_wrapper=self._client_wrapper)
+        self.report = ReportClient(client_wrapper=self._client_wrapper)
+        self.sql = SqlClient(client_wrapper=self._client_wrapper)
+
+
+class AsyncBaseAthena:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
+
+    Parameters:
+        - base_url: typing.Optional[str]. The base url to use for requests from the client.
+
+        - environment: AthenaEnvironment. The environment to use for requests from the client. from .environment import AthenaEnvironment
+
+                                          Defaults to AthenaEnvironment.DEFAULT
+
+        - api_key: str.
+
+        - timeout: typing.Optional[float]. The timeout to be used, in seconds, for requests by default the timeout is 60 seconds.
+
+        - httpx_client: typing.Optional[httpx.AsyncClient]. The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+    ---
+    from athena.client import AsyncAthena
+
+    client = AsyncAthena(
+        api_key="YOUR_API_KEY",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        api_key: str,
+        timeout: typing.Optional[float] = 60,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None
+    ):
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            api_key=api_key,
+            httpx_client=httpx.AsyncClient(timeout=timeout) if httpx_client is None else httpx_client,
+        )
+        self.message = AsyncMessageClient(client_wrapper=self._client_wrapper)
+        self.dataset = AsyncDatasetClient(client_wrapper=self._client_wrapper)
+        self.snippet = AsyncSnippetClient(client_wrapper=self._client_wrapper)
+        self.report = AsyncReportClient(client_wrapper=self._client_wrapper)
+        self.sql = AsyncSqlClient(client_wrapper=self._client_wrapper)
+
+
+def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

athena/client.py

@@ -4,13 +4,13 @@ import typing
 
 import httpx
 
-from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .base_client import BaseAthena, AsyncBaseAthena
 from .environment import AthenaEnvironment
 from .polling_message_client import \
         MessagePollingClient, AsyncMessagePollingClient
 
 
-class Athena:
+class Athena(BaseAthena):
     """
     Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
 
@@ -43,15 +43,18 @@ class Athena:
         timeout: typing.Optional[float] = 60,
         httpx_client: typing.Optional[httpx.Client] = None
     ):
-        self._client_wrapper = SyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
+        super().__init__(
+            base_url=base_url,
+            environment=environment,
             api_key=api_key,
-            httpx_client=httpx.Client(timeout=timeout) if httpx_client is None else httpx_client,
+            timeout=timeout,
+            httpx_client=httpx_client,
         )
-        self.message = MessagePollingClient(client_wrapper=self._client_wrapper)
+        self.message = MessagePollingClient(
+            client_wrapper=self._client_wrapper)
 
 
-class AsyncAthena:
+class AsyncAthena(AsyncBaseAthena):
     """
     Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propogate to these functions.
 
@@ -84,18 +87,13 @@ class AsyncAthena:
         timeout: typing.Optional[float] = 60,
         httpx_client: typing.Optional[httpx.AsyncClient] = None
     ):
-        self._client_wrapper = AsyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
+        super().__init__(
+            base_url=base_url,
+            environment=environment,
             api_key=api_key,
-            httpx_client=httpx.AsyncClient(timeout=timeout) if httpx_client is None else httpx_client,
+            timeout=timeout,
+            httpx_client=httpx_client,
         )
-        self.message = AsyncMessagePollingClient(client_wrapper=self._client_wrapper)
+        self.message = AsyncMessagePollingClient(
+            client_wrapper=self._client_wrapper)
 
-
-def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:
-    if base_url is not None:
-        return base_url
-    elif environment is not None:
-        return environment.value
-    else:
-        raise Exception("Please pass in either base_url or environment to construct the client")

athena/core/__init__.py

@@ -4,6 +4,7 @@ from .api_error import ApiError
 from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
 from .datetime_utils import serialize_datetime
 from .file import File, convert_file_dict_to_httpx_tuples
+from .http_client import AsyncHttpClient, HttpClient
 from .jsonable_encoder import jsonable_encoder
 from .remove_none_from_dict import remove_none_from_dict
 from .request_options import RequestOptions
@@ -11,8 +12,10 @@ from .request_options import RequestOptions
 __all__ = [
     "ApiError",
     "AsyncClientWrapper",
+    "AsyncHttpClient",
     "BaseClientWrapper",
     "File",
+    "HttpClient",
     "RequestOptions",
     "SyncClientWrapper",
     "convert_file_dict_to_httpx_tuples",

athena/core/client_wrapper.py

@@ -4,6 +4,8 @@ import typing
 
 import httpx
 
+from .http_client import AsyncHttpClient, HttpClient
+
 
 class BaseClientWrapper:
     def __init__(self, *, api_key: str, base_url: str):
@@ -14,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.24",
+            "X-Fern-SDK-Version": "0.1.26",
         }
         headers["X-API-KEY"] = self.api_key
         return headers
@@ -26,10 +28,10 @@ class BaseClientWrapper:
 class SyncClientWrapper(BaseClientWrapper):
     def __init__(self, *, api_key: str, base_url: str, httpx_client: httpx.Client):
         super().__init__(api_key=api_key, base_url=base_url)
-        self.httpx_client = httpx_client
+        self.httpx_client = HttpClient(httpx_client=httpx_client)
 
 
 class AsyncClientWrapper(BaseClientWrapper):
     def __init__(self, *, api_key: str, base_url: str, httpx_client: httpx.AsyncClient):
         super().__init__(api_key=api_key, base_url=base_url)
-        self.httpx_client = httpx_client
+        self.httpx_client = AsyncHttpClient(httpx_client=httpx_client)

athena/core/http_client.py

@@ -0,0 +1,125 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import asyncio
+import email.utils
+import re
+import time
+import typing
+from functools import wraps
+from random import random
+
+import httpx
+
+INITIAL_RETRY_DELAY_SECONDS = 0.5
+MAX_RETRY_DELAY_SECONDS = 10
+MAX_RETRY_DELAY_SECONDS_FROM_HEADER = 30
+
+
+def _parse_retry_after(response_headers: httpx.Headers) -> typing.Optional[float]:
+    """
+    This function parses the `Retry-After` header in a HTTP response and returns the number of seconds to wait.
+
+    Inspired by the urllib3 retry implementation.
+    """
+    retry_after_ms = response_headers.get("retry-after-ms")
+    if retry_after_ms is not None:
+        try:
+            return int(retry_after_ms) / 1000 if retry_after_ms > 0 else 0
+        except Exception:
+            pass
+
+    retry_after = response_headers.get("retry-after")
+    if retry_after is None:
+        return None
+
+    # Attempt to parse the header as an int.
+    if re.match(r"^\s*[0-9]+\s*$", retry_after):
+        seconds = float(retry_after)
+    # Fallback to parsing it as a date.
+    else:
+        retry_date_tuple = email.utils.parsedate_tz(retry_after)
+        if retry_date_tuple is None:
+            return None
+        if retry_date_tuple[9] is None:  # Python 2
+            # Assume UTC if no timezone was specified
+            # On Python2.7, parsedate_tz returns None for a timezone offset
+            # instead of 0 if no timezone is given, where mktime_tz treats
+            # a None timezone offset as local time.
+            retry_date_tuple = retry_date_tuple[:9] + (0,) + retry_date_tuple[10:]
+
+        retry_date = email.utils.mktime_tz(retry_date_tuple)
+        seconds = retry_date - time.time()
+
+    if seconds < 0:
+        seconds = 0
+
+    return seconds
+
+
+def _retry_timeout(response: httpx.Response, retries: int) -> float:
+    """
+    Determine the amount of time to wait before retrying a request.
+    This function begins by trying to parse a retry-after header from the response, and then proceeds to use exponential backoff
+    with a jitter to determine the number of seconds to wait.
+    """
+
+    # If the API asks us to wait a certain amount of time (and it's a reasonable amount), just do what it says.
+    retry_after = _parse_retry_after(response.headers)
+    if retry_after is not None and retry_after <= MAX_RETRY_DELAY_SECONDS_FROM_HEADER:
+        return retry_after
+
+    # Apply exponential backoff, capped at MAX_RETRY_DELAY_SECONDS.
+    retry_delay = min(INITIAL_RETRY_DELAY_SECONDS * pow(2.0, retries), MAX_RETRY_DELAY_SECONDS)
+
+    # Add a randomness / jitter to the retry delay to avoid overwhelming the server with retries.
+    timeout = retry_delay * (1 - 0.25 * random())
+    return timeout if timeout >= 0 else 0
+
+
+def _should_retry(response: httpx.Response) -> bool:
+    retriable_400s = [429, 408, 409]
+    return response.status_code >= 500 or response.status_code in retriable_400s
+
+
+class HttpClient:
+    def __init__(self, *, httpx_client: httpx.Client):
+        self.httpx_client = httpx_client
+
+    # Ensure that the signature of the `request` method is the same as the `httpx.Client.request` method
+    @wraps(httpx.Client.request)
+    def request(
+        self, *args: typing.Any, max_retries: int = 0, retries: int = 0, **kwargs: typing.Any
+    ) -> httpx.Response:
+        response = self.httpx_client.request(*args, **kwargs)
+        if _should_retry(response=response):
+            if max_retries > retries:
+                time.sleep(_retry_timeout(response=response, retries=retries))
+                return self.request(max_retries=max_retries, retries=retries + 1, *args, **kwargs)
+        return response
+
+    @wraps(httpx.Client.stream)
+    def stream(self, *args: typing.Any, max_retries: int = 0, retries: int = 0, **kwargs: typing.Any) -> typing.Any:
+        return self.httpx_client.stream(*args, **kwargs)
+
+
+class AsyncHttpClient:
+    def __init__(self, *, httpx_client: httpx.AsyncClient):
+        self.httpx_client = httpx_client
+
+    # Ensure that the signature of the `request` method is the same as the `httpx.Client.request` method
+    @wraps(httpx.AsyncClient.request)
+    async def request(
+        self, *args: typing.Any, max_retries: int = 0, retries: int = 0, **kwargs: typing.Any
+    ) -> httpx.Response:
+        response = await self.httpx_client.request(*args, **kwargs)
+        if _should_retry(response=response):
+            if max_retries > retries:
+                await asyncio.sleep(_retry_timeout(response=response, retries=retries))
+                return await self.request(max_retries=max_retries, retries=retries + 1, *args, **kwargs)
+        return response
+
+    @wraps(httpx.AsyncClient.request)
+    async def stream(
+        self, *args: typing.Any, max_retries: int = 0, retries: int = 0, **kwargs: typing.Any
+    ) -> typing.Any:
+        return self.httpx_client.stream(*args, **kwargs)

athena/core/request_options.py

@@ -16,6 +16,8 @@ class RequestOptions(typing.TypedDict):
     Attributes:
         - timeout_in_seconds: int. The number of seconds to await an API call before timing out.
 
+        - max_retries: int. The max number of retries to attempt if the API call fails.
+
         - additional_headers: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's header dict
 
         - additional_query_parameters: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's query parameters dict
@@ -24,6 +26,7 @@ class RequestOptions(typing.TypedDict):
     """
 
     timeout_in_seconds: NotRequired[int]
+    max_retries: NotRequired[int]
     additional_headers: NotRequired[typing.Dict[str, typing.Any]]
     additional_query_parameters: NotRequired[typing.Dict[str, typing.Any]]
     additional_body_parameters: NotRequired[typing.Dict[str, typing.Any]]

athena/dataset/client.py

@@ -50,6 +50,8 @@ class DatasetClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(GetDatasetsResponse, _response.json())  # type: ignore
@@ -93,6 +95,8 @@ class AsyncDatasetClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(GetDatasetsResponse, _response.json())  # type: ignore

athena/message/client.py

@@ -67,7 +67,7 @@ class MessageClient:
         """
         _request: typing.Dict[str, typing.Any] = {"content": content}
         if model is not OMIT:
-            _request["model"] = model.value if model is not None else None
+            _request["model"] = model
         if tools is not OMIT:
             _request["tools"] = tools
         if conversation_id is not OMIT:
@@ -97,6 +97,8 @@ class MessageClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(MessageOut, _response.json())  # type: ignore
@@ -141,6 +143,8 @@ class MessageClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(MessageOutDto, _response.json())  # type: ignore
@@ -195,7 +199,7 @@ class AsyncMessageClient:
         """
         _request: typing.Dict[str, typing.Any] = {"content": content}
         if model is not OMIT:
-            _request["model"] = model.value if model is not None else None
+            _request["model"] = model
         if tools is not OMIT:
             _request["tools"] = tools
         if conversation_id is not OMIT:
@@ -225,6 +229,8 @@ class AsyncMessageClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(MessageOut, _response.json())  # type: ignore
@@ -269,6 +275,8 @@ class AsyncMessageClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(MessageOutDto, _response.json())  # type: ignore

athena/report/client.py

@@ -76,6 +76,8 @@ class ReportClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(Report, _response.json())  # type: ignore
@@ -142,6 +144,8 @@ class AsyncReportClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(Report, _response.json())  # type: ignore

athena/snippet/client.py

@@ -50,6 +50,8 @@ class SnippetClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(GetSnippetsResponse, _response.json())  # type: ignore
@@ -93,6 +95,8 @@ class AsyncSnippetClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(GetSnippetsResponse, _response.json())  # type: ignore

athena/sql/client.py

@@ -70,6 +70,8 @@ class SqlClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(SqlResults, _response.json())  # type: ignore
@@ -130,6 +132,8 @@ class AsyncSqlClient:
             timeout=request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
             else 60,
+            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.parse_obj_as(SqlResults, _response.json())  # type: ignore

{athena_intelligence-0.1.24.dist-info → athena_intelligence-0.1.26.dist-info}/METADATA

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

{athena_intelligence-0.1.24.dist-info → athena_intelligence-0.1.26.dist-info}/RECORD

@@ -1,28 +1,30 @@
 athena/__init__.py,sha256=93_n6iw5UHLlzbN6H_LiJHsWeWu8RIA6LEB_-opfgRc,901
-athena/client.py,sha256=nFVJkayT6k0-DeqX-Ef7eVB-VHLHEq6wsqaZHy5Pp48,4004
-athena/core/__init__.py,sha256=95onSWXymaL0iV-nBsuKNgjh1LbsymAkRkx7POn1nc0,696
+athena/base_client.py,sha256=RfS3KAPYhyRK8sdakv_KXJGu-jv9D6iu0iclVFbnWuE,4792
+athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
+athena/core/__init__.py,sha256=RWfyDqkzWsf8e3VGc3NV60MovfJbg5XWzNFGB2DZ0hA,790
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=S7yhJhjjs6W50my1uC01J8jTBGr0gPicyUFNMJYZ07w,1089
+athena/core/client_wrapper.py,sha256=xJr-oADCmjCUkaDTBrhXbRUHA0QW6E6PATgnUaeE9u4,1198
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
+athena/core/http_client.py,sha256=LI0yP3jUyE0Ue7oyBcI9nyo1pljOwh9Y5ycTeIpKwOg,4882
 athena/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
 athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
-athena/core/request_options.py,sha256=9n3ZHWZkw60MGp7GbdjGEuGo73Pa-6JeusRGhS533aA,1297
+athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
 athena/dataset/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/dataset/client.py,sha256=GSEv6TAIxrfzSgO-xgI1R2xVFCcnju4DYcbRrf0_VPU,4198
+athena/dataset/client.py,sha256=pohi3ugsXM6BKz4iN__cIKaqCLGlZDkzdaxDEgYnBLY,4472
 athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
 athena/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
 athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
 athena/message/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/message/client.py,sha256=KUdDnXGBMxWsr9FQ_GjzJGprOBd1_rOwAuLY7QJNjJQ,11741
+athena/message/client.py,sha256=2jSeLD6LBDTWdYIHiz_iqdB78KX-nlzqi0NRosmyTgM,12215
 athena/polling_message_client.py,sha256=dmmycImvog2niEFFPo4rE5xMJHUlq9NqAr4xlFK6_Os,3998
 athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 athena/report/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/report/client.py,sha256=-g2icjmOjM67Elq5hCDENCVYbHfFPAlH4oc5jJnQn9c,6254
+athena/report/client.py,sha256=sGJDrgk_E1SPleRYNhvspmsz-G3FQwMW-3alFzZPquE,6528
 athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/snippet/client.py,sha256=tU7nOn_cefT3m6vM23UrT4qIIcgEUjvl_j2XgGI7iDs,4198
+athena/snippet/client.py,sha256=O1Q7ytPNEI4E1cGgu8Dv2oAzrGMhAEsmMb0LXOvsSSw,4472
 athena/sql/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/sql/client.py,sha256=vEv2DW47wuBgvtcKqoVtSa5wPD9u34ncvjVbqwQTIbQ,5961
+athena/sql/client.py,sha256=cKMrwMeBmtsKdRMSdz7b4zhTdHmNT9hAm-wI7fabvpo,6235
 athena/types/__init__.py,sha256=kSz8xv8p-5znJI-sg3b6yD0E0bi9pkI1J8F60D5hu78,918
 athena/types/dataset.py,sha256=LqLZn_Br9jGg4toU_6wMoZQMIgAea6Hdn11xJpgi_M8,966
 athena/types/get_datasets_response.py,sha256=gikRLR9f9jQSfbMJedBh9JicH_9ccJ851nUmIZXfEvg,926
@@ -38,6 +40,6 @@ athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,6
 athena/types/tools.py,sha256=Vfb-D9CYcPZH6MBATNODgfXjFyBpCs4qbkqpCMl7eBM,1277
 athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
 athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
-athena_intelligence-0.1.24.dist-info/METADATA,sha256=uhgDFypQRDFrSVm182j5uAFdI0JCrJ8pwL_E3bRvS_M,4738
-athena_intelligence-0.1.24.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.24.dist-info/RECORD,,
+athena_intelligence-0.1.26.dist-info/METADATA,sha256=x0iRKiTmjRf3IA9j2DsDxk7deH4XOAiFowDPbokxuwE,4738
+athena_intelligence-0.1.26.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.26.dist-info/RECORD,,