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

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/client_wrapper.py

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

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

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

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

@@ -1,9 +1,9 @@
 athena/__init__.py,sha256=93_n6iw5UHLlzbN6H_LiJHsWeWu8RIA6LEB_-opfgRc,901
 athena/base_client.py,sha256=RfS3KAPYhyRK8sdakv_KXJGu-jv9D6iu0iclVFbnWuE,4792
-athena/client.py,sha256=nFVJkayT6k0-DeqX-Ef7eVB-VHLHEq6wsqaZHy5Pp48,4004
+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=Inf7Leqn5tSmqObhDxR14dgeHGGysx5VKMaRRH6fDcI,1198
+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
@@ -40,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.25.dist-info/METADATA,sha256=N08JQ0f2xD8SqIOdWbVJW5jrX3f5HGot_zuN3XmaeL4,4738
-athena_intelligence-0.1.25.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.25.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,,