ragaai-catalyst 2.1.6.1__tar.gz → 2.1.6.2__tar.gz

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.6.1
+Version: 2.1.6.2
 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.10

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/pyproject.toml

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

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -92,7 +92,6 @@ class BaseTracer:
         self._upload_tasks = []
         self._is_uploading = False
         self._upload_completed_callback = None
-        self.timeout = self.user_details.get("timeout", 120)
         
         ensure_uploader_running()
 
@@ -315,8 +314,7 @@ class BaseTracer:
                 project_id=self.project_id,
                 dataset_name=self.dataset_name,
                 user_details=self.user_details,
-                base_url=self.base_url,
-                timeout=self.timeout
+                base_url=self.base_url
             )
             
             # For backward compatibility

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -48,7 +48,7 @@ from ragaai_catalyst.tracers.upload_traces import UploadTraces
 class AgenticTracing(
     BaseTracer, LLMTracerMixin, ToolTracerMixin, AgentTracerMixin, CustomTracerMixin
 ):
-    def __init__(self, user_detail, auto_instrumentation=None, timeout=120):
+    def __init__(self, user_detail, auto_instrumentation=None):
         # Initialize all parent classes
         self.user_interaction_tracer = UserInteractionTracer()
         LLMTracerMixin.__init__(self)
@@ -60,7 +60,7 @@ class AgenticTracing(
         self.project_id = user_detail["project_id"]
         self.trace_user_detail = user_detail["trace_user_detail"]
         self.base_url = f"{RagaAICatalyst.BASE_URL}"
-        self.timeout = timeout
+        self.timeout = 10
         
         # Add warning flag
         self._warning_shown = False

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -77,7 +77,7 @@ def get_executor():
 
 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, timeout=120) -> Dict[str, Any]:
+                  user_details: Dict[str, Any], base_url: str) -> Dict[str, Any]:
     """
     Process a single upload task
     
@@ -147,8 +147,7 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     json_file_path=filepath,
                     dataset_name=dataset_name,
                     project_name=project_name,
-                    base_url=base_url,
-                    timeout=timeout
+                    base_url=base_url
                 )
                 logger.info(f"Trace metrics uploaded: {response}")
             except Exception as e:
@@ -168,7 +167,6 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     dataset_name=dataset_name,
                     user_detail=user_details,
                     base_url=base_url,   
-                    timeout=timeout
                 )
                 upload_traces.upload_agentic_traces()
                 logger.info("Agentic traces uploaded successfully")
@@ -187,8 +185,7 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                     zip_path=zip_path,
                     project_name=project_name,
                     dataset_name=dataset_name,
-                    base_url=base_url,
-                    timeout=timeout
+                    base_url=base_url
                 )
                 logger.info(f"Code hash uploaded: {response}")
             except Exception as e:
@@ -218,7 +215,7 @@ def save_task_status(task_status: Dict[str, Any]):
     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, timeout=120):
+def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, dataset_name, user_details, base_url):
     """
     Submit a new upload task using futures.
     
@@ -262,8 +259,7 @@ def submit_upload_task(filepath, hash_id, zip_path, project_name, project_id, da
         project_id=project_id,
         dataset_name=dataset_name,
         user_details=user_details,
-        base_url=base_url,
-        timeout=timeout
+        base_url=base_url
     )
     
     # Store the future for later status checks

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py

@@ -24,7 +24,7 @@ class UploadAgenticTraces:
         self.dataset_name = dataset_name
         self.user_detail = user_detail
         self.base_url = base_url
-        self.timeout = 120
+        self.timeout = 30
 
 
     def _get_presigned_url(self):

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py

@@ -9,19 +9,19 @@ logger = logging.getLogger(__name__)
 from urllib.parse import urlparse, urlunparse
 import re
 
-def upload_code(hash_id, zip_path, project_name, dataset_name, base_url=None, timeout=120):
-    code_hashes_list = _fetch_dataset_code_hashes(project_name, dataset_name, base_url, timeout=timeout)
+def upload_code(hash_id, zip_path, project_name, dataset_name, base_url=None):
+    code_hashes_list = _fetch_dataset_code_hashes(project_name, dataset_name, base_url)
 
     if hash_id not in code_hashes_list:
-        presigned_url = _fetch_presigned_url(project_name, dataset_name, base_url, timeout=timeout)
-        _put_zip_presigned_url(project_name, presigned_url, zip_path, timeout=timeout)
+        presigned_url = _fetch_presigned_url(project_name, dataset_name, base_url)
+        _put_zip_presigned_url(project_name, presigned_url, zip_path)
 
-        response = _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url, timeout=timeout)
+        response = _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url)
         return response
     else:
         return "Code already exists"
 
-def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None, timeout=120):
+def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None):
     payload = {}
     headers = {
         "Authorization": f"Bearer {os.getenv('RAGAAI_CATALYST_TOKEN')}",
@@ -36,7 +36,7 @@ def _fetch_dataset_code_hashes(project_name, dataset_name, base_url=None, timeou
                                     endpoint, 
                                     headers=headers, 
                                     data=payload,
-                                    timeout=timeout)
+                                    timeout=99999)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [GET] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
@@ -66,7 +66,7 @@ def update_presigned_url(presigned_url, base_url):
     return presigned_url
 
 
-def _fetch_presigned_url(project_name, dataset_name, base_url=None, timeout=120):
+def _fetch_presigned_url(project_name, dataset_name, base_url=None):
     payload = json.dumps({
             "datasetName": dataset_name,
             "numFiles": 1,
@@ -87,7 +87,7 @@ def _fetch_presigned_url(project_name, dataset_name, base_url=None, timeout=120)
                                     endpoint, 
                                     headers=headers, 
                                     data=payload,
-                                    timeout=timeout)
+                                    timeout=99999)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [GET] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
@@ -102,7 +102,7 @@ def _fetch_presigned_url(project_name, dataset_name, base_url=None, timeout=120)
         logger.error(f"Failed to list datasets: {e}")
         raise
 
-def _put_zip_presigned_url(project_name, presignedUrl, filename, timeout=120):
+def _put_zip_presigned_url(project_name, presignedUrl, filename):
     headers = {
             "X-Project-Name": project_name,
             "Content-Type": "application/zip",
@@ -119,14 +119,14 @@ def _put_zip_presigned_url(project_name, presignedUrl, filename, timeout=120):
                                 presignedUrl, 
                                 headers=headers, 
                                 data=payload,
-                                timeout=timeout)
+                                timeout=99999)
     elapsed_ms = (time.time() - start_time) * 1000
     logger.debug(
         f"API Call: [PUT] {presignedUrl} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")
     if response.status_code != 200 or response.status_code != 201:
         return response, response.status_code
 
-def _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url=None, timeout=120):
+def _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url=None):
     payload = json.dumps({
         "datasetName": dataset_name,
         "codeHash": hash_id,
@@ -147,7 +147,7 @@ def _insert_code(dataset_name, hash_id, presigned_url, project_name, base_url=No
                                     endpoint, 
                                     headers=headers, 
                                     data=payload,
-                                    timeout=timeout)
+                                    timeout=99999)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/agentic_tracing/upload/upload_trace_metric.py

@@ -15,7 +15,7 @@ logging_level = (
 )
 
 
-def upload_trace_metric(json_file_path, dataset_name, project_name, base_url=None, timeout=120):
+def upload_trace_metric(json_file_path, dataset_name, project_name, base_url=None):
     try:
         with open(json_file_path, "r") as f:
             traces = json.load(f)
@@ -51,7 +51,7 @@ def upload_trace_metric(json_file_path, dataset_name, project_name, base_url=Non
                                     endpoint,
                                     headers=headers,
                                     data=payload,
-                                    timeout=timeout)
+                                    timeout=10)
         elapsed_ms = (time.time() - start_time) * 1000
         logger.debug(
             f"API Call: [POST] {endpoint} | Status: {response.status_code} | Time: {elapsed_ms:.2f}ms")

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py

@@ -14,7 +14,7 @@ class DynamicTraceExporter(SpanExporter):
     certain properties to be updated dynamically during execution.
     """
     
-    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120):
+    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost):
         """
         Initialize the DynamicTraceExporter.
         
@@ -33,8 +33,7 @@ class DynamicTraceExporter(SpanExporter):
             dataset_name=dataset_name,
             user_details=user_details,
             base_url=base_url,
-            custom_model_cost=custom_model_cost,
-            timeout=timeout
+            custom_model_cost=custom_model_cost
         )
         
         # Store the initial values

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -19,7 +19,7 @@ logging_level = (
 
 
 class RAGATraceExporter(SpanExporter):
-    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120):
+    def __init__(self, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost):
         self.trace_spans = dict()
         self.tmp_dir = tempfile.gettempdir()
         self.files_to_zip = files_to_zip
@@ -30,7 +30,6 @@ class RAGATraceExporter(SpanExporter):
         self.base_url = base_url
         self.custom_model_cost = custom_model_cost
         self.system_monitor = SystemMonitor(dataset_name)
-        self.timeout = timeout
 
     def export(self, spans):
         for span in spans:
@@ -123,8 +122,7 @@ class RAGATraceExporter(SpanExporter):
                 project_id=self.project_id,
                 dataset_name=self.dataset_name,
                 user_details=self.user_details,
-                base_url=self.base_url,
-                timeout=self.timeout
+                base_url=self.base_url
             )
 
         logger.info(f"Submitted upload task with ID: {self.upload_task_id}")

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst/tracers/tracer.py

@@ -37,6 +37,7 @@ logging_level = (
 
 class Tracer(AgenticTracing):
     NUM_PROJECTS = 99999
+    TIMEOUT = 10
     def __init__(
         self,
         project_name,
@@ -46,7 +47,7 @@ class Tracer(AgenticTracing):
         pipeline=None,
         metadata=None,
         description=None,
-        timeout=120,  # Default timeout of 120 seconds
+        upload_timeout=30,  # Default timeout of 30 seconds
         update_llm_cost=True,  # Parameter to control model cost updates
         auto_instrumentation={ # to control automatic instrumentation of different components
             'llm':True,
@@ -71,7 +72,7 @@ class Tracer(AgenticTracing):
             pipeline (dict, optional): The pipeline configuration. Defaults to None.
             metadata (dict, optional): The metadata. Defaults to None.
             description (str, optional): The description. Defaults to None.
-            timeout (int, optional): The upload timeout in seconds. Defaults to 30.
+            upload_timeout (int, optional): The upload timeout in seconds. Defaults to 30.
             update_llm_cost (bool, optional): Whether to update model costs from GitHub. Defaults to True.
         """
 
@@ -132,8 +133,9 @@ class Tracer(AgenticTracing):
         # self.metadata["total_tokens"] = 0
         self.pipeline = pipeline
         self.description = description
-        self.timeout = timeout
+        self.upload_timeout = upload_timeout
         self.base_url = f"{RagaAICatalyst.BASE_URL}"
+        self.timeout = 30
         self.num_projects = 99999
         self.start_time = datetime.datetime.now().astimezone().isoformat()
         self.model_cost_dict = model_cost
@@ -397,7 +399,7 @@ class Tracer(AgenticTracing):
                 'pipeline': self.pipeline,
                 'metadata': self.metadata,
                 'description': self.description,
-                'timeout': self.timeout
+                'upload_timeout': self.upload_timeout
             }
             
             # Reinitialize self with new dataset_name and stored parameters
@@ -613,11 +615,11 @@ class Tracer(AgenticTracing):
 
         This function uploads the traces generated by the RagaAICatalyst client to the RagaAICatalyst server. It uses the `aiohttp` library to make an asynchronous HTTP request to the server. The function first checks if the `RAGAAI_CATALYST_TOKEN` environment variable is set. If not, it raises a `ValueError` with the message "RAGAAI_CATALYST_TOKEN not found. Cannot upload traces.".
 
-        The function then uses the `asyncio.wait_for` function to wait for the `check_and_upload_files` method of the `raga_client` object to complete. The `check_and_upload_files` method is called with the `session` object and a list of file paths to be uploaded. The `timeout` parameter is set to the value of the `timeout` attribute of the `Tracer` object.
+        The function then uses the `asyncio.wait_for` function to wait for the `check_and_upload_files` method of the `raga_client` object to complete. The `check_and_upload_files` method is called with the `session` object and a list of file paths to be uploaded. The `timeout` parameter is set to the value of the `upload_timeout` attribute of the `Tracer` object.
 
         If the upload is successful, the function returns the string "Files uploaded successfully" if the `upload_stat` variable is truthy, otherwise it returns the string "No files to upload".
 
-        If the upload times out, the function returns a string with the message "Upload timed out after {self.timeout} seconds".
+        If the upload times out, the function returns a string with the message "Upload timed out after {self.upload_timeout} seconds".
 
         If any other exception occurs during the upload, the function returns a string with the message "Upload failed: {str(e)}", where `{str(e)}` is the string representation of the exception.
 
@@ -639,7 +641,7 @@ class Tracer(AgenticTracing):
                         session=session,
                         file_paths=[self.filespanx.sync_file],
                     ),
-                    timeout=self.timeout,
+                    timeout=self.upload_timeout,
                 )
                 return (
                     "Files uploaded successfully"
@@ -647,7 +649,7 @@ class Tracer(AgenticTracing):
                     else "No files to upload"
                 )
             except asyncio.TimeoutError:
-                return f"Upload timed out after {self.timeout} seconds"
+                return f"Upload timed out after {self.upload_timeout} seconds"
             except Exception as e:
                 return f"Upload failed: {str(e)}"
 
@@ -747,8 +749,7 @@ class Tracer(AgenticTracing):
             dataset_name=self.dataset_name,
             user_details=self.user_details,
             base_url=self.base_url,
-            custom_model_cost=self.model_custom_cost,
-            timeout=self.timeout
+            custom_model_cost=self.model_custom_cost
         )
         
         # Set up tracer provider

{ragaai_catalyst-2.1.6.1 → ragaai_catalyst-2.1.6.2}/ragaai_catalyst.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.6.1
+Version: 2.1.6.2
 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.10