ragaai-catalyst 2.1.5b35__py3-none-any.whl → 2.1.5b36__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -305,7 +305,7 @@ class BaseTracer:
 
             logger.debug("Base URL used for uploading: {}".format(self.base_url))
             
-            # Submit to background process for uploading
+            # Submit to background process for uploading using futures
             self.upload_task_id = submit_upload_task(
                 filepath=filepath,
                 hash_id=hash_id,
@@ -317,28 +317,28 @@ class BaseTracer:
                 base_url=self.base_url
             )
             
-            # # For backward compatibility
-            # self._is_uploading = True
+            # For backward compatibility
+            self._is_uploading = True
             
-            # # Start checking for completion if a callback is registered
-            # if self._upload_completed_callback:
-            #     # Start a thread to check status and call callback when complete
-            #     def check_status_and_callback():
-            #         status = self.get_upload_status()
-            #         if status.get("status") in ["completed", "failed"]:
-            #             self._is_uploading = False
-            #             # Execute callback
-            #             try:
-            #                 self._upload_completed_callback(self)
-            #             except Exception as e:
-            #                 logger.error(f"Error in upload completion callback: {e}")
-            #             return
+            # Start checking for completion if a callback is registered
+            if self._upload_completed_callback:
+                # Start a thread to check status and call callback when complete
+                def check_status_and_callback():
+                    status = self.get_upload_status()
+                    if status.get("status") in ["completed", "failed"]:
+                        self._is_uploading = False
+                        # Execute callback
+                        try:
+                            self._upload_completed_callback(self)
+                        except Exception as e:
+                            logger.error(f"Error in upload completion callback: {e}")
+                        return
                     
-            #         # Check again after a delay
-            #         threading.Timer(5.0, check_status_and_callback).start()
+                    # Check again after a delay
+                    threading.Timer(5.0, check_status_and_callback).start()
                 
-            #     # Start checking
-            #     threading.Timer(5.0, check_status_and_callback).start()
+                # Start checking
+                threading.Timer(5.0, check_status_and_callback).start()
             
             logger.info(f"Submitted upload task with ID: {self.upload_task_id}")
 

ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -17,6 +17,8 @@ from datetime import datetime
 import atexit
 import glob
 from logging.handlers import RotatingFileHandler
+import concurrent.futures
+from typing import Dict, Any, Optional
 
 # Set up logging
 log_dir = os.path.join(tempfile.gettempdir(), "ragaai_logs")
@@ -55,320 +57,175 @@ except ImportError:
 QUEUE_DIR = os.path.join(tempfile.gettempdir(), "ragaai_tasks")
 os.makedirs(QUEUE_DIR, exist_ok=True)
 
-# Clean up any stale processes
-def cleanup_stale_processes():
-    """Check for stale processes but allow active uploads to complete"""
-    pid_file = os.path.join(tempfile.gettempdir(), "trace_uploader.pid")
-    if os.path.exists(pid_file):
-        try:
-            with open(pid_file, "r") as f:
-                old_pid = int(f.read().strip())
-            try:
-                import psutil
-                if psutil.pid_exists(old_pid):
-                    p = psutil.Process(old_pid)
-                    if "trace_uploader.py" in " ".join(p.cmdline()):
-                        # Instead of terminating, just remove the PID file
-                        # This allows the process to finish its current uploads
-                        logger.info(f"Removing PID file for process {old_pid}")
-                        os.remove(pid_file)
-                        return
-            except Exception as e:
-                logger.warning(f"Error checking stale process: {e}")
-            os.remove(pid_file)
-        except Exception as e:
-            logger.warning(f"Error reading PID file: {e}")
-
-cleanup_stale_processes()
-
 # Status codes
 STATUS_PENDING = "pending"
 STATUS_PROCESSING = "processing"
 STATUS_COMPLETED = "completed"
 STATUS_FAILED = "failed"
 
-class UploadTask:
-    """Class representing a single upload task"""
-    
-    def __init__(self, task_id=None, **kwargs):
-        self.task_id = task_id or f"task_{int(time.time())}_{os.getpid()}_{hash(str(time.time()))}"
-        self.status = STATUS_PENDING
-        self.attempts = 0
-        self.max_attempts = 3
-        self.created_at = datetime.now().isoformat()
-        self.updated_at = self.created_at
-        self.error = None
-        
-        # Task details
-        self.filepath = kwargs.get("filepath")
-        self.hash_id = kwargs.get("hash_id")
-        self.zip_path = kwargs.get("zip_path")
-        self.project_name = kwargs.get("project_name")
-        self.project_id = kwargs.get("project_id")
-        self.dataset_name = kwargs.get("dataset_name")
-        self.user_details = kwargs.get("user_details", {})
-        self.base_url = kwargs.get("base_url")
-        
-    def to_dict(self):
-        """Convert task to dictionary for serialization"""
-        return {
-            "task_id": self.task_id,
-            "status": self.status,
-            "attempts": self.attempts,
-            "max_attempts": self.max_attempts,
-            "created_at": self.created_at,
-            "updated_at": self.updated_at,
-            "error": self.error,
-            "filepath": self.filepath,
-            "hash_id": self.hash_id,
-            "zip_path": self.zip_path,
-            "project_name": self.project_name,
-            "project_id": self.project_id,
-            "dataset_name": self.dataset_name,
-            "user_details": self.user_details,
-            "base_url": self.base_url
-        }
-        
-    @classmethod
-    def from_dict(cls, data):
-        """Create task from dictionary"""
-        task = cls(task_id=data.get("task_id"))
-        task.status = data.get("status", STATUS_PENDING)
-        task.attempts = data.get("attempts", 0)
-        task.max_attempts = data.get("max_attempts", 3)
-        task.created_at = data.get("created_at")
-        task.updated_at = data.get("updated_at")
-        task.error = data.get("error")
-        task.filepath = data.get("filepath")
-        task.hash_id = data.get("hash_id")
-        task.zip_path = data.get("zip_path")
-        task.project_name = data.get("project_name")
-        task.project_id = data.get("project_id")
-        task.dataset_name = data.get("dataset_name")
-        task.user_details = data.get("user_details", {})
-        task.base_url = data.get("base_url")    
-        return task
-        
-    def update_status(self, status, error=None):
-        """Update task status"""
-        self.status = status
-        self.updated_at = datetime.now().isoformat()
-        if error:
-            self.error = str(error)
-        self.save()
-        
-    def increment_attempts(self):
-        """Increment the attempt counter"""
-        self.attempts += 1
-        self.updated_at = datetime.now().isoformat()
-        self.save()
-        
-    def save(self):
-        """Save task to disk"""
-        task_path = os.path.join(QUEUE_DIR, f"{self.task_id}.json")
-        with open(task_path, "w") as f:
-            json.dump(self.to_dict(), f, indent=2)
-            
-    def delete(self):
-        """Delete task file from disk"""
-        task_path = os.path.join(QUEUE_DIR, f"{self.task_id}.json")
-        if os.path.exists(task_path):
-            os.remove(task_path)
-            
-    @staticmethod
-    def list_pending_tasks():
-        """List all pending tasks"""
-        tasks = []
-        #logger.info("Listing pending tasks from queue directory: {}".format(QUEUE_DIR))
-        for filename in os.listdir(QUEUE_DIR):
-            if filename.endswith(".json"):
-                try:
-                    with open(os.path.join(QUEUE_DIR, filename), "r") as f:
-                        task_data = json.load(f)
-                        task = UploadTask.from_dict(task_data)
-                        if task.status in [STATUS_PENDING, STATUS_FAILED] and task.attempts < task.max_attempts:
-                            # Verify files still exist
-                            if (not task.filepath or os.path.exists(task.filepath)) and \
-                               (not task.zip_path or os.path.exists(task.zip_path)):
-                                tasks.append(task)
-                            else:
-                                # Files missing, mark as failed
-                                task.update_status(STATUS_FAILED, "Required files missing")
-                except Exception as e:
-                    logger.error(f"Error loading task {filename}: {e}")
-        return tasks
+# Global executor for handling uploads
+_executor = None
+# Dictionary to track futures and their associated task IDs
+_futures: Dict[str, Any] = {}
 
+def get_executor():
+    """Get or create the thread pool executor"""
+    global _executor
+    if _executor is None:
+        _executor = concurrent.futures.ThreadPoolExecutor(max_workers=8, thread_name_prefix="trace_uploader")
+    return _executor
 
-class TraceUploader:
-    """
-    Trace uploader process
-    Handles the actual upload work in a separate process
+def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str, 
+                  project_name: str, project_id: str, dataset_name: str, 
+                  user_details: Dict[str, Any], base_url: str) -> Dict[str, Any]:
     """
+    Process a single upload task
     
-    def __init__(self):
-        self.running = True
-        self.processing = False
-        
-    def start(self):
-        """Start the uploader loop"""
-        logger.info("Trace uploader starting")
-        
-        # Register signal handlers
-        signal.signal(signal.SIGTERM, self.handle_signal)
-        signal.signal(signal.SIGINT, self.handle_signal)
+    Args:
+        task_id: Unique identifier for the task
+        filepath: Path to the trace file
+        hash_id: Hash ID for the code
+        zip_path: Path to the code zip file
+        project_name: Project name
+        project_id: Project ID
+        dataset_name: Dataset name
+        user_details: User details dictionary
+        base_url: Base URL for API calls
         
-        # Register cleanup
-        atexit.register(self.cleanup)
-        
-        # Main processing loop
-        while self.running:
-            try:
-                # Get pending tasks
-                tasks = UploadTask.list_pending_tasks()
-                if tasks:
-                    logger.info(f"Found {len(tasks)} pending tasks")
-                    for task in tasks:
-                        if not self.running:
-                            break
-                        self.process_task(task)
-                else:
-                    # No tasks, sleep before checking again
-                    time.sleep(5)
-            except Exception as e:
-                logger.error(f"Error in uploader loop: {e}")
-                time.sleep(5)
-                
-        logger.info("Trace uploader stopped")
-        
-    def process_task(self, task):
-        """Process a single upload task"""
-        logger.info(f"Starting to process task {task.task_id}")
-        logger.debug(f"Task details: {task.to_dict()}")
-
+    Returns:
+        Dict containing status and any error information
+    """
+    logger.info(f"Processing upload task {task_id}")
+    result = {
+        "task_id": task_id,
+        "status": STATUS_PROCESSING,
+        "error": None,
+        "start_time": datetime.now().isoformat()
+    }
+    
+    # Save initial status to file
+    save_task_status(result)
+    
+    try:
         # Check if file exists
-        if not os.path.exists(task.filepath):
-            error_msg = f"Task filepath does not exist: {task.filepath}"
+        if not os.path.exists(filepath):
+            error_msg = f"Task filepath does not exist: {filepath}"
             logger.error(error_msg)
-            task.update_status(STATUS_FAILED, error_msg)
-            return
+            result["status"] = STATUS_FAILED
+            result["error"] = error_msg
+            save_task_status(result)
+            return result
 
         if not IMPORTS_AVAILABLE:
-            logger.warning(f"Test mode: Simulating processing of task {task.task_id}")
-            #time.sleep(2)  # Simulate work
-            #task.update_status(STATUS_COMPLETED)
-            return
+            logger.warning(f"Test mode: Simulating processing of task {task_id}")
+            time.sleep(2)  # Simulate work
+            result["status"] = STATUS_COMPLETED
+            save_task_status(result)
+            return result
             
-        logger.info(f"Processing task {task.task_id} (attempt {task.attempts+1}/{task.max_attempts})")
-        self.processing = True
-        task.update_status(STATUS_PROCESSING)
-        task.increment_attempts()
-        
-        # Log memory state for debugging
-        try:
-            import psutil
-            process = psutil.Process()
-            logger.debug(f"Memory usage before processing: {process.memory_info().rss / 1024 / 1024:.2f} MB")
-        except ImportError:
-            pass
-        
+        # Step 1: Create dataset schema
+        logger.info(f"Creating dataset schema for {dataset_name} with base_url: {base_url}")
         try:
-            # Step 1: Create dataset schema
-            logger.info(f"Creating dataset schema for {task.dataset_name} with base_url: {task.base_url}")
             response = create_dataset_schema_with_trace(
-                dataset_name=task.dataset_name,
-                project_name=task.project_name,
-                base_url=task.base_url
+                dataset_name=dataset_name,
+                project_name=project_name,
+                base_url=base_url
             )
             logger.info(f"Dataset schema created: {response}")
-            
-            # Step 2: Upload trace metrics
-            if task.filepath and os.path.exists(task.filepath):
-                logger.info(f"Uploading trace metrics for {task.filepath}")
-                try:
-                    response = upload_trace_metric(
-                        json_file_path=task.filepath,
-                        dataset_name=task.dataset_name,
-                        project_name=task.project_name,
-                        base_url=task.base_url
-                    )
-                    logger.info(f"Trace metrics uploaded: {response}")
-                except Exception as e:
-                    logger.error(f"Error uploading trace metrics: {e}")
-                    # Continue with other uploads
-            else:
-                logger.warning(f"Trace file {task.filepath} not found, skipping metrics upload")
-            
-            # Step 3: Upload agentic traces
-            if task.filepath and os.path.exists(task.filepath):
-                logger.info(f"Uploading agentic traces for {task.filepath}")
-                try:
-                    upload_traces = UploadAgenticTraces(
-                        json_file_path=task.filepath,
-                        project_name=task.project_name,
-                        project_id=task.project_id,
-                        dataset_name=task.dataset_name,
-                        user_detail=task.user_details,
-                        base_url=task.base_url,   
-                    )
-                    upload_traces.upload_agentic_traces()
-                    logger.info("Agentic traces uploaded successfully")
-                except Exception as e:
-                    logger.error(f"Error uploading agentic traces: {e}")
-                    # Continue with code upload
-            else:
-                logger.warning(f"Trace file {task.filepath} not found, skipping traces upload")
-            
-            # Step 4: Upload code hash
-            if task.hash_id and task.zip_path and os.path.exists(task.zip_path):
-                logger.info(f"Uploading code hash {task.hash_id}")
-                try:
-                    response = upload_code(
-                        hash_id=task.hash_id,
-                        zip_path=task.zip_path,
-                        project_name=task.project_name,
-                        dataset_name=task.dataset_name,
-                        base_url=task.base_url
-                    )
-                    logger.info(f"Code hash uploaded: {response}")
-                except Exception as e:
-                    logger.error(f"Error uploading code hash: {e}")
-            else:
-                logger.warning(f"Code zip {task.zip_path} not found, skipping code upload")
-            
-            # Mark task as completed
-            task.update_status(STATUS_COMPLETED)
-            logger.info(f"Task {task.task_id} completed successfully")
-            
-            # Clean up task file
-            task.delete()
-            
         except Exception as e:
-            logger.error(f"Error processing task {task.task_id}: {e}")
-            if task.attempts >= task.max_attempts:
-                task.update_status(STATUS_FAILED, str(e))
-                logger.error(f"Task {task.task_id} failed after {task.attempts} attempts")
-            else:
-                task.update_status(STATUS_PENDING, str(e))
-                logger.warning(f"Task {task.task_id} will be retried (attempt {task.attempts}/{task.max_attempts})")
-        finally:
-            self.processing = False
+            logger.error(f"Error creating dataset schema: {e}")
+            # Continue with other steps
             
-    def handle_signal(self, signum, frame):
-        """Handle termination signals"""
-        logger.info(f"Received signal {signum}, shutting down gracefully")
-        self.running = False
+        # Step 2: Upload trace metrics
+        if filepath and os.path.exists(filepath):
+            logger.info(f"Uploading trace metrics for {filepath}")
+            try:
+                response = upload_trace_metric(
+                    json_file_path=filepath,
+                    dataset_name=dataset_name,
+                    project_name=project_name,
+                    base_url=base_url
+                )
+                logger.info(f"Trace metrics uploaded: {response}")
+            except Exception as e:
+                logger.error(f"Error uploading trace metrics: {e}")
+                # Continue with other uploads
+        else:
+            logger.warning(f"Trace file {filepath} not found, skipping metrics upload")
         
-    def cleanup(self):
-        """Cleanup before exit"""
-        logger.info("Performing cleanup before exit")
-        self.running = False
+        # Step 3: Upload agentic traces
+        if filepath and os.path.exists(filepath):
+            logger.info(f"Uploading agentic traces for {filepath}")
+            try:
+                upload_traces = UploadAgenticTraces(
+                    json_file_path=filepath,
+                    project_name=project_name,
+                    project_id=project_id,
+                    dataset_name=dataset_name,
+                    user_detail=user_details,
+                    base_url=base_url,   
+                )
+                upload_traces.upload_agentic_traces()
+                logger.info("Agentic traces uploaded successfully")
+            except Exception as e:
+                logger.error(f"Error uploading agentic traces: {e}")
+                # Continue with code upload
+        else:
+            logger.warning(f"Trace file {filepath} not found, skipping traces upload")
+        
+        # Step 4: Upload code hash
+        if hash_id and zip_path and os.path.exists(zip_path):
+            logger.info(f"Uploading code hash {hash_id}")
+            try:
+                response = upload_code(
+                    hash_id=hash_id,
+                    zip_path=zip_path,
+                    project_name=project_name,
+                    dataset_name=dataset_name,
+                    base_url=base_url
+                )
+                logger.info(f"Code hash uploaded: {response}")
+            except Exception as e:
+                logger.error(f"Error uploading code hash: {e}")
+        else:
+            logger.warning(f"Code zip {zip_path} not found, skipping code upload")
         
+        # Mark task as completed
+        result["status"] = STATUS_COMPLETED
+        result["end_time"] = datetime.now().isoformat()
+        logger.info(f"Task {task_id} completed successfully")
+        
+    except Exception as e:
+        logger.error(f"Error processing task {task_id}: {e}")
+        result["status"] = STATUS_FAILED
+        result["error"] = str(e)
+        result["end_time"] = datetime.now().isoformat()
+        
+    # Save final status
+    save_task_status(result)
+    return result
+
+def save_task_status(task_status: Dict[str, Any]):
+    """Save task status to a file"""
+    task_id = task_status["task_id"]
+    status_path = os.path.join(QUEUE_DIR, f"{task_id}_status.json")
+    with open(status_path, "w") as f:
+        json.dump(task_status, f, indent=2)
 
 def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, dataset_name, user_details, base_url):
     """
-    Submit a new upload task to the queue.
-    This function can be called from the main application.
+    Submit a new upload task using futures.
     
+    Args:
+        filepath: Path to the trace file
+        hash_id: Hash ID for the code
+        zip_path: Path to the code zip file
+        project_name: Project name
+        project_id: Project ID
+        dataset_name: Dataset name
+        user_details: User details dictionary
+        base_url: Base URL for API calls
+        
     Returns:
         str: Task ID
     """
@@ -380,11 +237,18 @@ def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, da
         logger.error(f"Trace file not found: {filepath}")
         return None
 
-    # Create task with absolute path to the trace file
+    # Create absolute path to the trace file
     filepath = os.path.abspath(filepath)
     logger.debug(f"Using absolute filepath: {filepath}")
 
-    task = UploadTask(
+    # Generate a unique task ID
+    task_id = f"task_{int(time.time())}_{os.getpid()}_{hash(str(time.time()))}"
+    
+    # Submit the task to the executor
+    executor = get_executor()
+    future = executor.submit(
+        process_upload,
+        task_id=task_id,
         filepath=filepath,
         hash_id=hash_id,
         zip_path=zip_path,
@@ -395,290 +259,97 @@ def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, da
         base_url=base_url
     )
     
-    # Save the task with proper error handling
-    task_path = os.path.join(QUEUE_DIR, f"{task.task_id}.json")
-    logger.debug(f"Saving task to: {task_path}")
+    # Store the future for later status checks
+    _futures[task_id] = future
     
-    try:
-        # Ensure queue directory exists
-        os.makedirs(QUEUE_DIR, exist_ok=True)
-        
-        with open(task_path, "w") as f:
-            json.dump(task.to_dict(), f, indent=2)
-            
-        logger.info(f"Task {task.task_id} created successfully for trace file: {filepath}")
-    except Exception as e:
-        logger.error(f"Error creating task file: {e}", exc_info=True)
-        return None
-    
-    # Ensure uploader process is running
-    logger.info("Starting uploader process...")
-    pid = ensure_uploader_running()
-    if pid:
-        logger.info(f"Uploader process running with PID {pid}")
-    else:
-        logger.warning("Failed to start uploader process, but task was queued")
+    # Create initial status
+    initial_status = {
+        "task_id": task_id,
+        "status": STATUS_PENDING,
+        "error": None,
+        "start_time": datetime.now().isoformat()
+    }
+    save_task_status(initial_status)
     
-    return task.task_id
-
+    return task_id
 
 def get_task_status(task_id):
     """
     Get the status of a task by ID.
-    This function can be called from the main application.
     
+    Args:
+        task_id: Task ID to check
+        
     Returns:
         dict: Task status information
     """
-    task_path = os.path.join(QUEUE_DIR, f"{task_id}.json")
-    if not os.path.exists(task_path):
-        # Check if it might be in completed directory
-        completed_path = os.path.join(QUEUE_DIR, "completed", f"{task_id}.json")
-        if os.path.exists(completed_path):
-            with open(completed_path, "r") as f:
+    logger.debug(f"Getting status for task {task_id}")
+    
+    # Check if we have a future for this task
+    future = _futures.get(task_id)
+    
+    # If we have a future, check its status
+    if future:
+        if future.done():
+            try:
+                # Get the result (this will re-raise any exception that occurred)
+                result = future.result(timeout=0)
+                return result
+            except concurrent.futures.TimeoutError:
+                return {"status": STATUS_PROCESSING, "error": None}
+            except Exception as e:
+                logger.error(f"Error retrieving future result for task {task_id}: {e}")
+                return {"status": STATUS_FAILED, "error": str(e)}
+        else:
+            return {"status": STATUS_PROCESSING, "error": None}
+    
+    # If we don't have a future, try to read from the status file
+    status_path = os.path.join(QUEUE_DIR, f"{task_id}_status.json")
+    if os.path.exists(status_path):
+        try:
+            with open(status_path, "r") as f:
                 return json.load(f)
-        return {"status": "unknown", "error": "Task not found"}
+        except Exception as e:
+            logger.error(f"Error reading status file for task {task_id}: {e}")
+            return {"status": "unknown", "error": f"Error reading status: {e}"}
     
-    with open(task_path, "r") as f:
-        return json.load(f)
+    return {"status": "unknown", "error": "Task not found"}
 
+def shutdown():
+    """Shutdown the executor"""
+    global _executor
+    if _executor:
+        logger.info("Shutting down executor")
+        _executor.shutdown(wait=False)
+        _executor = None
 
+# Register shutdown handler
+atexit.register(shutdown)
+
+# For backward compatibility
 def ensure_uploader_running():
     """
-    Ensure the uploader process is running.
-    Starts it if not already running.
+    Ensure the uploader is running.
+    This is a no-op in the futures implementation, but kept for API compatibility.
     """
-    logger.info("Checking if uploader process is running...")
-    
-    # Check if we can find a running process
-    pid_file = os.path.join(tempfile.gettempdir(), "trace_uploader.pid")
-    logger.debug(f"PID file location: {pid_file}")
-    
-    if os.path.exists(pid_file):
-        try:
-            with open(pid_file, "r") as f:
-                pid_str = f.read().strip()
-                logger.debug(f"Read PID from file: {pid_str}")
-                pid = int(pid_str)
-            
-            # Check if process is actually running
-            # Use platform-specific process check
-            is_running = False
-            try:
-                if os.name == 'posix':  # Unix/Linux/Mac
-                    logger.debug(f"Checking process {pid} on Unix/Mac")
-                    os.kill(pid, 0)  # This raises an exception if process doesn't exist
-                    is_running = True
-                else:  # Windows
-                    logger.debug(f"Checking process {pid} on Windows")
-                    import ctypes
-                    kernel32 = ctypes.windll.kernel32
-                    SYNCHRONIZE = 0x00100000
-                    process = kernel32.OpenProcess(SYNCHRONIZE, False, pid)
-                    if process:
-                        kernel32.CloseHandle(process)
-                        is_running = True
-            except (ImportError, AttributeError) as e:
-                logger.debug(f"Platform-specific check failed: {e}, falling back to cross-platform check")
-                # Fall back to cross-platform check
-                try:
-                    import psutil
-                    is_running = psutil.pid_exists(pid)
-                    logger.debug(f"psutil check result: {is_running}")
-                except ImportError:
-                    logger.debug("psutil not available, using basic process check")
-                    # If psutil is not available, make a best guess
-                    try:
-                        os.kill(pid, 0)
-                        is_running = True
-                    except Exception as e:
-                        logger.debug(f"Basic process check failed: {e}")
-                        is_running = False
-            
-            if is_running:
-                logger.debug(f"Uploader process already running with PID {pid}")
-                return pid
-        except (ProcessLookupError, ValueError, PermissionError):
-            # Process not running or other error, remove stale PID file
-            try:
-                os.remove(pid_file)
-            except:
-                pass
-    
-    # Start new process
-    logger.info("Starting new uploader process")
-    
-    # Get the path to this script
-    script_path = os.path.abspath(__file__)
-    
-    # Start detached process in a platform-specific way
-    try:
-        # First, try the preferred method for each platform
-        if os.name == 'posix':  # Unix/Linux/Mac
-            import subprocess
-            
-            # Use double fork method on Unix systems
-            try:
-                # First fork
-                pid = os.fork()
-                if pid > 0:
-                    # Parent process, return
-                    return pid
-                    
-                # Decouple from parent environment
-                os.chdir('/')
-                os.setsid()
-                os.umask(0)
-                
-                # Second fork
-                pid = os.fork()
-                if pid > 0:
-                    # Exit from second parent
-                    os._exit(0)
-                    
-                # Redirect standard file descriptors
-                sys.stdout.flush()
-                sys.stderr.flush()
-                si = open(os.devnull, 'r')
-                so = open(os.path.join(tempfile.gettempdir(), 'trace_uploader_stdout.log'), 'a+')
-                se = open(os.path.join(tempfile.gettempdir(), 'trace_uploader_stderr.log'), 'a+')
-                os.dup2(si.fileno(), sys.stdin.fileno())
-                os.dup2(so.fileno(), sys.stdout.fileno())
-                os.dup2(se.fileno(), sys.stderr.fileno())
-                
-                # Execute the daemon process
-                os.execl(sys.executable, sys.executable, script_path, '--daemon')
-                
-            except (AttributeError, OSError):
-                # Fork not available, try subprocess
-                process = subprocess.Popen(
-                    [sys.executable, script_path, "--daemon"],
-                    stdout=subprocess.PIPE,
-                    stderr=subprocess.PIPE,
-                    stdin=subprocess.PIPE,
-                    start_new_session=True  # Detach from parent
-                )
-                pid = process.pid
-                
-        else:  # Windows
-            import subprocess
-            # Use the DETACHED_PROCESS flag on Windows
-            startupinfo = subprocess.STARTUPINFO()
-            startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
-            startupinfo.wShowWindow = 0  # SW_HIDE
-            
-            # Windows-specific flags
-            DETACHED_PROCESS = 0x00000008
-            CREATE_NO_WINDOW = 0x08000000
-            
-            process = subprocess.Popen(
-                [sys.executable, script_path, "--daemon"],
-                stdout=subprocess.PIPE,
-                stderr=subprocess.PIPE,
-                stdin=subprocess.PIPE,
-                startupinfo=startupinfo,
-                creationflags=DETACHED_PROCESS | CREATE_NO_WINDOW
-            )
-            pid = process.pid
-        
-        # Write PID to file
-        with open(pid_file, "w") as f:
-            f.write(str(pid))
-            
-        logger.info(f"Started uploader process with PID {pid}")
-        return pid
-        
-    except Exception as e:
-        logger.error(f"Error starting uploader process using primary method: {e}")
-        
-        # Fallback method using multiprocessing (works on most platforms)
-        try:
-            logger.info("Trying fallback method with multiprocessing")
-            import multiprocessing
-            
-            def run_uploader():
-                """Run the uploader in a separate process"""
-                # Redirect output
-                sys.stdout = open(os.path.join(tempfile.gettempdir(), 'trace_uploader_stdout.log'), 'a+')
-                sys.stderr = open(os.path.join(tempfile.gettempdir(), 'trace_uploader_stderr.log'), 'a+')
-                
-                # Run daemon
-                run_daemon()
-                
-            # Start process
-            process = multiprocessing.Process(target=run_uploader)
-            process.daemon = True  # Daemonize it
-            process.start()
-            pid = process.pid
-            
-            # Write PID to file
-            with open(pid_file, "w") as f:
-                f.write(str(pid))
-                
-            logger.info(f"Started uploader process with fallback method, PID {pid}")
-            return pid
-            
-        except Exception as e2:
-            logger.error(f"Error starting uploader process using fallback method: {e2}")
-            
-            # Super fallback - run in the current process if all else fails
-            # This is not ideal but better than failing completely
-            try:
-                logger.warning("Using emergency fallback - running in current process thread")
-                import threading
-                
-                thread = threading.Thread(target=run_daemon, daemon=True)
-                thread.start()
-                
-                # No real PID since it's a thread, but we'll create a marker file
-                with open(pid_file, "w") as f:
-                    f.write(f"thread_{id(thread)}")
-                    
-                return None
-            except Exception as e3:
-                logger.error(f"All methods failed to start uploader: {e3}")
-                return None
-
+    get_executor()  # Just ensure the executor is created
+    return True
 
+# For backward compatibility with the old daemon mode
 def run_daemon():
-    """Run the uploader as a daemon process"""
-    # Write PID to file
-    pid_file = os.path.join(tempfile.gettempdir(), "trace_uploader.pid")
-    with open(pid_file, "w") as f:
-        f.write(str(os.getpid()))
-        
-    try:
-        uploader = TraceUploader()
-        uploader.start()
-    finally:
-        # Clean up PID file
-        if os.path.exists(pid_file):
-            os.remove(pid_file)
-
+    """
+    Run the uploader as a daemon process.
+    This is a no-op in the futures implementation, but kept for API compatibility.
+    """
+    logger.info("Daemon mode not needed in futures implementation")
+    return
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Trace uploader process")
     parser.add_argument("--daemon", action="store_true", help="Run as daemon process")
-    parser.add_argument("--test", action="store_true", help="Submit a test task")
     args = parser.parse_args()
     
     if args.daemon:
-        run_daemon()
-    elif args.test:
-        # Submit a test task
-        test_file = os.path.join(tempfile.gettempdir(), "test_trace.json")
-        with open(test_file, "w") as f:
-            f.write("{}")
-        
-        task_id = submit_upload_task(
-            filepath=test_file,
-            hash_id="test_hash",
-            zip_path=test_file,
-            project_name="test_project",
-            project_id="test_id",
-            dataset_name="test_dataset",
-            user_details={"id": "test_user"}
-        )
-        print(f"Submitted test task with ID: {task_id}")
+        logger.info("Daemon mode not needed in futures implementation")
     else:
-        print("Use --daemon to run as daemon or --test to submit a test task")
+        logger.info("Interactive mode not needed in futures implementation")

{ragaai_catalyst-2.1.5b35.dist-info → ragaai_catalyst-2.1.5b36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b35
+Version: 2.1.5b36
 Summary: RAGA AI CATALYST
 Author-email: Kiran Scaria <kiran.scaria@raga.ai>, Kedar Gaikwad <kedar.gaikwad@raga.ai>, Dushyant Mahajan <dushyant.mahajan@raga.ai>, Siddhartha Kosti <siddhartha.kosti@raga.ai>, Ritika Goel <ritika.goel@raga.ai>, Vijay Chaurasia <vijay.chaurasia@raga.ai>, Tushar Kumar <tushar.kumar@raga.ai>
 Requires-Python: <3.13,>=3.9

{ragaai_catalyst-2.1.5b35.dist-info → ragaai_catalyst-2.1.5b36.dist-info}/RECORD

@@ -45,7 +45,7 @@ ragaai_catalyst/tracers/agentic_tracing/tests/ai_travel_agent.py,sha256=S4rCcKzU
 ragaai_catalyst/tracers/agentic_tracing/tests/unique_decorator_test.py,sha256=Xk1cLzs-2A3dgyBwRRnCWs7Eubki40FVonwd433hPN8,4805
 ragaai_catalyst/tracers/agentic_tracing/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=LzbsHvELwBmH8ObFomJRhiQ98b6MEi18irm0DPiplt0,29743
-ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=_C2j97NEH1x3Hnn63LowcS39IdfIUmwbDiwWFLlOgcc,54275
+ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=Kmy1kgwy19e7MuMMq9GPUq9VXpJV2bXeaIhx8UxX5Sc,54251
 ragaai_catalyst/tracers/agentic_tracing/tracers/custom_tracer.py,sha256=OBJJjFSvwRjCGNJyqX3yIfC1W05ZN2QUXasCJ4gmCjQ,13930
 ragaai_catalyst/tracers/agentic_tracing/tracers/langgraph_tracer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=z-qzmCQCkhyW0aLDUR_rNq4pmxhAaVhNY-kZQsox-Ws,50221
@@ -54,7 +54,7 @@ ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=m8CxYkl
 ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=xxrliKPfdfbIZRZqMnUewsaTD8_Hv0dbuoBivNZGD4U,21674
 ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py,sha256=bhSUhNQCuJXKjgJAXhjKEYjnHMpYN90FSZdR84fNIKU,4614
 ragaai_catalyst/tracers/agentic_tracing/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=pklyXsz1_tKsU-Kon0ZxsV11BGRLV66bCh8UG9tfrhc,26454
+ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=uRuZlo5XFoBK-a9tYI5ckhSRiXCaUf9fsE2Plu_W1XI,12282
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=icycLgfA0734xxoM1rTMG_iIrI3iM94th8RQggJ7sSw,8541
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=aw_eHhUYRbR_9IbIkNjYb7NOsmETD3k1p4a6gxaGI7Q,6462
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_local_metric.py,sha256=m1O8lKpxKwtHofXLW3fTHX5yfqDW5GxoveARlg5cTw4,2571
@@ -87,8 +87,8 @@ ragaai_catalyst/tracers/utils/extraction_logic_llama_index.py,sha256=ZhPs0YhVtB8
 ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=XS2_x2qneqEx9oAighLg-LRiueWcESLwIC2r7eJT-Ww,3117
 ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256=C3uwkibJ08C9sOX-54kulZYmJlIpZ-SQpfE6HNGrjbM,343502
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.5b35.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b35.dist-info/METADATA,sha256=lNh8SY9ADk3KBvTywahh147oPxI96bSdsTfj_g7wHoA,21884
-ragaai_catalyst-2.1.5b35.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-ragaai_catalyst-2.1.5b35.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b35.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b36.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b36.dist-info/METADATA,sha256=zhMfu6uQAQsus_PGtduzzY8JtrbH_ZEaNStdUJs80VQ,21884
+ragaai_catalyst-2.1.5b36.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+ragaai_catalyst-2.1.5b36.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b36.dist-info/RECORD,,