lucidicai 2.1.2__py3-none-any.whl → 2.1.3__py3-none-any.whl

lucidicai/__init__.py

@@ -297,7 +297,7 @@ get_error_history = error_boundary.get_error_history
 clear_error_history = error_boundary.clear_error_history
 
 # Version
-__version__ = "2.1.2"
+__version__ = "2.1.3"
 
 # Apply error boundary wrapping to all SDK functions
 from .sdk.error_boundary import wrap_sdk_function

lucidicai/api/client.py

@@ -1,15 +1,13 @@
 """Pure HTTP client for Lucidic API communication.
 
-This module contains only the HTTP client logic, separated from
-session management and other concerns.
+This module contains only the HTTP client logic using httpx,
+supporting both synchronous and asynchronous operations.
 """
-import json
+import asyncio
+from datetime import datetime, timezone
 from typing import Any, Dict, Optional
-from urllib.parse import urlencode
 
-import requests
-from requests.adapters import HTTPAdapter
-from urllib3.util import Retry
+import httpx
 
 from ..core.config import SDKConfig, get_config
 from ..core.errors import APIKeyVerificationError
@@ -17,7 +15,7 @@ from ..utils.logger import debug, info, warning, error, mask_sensitive, truncate
 
 
 class HttpClient:
-    """HTTP client for API communication."""
+    """HTTP client for API communication with sync and async support."""
     
     def __init__(self, config: Optional[SDKConfig] = None):
         """Initialize the HTTP client.
@@ -28,36 +26,26 @@ class HttpClient:
         self.config = config or get_config()
         self.base_url = self.config.network.base_url
         
-        # Create session with connection pooling
-        self.session = requests.Session()
+        # Build default headers
+        self._headers = self._build_headers()
         
-        # Configure retries
-        retry_cfg = Retry(
-            total=self.config.network.max_retries,
-            backoff_factor=self.config.network.backoff_factor,
-            status_forcelist=[502, 503, 504],
-            allowed_methods=["GET", "POST", "PUT", "DELETE"],
-        )
+        # Transport configuration for connection pooling and retries
+        self._transport_kwargs = {
+            "retries": self.config.network.max_retries,
+        }
         
-        # Configure adapter with connection pooling
-        adapter = HTTPAdapter(
-            max_retries=retry_cfg,
-            pool_connections=self.config.network.connection_pool_size,
-            pool_maxsize=self.config.network.connection_pool_maxsize
+        # Connection limits for pooling
+        self._limits = httpx.Limits(
+            max_connections=self.config.network.connection_pool_maxsize,
+            max_keepalive_connections=self.config.network.connection_pool_size,
         )
         
-        self.session.mount("https://", adapter)
-        self.session.mount("http://", adapter)
-        
-        # Set headers
-        self._update_headers()
-        
-        # Verify API key if configured
-        if self.config.api_key:
-            self._verify_api_key()
+        # Lazy-initialized clients
+        self._sync_client: Optional[httpx.Client] = None
+        self._async_client: Optional[httpx.AsyncClient] = None
     
-    def _update_headers(self) -> None:
-        """Update session headers with authentication."""
+    def _build_headers(self) -> Dict[str, str]:
+        """Build default headers for requests."""
         headers = {
             "User-Agent": "lucidic-sdk/2.0",
             "Content-Type": "application/json"
@@ -69,24 +57,91 @@ class HttpClient:
         if self.config.agent_id:
             headers["x-agent-id"] = self.config.agent_id
         
-        self.session.headers.update(headers)
+        return headers
+    
+    @property
+    def sync_client(self) -> httpx.Client:
+        """Get or create the synchronous HTTP client."""
+        if self._sync_client is None or self._sync_client.is_closed:
+            transport = httpx.HTTPTransport(**self._transport_kwargs)
+            self._sync_client = httpx.Client(
+                base_url=self.base_url,
+                headers=self._headers,
+                timeout=httpx.Timeout(self.config.network.timeout),
+                limits=self._limits,
+                transport=transport,
+            )
+        return self._sync_client
+    
+    @property
+    def async_client(self) -> httpx.AsyncClient:
+        """Get or create the asynchronous HTTP client."""
+        if self._async_client is None or self._async_client.is_closed:
+            transport = httpx.AsyncHTTPTransport(**self._transport_kwargs)
+            self._async_client = httpx.AsyncClient(
+                base_url=self.base_url,
+                headers=self._headers,
+                timeout=httpx.Timeout(self.config.network.timeout),
+                limits=self._limits,
+                transport=transport,
+            )
+        return self._async_client
+    
+    def _add_timestamp(self, data: Optional[Dict[str, Any]]) -> Dict[str, Any]:
+        """Add current_time to request data."""
+        if data is None:
+            data = {}
+        data["current_time"] = datetime.now(timezone.utc).isoformat()
+        return data
     
-    def _verify_api_key(self) -> None:
-        """Verify the API key with the backend."""
-        debug("[HTTP] Verifying API key")
+    def _handle_response(self, response: httpx.Response) -> Dict[str, Any]:
+        """Handle HTTP response and parse JSON.
+        
+        Args:
+            response: httpx Response object
+            
+        Returns:
+            Response data as dictionary
+            
+        Raises:
+            APIKeyVerificationError: On 401 Unauthorized responses
+            httpx.HTTPStatusError: On other HTTP errors
+        """
+        # Log and raise for HTTP errors
+        if not response.is_success:
+            try:
+                error_data = response.json()
+                error_msg = error_data.get('detail', response.text)
+            except Exception:
+                error_msg = response.text
+            
+            error(f"[HTTP] Error {response.status_code}: {error_msg}")
+            
+            # Raise specific error for authentication/authorization failures
+            if response.status_code in (401, 403):
+                raise APIKeyVerificationError(f"Authentication failed: {error_msg}")
+        
+        response.raise_for_status()
+        
+        # Parse JSON response
         try:
-            response = self.get("verifyapikey")
-            # Backend returns 200 OK for valid key, check if we got a response
-            if response is None:
-                raise APIKeyVerificationError("No response from API")
-            info("[HTTP] API key verified successfully")
-        except APIKeyVerificationError:
-            raise
-        except requests.RequestException as e:
-            raise APIKeyVerificationError(f"Could not verify API key: {e}")
+            data = response.json()
+        except ValueError:
+            # For empty responses (like verifyapikey), return success
+            if response.status_code == 200 and not response.text:
+                data = {"success": True}
+            else:
+                # Return text if not JSON
+                data = {"response": response.text}
+        
+        debug(f"[HTTP] Response ({response.status_code}): {truncate_data(data)}")
+        
+        return data
+    
+    # ==================== Synchronous Methods ====================
     
     def get(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
-        """Make a GET request.
+        """Make a synchronous GET request.
         
         Args:
             endpoint: API endpoint (without base URL)
@@ -98,7 +153,7 @@ class HttpClient:
         return self.request("GET", endpoint, params=params)
     
     def post(self, endpoint: str, data: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
-        """Make a POST request.
+        """Make a synchronous POST request.
         
         Args:
             endpoint: API endpoint (without base URL)
@@ -107,15 +162,11 @@ class HttpClient:
         Returns:
             Response data as dictionary
         """
-        # Add current_time to all POST requests like TypeScript SDK does
-        from datetime import datetime, timezone
-        if data is None:
-            data = {}
-        data["current_time"] = datetime.now(timezone.utc).isoformat()
+        data = self._add_timestamp(data)
         return self.request("POST", endpoint, json=data)
     
     def put(self, endpoint: str, data: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
-        """Make a PUT request.
+        """Make a synchronous PUT request.
         
         Args:
             endpoint: API endpoint (without base URL)
@@ -124,15 +175,11 @@ class HttpClient:
         Returns:
             Response data as dictionary
         """
-        # Add current_time to all PUT requests like TypeScript SDK does
-        from datetime import datetime, timezone
-        if data is None:
-            data = {}
-        data["current_time"] = datetime.now(timezone.utc).isoformat()
+        data = self._add_timestamp(data)
         return self.request("PUT", endpoint, json=data)
     
     def delete(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
-        """Make a DELETE request.
+        """Make a synchronous DELETE request.
         
         Args:
             endpoint: API endpoint (without base URL)
@@ -151,68 +198,160 @@ class HttpClient:
         json: Optional[Dict[str, Any]] = None,
         **kwargs
     ) -> Dict[str, Any]:
-        """Make an HTTP request.
+        """Make a synchronous HTTP request.
         
         Args:
             method: HTTP method
             endpoint: API endpoint (without base URL)
             params: Query parameters
             json: Request body (for POST/PUT)
-            **kwargs: Additional arguments for requests
+            **kwargs: Additional arguments for httpx
             
         Returns:
             Response data as dictionary
             
         Raises:
-            requests.RequestException: On HTTP errors
+            httpx.HTTPError: On HTTP errors
         """
-        url = f"{self.base_url}/{endpoint}"
+        url = f"/{endpoint}"
         
         # Log request details
-        debug(f"[HTTP] {method} {url}")
+        debug(f"[HTTP] {method} {self.base_url}{url}")
         if params:
             debug(f"[HTTP] Query params: {mask_sensitive(params)}")
         if json:
             debug(f"[HTTP] Request body: {truncate_data(mask_sensitive(json))}")
         
-        response = self.session.request(
+        response = self.sync_client.request(
             method=method,
             url=url,
             params=params,
             json=json,
-            timeout=self.config.network.timeout,
             **kwargs
         )
         
-        # Raise for HTTP errors with more detail
-        if not response.ok:
-            # Try to get error details from response
-            try:
-                error_data = response.json()
-                error_msg = error_data.get('detail', response.text)
-            except:
-                error_msg = response.text
+        return self._handle_response(response)
+    
+    # ==================== Asynchronous Methods ====================
+    
+    async def aget(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Make an asynchronous GET request.
+        
+        Args:
+            endpoint: API endpoint (without base URL)
+            params: Query parameters
             
-            error(f"[HTTP] Error {response.status_code}: {error_msg}")
+        Returns:
+            Response data as dictionary
+        """
+        return await self.arequest("GET", endpoint, params=params)
+    
+    async def apost(self, endpoint: str, data: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Make an asynchronous POST request.
         
-        response.raise_for_status()
+        Args:
+            endpoint: API endpoint (without base URL)
+            data: Request body data
+            
+        Returns:
+            Response data as dictionary
+        """
+        data = self._add_timestamp(data)
+        return await self.arequest("POST", endpoint, json=data)
+    
+    async def aput(self, endpoint: str, data: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Make an asynchronous PUT request.
         
-        # Parse JSON response
-        try:
-            data = response.json()
-        except ValueError:
-            # For empty responses (like verifyapikey), return success
-            if response.status_code == 200 and not response.text:
-                data = {"success": True}
-            else:
-                # Return text if not JSON
-                data = {"response": response.text}
+        Args:
+            endpoint: API endpoint (without base URL)
+            data: Request body data
+            
+        Returns:
+            Response data as dictionary
+        """
+        data = self._add_timestamp(data)
+        return await self.arequest("PUT", endpoint, json=data)
+    
+    async def adelete(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Make an asynchronous DELETE request.
         
-        debug(f"[HTTP] Response ({response.status_code}): {truncate_data(data)}")
+        Args:
+            endpoint: API endpoint (without base URL)
+            params: Query parameters
+            
+        Returns:
+            Response data as dictionary
+        """
+        return await self.arequest("DELETE", endpoint, params=params)
+    
+    async def arequest(
+        self,
+        method: str,
+        endpoint: str,
+        params: Optional[Dict[str, Any]] = None,
+        json: Optional[Dict[str, Any]] = None,
+        **kwargs
+    ) -> Dict[str, Any]:
+        """Make an asynchronous HTTP request.
         
-        return data
+        Args:
+            method: HTTP method
+            endpoint: API endpoint (without base URL)
+            params: Query parameters
+            json: Request body (for POST/PUT)
+            **kwargs: Additional arguments for httpx
+            
+        Returns:
+            Response data as dictionary
+            
+        Raises:
+            httpx.HTTPError: On HTTP errors
+        """
+        url = f"/{endpoint}"
+        
+        # Log request details
+        debug(f"[HTTP] {method} {self.base_url}{url}")
+        if params:
+            debug(f"[HTTP] Query params: {mask_sensitive(params)}")
+        if json:
+            debug(f"[HTTP] Request body: {truncate_data(mask_sensitive(json))}")
+        
+        response = await self.async_client.request(
+            method=method,
+            url=url,
+            params=params,
+            json=json,
+            **kwargs
+        )
+        
+        return self._handle_response(response)
+    
+    # ==================== Lifecycle Methods ====================
     
     def close(self) -> None:
-        """Close the HTTP session."""
-        if self.session:
-            self.session.close()
+        """Close the synchronous HTTP client."""
+        if self._sync_client is not None and not self._sync_client.is_closed:
+            self._sync_client.close()
+            self._sync_client = None
+    
+    async def aclose(self) -> None:
+        """Close the asynchronous HTTP client."""
+        if self._async_client is not None and not self._async_client.is_closed:
+            await self._async_client.aclose()
+            self._async_client = None
+    
+    def __enter__(self) -> "HttpClient":
+        """Context manager entry for sync client."""
+        return self
+    
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+        """Context manager exit for sync client."""
+        self.close()
+    
+    async def __aenter__(self) -> "HttpClient":
+        """Async context manager entry."""
+        return self
+    
+    async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
+        """Async context manager exit."""
+        await self.aclose()

lucidicai/sdk/features/dataset.py

@@ -1,9 +1,5 @@
-import os
 import logging
 from typing import Optional, Dict, List, Any
-from dotenv import load_dotenv
-
-from ...core.errors import APIKeyVerificationError
 
 logger = logging.getLogger("Lucidic")
 
@@ -36,37 +32,15 @@ def get_dataset(
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
         ValueError: If dataset_id is not provided.
     """
-    load_dotenv()
-    
     # Validation
     if not dataset_id:
         raise ValueError("Dataset ID is required")
     
-    # Get credentials
-    if api_key is None:
-        api_key = os.getenv("LUCIDIC_API_KEY", None)
-        if api_key is None:
-            raise APIKeyVerificationError(
-                "Make sure to either pass your API key into get_dataset() or set the LUCIDIC_API_KEY environment variable."
-            )
-    
-    if agent_id is None:
-        agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-        if agent_id is None:
-            raise APIKeyVerificationError(
-                "Lucidic agent ID not specified. Make sure to either pass your agent ID into get_dataset() or set the LUCIDIC_AGENT_ID environment variable."
-            )
-    
-    # Get HTTP client
-    from ..init import get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
+    from ..init import ensure_http_and_resources, get_http
     
+    # Ensure HTTP client is initialized and stored in SDK state
+    ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     http = get_http()
-    if not http:
-        # Create a new HTTP client if needed
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
     
     # Make request to get dataset
     response = http.get('getdataset', {'dataset_id': dataset_id})
@@ -126,31 +100,9 @@ def list_datasets(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        # Get credentials
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].list_datasets(agent_id)
 
 
@@ -179,30 +131,9 @@ def create_dataset(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].create_dataset(name, description, tags, suggested_flag_config, agent_id)
 
 
@@ -233,29 +164,9 @@ def update_dataset(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
+
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
 
     kwargs = {}
     if name is not None:
@@ -289,30 +200,9 @@ def delete_dataset(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].delete_dataset(dataset_id)
 
 
@@ -349,30 +239,9 @@ def create_dataset_item(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].create_item(
         dataset_id, name, input_data,
         expected_output=expected_output,
@@ -404,30 +273,9 @@ def get_dataset_item(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].get_item(dataset_id, item_id)
 
 
@@ -466,29 +314,9 @@ def update_dataset_item(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
+
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
 
     kwargs = {}
     if name is not None:
@@ -530,30 +358,9 @@ def delete_dataset_item(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].delete_item(dataset_id, item_id)
 
 
@@ -578,28 +385,7 @@ def list_dataset_item_sessions(
     Raises:
         APIKeyVerificationError: If API key or agent ID is missing or invalid.
     """
-    from ..init import get_resources, get_http
-    from ...core.config import SDKConfig
-    from ...api.client import HttpClient
-
-    # Get or create resources
-    resources = get_resources()
-    if not resources or 'datasets' not in resources:
-        load_dotenv()
-
-        if api_key is None:
-            api_key = os.getenv("LUCIDIC_API_KEY", None)
-            if api_key is None:
-                raise APIKeyVerificationError(
-                    "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
-                )
-
-        if agent_id is None:
-            agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
-
-        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
-        http = HttpClient(config)
-        from ...api.resources.dataset import DatasetResource
-        resources = {'datasets': DatasetResource(http)}
+    from ..init import ensure_http_and_resources
 
+    resources = ensure_http_and_resources(api_key=api_key, agent_id=agent_id)
     return resources['datasets'].list_item_sessions(dataset_id, item_id)

lucidicai/sdk/init.py

@@ -353,6 +353,75 @@ def get_resources() -> dict:
     return _sdk_state.resources
 
 
+def set_http(http: HttpClient) -> None:
+    """Set the HTTP client instance in SDK state."""
+    global _sdk_state
+    _sdk_state.http = http
+
+
+def set_resources(resources: dict) -> None:
+    """Set API resource instances in SDK state."""
+    global _sdk_state
+    _sdk_state.resources = resources
+
+
+def ensure_http_and_resources(api_key: Optional[str] = None, agent_id: Optional[str] = None) -> dict:
+    """Ensure HTTP client and resources are initialized, creating them if needed.
+    
+    This function checks if the HTTP client and resources already exist in SDK state.
+    If not, it creates them and stores them in SDK state for reuse.
+    
+    Args:
+        api_key: API key (uses env if not provided)
+        agent_id: Agent ID (uses env if not provided)
+        
+    Returns:
+        Dictionary of API resources with 'datasets' key
+        
+    Raises:
+        APIKeyVerificationError: If API key is not available
+    """
+    global _sdk_state
+    
+    # If we already have resources with datasets, return them
+    if _sdk_state.resources and 'datasets' in _sdk_state.resources:
+        return _sdk_state.resources
+    
+    # Need to create HTTP client and resources
+    from dotenv import load_dotenv
+    import os
+    from ..core.errors import APIKeyVerificationError
+    
+    load_dotenv()
+    
+    # Get credentials
+    if api_key is None:
+        api_key = os.getenv("LUCIDIC_API_KEY", None)
+        if api_key is None:
+            raise APIKeyVerificationError(
+                "Make sure to either pass your API key or set the LUCIDIC_API_KEY environment variable."
+            )
+    
+    if agent_id is None:
+        agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
+    
+    # Create or reuse HTTP client
+    if not _sdk_state.http:
+        debug("[SDK] Creating HTTP client for standalone use")
+        config = SDKConfig.from_env(api_key=api_key, agent_id=agent_id)
+        _sdk_state.http = HttpClient(config)
+    
+    # Create resources if not already present
+    if not _sdk_state.resources:
+        _sdk_state.resources = {}
+    
+    if 'datasets' not in _sdk_state.resources:
+        debug("[SDK] Creating DatasetResource for standalone use")
+        _sdk_state.resources['datasets'] = DatasetResource(_sdk_state.http)
+    
+    return _sdk_state.resources
+
+
 def get_tracer_provider() -> Optional[TracerProvider]:
     """Get the tracer provider instance."""
     return _sdk_state.tracer_provider

lucidicai/utils/images.py

@@ -11,7 +11,7 @@ import logging
 import threading
 from typing import List, Dict, Any, Optional, Tuple, Union
 from PIL import Image
-import requests
+import httpx
 
 logger = logging.getLogger("Lucidic")
 
@@ -299,9 +299,9 @@ class ImageUploader:
             image_obj, content_type = ImageHandler.prepare_for_upload(image_data, format)
         
         # Upload to S3
-        upload_response = requests.put(
+        upload_response = httpx.put(
             url,
-            data=image_obj.getvalue() if hasattr(image_obj, 'getvalue') else image_obj,
+            content=image_obj.getvalue() if hasattr(image_obj, 'getvalue') else image_obj,
             headers={"Content-Type": content_type}
         )
         upload_response.raise_for_status()

lucidicai/utils/queue.py

@@ -9,7 +9,7 @@ import json
 import queue
 import threading
 import time
-import requests
+import httpx
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional, Set, Tuple
@@ -362,7 +362,7 @@ class EventQueue:
     def _upload_blob(self, blob_url: str, data: bytes) -> None:
         """Upload compressed blob to presigned URL."""
         headers = {"Content-Type": "application/json", "Content-Encoding": "gzip"}
-        resp = requests.put(blob_url, data=data, headers=headers)
+        resp = httpx.put(blob_url, content=data, headers=headers)
         resp.raise_for_status()
 
     @staticmethod

{lucidicai-2.1.2.dist-info → lucidicai-2.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lucidicai
-Version: 2.1.2
+Version: 2.1.3
 Summary: Lucidic AI Python SDK
 Author: Andy Liang
 Author-email: andy@lucidic.ai

{lucidicai-2.1.2.dist-info → lucidicai-2.1.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-lucidicai/__init__.py,sha256=WwZ3A73h0Ttk1xgiiCt9yM-zXD1vIvQNg6XLTbMYQLY,12757
+lucidicai/__init__.py,sha256=vujMytXGPFrZRxIZtvJbo81f1AtgyZpMVIhLyUX32v0,12757
 lucidicai/action.py,sha256=sPRd1hTIVXDqnvG9ZXWEipUFh0bsXcE0Fm7RVqmVccM,237
 lucidicai/client.py,sha256=IIhlY6Mfwy47FeMxzpvIygCaqcI1FnqiXiVU6M4QEiE,22327
 lucidicai/constants.py,sha256=zN8O7TjoRHRlaGa9CZUWppS73rhzKGwaEkF9XMTV0Cg,1160
@@ -18,7 +18,7 @@ lucidicai/state.py,sha256=4Tb1X6l2or6w_e62FYSuEeghAv3xXm5gquKwzCpvdok,235
 lucidicai/step.py,sha256=_oBIyTBZBvNkUkYHIrwWd75KMSlMtR9Ws2Lo71Lyff8,2522
 lucidicai/streaming.py,sha256=QOLAzhwxetvx711J8VcphY5kXWPJz9XEBJrmHveRKMc,9796
 lucidicai/api/__init__.py,sha256=UOYuFZupG0TgzMAxbLNgpodDXhDRXBgMva8ZblgBN9Y,31
-lucidicai/api/client.py,sha256=czD3sg4wgyGQTVVlnSi3wpeCt90_D4eOuK8nBlpiv4U,7276
+lucidicai/api/client.py,sha256=RsJXaQn3ccxPcCKYyxoAR5268AhUVkkx1SvTX5_pFbI,12089
 lucidicai/api/resources/__init__.py,sha256=Wc8-JfL82wkE7eB8PHplqYvaEG2oXNXXhRyEPeduJeE,27
 lucidicai/api/resources/dataset.py,sha256=6UnMUd-y__TOAjUJLjbc0lZJRTy_gHkyoE82OvjFoN4,5583
 lucidicai/api/resources/event.py,sha256=GyyNL3_k53EbmvTdgJEABexiuJnoX61hxWey7DYmlYY,2434
@@ -50,10 +50,10 @@ lucidicai/sdk/decorators.py,sha256=B5BXG9Sn5ruUkxFq10L1rrCR_wzYUPlYeu5aqyXetMM,8
 lucidicai/sdk/error_boundary.py,sha256=IPr5wS9rS7ZQNgEaBwK53UaixAm6L2rijKKFfxcxjUI,9190
 lucidicai/sdk/event.py,sha256=NiPcnPzYCU0VlFbBk93LD88wqAYmnglV64nQb2XteOs,3747
 lucidicai/sdk/event_builder.py,sha256=oMvt39m07ZLmPllJTWwRxpinJUz9_AD17yNE6wQRoDA,10423
-lucidicai/sdk/init.py,sha256=tfpTRZLT317xNC_GE2OKqRR02Nj3s3a12CJELc-vVAE,12923
+lucidicai/sdk/init.py,sha256=lUEChS3YKiwKWFLZnJIs-VXuicLZPiBXr8_UexqjKhE,15204
 lucidicai/sdk/shutdown_manager.py,sha256=I5ylR96QHQ_SfP1euAiM0qQ-I7upCPMW1HUNvoj7hCw,12090
 lucidicai/sdk/features/__init__.py,sha256=23KUF2EZBzsaH9JUFDGNXZb_3PSfc35VZfD59gAfyR0,26
-lucidicai/sdk/features/dataset.py,sha256=qFGnu8Wm1yhaflBhtm-5veN-KaoxGLBL5xWEifkrsY0,19416
+lucidicai/sdk/features/dataset.py,sha256=DfoxHrfN86Hu-H0vAO0OHGwaevaGaXm26Z5wpfuXDSY,11814
 lucidicai/sdk/features/feature_flag.py,sha256=SzuzHiVnbticD6Ojn0_i9xQKui2s9QUFPJ7LixzAtf4,13844
 lucidicai/telemetry/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lucidicai/telemetry/base_provider.py,sha256=nrZVr4Y9xcAiMn4uAN3t3k6DlHNTvlXrA4qQg7lANOQ,544
@@ -78,10 +78,10 @@ lucidicai/telemetry/utils/model_pricing.py,sha256=Dxi6e0WjcIyCTkVX7K7f0pJ5rPu7nS
 lucidicai/telemetry/utils/text_storage.py,sha256=L62MMJ8E23TDqDTUv2aRntdKMCItsXV7XjY6cFwx2DE,1503
 lucidicai/telemetry/utils/universal_image_interceptor.py,sha256=vARgMk1hVSF--zfi5b8qBpJJOESuD17YlH9xqxmB9Uw,15954
 lucidicai/utils/__init__.py,sha256=ZiGtmJaF0ph9iIFIgQiAreVuYM_1o7qu9VySK1NblTw,22
-lucidicai/utils/images.py,sha256=YHFjeKHRxzWu0IsuNwKw303egPsd99AShaD4WND1lJk,12325
+lucidicai/utils/images.py,sha256=HzPpv-Xs9k9VaDD1daT5fdZvoNf5ke9-E1awmkzAADc,12322
 lucidicai/utils/logger.py,sha256=R3B3gSee64F6UVHUrShihBq_O7W7bgfrBiVDXTO3Isg,4777
-lucidicai/utils/queue.py,sha256=tZYPAUHRAK_uyE8Mk4PloObsBcfITurHwIlNHr3gMFU,17326
-lucidicai-2.1.2.dist-info/METADATA,sha256=ZcL5YaquJn3D5MyMf7UqZTzrP9m5E4XAhEJQJAuOXTs,902
-lucidicai-2.1.2.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-lucidicai-2.1.2.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
-lucidicai-2.1.2.dist-info/RECORD,,
+lucidicai/utils/queue.py,sha256=8DQwnGw7pINEJ0dNSkB0PhdPW-iBQQ-YZg23poe4umE,17323
+lucidicai-2.1.3.dist-info/METADATA,sha256=A5SQWuHc48k5n0Dg9KFI52E2SLe0su9vqHEC8jWIi5A,902
+lucidicai-2.1.3.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+lucidicai-2.1.3.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
+lucidicai-2.1.3.dist-info/RECORD,,