ragaai-catalyst 2.1.5b33__tar.gz → 2.1.5b35__tar.gz

{ragaai_catalyst-2.1.5b33/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.5b35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b33
+Version: 2.1.5b35
 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.5b33 → ragaai_catalyst-2.1.5b35}/pyproject.toml

@@ -8,7 +8,7 @@ description = "RAGA AI CATALYST"
 readme = "README.md"
 requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.5.b33"
+version = "2.1.5.b35"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/ragaai_catalyst.py

@@ -5,7 +5,9 @@ import time
 from typing import Dict, Optional, Union
 import re
 logger = logging.getLogger("RagaAICatalyst")
-
+logging_level = (
+    logger.setLevel(logging.DEBUG) if os.getenv("DEBUG") == "1" else logging.INFO
+)
 
 class RagaAICatalyst:
     BASE_URL = None

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/synthetic_data_generation.py

@@ -78,7 +78,9 @@ class SyntheticDataGeneration:
             "No connection adapters", 
             "Required API Keys are not set",
             "litellm.BadRequestError",
-            "litellm.AuthenticationError"]
+            "litellm.AuthenticationError", 
+            "Max retries exceeded"
+            ]
         
         for _ in range(num_batches):
             current_batch_size = min(BATCH_SIZE, n - len(all_responses))

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -92,6 +92,8 @@ class BaseTracer:
         self._upload_tasks = []
         self._is_uploading = False
         self._upload_completed_callback = None
+        
+        ensure_uploader_running()
 
     def _get_system_info(self) -> SystemInfo:
         return self.system_monitor.get_system_info()
@@ -1294,4 +1296,4 @@ class BaseTracer:
             print(f"Error during direct upload: {e}")
             import traceback
             traceback.print_exc()
-            return False
+            return False

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -16,17 +16,26 @@ import queue
 from datetime import datetime
 import atexit
 import glob
+from logging.handlers import RotatingFileHandler
 
 # Set up logging
 log_dir = os.path.join(tempfile.gettempdir(), "ragaai_logs")
 os.makedirs(log_dir, exist_ok=True)
 
+# Define maximum file size (e.g., 5 MB) and backup count
+max_file_size = 5 * 1024 * 1024  # 5 MB
+backup_count = 1  # Number of backup files to keep
+
 logging.basicConfig(
     level=logging.DEBUG,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     handlers=[
         logging.StreamHandler(),
-        logging.FileHandler(os.path.join(log_dir, "trace_uploader.log"))
+        RotatingFileHandler(
+            os.path.join(log_dir, "trace_uploader.log"),
+            maxBytes=max_file_size,
+            backupCount=backup_count
+        )
     ]
 )
 logger = logging.getLogger("trace_uploader")
@@ -170,7 +179,7 @@ class UploadTask:
     def list_pending_tasks():
         """List all pending tasks"""
         tasks = []
-        logger.info("Listing pending tasks from queue directory: {}".format(QUEUE_DIR))
+        #logger.info("Listing pending tasks from queue directory: {}".format(QUEUE_DIR))
         for filename in os.listdir(QUEUE_DIR):
             if filename.endswith(".json"):
                 try:
@@ -245,8 +254,8 @@ class TraceUploader:
 
         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)
+            #time.sleep(2)  # Simulate work
+            #task.update_status(STATUS_COMPLETED)
             return
             
         logger.info(f"Processing task {task.task_id} (attempt {task.attempts+1}/{task.max_attempts})")

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -62,6 +62,8 @@ class UploadAgenticTraces:
     def update_presigned_url(self, presigned_url, base_url):
         """Replaces the domain (and port, if applicable) of the presigned URL 
         with that of the base URL only if the base URL contains 'localhost' or an IP address."""
+        #To Do: If Proxy URL has domain name how do we handle such cases
+        
         presigned_parts = urlparse(presigned_url)
         base_parts = urlparse(base_url)
         # Check if base_url contains localhost or an IP address

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -52,6 +52,8 @@ def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None):
 
 def update_presigned_url(presigned_url, base_url):
     """Replaces the domain (and port, if applicable) of the presigned URL with that of the base URL."""
+    #To Do: If Proxy URL has domain name how do we handle such cases? Engineering Dependency.
+    
     presigned_parts = urlparse(presigned_url)
     base_parts = urlparse(base_url)
     # Check if base_url contains localhost or an IP address
@@ -92,7 +94,7 @@ def _fetch_presigned_url(project_name, dataset_name, base_url=None):
 
         if response.status_code == 200:
             presigned_url = response.json()["data"]["presignedUrls"][0]
-            presigned_url = update_presigned_url(presigned_url,RagaAICatalyst.BASE_URL)
+            presigned_url = update_presigned_url(presigned_url,url_base)
             return presigned_url
         else:
             raise Exception(f"Failed to fetch code hashes: {response.json()['message']}")

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35}/ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py

@@ -15,6 +15,10 @@ import logging
 logger = logging.getLogger(__name__)
 
 def get_model_cost():
+    """Load model costs from a JSON file. 
+    Note: This file should be updated periodically or whenever a new package is created to ensure accurate cost calculations.
+    To Do: Implement to do this automatically.
+    """
     file="model_prices_and_context_window_backup.json"
     d={}
     with resources.open_text("ragaai_catalyst.tracers.utils", file) as f:

{ragaai_catalyst-2.1.5b33 → ragaai_catalyst-2.1.5b35/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b33
+Version: 2.1.5b35
 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