ragaai-catalyst 2.1.7b0__py3-none-any.whl → 2.1.7b1__py3-none-any.whl

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

@@ -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.dist-info → ragaai_catalyst-2.1.7b1.dist-info}/RECORD

@@ -31,7 +31,7 @@ ragaai_catalyst/tracers/distributed.py,sha256=MwlBwIxCAng-OI-7Ove_rkE1mTLeuW4Jw-
 ragaai_catalyst/tracers/langchain_callback.py,sha256=CB75zzG3-DkYTELj0vI1MOHQTY0MuQJfoHIXz9Cl8S8,34568
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
 ragaai_catalyst/tracers/llamaindex_instrumentation.py,sha256=Ys_jLkvVqo12bKgXDmkp4TxJu9HkBATrFE8cIcTYxWw,14329
-ragaai_catalyst/tracers/tracer.py,sha256=16fu4zAPP8N8xzBtcbvXlWaR8g2GIC8BhOJwcDrp4vo,36993
+ragaai_catalyst/tracers/tracer.py,sha256=YZB3l55EXLog_rrAMb2jc5pvhQ63qtCc9UZcY7Yqmxo,37134
 ragaai_catalyst/tracers/upload_traces.py,sha256=PEE_JhAmOAMKyb-pl4ZoFWhIePxJm1zs93crrk94iEg,5887
 ragaai_catalyst/tracers/agentic_tracing/README.md,sha256=X4QwLb7-Jg7GQMIXj-SerZIgDETfw-7VgYlczOR8ZeQ,4508
 ragaai_catalyst/tracers/agentic_tracing/__init__.py,sha256=yf6SKvOPSpH-9LiKaoLKXwqj5sez8F_5wkOb91yp0oE,260
@@ -74,10 +74,10 @@ ragaai_catalyst/tracers/agentic_tracing/utils/trace_utils.py,sha256=W7Nw-IpugejI
 ragaai_catalyst/tracers/agentic_tracing/utils/unique_decorator.py,sha256=G027toV-Km20JjKrc-Y_PilQ8ABEKrBvvzgLTnqVg7I,5819
 ragaai_catalyst/tracers/agentic_tracing/utils/zip_list_of_unique_files.py,sha256=4TeCGsFF26249fV6dJHLTZDrRa93SG9oer4rudoF8Y4,19443
 ragaai_catalyst/tracers/exporters/__init__.py,sha256=wQbaqyeIjVZxYprHCKZ9BeiqxeXYBKjzEgP79LWNxCU,293
-ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py,sha256=l5wYX80lmMa39iOgZ7wopcH5km8t-koERL7B9wFy8Hg,5433
+ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py,sha256=eR_bbWMg_q8g9SzutrOZA24Bptr0BWpauZKVWfiCM1c,5910
 ragaai_catalyst/tracers/exporters/file_span_exporter.py,sha256=RgGteu-NVGprXKkynvyIO5yOjpbtA41R3W_NzCjnkwE,6445
 ragaai_catalyst/tracers/exporters/raga_exporter.py,sha256=6xvjWXyh8XPkHKSLLmAZUQSvwuyY17ov8pv2VdfI0qA,17875
-ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py,sha256=5TJfwD5LBLD3D9i_mrvkx2-Jp8JtWUgKTqnmAo2toCw,11364
+ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py,sha256=jfJr836ahLyo2qYrP5Hmdx0XkZt8lEhlKGHIP7PNmGs,11463
 ragaai_catalyst/tracers/instrumentators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/convert_langchain_callbacks_output.py,sha256=e0URuRWCdzpxuBLfL82FOTMjbRuDAkW8aIRi7s7Nocc,1655
@@ -88,8 +88,8 @@ ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256
 ragaai_catalyst/tracers/utils/rag_trace_json_converter.py,sha256=XSanJ3ChRn_Nqj5mlJhy-jFUCAhkw6lBuhhzUx6Mo9k,11239
 ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=E0_QfciQMMpCtQYrNB4l8HJhlaFalr5bkMqkVRgQahY,14073
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.7b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.7b0.dist-info/METADATA,sha256=Ccfgxo8Ehm7CeGQRZrFsDYPb_RsTymn6B0oG_y9JuZE,22139
-ragaai_catalyst-2.1.7b0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ragaai_catalyst-2.1.7b0.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.7b0.dist-info/RECORD,,
+ragaai_catalyst-2.1.7b1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.7b1.dist-info/METADATA,sha256=t0urAW0WO0OddkYhXjrQt925JQcbdqPWKnv0urR-mwc,22139
+ragaai_catalyst-2.1.7b1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+ragaai_catalyst-2.1.7b1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.7b1.dist-info/RECORD,,