indoxrouter 0.1.8__py3-none-any.whl → 0.1.11__py3-none-any.whl

indoxrouter/client.py

@@ -5,6 +5,9 @@ This module provides a client for interacting with the IndoxRouter API, which se
 interface to multiple AI providers and models. The client handles authentication, rate limiting,
 error handling, and provides a standardized response format across different AI services.
 
+IMPORTANT: The IndoxRouter server now supports only cookie-based authentication. This client
+automatically handles authentication by exchanging your API key for a JWT token through the login endpoint.
+
 The Client class offers methods for:
 - Authentication and session management
 - Making API requests with automatic token refresh
@@ -71,6 +74,7 @@ from .constants import (
     IMAGE_ENDPOINT,
     MODEL_ENDPOINT,
     USAGE_ENDPOINT,
+    USE_COOKIES,
 )
 
 logger = logging.getLogger(__name__)
@@ -84,7 +88,6 @@ class Client:
     def __init__(
         self,
         api_key: Optional[str] = None,
-        base_url: Optional[str] = None,
         timeout: int = DEFAULT_TIMEOUT,
     ):
         """
@@ -93,19 +96,91 @@ class Client:
         Args:
             api_key: API key for authentication. If not provided, the client will look for the
                 INDOX_ROUTER_API_KEY environment variable.
-            base_url: Custom base URL for the API. If not provided, the default base URL will be used.
             timeout: Request timeout in seconds.
         """
+
+        use_cookies = USE_COOKIES
         self.api_key = api_key or os.environ.get("INDOX_ROUTER_API_KEY")
         if not self.api_key:
             raise ValueError(
                 "API key must be provided either as an argument or as the INDOX_ROUTER_API_KEY environment variable."
             )
 
-        self.base_url = base_url or DEFAULT_BASE_URL
+        self.base_url = DEFAULT_BASE_URL
         self.timeout = timeout
+        self.use_cookies = use_cookies
         self.session = requests.Session()
-        self.session.headers.update({"Authorization": f"Bearer {self.api_key}"})
+
+        # Authenticate and get JWT tokens
+        self._authenticate()
+
+    def _authenticate(self):
+        """
+        Authenticate with the server and get JWT tokens.
+        This uses the /auth/token endpoint to get JWT tokens using the API key.
+        """
+        try:
+            # First try using the API key as a username with empty password (API key flow)
+            logger.debug("Authenticating with API key as username")
+            response = self.session.post(
+                f"{self.base_url}/api/v1/auth/token",
+                data={
+                    "username": self.api_key,
+                    "password": "",
+                },
+                timeout=self.timeout,
+            )
+
+            if response.status_code != 200:
+                # Try using email:api_key format if the first method fails
+                logger.debug(
+                    "First auth method failed, trying with API key as password"
+                )
+                response = self.session.post(
+                    f"{self.base_url}/api/v1/auth/login",
+                    json={
+                        "username": "pip_client",
+                        "password": self.api_key,
+                    },
+                    timeout=self.timeout,
+                )
+
+            if response.status_code != 200:
+                error_data = {}
+                try:
+                    error_data = response.json()
+                except:
+                    error_data = {"detail": response.text}
+
+                raise AuthenticationError(
+                    f"Authentication failed: {error_data.get('detail', 'Unknown error')}"
+                )
+
+            # At this point, the cookies should be set in the session
+            logger.debug("Authentication successful")
+
+            # Check if we have the cookies we need
+            if "access_token" not in self.session.cookies:
+                logger.warning(
+                    "Authentication succeeded but no access_token cookie was set"
+                )
+
+        except requests.RequestException as e:
+            logger.error(f"Authentication request failed: {str(e)}")
+            raise NetworkError(f"Network error during authentication: {str(e)}")
+
+    def _get_domain(self):
+        """
+        Extract domain from the base URL for cookie setting.
+        """
+        try:
+            from urllib.parse import urlparse
+
+            parsed_url = urlparse(self.base_url)
+            return parsed_url.netloc
+        except Exception:
+            # If parsing fails, return a default value
+            return ""
 
     def enable_debug(self, level=logging.DEBUG):
         """
@@ -177,6 +252,24 @@ class Client:
             if stream:
                 return response
 
+            # Check if we need to reauthenticate (401 Unauthorized)
+            if response.status_code == 401:
+                logger.debug("Received 401, attempting to reauthenticate")
+                self._authenticate()
+
+                # Retry the request after reauthentication
+                response = self.session.request(
+                    method,
+                    url,
+                    headers=headers,
+                    json=data,
+                    timeout=self.timeout,
+                    stream=stream,
+                )
+
+                if stream:
+                    return response
+
             response.raise_for_status()
             return response.json()
         except requests.HTTPError as e:

indoxrouter/constants.py

@@ -7,6 +7,7 @@ DEFAULT_API_VERSION = "v1"
 DEFAULT_BASE_URL = "https://api.indox.org"  # Production server URL with HTTPS
 # DEFAULT_BASE_URL = "http://localhost:8000"  # Local development server
 DEFAULT_TIMEOUT = 60
+USE_COOKIES = True  # Always use cookie-based authentication
 
 # Default models
 DEFAULT_MODEL = "openai/gpt-4o-mini"

indoxrouter/test_api_key.py

@@ -0,0 +1,98 @@
+#!/usr/bin/env python
+"""
+Test script for the IndoxRouter client.
+This script tests authentication and basic functionality of the client.
+"""
+
+import os
+import sys
+import logging
+from indoxrouter import Client, AuthenticationError, NetworkError
+
+# Set up logging
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+)
+logger = logging.getLogger("indoxrouter_test")
+
+
+def test_client(api_key=None, debug=False):
+    """
+    Test the IndoxRouter client with the provided API key.
+
+    Args:
+        api_key: API key to use for authentication
+        debug: Whether to enable debug logging
+    """
+    if debug:
+        logging.getLogger("indoxrouter").setLevel(logging.DEBUG)
+
+    logger.info("Testing IndoxRouter client...")
+
+    # If no API key is provided, try to get it from environment variable
+    if not api_key:
+        api_key = os.environ.get("INDOX_ROUTER_API_KEY")
+        if not api_key:
+            logger.error(
+                "No API key provided. Please provide an API key as an argument or set the INDOX_ROUTER_API_KEY environment variable."
+            )
+            sys.exit(1)
+
+    try:
+        # Initialize client
+        logger.info("Initializing client...")
+        client = Client(api_key=api_key)
+
+        # Test connection
+        logger.info("Testing connection...")
+        connection_info = client.test_connection()
+        logger.info(f"Connection test: {connection_info['status']}")
+
+        # Try to get available models
+        logger.info("Fetching available models...")
+        models = client.models()
+
+        # Display some models
+        providers = [p.get("name") for p in models]
+        logger.info(f"Available providers: {', '.join(providers)}")
+
+        # Try a simple chat completion
+        logger.info("Testing chat completion...")
+        response = client.chat(
+            messages=[{"role": "user", "content": "Hello, who are you?"}],
+            model="openai/gpt-3.5-turbo",
+            max_tokens=30,
+        )
+
+        logger.info("Chat completion successful!")
+        logger.info(f"Response: {response['choices'][0]['message']['content']}")
+
+        logger.info("All tests passed! The client is working correctly.")
+
+    except AuthenticationError as e:
+        logger.error(f"Authentication error: {e}")
+        logger.error("Please check that your API key is correct.")
+        sys.exit(1)
+    except NetworkError as e:
+        logger.error(f"Network error: {e}")
+        logger.error(
+            "Please check your internet connection and that the IndoxRouter server is accessible."
+        )
+        sys.exit(1)
+    except Exception as e:
+        logger.error(f"Unexpected error: {e}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    import argparse
+
+    parser = argparse.ArgumentParser(description="Test the IndoxRouter client")
+    parser.add_argument("--api-key", "-k", help="API key to use for authentication")
+    parser.add_argument(
+        "--debug", "-d", action="store_true", help="Enable debug logging"
+    )
+
+    args = parser.parse_args()
+
+    test_client(api_key=args.api_key, debug=args.debug)

{indoxrouter-0.1.8.dist-info → indoxrouter-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: indoxrouter
-Version: 0.1.8
+Version: 0.1.11
 Summary: A unified client for various AI providers
 Home-page: https://github.com/indoxrouter/indoxrouter
 Author: indoxRouter Team
@@ -45,7 +45,7 @@ A unified client for various AI providers, including OpenAI, Anthropic, Google,
 - **Unified API**: Access multiple AI providers through a single API
 - **Simple Interface**: Easy-to-use methods for chat, completion, embeddings, and image generation
 - **Error Handling**: Standardized error handling across providers
-- **Authentication**: Automatic token management
+- **Authentication**: Secure cookie-based authentication
 
 ## Installation
 
@@ -60,7 +60,7 @@ pip install indoxrouter
 ```python
 from indoxrouter import Client
 
-# Initialize with API key (default connects to localhost:8000)
+# Initialize with API key
 client = Client(api_key="your_api_key")
 
 # Using environment variables
@@ -68,6 +68,42 @@ client = Client(api_key="your_api_key")
 import os
 os.environ["INDOX_ROUTER_API_KEY"] = "your_api_key"
 client = Client()
+
+# Connect to a custom server
+client = Client(
+    api_key="your_api_key",
+    base_url="https://your-indoxrouter-server.com"
+)
+```
+
+### Authentication
+
+IndoxRouter uses cookie-based authentication with JWT tokens. The client handles this automatically by:
+
+1. Taking your API key and exchanging it for JWT tokens using the server's authentication endpoints
+2. Storing the JWT tokens in cookies
+3. Using the cookies for subsequent requests
+4. Automatically refreshing tokens when they expire
+
+```python
+# Authentication is handled automatically when creating the client
+client = Client(api_key="your_api_key")
+```
+
+### Testing Your API Key
+
+The package includes a test script to verify your API key and connection:
+
+```bash
+# Run the test script with your API key
+python -m indoxrouter.test_api_key --api-key YOUR_API_KEY
+
+# Or set the environment variable and run
+export INDOX_ROUTER_API_KEY=YOUR_API_KEY
+python -m indoxrouter.test_api_key
+
+# To see detailed debugging information
+python -m indoxrouter.test_api_key --debug
 ```
 
 ### Chat Completions

indoxrouter-0.1.11.dist-info/RECORD

@@ -0,0 +1,9 @@
+indoxrouter/__init__.py,sha256=28pdx482uGFF_S1msov0LTTGsFTvVBKRqMkDmoXWUBY,1416
+indoxrouter/client.py,sha256=Bt8JuU7G_TSX2sVn3Voi6UBXeZnmG-vLCGniAGh_r5U,27387
+indoxrouter/constants.py,sha256=D-l5qhEmeDIXOEfogvf6lJI1Sw_fXSKZy-Nj_l_MV7E,1184
+indoxrouter/exceptions.py,sha256=0ULxtK9va4718PGTO5VoClXYEJeojpiM-7AganeiZZ4,1263
+indoxrouter/test_api_key.py,sha256=u4fUTJwxC7-kTrLdCq8u5a1Mx66jeExfsBlHLhyzNmA,3229
+indoxrouter-0.1.11.dist-info/METADATA,sha256=3SEOZwK_ZKPQCfZc7BRKYQ8q3u4meMtmSIbSY4VYZxg,6004
+indoxrouter-0.1.11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+indoxrouter-0.1.11.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
+indoxrouter-0.1.11.dist-info/RECORD,,

indoxrouter-0.1.8.dist-info/RECORD

@@ -1,8 +0,0 @@
-indoxrouter/__init__.py,sha256=28pdx482uGFF_S1msov0LTTGsFTvVBKRqMkDmoXWUBY,1416
-indoxrouter/client.py,sha256=1XhBiz6CBoN_jNgKmmHvyXecdHZftDckssY2lAir_tA,24044
-indoxrouter/constants.py,sha256=m5EW5DOxjpvjwj_u2AxMf_5RuPpjpx2Iih35b0E0LEY,1123
-indoxrouter/exceptions.py,sha256=0ULxtK9va4718PGTO5VoClXYEJeojpiM-7AganeiZZ4,1263
-indoxrouter-0.1.8.dist-info/METADATA,sha256=HZqWq-PhdEuCfSzROo7L5rQPCvnbSJVur4CKLjMqD98,4971
-indoxrouter-0.1.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-indoxrouter-0.1.8.dist-info/top_level.txt,sha256=v6FGWkw0QAnXhyYtnXLI1cxzna0iveNvZUotVzCWabM,12
-indoxrouter-0.1.8.dist-info/RECORD,,