airia 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

{airia-0.1.9.dist-info → airia-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.9
+Version: 0.1.11
 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",
@@ -268,7 +329,7 @@ When using streaming mode (`async_output=True`), the API returns Server-Sent Eve
 
 ### Available Message Types
 
-The streaming response includes various message types defined in `airia.types`. Here are the key ones:
+The streaming response includes various message types defined in `airia.types.sse`. Here are the key ones:
 
 - `AgentModelStreamFragmentMessage` - Contains actual LLM output chunks
 - `AgentModelStreamStartMessage` - Indicates LLM streaming has started
@@ -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,117 @@ config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
 print(f"Pipeline Name: {config.agent.name}")
 ```
 
+## Conversation Management
+
+You can create and manage conversations using the `create_conversation` method. Conversations allow you to organize and persist interactions within the Airia platform.
+
+### Creating Conversations
+
+#### Synchronous Usage
+
+```python
+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")
+
+# Create a basic conversation
+conversation = client.create_conversation(
+    user_id="user_123"
+)
+print(f"Created conversation: {conversation.conversation_id}")
+print(f"WebSocket URL: {conversation.websocket_url}")
+
+# Create a conversation with all options
+conversation = client.create_conversation(
+    user_id="user_123",
+    title="My Research Session",
+    deployment_id="deployment_456", 
+    data_source_files={"documents": ["doc1.pdf", "doc2.txt"]},
+    is_bookmarked=True
+)
+print(f"Created bookmarked conversation: {conversation.conversation_id}")
+```
+
+#### Asynchronous Usage
+
+```python
+import asyncio
+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")
+    
+    # Create a basic conversation
+    conversation = await client.create_conversation(
+        user_id="user_123"
+    )
+    print(f"Created conversation: {conversation.conversation_id}")
+    
+    # Create a conversation with all options
+    conversation = await client.create_conversation(
+        user_id="user_123",
+        title="My Research Session",
+        deployment_id="deployment_456",
+        data_source_files={"documents": ["doc1.pdf", "doc2.txt"]},
+        is_bookmarked=True
+    )
+    print(f"Created bookmarked conversation: {conversation.conversation_id}")
+
+asyncio.run(main())
+```
+
+### Conversation Parameters
+
+- **user_id** (required): The unique identifier of the user creating the conversation
+- **title** (optional): A descriptive title for the conversation
+- **deployment_id** (optional): The unique identifier of the deployment to associate with the conversation
+- **data_source_files** (optional): Configuration for data source files to be associated with the conversation
+- **is_bookmarked** (optional): Whether the conversation should be bookmarked (defaults to False)
+- **correlation_id** (optional): A unique identifier for request tracing and logging
+
+### Response Fields
+
+The `create_conversation` method returns a `CreateConversationResponse` object containing:
+
+- **conversation_id**: The unique identifier of the created conversation
+- **user_id**: The user ID associated with the conversation
+- **websocket_url**: The WebSocket URL for real-time communication
+- **deployment_id**: The deployment ID associated with the conversation
+- **icon_id** and **icon_url**: Optional conversation icon information
+- **description**: Optional conversation description
+- **space_name**: Optional workspace or space 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 +652,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.11.dist-info/RECORD

@@ -0,0 +1,23 @@
+airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
+airia/constants.py,sha256=fHZNUbvEP_x567uFI1qnA2DT8dOzQ07ojdawdPwIrIA,292
+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=7EOYRWT_DWTXKZuvWr3n9xkvws4DseiiCvhhpAWVKQs,29600
+airia/client/base_client.py,sha256=qG78HclndjAGsn2DLEdvog-V7AR1yt7bEakSYDsi-xM,10898
+airia/client/sync_client.py,sha256=AwJKBaKuJvo_2Z-0dfZva_9BvZ3lFUetwoRmuFGEw94,28579
+airia/types/_api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
+airia/types/_request_data.py,sha256=WUvf7fixhBPWLxKbc2VY3Y32BOUbLmXktY58mINuq3A,243
+airia/types/api/__init__.py,sha256=ay9ncmHuyfTZtwwUELcv1U2WalQQFqSBfL-XoRnGw04,596
+airia/types/api/conversations.py,sha256=EYNp1XgydFSFk-f_FhDei41Gw155vUBUYI0Pwn9yDcU,535
+airia/types/api/get_pipeline_config.py,sha256=BLz4HS94EYmdwq37KHda2XJ5KV7tJPP43jHmPbO8-NE,6647
+airia/types/api/get_projects.py,sha256=kWmlN0VMgmQ9WP9ms245MAAEmsHhkpwWUQJ_XJrhIRI,1362
+airia/types/api/pipeline_execution.py,sha256=TBtdpPX3bJnmML7wcrHBRZjN4zAHQXctdOjQCXBdyrU,740
+airia/types/sse/__init__.py,sha256=BVrFyLdRI7-eeFLtbDIingLU5L6kQR1yyu4hXVRlEuE,1581
+airia/types/sse/sse_messages.py,sha256=wSdowY07AjEO8R73SJrFPJtkfIBS4satUpNytjKQq2U,8305
+airia/utils/sse_parser.py,sha256=Zij-BEBdNtc_RrSppeKCLX0pmiFNdgFGHDW7_KiDz50,2782
+airia-0.1.11.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.11.dist-info/METADATA,sha256=nwpvwFbP8VGNqpvnYr2OzHxUMGMFIiKV8VPwjxiw7Ic,19691
+airia-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.11.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.11.dist-info/RECORD,,

airia-0.1.9.dist-info/RECORD

@@ -1,20 +0,0 @@
-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=jILf-aNgIw3YY4VFRxzkDoRAWcFqzdJ3fom9Up8nYYo,27010
-airia/client/base_client.py,sha256=yNo_N4I08fHXtRm9zMZufSKlA6peNB8b2m8zhKK3b9A,8475
-airia/client/sync_client.py,sha256=NP-mfiGtLSnxJcJ8Ymt9y8K5fHrG4I7yyhdFaPk6gcA,26068
-airia/types/__init__.py,sha256=Tr1xm5bQrE7yvw0Bo-goNmeh7Rl0QiXUabg7dsJ55dk,2227
-airia/types/api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
-airia/types/request_data.py,sha256=WUvf7fixhBPWLxKbc2VY3Y32BOUbLmXktY58mINuq3A,243
-airia/types/sse_messages.py,sha256=wSdowY07AjEO8R73SJrFPJtkfIBS4satUpNytjKQq2U,8305
-airia/types/api/get_pipeline_config.py,sha256=BLz4HS94EYmdwq37KHda2XJ5KV7tJPP43jHmPbO8-NE,6647
-airia/types/api/get_projects.py,sha256=kWmlN0VMgmQ9WP9ms245MAAEmsHhkpwWUQJ_XJrhIRI,1362
-airia/types/api/pipeline_execution.py,sha256=4zGM5W4aXiAibAdgWQCIZBC_blW6QYAlAHoVUMbnCF0,752
-airia/utils/sse_parser.py,sha256=h3TcBvXqUIngTqgY6yYxZCGLnC1eI6meQzYr13aFAb8,2791
-airia-0.1.9.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.9.dist-info/METADATA,sha256=qQ_S_ut5wUtvkiGKH-DPQzzJtbR4FlS9Rw_-0xzugoI,13218
-airia-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.9.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.9.dist-info/RECORD,,