airia 0.1.9__tar.gz → 0.1.10__tar.gz

{airia-0.1.9 → airia-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.9
+Version: 0.1.10
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT
@@ -43,7 +43,9 @@ Airia Python API Library that provides a clean and intuitive interface to intera
 - **Gateway Support**: Seamlessly integrate with OpenAI and Anthropic services through Airia gateways
 - **Error Handling**: Comprehensive error handling with custom exceptions
 - **Logging**: Built-in configurable logging with correlation ID support for request tracing
-- **API Key Management**: Flexible API key configuration via parameters or environment variables
+- **Flexible Authentication**: Support for both API keys and bearer tokens with flexible configuration
+- **API Key Management**: API key configuration via parameters or environment variables
+- **Bearer Token Support**: Bearer token authentication for ephemeral, short-lived credentials
 
 ## Installation
 
@@ -181,17 +183,38 @@ This will create both wheel and source distribution in the `dist/` directory.
 ```python
 from airia import AiriaClient
 
+# API Key Authentication
 client = AiriaClient(
     base_url="https://api.airia.ai",        # Default: "https://api.airia.ai"
-    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)
+    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)
+)
+
+# Bearer Token Authentication
+client = AiriaClient(
+    base_url="https://api.airia.ai",        # Default: "https://api.airia.ai"
+    bearer_token="your_bearer_token",       # Must be provided explicitly (no env var fallback)
+    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)
+)
+
+# Convenience method for bearer token
+client = AiriaClient.with_bearer_token(
+    bearer_token="your_bearer_token",
+    base_url="https://api.airia.ai",        # Optional, uses default if not provided
+    timeout=30.0,                           # Optional, uses default if not provided
+    log_requests=False,                     # Optional, uses default if not provided
+    custom_logger=None                      # Optional, uses default if not provided
 )
 ```
 
 ### Synchronous Usage
 
+#### With API Key
+
 ```python
 from airia import AiriaClient
 
@@ -207,6 +230,23 @@ response = client.execute_pipeline(
 print(response.result)
 ```
 
+#### With Bearer Token
+
+```python
+from airia import AiriaClient
+
+# Initialize client with bearer token
+client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
+
+# Execute a pipeline
+response = client.execute_pipeline(
+    pipeline_id="your_pipeline_id",
+    user_input="Tell me about quantum computing"
+)
+
+print(response.result)
+```
+
 #### Synchronous Streaming
 
 ```python
@@ -214,6 +254,7 @@ from airia import AiriaClient
 
 # Initialize client (API key can be passed directly or via AIRIA_API_KEY environment variable)
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 # Execute a pipeline
 response = client.execute_pipeline(
@@ -228,6 +269,8 @@ for c in response.stream:
 
 ### Asynchronous Usage
 
+#### With API Key
+
 ```python
 import asyncio
 from airia import AiriaAsyncClient
@@ -243,6 +286,23 @@ async def main():
 asyncio.run(main())
 ```
 
+#### With Bearer Token
+
+```python
+import asyncio
+from airia import AiriaAsyncClient
+
+async def main():
+    client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
+    response = await client.execute_pipeline(
+        pipeline_id="your_pipeline_id",
+        user_input="Tell me about quantum computing"
+    )
+    print(response.result)
+
+asyncio.run(main())
+```
+
 #### Asynchronous Streaming
 
 ```python
@@ -251,6 +311,7 @@ from airia import AiriaAsyncClient
 
 async def main():
     client = AiriaAsyncClient(api_key="your_api_key")
+    # Or with bearer token: client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
     response = await client.execute_pipeline(
         pipeline_id="your_pipeline_id",
         user_input="Tell me about quantum computing",
@@ -320,6 +381,7 @@ from airia import AiriaClient
 from airia.types import AgentModelStreamFragmentMessage
 
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 response = client.execute_pipeline(
     pipeline_id="your_pipeline_id",
@@ -343,6 +405,7 @@ You can retrieve detailed configuration information about a pipeline using the `
 from airia import AiriaClient
 
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 # Get pipeline configuration
 config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
@@ -351,10 +414,35 @@ config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
 print(f"Pipeline Name: {config.agent.name}")
 ```
 
+## Authentication Methods
+
+Airia supports two authentication methods:
+
+### API Keys
+- Can be passed as a parameter or via `AIRIA_API_KEY` environment variable
+- Support gateway functionality (OpenAI and Anthropic gateways)
+- Suitable for long-term, persistent authentication
+
+### Bearer Tokens
+- Must be provided explicitly (no environment variable fallback)
+- **Important**: Bearer tokens cannot be used with gateway functionality
+- Suitable for ephemeral, short-lived authentication scenarios
+- Ideal for temporary access or programmatic token generation
+
+```python
+# ✅ API key with gateway support
+client = AiriaClient.with_openai_gateway(api_key="your_api_key")
+
+# ❌ Bearer token with gateway - NOT SUPPORTED
+# client = AiriaClient.with_openai_gateway(bearer_token="token")  # This won't work
+```
+
 ## Gateway Usage
 
 Airia provides gateway capabilities for popular AI services like OpenAI and Anthropic, allowing you to use your Airia API key with these services.
 
+> **Note**: Gateway functionality requires API key authentication. Bearer tokens are not supported for gateway usage.
+
 ### OpenAI Gateway
 
 ```python
@@ -482,7 +570,11 @@ console_logger = configure_logging(
 The SDK uses custom exceptions to provide clear error messages:
 
 ```python
-from airia import AiriaAPIError
+from airia import AiriaAPIError, AiriaClient
+
+# Works with both API keys and bearer tokens
+client = AiriaClient(api_key="your_api_key")
+# Or: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 try:
     response = client.execute_pipeline(

{airia-0.1.9 → airia-0.1.10}/README.md

@@ -13,7 +13,9 @@ Airia Python API Library that provides a clean and intuitive interface to intera
 - **Gateway Support**: Seamlessly integrate with OpenAI and Anthropic services through Airia gateways
 - **Error Handling**: Comprehensive error handling with custom exceptions
 - **Logging**: Built-in configurable logging with correlation ID support for request tracing
-- **API Key Management**: Flexible API key configuration via parameters or environment variables
+- **Flexible Authentication**: Support for both API keys and bearer tokens with flexible configuration
+- **API Key Management**: API key configuration via parameters or environment variables
+- **Bearer Token Support**: Bearer token authentication for ephemeral, short-lived credentials
 
 ## Installation
 
@@ -151,17 +153,38 @@ This will create both wheel and source distribution in the `dist/` directory.
 ```python
 from airia import AiriaClient
 
+# API Key Authentication
 client = AiriaClient(
     base_url="https://api.airia.ai",        # Default: "https://api.airia.ai"
-    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)
+    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)
+)
+
+# Bearer Token Authentication
+client = AiriaClient(
+    base_url="https://api.airia.ai",        # Default: "https://api.airia.ai"
+    bearer_token="your_bearer_token",       # Must be provided explicitly (no env var fallback)
+    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)
+)
+
+# Convenience method for bearer token
+client = AiriaClient.with_bearer_token(
+    bearer_token="your_bearer_token",
+    base_url="https://api.airia.ai",        # Optional, uses default if not provided
+    timeout=30.0,                           # Optional, uses default if not provided
+    log_requests=False,                     # Optional, uses default if not provided
+    custom_logger=None                      # Optional, uses default if not provided
 )
 ```
 
 ### Synchronous Usage
 
+#### With API Key
+
 ```python
 from airia import AiriaClient
 
@@ -177,6 +200,23 @@ response = client.execute_pipeline(
 print(response.result)
 ```
 
+#### With Bearer Token
+
+```python
+from airia import AiriaClient
+
+# Initialize client with bearer token
+client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
+
+# Execute a pipeline
+response = client.execute_pipeline(
+    pipeline_id="your_pipeline_id",
+    user_input="Tell me about quantum computing"
+)
+
+print(response.result)
+```
+
 #### Synchronous Streaming
 
 ```python
@@ -184,6 +224,7 @@ from airia import AiriaClient
 
 # Initialize client (API key can be passed directly or via AIRIA_API_KEY environment variable)
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 # Execute a pipeline
 response = client.execute_pipeline(
@@ -198,6 +239,8 @@ for c in response.stream:
 
 ### Asynchronous Usage
 
+#### With API Key
+
 ```python
 import asyncio
 from airia import AiriaAsyncClient
@@ -213,6 +256,23 @@ async def main():
 asyncio.run(main())
 ```
 
+#### With Bearer Token
+
+```python
+import asyncio
+from airia import AiriaAsyncClient
+
+async def main():
+    client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
+    response = await client.execute_pipeline(
+        pipeline_id="your_pipeline_id",
+        user_input="Tell me about quantum computing"
+    )
+    print(response.result)
+
+asyncio.run(main())
+```
+
 #### Asynchronous Streaming
 
 ```python
@@ -221,6 +281,7 @@ from airia import AiriaAsyncClient
 
 async def main():
     client = AiriaAsyncClient(api_key="your_api_key")
+    # Or with bearer token: client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
     response = await client.execute_pipeline(
         pipeline_id="your_pipeline_id",
         user_input="Tell me about quantum computing",
@@ -290,6 +351,7 @@ from airia import AiriaClient
 from airia.types import AgentModelStreamFragmentMessage
 
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 response = client.execute_pipeline(
     pipeline_id="your_pipeline_id",
@@ -313,6 +375,7 @@ You can retrieve detailed configuration information about a pipeline using the `
 from airia import AiriaClient
 
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 # Get pipeline configuration
 config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
@@ -321,10 +384,35 @@ config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
 print(f"Pipeline Name: {config.agent.name}")
 ```
 
+## Authentication Methods
+
+Airia supports two authentication methods:
+
+### API Keys
+- Can be passed as a parameter or via `AIRIA_API_KEY` environment variable
+- Support gateway functionality (OpenAI and Anthropic gateways)
+- Suitable for long-term, persistent authentication
+
+### Bearer Tokens
+- Must be provided explicitly (no environment variable fallback)
+- **Important**: Bearer tokens cannot be used with gateway functionality
+- Suitable for ephemeral, short-lived authentication scenarios
+- Ideal for temporary access or programmatic token generation
+
+```python
+# ✅ API key with gateway support
+client = AiriaClient.with_openai_gateway(api_key="your_api_key")
+
+# ❌ Bearer token with gateway - NOT SUPPORTED
+# client = AiriaClient.with_openai_gateway(bearer_token="token")  # This won't work
+```
+
 ## Gateway Usage
 
 Airia provides gateway capabilities for popular AI services like OpenAI and Anthropic, allowing you to use your Airia API key with these services.
 
+> **Note**: Gateway functionality requires API key authentication. Bearer tokens are not supported for gateway usage.
+
 ### OpenAI Gateway
 
 ```python
@@ -452,7 +540,11 @@ console_logger = configure_logging(
 The SDK uses custom exceptions to provide clear error messages:
 
 ```python
-from airia import AiriaAPIError
+from airia import AiriaAPIError, AiriaClient
+
+# Works with both API keys and bearer tokens
+client = AiriaClient(api_key="your_api_key")
+# Or: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 try:
     response = client.execute_pipeline(

{airia-0.1.9 → airia-0.1.10}/airia/client/async_client.py

@@ -6,6 +6,7 @@ from urllib.parse import urljoin
 import aiohttp
 import loguru
 
+from ..constants import DEFAULT_ANTHROPIC_GATEWAY_URL, DEFAULT_BASE_URL, DEFAULT_OPENAI_GATEWAY_URL, DEFAULT_TIMEOUT
 from ..exceptions import AiriaAPIError
 from ..types import (
     ApiVersion,
@@ -26,9 +27,10 @@ class AiriaAsyncClient(AiriaBaseClient):
 
     def __init__(
         self,
-        base_url: str = "https://api.airia.ai/",
+        base_url: str = DEFAULT_BASE_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        bearer_token: Optional[str] = None,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
     ):
@@ -38,6 +40,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         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.
+            bearer_token: Bearer token for authentication. Must be provided explicitly (no environment variable fallback).
             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.
@@ -45,6 +48,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         super().__init__(
             base_url=base_url,
             api_key=api_key,
+            bearer_token=bearer_token,
             timeout=timeout,
             log_requests=log_requests,
             custom_logger=custom_logger,
@@ -80,10 +84,10 @@ class AiriaAsyncClient(AiriaBaseClient):
     @classmethod
     def with_openai_gateway(
         cls,
-        base_url: str = "https://api.airia.ai/",
-        gateway_url: str = "https://gateway.airia.ai/openai/v1",
+        base_url: str = DEFAULT_BASE_URL,
+        gateway_url: str = DEFAULT_OPENAI_GATEWAY_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
         **kwargs,
@@ -102,22 +106,22 @@ class AiriaAsyncClient(AiriaBaseClient):
         """
         from openai import AsyncOpenAI
 
-        api_key = cls._get_api_key(api_key)
+        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
         cls.openai = AsyncOpenAI(
-            api_key=api_key,
+            api_key=client.api_key,
             base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(base_url, api_key, timeout, log_requests, custom_logger)
+        return client
 
     @classmethod
     def with_anthropic_gateway(
         cls,
-        base_url: str = "https://api.airia.ai/",
-        gateway_url: str = "https://gateway.airia.ai/anthropic",
+        base_url: str = DEFAULT_BASE_URL,
+        gateway_url: str = DEFAULT_ANTHROPIC_GATEWAY_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
         **kwargs,
@@ -136,14 +140,41 @@ class AiriaAsyncClient(AiriaBaseClient):
         """
         from anthropic import AsyncAnthropic
 
-        api_key = cls._get_api_key(api_key)
+        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
         cls.anthropic = AsyncAnthropic(
-            api_key=api_key,
+            api_key=client.api_key,
             base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(base_url, api_key, timeout, log_requests, custom_logger)
+        return client
+
+    @classmethod
+    def with_bearer_token(
+        cls,
+        bearer_token: str,
+        base_url: str = DEFAULT_BASE_URL,
+        timeout: float = DEFAULT_TIMEOUT,
+        log_requests: bool = False,
+        custom_logger: Optional["loguru.Logger"] = None,
+    ):
+        """
+        Initialize the asynchronous Airia API client with bearer token authentication.
+
+        Args:
+            bearer_token: Bearer token for authentication.
+            base_url: Base URL of the Airia API.
+            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.
+        """
+        return cls(
+            base_url=base_url,
+            bearer_token=bearer_token,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
 
     def _handle_exception(
         self, e: aiohttp.ClientResponseError, url: str, correlation_id: str
@@ -159,12 +190,12 @@ class AiriaAsyncClient(AiriaBaseClient):
         # Extract error details from response
         error_message = e.message
 
-        # Make sure API key is not included in error messages
-        sanitized_message = (
-            error_message.replace(self.api_key, "[REDACTED]")
-            if self.api_key in error_message
-            else error_message
-        )
+        # Make sure sensitive auth information is not included in error messages
+        sanitized_message = error_message
+        if self.api_key and self.api_key in sanitized_message:
+            sanitized_message = sanitized_message.replace(self.api_key, "[REDACTED]")
+        if self.bearer_token and self.bearer_token in sanitized_message:
+            sanitized_message = sanitized_message.replace(self.bearer_token, "[REDACTED]")
 
         # Raise custom exception with status code and sanitized message
         raise AiriaAPIError(status_code=e.status, message=sanitized_message) from e

{airia-0.1.9 → airia-0.1.10}/airia/client/base_client.py

@@ -5,6 +5,7 @@ from urllib.parse import urljoin
 
 import loguru
 
+from ..constants import DEFAULT_BASE_URL, DEFAULT_TIMEOUT
 from ..logs import configure_logging, set_correlation_id
 from ..types import ApiVersion, RequestData
 
@@ -17,9 +18,10 @@ class AiriaBaseClient:
 
     def __init__(
         self,
-        base_url: str = "https://api.airia.ai/",
+        base_url: str = DEFAULT_BASE_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        bearer_token: Optional[str] = None,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
     ):
@@ -28,12 +30,13 @@ class AiriaBaseClient:
 
         Args:
             api_key: API key for authentication. If not provided, will attempt to use AIRIA_API_KEY environment variable.
+            bearer_token: Bearer token for authentication. Must be provided explicitly (no environment variable fallback).
             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.
         """
-        # Resolve API key: parameter takes precedence over environment variable
-        self.api_key = self.__class__._get_api_key(api_key)
+        # Resolve authentication credentials
+        self.api_key, self.bearer_token = self.__class__._resolve_auth_credentials(api_key, bearer_token)
 
         # Store configuration
         self.base_url = base_url
@@ -44,27 +47,43 @@ class AiriaBaseClient:
         self.logger = configure_logging() if custom_logger is None else custom_logger
 
     @staticmethod
-    def _get_api_key(api_key: Optional[str] = None):
+    def _resolve_auth_credentials(api_key: Optional[str] = None, bearer_token: Optional[str] = None):
         """
-        Get the API key from either the provided parameter or environment variable.
+        Resolve authentication credentials from parameters and environment variables.
 
         Args:
             api_key (Optional[str]): The API key provided as a parameter. Defaults to None.
+            bearer_token (Optional[str]): The bearer token provided as a parameter. Defaults to None.
 
         Returns:
-            str: The resolved API key.
+            tuple: (api_key, bearer_token) - exactly one will be non-None
 
         Raises:
-            ValueError: If no API key is provided through either method.
+            ValueError: If no authentication method is provided or if both are provided.
         """
-        api_key = api_key or os.environ.get("AIRIA_API_KEY")
-
-        if not api_key:
+        # Check for explicit conflict first
+        if api_key and bearer_token:
             raise ValueError(
-                "API key must be provided either as a parameter or through the AIRIA_API_KEY environment variable."
+                "Cannot provide both api_key and bearer_token. Please use only one authentication method."
             )
-
-        return api_key
+        
+        # If bearer token is explicitly provided, use it exclusively
+        if bearer_token:
+            return None, bearer_token
+        
+        # If API key is explicitly provided, use it exclusively
+        if api_key:
+            return api_key, None
+            
+        # If neither is provided explicitly, fall back to environment variable
+        resolved_api_key = os.environ.get("AIRIA_API_KEY")
+        if resolved_api_key:
+            return resolved_api_key, None
+        
+        # No authentication method found
+        raise ValueError(
+            "Authentication required. Provide either api_key (or set AIRIA_API_KEY environment variable) or bearer_token."
+        )
 
     def _prepare_request(
         self,
@@ -76,13 +95,18 @@ class AiriaBaseClient:
         # Set correlation ID if provided or generate a new one
         correlation_id = set_correlation_id(correlation_id)
 
-        # Add the X-API-KEY header and correlation ID
+        # Set up base headers
         headers = {
-            "X-API-KEY": self.api_key,
             "X-Correlation-ID": correlation_id,
             "Content-Type": "application/json",
         }
 
+        # Add authentication header based on the method used
+        if self.api_key:
+            headers["X-API-KEY"] = self.api_key
+        elif self.bearer_token:
+            headers["Authorization"] = f"Bearer {self.bearer_token}"
+
         # Log the request if enabled
         if self.log_requests:
             # Create a sanitized copy of headers and params for logging
@@ -92,6 +116,8 @@ class AiriaBaseClient:
             # Filter out sensitive headers
             if "X-API-KEY" in log_headers:
                 log_headers["X-API-KEY"] = "[REDACTED]"
+            if "Authorization" in log_headers:
+                log_headers["Authorization"] = "[REDACTED]"
 
             # Process payload for logging
             log_payload = payload.copy() if payload is not None else {}

{airia-0.1.9 → airia-0.1.10}/airia/client/sync_client.py

@@ -4,6 +4,7 @@ from urllib.parse import urljoin
 import loguru
 import requests
 
+from ..constants import DEFAULT_ANTHROPIC_GATEWAY_URL, DEFAULT_BASE_URL, DEFAULT_OPENAI_GATEWAY_URL, DEFAULT_TIMEOUT
 from ..exceptions import AiriaAPIError
 from ..types import (
     ApiVersion,
@@ -24,9 +25,10 @@ class AiriaClient(AiriaBaseClient):
 
     def __init__(
         self,
-        base_url: str = "https://api.airia.ai/",
+        base_url: str = DEFAULT_BASE_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        bearer_token: Optional[str] = None,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
     ):
@@ -36,6 +38,7 @@ class AiriaClient(AiriaBaseClient):
         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.
+            bearer_token: Bearer token for authentication. Must be provided explicitly (no environment variable fallback).
             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.
@@ -43,6 +46,7 @@ class AiriaClient(AiriaBaseClient):
         super().__init__(
             base_url=base_url,
             api_key=api_key,
+            bearer_token=bearer_token,
             timeout=timeout,
             log_requests=log_requests,
             custom_logger=custom_logger,
@@ -55,10 +59,10 @@ 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",
+        base_url: str = DEFAULT_BASE_URL,
+        gateway_url: str = DEFAULT_OPENAI_GATEWAY_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
         **kwargs,
@@ -77,22 +81,22 @@ class AiriaClient(AiriaBaseClient):
         """
         from openai import OpenAI
 
-        api_key = cls._get_api_key(api_key)
+        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
         cls.openai = OpenAI(
-            api_key=api_key,
+            api_key=client.api_key,
             base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(base_url, api_key, timeout, log_requests, custom_logger)
+        return client
 
     @classmethod
     def with_anthropic_gateway(
         cls,
-        base_url: str = "https://api.airia.ai/",
-        gateway_url: str = "https://gateway.airia.ai/anthropic",
+        base_url: str = DEFAULT_BASE_URL,
+        gateway_url: str = DEFAULT_ANTHROPIC_GATEWAY_URL,
         api_key: Optional[str] = None,
-        timeout: float = 30.0,
+        timeout: float = DEFAULT_TIMEOUT,
         log_requests: bool = False,
         custom_logger: Optional["loguru.Logger"] = None,
         **kwargs,
@@ -111,14 +115,41 @@ class AiriaClient(AiriaBaseClient):
         """
         from anthropic import Anthropic
 
-        api_key = cls._get_api_key(api_key)
+        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
         cls.anthropic = Anthropic(
-            api_key=api_key,
+            api_key=client.api_key,
             base_url=gateway_url,
             **kwargs,
         )
 
-        return cls(base_url, api_key, timeout, log_requests, custom_logger)
+        return client
+
+    @classmethod
+    def with_bearer_token(
+        cls,
+        bearer_token: str,
+        base_url: str = DEFAULT_BASE_URL,
+        timeout: float = DEFAULT_TIMEOUT,
+        log_requests: bool = False,
+        custom_logger: Optional["loguru.Logger"] = None,
+    ):
+        """
+        Initialize the synchronous Airia API client with bearer token authentication.
+
+        Args:
+            bearer_token: Bearer token for authentication.
+            base_url: Base URL of the Airia API.
+            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.
+        """
+        return cls(
+            base_url=base_url,
+            bearer_token=bearer_token,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
 
     def _handle_exception(self, e: requests.HTTPError, url: str, correlation_id: str):
         # Log the error response if enabled
@@ -141,12 +172,12 @@ class AiriaClient(AiriaBaseClient):
             # If JSON parsing fails or expected keys are missing
             error_message = f"API request failed: {str(e)}"
 
-        # Make sure API key is not included in error messages
-        sanitized_message = (
-            error_message.replace(self.api_key, "[REDACTED]")
-            if self.api_key in error_message
-            else error_message
-        )
+        # Make sure sensitive auth information is not included in error messages
+        sanitized_message = error_message
+        if self.api_key and self.api_key in sanitized_message:
+            sanitized_message = sanitized_message.replace(self.api_key, "[REDACTED]")
+        if self.bearer_token and self.bearer_token in sanitized_message:
+            sanitized_message = sanitized_message.replace(self.bearer_token, "[REDACTED]")
 
         # Raise custom exception with status code and sanitized message
         raise AiriaAPIError(

airia-0.1.10/airia/constants.py

@@ -0,0 +1,9 @@
+"""Constants used throughout the Airia SDK."""
+
+# Default API endpoints
+DEFAULT_BASE_URL = "https://api.airia.ai/"
+DEFAULT_OPENAI_GATEWAY_URL = "https://gateway.airia.ai/openai/v1"
+DEFAULT_ANTHROPIC_GATEWAY_URL = "https://gateway.airia.ai/anthropic"
+
+# Default timeouts
+DEFAULT_TIMEOUT = 30.0

{airia-0.1.9 → airia-0.1.10}/airia.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.9
+Version: 0.1.10
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT
@@ -43,7 +43,9 @@ Airia Python API Library that provides a clean and intuitive interface to intera
 - **Gateway Support**: Seamlessly integrate with OpenAI and Anthropic services through Airia gateways
 - **Error Handling**: Comprehensive error handling with custom exceptions
 - **Logging**: Built-in configurable logging with correlation ID support for request tracing
-- **API Key Management**: Flexible API key configuration via parameters or environment variables
+- **Flexible Authentication**: Support for both API keys and bearer tokens with flexible configuration
+- **API Key Management**: API key configuration via parameters or environment variables
+- **Bearer Token Support**: Bearer token authentication for ephemeral, short-lived credentials
 
 ## Installation
 
@@ -181,17 +183,38 @@ This will create both wheel and source distribution in the `dist/` directory.
 ```python
 from airia import AiriaClient
 
+# API Key Authentication
 client = AiriaClient(
     base_url="https://api.airia.ai",        # Default: "https://api.airia.ai"
-    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)
+    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)
+)
+
+# Bearer Token Authentication
+client = AiriaClient(
+    base_url="https://api.airia.ai",        # Default: "https://api.airia.ai"
+    bearer_token="your_bearer_token",       # Must be provided explicitly (no env var fallback)
+    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)
+)
+
+# Convenience method for bearer token
+client = AiriaClient.with_bearer_token(
+    bearer_token="your_bearer_token",
+    base_url="https://api.airia.ai",        # Optional, uses default if not provided
+    timeout=30.0,                           # Optional, uses default if not provided
+    log_requests=False,                     # Optional, uses default if not provided
+    custom_logger=None                      # Optional, uses default if not provided
 )
 ```
 
 ### Synchronous Usage
 
+#### With API Key
+
 ```python
 from airia import AiriaClient
 
@@ -207,6 +230,23 @@ response = client.execute_pipeline(
 print(response.result)
 ```
 
+#### With Bearer Token
+
+```python
+from airia import AiriaClient
+
+# Initialize client with bearer token
+client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
+
+# Execute a pipeline
+response = client.execute_pipeline(
+    pipeline_id="your_pipeline_id",
+    user_input="Tell me about quantum computing"
+)
+
+print(response.result)
+```
+
 #### Synchronous Streaming
 
 ```python
@@ -214,6 +254,7 @@ from airia import AiriaClient
 
 # Initialize client (API key can be passed directly or via AIRIA_API_KEY environment variable)
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 # Execute a pipeline
 response = client.execute_pipeline(
@@ -228,6 +269,8 @@ for c in response.stream:
 
 ### Asynchronous Usage
 
+#### With API Key
+
 ```python
 import asyncio
 from airia import AiriaAsyncClient
@@ -243,6 +286,23 @@ async def main():
 asyncio.run(main())
 ```
 
+#### With Bearer Token
+
+```python
+import asyncio
+from airia import AiriaAsyncClient
+
+async def main():
+    client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
+    response = await client.execute_pipeline(
+        pipeline_id="your_pipeline_id",
+        user_input="Tell me about quantum computing"
+    )
+    print(response.result)
+
+asyncio.run(main())
+```
+
 #### Asynchronous Streaming
 
 ```python
@@ -251,6 +311,7 @@ from airia import AiriaAsyncClient
 
 async def main():
     client = AiriaAsyncClient(api_key="your_api_key")
+    # Or with bearer token: client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
     response = await client.execute_pipeline(
         pipeline_id="your_pipeline_id",
         user_input="Tell me about quantum computing",
@@ -320,6 +381,7 @@ from airia import AiriaClient
 from airia.types import AgentModelStreamFragmentMessage
 
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 response = client.execute_pipeline(
     pipeline_id="your_pipeline_id",
@@ -343,6 +405,7 @@ You can retrieve detailed configuration information about a pipeline using the `
 from airia import AiriaClient
 
 client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 # Get pipeline configuration
 config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
@@ -351,10 +414,35 @@ config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
 print(f"Pipeline Name: {config.agent.name}")
 ```
 
+## Authentication Methods
+
+Airia supports two authentication methods:
+
+### API Keys
+- Can be passed as a parameter or via `AIRIA_API_KEY` environment variable
+- Support gateway functionality (OpenAI and Anthropic gateways)
+- Suitable for long-term, persistent authentication
+
+### Bearer Tokens
+- Must be provided explicitly (no environment variable fallback)
+- **Important**: Bearer tokens cannot be used with gateway functionality
+- Suitable for ephemeral, short-lived authentication scenarios
+- Ideal for temporary access or programmatic token generation
+
+```python
+# ✅ API key with gateway support
+client = AiriaClient.with_openai_gateway(api_key="your_api_key")
+
+# ❌ Bearer token with gateway - NOT SUPPORTED
+# client = AiriaClient.with_openai_gateway(bearer_token="token")  # This won't work
+```
+
 ## Gateway Usage
 
 Airia provides gateway capabilities for popular AI services like OpenAI and Anthropic, allowing you to use your Airia API key with these services.
 
+> **Note**: Gateway functionality requires API key authentication. Bearer tokens are not supported for gateway usage.
+
 ### OpenAI Gateway
 
 ```python
@@ -482,7 +570,11 @@ console_logger = configure_logging(
 The SDK uses custom exceptions to provide clear error messages:
 
 ```python
-from airia import AiriaAPIError
+from airia import AiriaAPIError, AiriaClient
+
+# Works with both API keys and bearer tokens
+client = AiriaClient(api_key="your_api_key")
+# Or: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
 
 try:
     response = client.execute_pipeline(

{airia-0.1.9 → airia-0.1.10}/airia.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 pyproject.toml
 airia/__init__.py
+airia/constants.py
 airia/exceptions.py
 airia/logs.py
 airia.egg-info/PKG-INFO
@@ -22,6 +23,7 @@ airia/types/api/get_projects.py
 airia/types/api/pipeline_execution.py
 airia/utils/sse_parser.py
 tests/test_anthropic_gateway.py
+tests/test_bearer_token_auth.py
 tests/test_execute_pipeline.py
 tests/test_get_active_pipelines_ids.py
 tests/test_get_pipeline_config.py

{airia-0.1.9 → airia-0.1.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "airia"
-version = "0.1.9"
+version = "0.1.10"
 description = "Python SDK for Airia API"
 license = { text = "MIT" }
 authors = [{ name = "Airia LLC", email = "support@airia.com" }]

airia-0.1.10/tests/test_bearer_token_auth.py

@@ -0,0 +1,90 @@
+import os
+import pytest
+import pytest_asyncio
+from unittest.mock import patch, MagicMock
+
+from airia import AiriaAsyncClient, AiriaClient
+
+
+class TestBearerTokenAuthentication:
+    """Tests for bearer token authentication functionality."""
+
+    def test_sync_client_with_bearer_token(self):
+        """Test synchronous client initialization with bearer token."""
+        bearer_token = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.test.token"
+        client = AiriaClient(bearer_token=bearer_token)
+        
+        assert client.bearer_token == bearer_token
+        assert client.api_key is None
+
+    def test_sync_client_with_bearer_token_class_method(self):
+        """Test synchronous client initialization using with_bearer_token class method."""
+        bearer_token = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.test.token"
+        client = AiriaClient.with_bearer_token(bearer_token)
+        
+        assert client.bearer_token == bearer_token
+        assert client.api_key is None
+
+    def test_explicit_api_key_and_bearer_token_conflict(self):
+        """Test that providing both api_key and bearer_token explicitly raises an error."""
+        with pytest.raises(ValueError, match="Cannot provide both api_key and bearer_token"):
+            AiriaClient(api_key="test_key", bearer_token="test_token")
+
+    def test_no_auth_provided_raises_error(self):
+        """Test that providing no authentication raises an error."""
+        with patch.dict(os.environ, {}, clear=True):
+            with pytest.raises(ValueError, match="Authentication required"):
+                AiriaClient()
+
+    def test_bearer_token_auth_header_generation(self):
+        """Test that bearer token generates correct Authorization header."""
+        bearer_token = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.test.token"
+        client = AiriaClient(bearer_token=bearer_token)
+        
+        request_data = client._prepare_request("https://test.com")
+        
+        assert "Authorization" in request_data.headers
+        assert request_data.headers["Authorization"] == f"Bearer {bearer_token}"
+        assert "X-API-KEY" not in request_data.headers
+
+    def test_api_key_auth_header_generation(self):
+        """Test that API key generates correct X-API-KEY header."""
+        api_key = "test_api_key"
+        client = AiriaClient(api_key=api_key)
+        
+        request_data = client._prepare_request("https://test.com")
+        
+        assert "X-API-KEY" in request_data.headers
+        assert request_data.headers["X-API-KEY"] == api_key
+        assert "Authorization" not in request_data.headers
+
+    def test_bearer_token_logging_redaction(self):
+        """Test that bearer token is redacted in logs."""
+        bearer_token = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.test.token"
+        client = AiriaClient(bearer_token=bearer_token, log_requests=True)
+        
+        # Mock the logger to capture log output
+        with patch.object(client, 'logger') as mock_logger:
+            request_data = client._prepare_request("https://test.com")
+            
+            # Check that the log call was made and bearer token was redacted
+            mock_logger.info.assert_called_once()
+            log_call_args = mock_logger.info.call_args[0][0]
+            assert "[REDACTED]" in log_call_args
+            assert bearer_token not in log_call_args
+
+    def test_bearer_token_error_sanitization(self):
+        """Test that bearer token is sanitized in error messages."""
+        bearer_token = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.test.token"
+        client = AiriaClient(bearer_token=bearer_token)
+        
+        # Test the sanitization logic directly
+        error_message = f"Invalid token: {bearer_token}"
+        sanitized_message = error_message
+        if client.bearer_token and client.bearer_token in sanitized_message:
+            sanitized_message = sanitized_message.replace(client.bearer_token, "[REDACTED]")
+        
+        assert bearer_token not in sanitized_message
+        assert "[REDACTED]" in sanitized_message
+
+