ragaai-catalyst 2.1.7.5b4__py3-none-any.whl → 2.1.7.5b6__py3-none-any.whl

ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -100,21 +100,27 @@ class RAGATraceExporter(SpanExporter):
             if self.post_processor!=None and self.tracer_type != "langchain":
                 ragaai_trace_details['trace_file_path'] = self.post_processor(ragaai_trace_details['trace_file_path'])
             if self.tracer_type == "langchain":
-                # Check if we're already in an event loop
-                try:
-                    loop = asyncio.get_event_loop()
-                    if loop.is_running():
-                        # We're in a running event loop (like in Colab/Jupyter)
-                        # Create a future and run the coroutine
-                        future = asyncio.ensure_future(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id, self.post_processor))
-                        # We don't wait for it to complete as this would block the event loop
-                        logger.info(f"Scheduled async upload for trace {trace_id} in existing event loop")
-                    else:
-                        # No running event loop, use asyncio.run()
-                        asyncio.run(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id, self.post_processor))
-                except RuntimeError:
-                    # No event loop exists, create one
-                    asyncio.run(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id, self.post_processor))
+                import threading
+                
+                def upload_in_thread():
+                    try:
+                        thread_loop = asyncio.new_event_loop()
+                        asyncio.set_event_loop(thread_loop)
+                        try:
+                            thread_loop.run_until_complete(
+                                self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id, self.post_processor)
+                            )
+                            logger.info(f"Successfully completed upload for trace {trace_id} in dedicated thread")
+                        finally:
+                            thread_loop.close()
+                    except Exception as e:
+                        logger.error(f"Error in upload thread for trace {trace_id}: {str(e)}")
+                
+                upload_thread = threading.Thread(target=upload_in_thread)
+                upload_thread.daemon = True
+                upload_thread.start()
+                logger.info(f"Started dedicated upload thread for trace {trace_id}")
+                
             else:
                 self.upload_trace(ragaai_trace_details, trace_id)
         except Exception as e: 

ragaai_catalyst/tracers/tracer.py

@@ -535,11 +535,18 @@ class Tracer(AgenticTracing):
             'max_upload_workers': self.max_upload_workers
         }
 
+        # Save the model_custom_cost before reinitialization
+        saved_model_custom_cost = self.model_custom_cost.copy()
+
         # Reinitialize self with new external_id and stored parameters
         self.__init__(
             external_id=external_id,
             **current_params
         )
+        
+        # Restore the model_custom_cost after reinitialization
+        self.model_custom_cost = saved_model_custom_cost
+        self.dynamic_exporter.custom_model_cost = self.model_custom_cost
 
     
 

ragaai_catalyst/tracers/upload_traces.py

@@ -37,7 +37,8 @@ class UploadTraces:
             "model_name": {"columnType": "metadata"},
             "total_cost": {"columnType": "metadata", "dataType": "numerical"},
             "total_latency": {"columnType": "metadata", "dataType": "numerical"},
-            "error": {"columnType": "metadata"}
+            "error": {"columnType": "metadata"},
+            "externalId": {"columnType": "externalId"}
         }
 
         if additional_metadata_keys:

{ragaai_catalyst-2.1.7.5b4.dist-info → ragaai_catalyst-2.1.7.5b6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7.5b4
+Version: 2.1.7.5b6
 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.2,>=3.10

{ragaai_catalyst-2.1.7.5b4.dist-info → ragaai_catalyst-2.1.7.5b6.dist-info}/RECORD

@@ -31,8 +31,8 @@ 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=W5dS2h4IQrqbE_rgFY-1CrVCSs7WSFrDCny9_RASyjY,43593
-ragaai_catalyst/tracers/upload_traces.py,sha256=yfg1vHP5quz2Vb8xkas_vzXwIrTVqRwrmFNCTrWvFdg,6938
+ragaai_catalyst/tracers/tracer.py,sha256=w9Poa2dsN3WNMPEH_v4M7SqFF_MQp_iSam4A-qcoAXM,43921
+ragaai_catalyst/tracers/upload_traces.py,sha256=w1clGGfdOMpStUJX40NAlxe6dcFdN4pwcezyII0bGYA,6994
 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
 ragaai_catalyst/tracers/agentic_tracing/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -77,7 +77,7 @@ ragaai_catalyst/tracers/exporters/__init__.py,sha256=wQbaqyeIjVZxYprHCKZ9BeiqxeX
 ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py,sha256=v2V-y1l9t9gQQbqoi2jlEhm1xRQjR5yTyr6rxSriXCE,6303
 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=C88pZ_n4mNv1Fc3qOOOm5JQdoTvQzHcjS7mSImslQ0I,13313
+ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py,sha256=K5dy54bgUjYMDNT48qBuwx8ta245yQPIczCkFviAyd4,13290
 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=SehrD7q8ytAiUYoWr406b4mWs3Lk0Rcy6Ekkihh22TI,1703
@@ -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=adCKk7Nj8307XYYg2sB-QT-66OShOs2iTGwNVwqbHig,19373
 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.7.5b4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.7.5b4.dist-info/METADATA,sha256=Z0X9VzL8SIUjeobOKLCf70qmiBwp79VQwyTTbCt6MxQ,17607
-ragaai_catalyst-2.1.7.5b4.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-ragaai_catalyst-2.1.7.5b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.7.5b4.dist-info/RECORD,,
+ragaai_catalyst-2.1.7.5b6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.7.5b6.dist-info/METADATA,sha256=xpZy8rtoEBjykChpgpFoIq_CfcqftU4yxR9ExxiROK4,17607
+ragaai_catalyst-2.1.7.5b6.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+ragaai_catalyst-2.1.7.5b6.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.7.5b6.dist-info/RECORD,,

{ragaai_catalyst-2.1.7.5b4.dist-info → ragaai_catalyst-2.1.7.5b6.dist-info}/WHEEL

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