athena-intelligence 0.1.25__tar.gz → 0.1.26__tar.gz

{athena_intelligence-0.1.25 → athena_intelligence-0.1.26}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.25"
+version = "0.1.26"
 description = ""
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.25 → athena_intelligence-0.1.26}/src/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_intelligence-0.1.25 → athena_intelligence-0.1.26}/src/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