ragaai-catalyst 2.1.5b33__py3-none-any.whl → 2.1.5b34__py3-none-any.whl

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/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/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/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/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.dist-info → ragaai_catalyst-2.1.5b34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b33
+Version: 2.1.5b34
 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.dist-info → ragaai_catalyst-2.1.5b34.dist-info}/RECORD

@@ -8,7 +8,7 @@ ragaai_catalyst/guardrails_manager.py,sha256=DILMOAASK57FH9BLq_8yC1AQzRJ8McMFLwC
 ragaai_catalyst/internal_api_completion.py,sha256=DdICI5yfEudiOAIC8L4oxH0Qz7kX-BZCdo9IWsi2gNo,2965
 ragaai_catalyst/prompt_manager.py,sha256=W8ypramzOprrJ7-22d5vkBXIuIQ8v9XAzKDGxKsTK28,16550
 ragaai_catalyst/proxy_call.py,sha256=CHxldeceZUaLU-to_hs_Kf1z_b2vHMssLS_cOBedu78,5499
-ragaai_catalyst/ragaai_catalyst.py,sha256=l6aoBEXMAxPzdsd0X04SeHCLuD5tBR-9qZTq3hiD5ls,19479
+ragaai_catalyst/ragaai_catalyst.py,sha256=1FaeK_VZpJLQ1ZqEWpMyI8J8M2MI0abLLLDFWY9W-4A,19580
 ragaai_catalyst/redteaming_old.py,sha256=W2d89Ok8W-C8g7TBM3fDIFLof3q9FuYSr0jcryH2XQo,7097
 ragaai_catalyst/synthetic_data_generation.py,sha256=rJPWj6luKMa6CTs1cEAmtnZhUMEQsr67O_C4jG47dMQ,37547
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
@@ -54,9 +54,9 @@ 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=JZDnZcDCnFu3WbOYj6b2dSw_1AHXFo--iJxObhJF7yA,26157
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=FWrijnaKV7o37gcLVe84OQT4Y6f4Jy8yxqqihJBsyws,8459
-ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=dUBxtadRUX5QFBzzHStjFI61Icedk-QAj9xgeCpjSqU,6378
+ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py,sha256=pklyXsz1_tKsU-Kon0ZxsV11BGRLV66bCh8UG9tfrhc,26454
+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
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_trace_metric.py,sha256=dmgcFZtCUpTGV2GayKQ9znaFRDLlXY8NW6RiyNwerOk,4278
 ragaai_catalyst/tracers/agentic_tracing/utils/__init__.py,sha256=XdB3X_ufe4RVvGorxSqAiB9dYv4UD7Hvvuw3bsDUppY,60
@@ -65,7 +65,7 @@ ragaai_catalyst/tracers/agentic_tracing/utils/create_dataset_schema.py,sha256=xH
 ragaai_catalyst/tracers/agentic_tracing/utils/file_name_tracker.py,sha256=YG601l1a29ov9VPu9Vl4RXxgL7l16k54_WWnoTNoG58,2064
 ragaai_catalyst/tracers/agentic_tracing/utils/generic.py,sha256=WwXT01xmp8MSr7KinuDCSK9a1ifpLcT7ajFkvYviG_A,1190
 ragaai_catalyst/tracers/agentic_tracing/utils/get_user_trace_metrics.py,sha256=vPZ4dn4EHFW0kqd1GyRpsYXbfrRrd0DXCmh-pzsDBNE,1109
-ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py,sha256=-XhdAkMULNfR7kZUE7-YK_YYCxFU5NEEiVZieRNX8W8,21324
+ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py,sha256=Z01wEEGVlWx-45rHPCKDmeusfpjdS2-LXKYNhsAgjJE,21548
 ragaai_catalyst/tracers/agentic_tracing/utils/model_costs.json,sha256=2tzGw_cKCTPcfjEm7iGvFE6pTw7gMTPzeBov_MTaXNY,321336
 ragaai_catalyst/tracers/agentic_tracing/utils/span_attributes.py,sha256=qmODERcFZhc8MX24boFCXkkh6sJ-vZngRHPvxhyWFeE,4347
 ragaai_catalyst/tracers/agentic_tracing/utils/supported_llm_provider.toml,sha256=LvFDivDIE96Zasp-fgDEqUJ5GEQZUawQucR3aOcSUTY,926
@@ -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.5b33.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b33.dist-info/METADATA,sha256=KHP_0o1qGuahcTZ4ZIY3idwT9Ef4hh-2CpPJ9QMrc2M,21884
-ragaai_catalyst-2.1.5b33.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
-ragaai_catalyst-2.1.5b33.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b33.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b34.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b34.dist-info/METADATA,sha256=Nuu6aspLxz8lsGv1UmUlDvEAEwK4bYvWh9PHKQcoRsg,21884
+ragaai_catalyst-2.1.5b34.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+ragaai_catalyst-2.1.5b34.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b34.dist-info/RECORD,,

{ragaai_catalyst-2.1.5b33.dist-info → ragaai_catalyst-2.1.5b34.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.1)
+Generator: setuptools (75.8.2)
 Root-Is-Purelib: true
 Tag: py3-none-any