airia 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

airia/client/async_client.py

@@ -21,6 +21,7 @@ class AiriaAsyncClient(AiriaBaseClient):
 
     def __init__(
         self,
+        base_url: str = "https://api.airia.ai/",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -30,20 +31,29 @@ class AiriaAsyncClient(AiriaBaseClient):
         Initialize the asynchronous Airia API client.
 
         Args:
+            base_url: Base URL of the Airia API.
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
             timeout: Request timeout in seconds.
             log_requests: Whether to log API requests and responses. Default is False.
             custom_logger: Optional custom logger object to use for logging. If not provided, will use a default logger when `log_requests` is True.
         """
-        super().__init__(api_key, timeout, log_requests, custom_logger)
+        super().__init__(
+            base_url=base_url,
+            api_key=api_key,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
 
         # Session will be initialized in __aenter__
         self.session = None
         self.headers = {"Content-Type": "application/json"}
-    
+
     @classmethod
     def with_openai_gateway(
         cls,
+        base_url: str = "https://api.airia.ai/",
+        gateway_url: str = "https://gateway.airia.ai/openai/v1",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -54,6 +64,8 @@ class AiriaAsyncClient(AiriaBaseClient):
         Initialize the asynchronous Airia API client with AsyncOpenAI gateway capabilities.
 
         Args:
+            base_url: Base URL of the Airia API.
+            gateway_url: Base URL of the Airia Gateway API.
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
             timeout: Request timeout in seconds.
             log_requests: Whether to log API requests and responses. Default is False.
@@ -65,15 +77,17 @@ class AiriaAsyncClient(AiriaBaseClient):
         api_key = cls._get_api_key(api_key)
         cls.openai = AsyncOpenAI(
             api_key=api_key,
-            base_url="https://gateway.airia.ai/openai/v1",
+            base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(api_key, timeout, log_requests, custom_logger)
+        return cls(base_url, api_key, timeout, log_requests, custom_logger)
 
     @classmethod
     def with_anthropic_gateway(
         cls,
+        base_url: str = "https://api.airia.ai/",
+        gateway_url: str = "https://gateway.airia.ai/anthropic",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -84,6 +98,8 @@ class AiriaAsyncClient(AiriaBaseClient):
         Initialize the asynchronous Airia API client with AsyncAnthropic gateway capabilities.
 
         Args:
+            base_url: Base URL of the Airia API.
+            gateway_url: Base URL of the Airia Gateway API.
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
             timeout: Request timeout in seconds.
             log_requests: Whether to log API requests and responses. Default is False.
@@ -95,11 +111,11 @@ class AiriaAsyncClient(AiriaBaseClient):
         api_key = cls._get_api_key(api_key)
         cls.anthropic = AsyncAnthropic(
             api_key=api_key,
-            base_url="https://gateway.airia.ai/anthropic",
+            base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(api_key, timeout, log_requests, custom_logger)
+        return cls(base_url, api_key, timeout, log_requests, custom_logger)
 
     async def __aenter__(self):
         """Async context manager entry point."""

airia/client/base_client.py

@@ -16,6 +16,7 @@ class AiriaBaseClient:
 
     def __init__(
         self,
+        base_url: str = "https://api.airia.ai/",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -34,7 +35,7 @@ class AiriaBaseClient:
         self.api_key = self.__class__._get_api_key(api_key)
 
         # Store configuration
-        self.base_url = "https://api.airia.ai/"
+        self.base_url = base_url
         self.timeout = timeout
         self.log_requests = log_requests
 

airia/client/sync_client.py

@@ -21,6 +21,7 @@ class AiriaClient(AiriaBaseClient):
 
     def __init__(
         self,
+        base_url: str = "https://api.airia.ai/",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -30,12 +31,19 @@ class AiriaClient(AiriaBaseClient):
         Initialize the synchronous Airia API client.
 
         Args:
+            base_url: Base URL of the Airia API.
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
             timeout: Request timeout in seconds.
             log_requests: Whether to log API requests and responses. Default is False.
             custom_logger: Optional custom logger object to use for logging. If not provided, will use a default logger when `log_requests` is True.
         """
-        super().__init__(api_key, timeout, log_requests, custom_logger)
+        super().__init__(
+            base_url=base_url,
+            api_key=api_key,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
 
         # Initialize session for synchronous requests
         self.session = requests.Session()
@@ -44,6 +52,8 @@ class AiriaClient(AiriaBaseClient):
     @classmethod
     def with_openai_gateway(
         cls,
+        base_url: str = "https://api.airia.ai/",
+        gateway_url: str = "https://gateway.airia.ai/openai/v1",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -54,6 +64,8 @@ class AiriaClient(AiriaBaseClient):
         Initialize the synchronous Airia API client with OpenAI gateway capabilities.
 
         Args:
+            base_url: Base URL of the Airia API.
+            gateway_url: Base URL of the Airia Gateway API.
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
             timeout: Request timeout in seconds.
             log_requests: Whether to log API requests and responses. Default is False.
@@ -65,15 +77,17 @@ class AiriaClient(AiriaBaseClient):
         api_key = cls._get_api_key(api_key)
         cls.openai = OpenAI(
             api_key=api_key,
-            base_url="https://gateway.airia.ai/openai/v1",
+            base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(api_key, timeout, log_requests, custom_logger)
+        return cls(base_url, api_key, timeout, log_requests, custom_logger)
 
     @classmethod
     def with_anthropic_gateway(
         cls,
+        base_url: str = "https://api.airia.ai/",
+        gateway_url: str = "https://gateway.airia.ai/anthropic",
         api_key: Optional[str] = None,
         timeout: float = 30.0,
         log_requests: bool = False,
@@ -84,6 +98,8 @@ class AiriaClient(AiriaBaseClient):
         Initialize the synchronous Airia API client with Anthropic gateway capabilities.
 
         Args:
+            base_url: Base URL of the Airia API.
+            gateway_url: Base URL of the Airia Gateway API.
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
             timeout: Request timeout in seconds.
             log_requests: Whether to log API requests and responses. Default is False.
@@ -95,11 +111,11 @@ class AiriaClient(AiriaBaseClient):
         api_key = cls._get_api_key(api_key)
         cls.anthropic = Anthropic(
             api_key=api_key,
-            base_url="https://gateway.airia.ai/anthropic",
+            base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(api_key, timeout, log_requests, custom_logger)
+        return cls(base_url, api_key, timeout, log_requests, custom_logger)
 
     def _handle_exception(self, e: requests.HTTPError, url: str, correlation_id: str):
         # Log the error response if enabled

{airia-0.1.4.dist-info → airia-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.4
+Version: 0.1.5
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT
@@ -176,6 +176,20 @@ This will create both wheel and source distribution in the `dist/` directory.
 
 ## Quick Start
 
+### Client Instantiation
+
+```python
+from airia import AiriaClient
+
+client = AiriaClient(
+    base_url="https://api.airia.com",        # Default: "https://api.airia.com"
+    api_key=None,                  # Or set AIRIA_API_KEY environment variable
+    timeout=30.0,                            # Request timeout in seconds (default: 30.0)
+    log_requests=False,                       # Enable request/response logging (default: False)
+    custom_logger=None                       # Use custom logger (default: None - uses built-in)
+)
+```
+
 ### Synchronous Usage
 
 ```python
@@ -208,7 +222,7 @@ response = client.execute_pipeline(
     async_output=True
 )
 
-for c in resp.stream:
+for c in response.stream:
     print(c, end="")
 ```
 
@@ -244,7 +258,7 @@ async def main():
             user_input="Tell me about quantum computing",
             async_output=True
         )
-        async for c in resp.stream:
+        async for c in response.stream:
             print(c, end="")
 
 asyncio.run(main())
@@ -292,6 +306,8 @@ response = client.anthropic.messages.create(
 print(response.content[0].text)
 ```
 
+You can set the Gateway URL by passing the `gateway_url` parameter when using the gateway constructors. The default values are `https://gateway.airia.ai/openai/v1` for OpenAI and `https://gateway.airia.ai/anthropic` for Anthropic.
+
 ### Asynchronous Gateway Usage
 
 Both gateways also support asynchronous usage:

{airia-0.1.4.dist-info → airia-0.1.5.dist-info}/RECORD

@@ -2,15 +2,15 @@ airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
 airia/exceptions.py,sha256=4Z55n-cRJrtTa5-pZBIK2oZD4-Z99aUtKx_kfTFYY5o,1146
 airia/logs.py,sha256=17YZ4IuzOF0m5bgofj9-QYlJ2BYR2kRZbBVQfFSLFEk,5441
 airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
-airia/client/async_client.py,sha256=Xd1GAGxQvXVh0-hH8vE4HOeaNY_Fawh5iRxeS26CInY,18173
-airia/client/base_client.py,sha256=3oUcCWWq-DmPbiJke_HHPXm9UzDvD-MzrnoiKnBjcdk,6810
-airia/client/sync_client.py,sha256=p_auqkfyP-9YuCoppl43SyMdKek30kA6M05QI0LYunE,17689
+airia/client/async_client.py,sha256=4JU6RePXjznulnQXkfo9ZVX3zLeUyclOgxwlaM7-5Js,18816
+airia/client/base_client.py,sha256=IEVsskGpiEL3jw7YLc96YWUZ5rpUZula52IZHe2D8mE,6844
+airia/client/sync_client.py,sha256=mCLViWkW-z8QhCy8CmlEsT8dssBb1g6V2B3fAYgYU1A,18336
 airia/types/__init__.py,sha256=OOFtJ0VIbO98px89u7cq645iL7CYDOel43g85IAjRRw,562
 airia/types/api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
 airia/types/pipeline_execution.py,sha256=obp8KOz-ShNxEciF1YQCSpHXPoJUyEa_9uPv-VHf6YA,699
 airia/types/request_data.py,sha256=RbVPWPRIYuT7FaolJKn19IVzuQKbLM4VhXM5r7UXTUY,186
-airia-0.1.4.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.4.dist-info/METADATA,sha256=VOWxozh4xZXAHUXH6_ZyCUTIU95LXxly3Yfs-ayi0sk,9966
-airia-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.4.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.4.dist-info/RECORD,,
+airia-0.1.5.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.5.dist-info/METADATA,sha256=WIPLholUrToVlvO4K269qOSYHT4fD7XHpVSrjowOF6A,10745
+airia-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.5.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.5.dist-info/RECORD,,