ragaai-catalyst 2.1.7b0__tar.gz → 2.1.7b1__tar.gz

{ragaai_catalyst-2.1.7b0/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.7b1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7b0
+Version: 2.1.7b1
 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.7b0 → ragaai_catalyst-2.1.7b1}/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.7.beta.0"
+version = "2.1.7.beta.1"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.7b0 → ragaai_catalyst-2.1.7b1}/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, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None):
+    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30):
         """
         Initialize the DynamicTraceExporter.
         
@@ -27,6 +27,7 @@ class DynamicTraceExporter(SpanExporter):
             user_details: User details
             base_url: Base URL for API
             post_processor: Post processing function before uploading trace
+            max_upload_workers: Maximum number of upload workers
         """
         self._exporter = RAGATraceExporter(
             tracer_type=tracer_type,
@@ -38,7 +39,8 @@ class DynamicTraceExporter(SpanExporter):
             base_url=base_url,
             custom_model_cost=custom_model_cost,
             timeout=timeout,
-            post_processor= post_processor
+            post_processor= post_processor,
+            max_upload_workers = max_upload_workers
         )
         
         # Store the initial values
@@ -50,6 +52,7 @@ class DynamicTraceExporter(SpanExporter):
         self._base_url = base_url
         self._custom_model_cost = custom_model_cost
         self._post_processor = post_processor
+        self._max_upload_workers = max_upload_workers
 
     
     def export(self, spans):
@@ -107,6 +110,7 @@ class DynamicTraceExporter(SpanExporter):
         self._exporter.base_url = self._base_url
         self._exporter.custom_model_cost = self._custom_model_cost
         self._exporter.post_processor = self._post_processor
+        self._exporter.max_upload_workers = self._max_upload_workers
     
     # Getter and setter methods for dynamic properties
     
@@ -165,3 +169,11 @@ class DynamicTraceExporter(SpanExporter):
     @custom_model_cost.setter
     def custom_model_cost(self, value):
         self._custom_model_cost = value
+    
+    @property
+    def max_upload_workers(self):
+        return self._max_upload_workers
+    
+    @max_upload_workers.setter
+    def max_upload_workers(self, value):
+        self._max_upload_workers = value

{ragaai_catalyst-2.1.7b0 → ragaai_catalyst-2.1.7b1}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -26,7 +26,7 @@ logging_level = (
 
 
 class RAGATraceExporter(SpanExporter):
-    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None):
+    def __init__(self, tracer_type, files_to_zip, project_name, project_id, dataset_name, user_details, base_url, custom_model_cost, timeout=120, post_processor = None, max_upload_workers = 30):
         self.trace_spans = dict()
         self.tmp_dir = tempfile.gettempdir()
         self.tracer_type = tracer_type
@@ -40,6 +40,7 @@ class RAGATraceExporter(SpanExporter):
         self.system_monitor = SystemMonitor(dataset_name)
         self.timeout = timeout
         self.post_processor = post_processor
+        self.max_upload_workers = max_upload_workers
 
     def export(self, spans):
         for span in spans:
@@ -189,7 +190,7 @@ class RAGATraceExporter(SpanExporter):
                 json.dump(ragaai_trace, f, indent=2)
             
             # Create a ThreadPoolExecutor with max_workers=30
-            with concurrent.futures.ThreadPoolExecutor(max_workers=30) as executor:
+            with concurrent.futures.ThreadPoolExecutor(max_workers=self.max_upload_workers) as executor:
                 # Create a partial function with all the necessary arguments
                 upload_func = partial(
                     UploadTraces(

{ragaai_catalyst-2.1.7b0 → ragaai_catalyst-2.1.7b1}/ragaai_catalyst/tracers/tracer.py

@@ -59,6 +59,7 @@ class Tracer(AgenticTracing):
         },
         interval_time=2,
         # auto_instrumentation=True/False  # to control automatic instrumentation of everything
+        max_upload_workers=30
 
     ):
         """
@@ -141,6 +142,7 @@ class Tracer(AgenticTracing):
         self.user_context = ""  # Initialize user_context to store context from add_context
         self.file_tracker = TrackName()
         self.post_processor = None
+        self.max_upload_workers = max_upload_workers
         
         try:
             response = requests.get(
@@ -763,7 +765,8 @@ class Tracer(AgenticTracing):
             base_url=self.base_url,
             custom_model_cost=self.model_custom_cost,
             timeout = self.timeout,
-            post_processor= self.post_processor
+            post_processor= self.post_processor,
+            max_upload_workers = self.max_upload_workers
         )
         
         # Set up tracer provider

{ragaai_catalyst-2.1.7b0 → ragaai_catalyst-2.1.7b1/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7b0
+Version: 2.1.7b1
 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