ragaai-catalyst 2.1.7.4b2__tar.gz → 2.1.7.4b3__tar.gz

{ragaai_catalyst-2.1.7.4b2/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.7.4b3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7.4b2
+Version: 2.1.7.4b3
 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.4b2 → ragaai_catalyst-2.1.7.4b3}/pyproject.toml

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

{ragaai_catalyst-2.1.7.4b2 → ragaai_catalyst-2.1.7.4b3}/ragaai_catalyst/tracers/agentic_tracing/upload/trace_uploader.py

@@ -157,7 +157,7 @@ def process_upload(task_id: str, filepath: str, hash_id: str, zip_path: str,
                 )
                 logger.info(f"Trace metrics uploaded: {response}")
             except Exception as e:
-                logger.error(f"Error uploading trace metrics: {e}")
+                logger.error(f"Error uploading trace trace uploader metrics: {e}")
                 # Continue with other uploads
         else:
             logger.warning(f"Trace file {filepath} not found, skipping metrics upload")
@@ -364,4 +364,4 @@ if __name__ == "__main__":
     if args.daemon:
         logger.info("Daemon mode not needed in futures implementation")
     else:
-        logger.info("Interactive mode not needed in futures implementation")
+        logger.info("Interactive mode not needed in futures implementation")

{ragaai_catalyst-2.1.7.4b2 → ragaai_catalyst-2.1.7.4b3}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -89,7 +89,7 @@ class RAGATraceExporter(SpanExporter):
         except Exception as e:
             print(f"Error converting trace {trace_id}: {e}")
             return  # Exit early if conversion fails
-            
+
         # Check if trace details are None (conversion failed)
         if ragaai_trace_details is None:
             logger.error(f"Cannot upload trace {trace_id}: conversion failed and returned None")
@@ -97,7 +97,7 @@ class RAGATraceExporter(SpanExporter):
             
         # Upload the trace if upload_trace function is provided
         try:
-            if self.post_processor!=None:
+            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
@@ -106,18 +106,18 @@ class RAGATraceExporter(SpanExporter):
                     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))
+                        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))
+                        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))
+                    asyncio.run(self.upload_rag_trace(ragaai_trace_details, additional_metadata, trace_id, self.post_processor))
             else:
                 self.upload_trace(ragaai_trace_details, trace_id)
-        except Exception as e:
+        except Exception as e: 
             print(f"Error uploading trace {trace_id}: {e}")
 
     def prepare_trace(self, spans, trace_id):
@@ -206,13 +206,16 @@ class RAGATraceExporter(SpanExporter):
 
         logger.info(f"Submitted upload task with ID: {self.upload_task_id}")
     
-    async def upload_rag_trace(self, ragaai_trace, additional_metadata, trace_id):
+    async def upload_rag_trace(self, ragaai_trace, additional_metadata, trace_id, post_processor=None):
         try:
             ragaai_trace[0]['external_id'] = self.external_id
             trace_file_path = os.path.join(self.tmp_dir, f"{trace_id}.json")
             with open(trace_file_path, 'w') as f:
                 json.dump(ragaai_trace, f, indent=2)
             logger.info(f"Trace file saved at {trace_file_path}")
+            if self.post_processor!=None:
+                trace_file_path = self.post_processor(trace_file_path)
+                logger.info(f"After post processing Trace file saved at {trace_file_path}")
 
             # Create a ThreadPoolExecutor with max_workers=30
             with concurrent.futures.ThreadPoolExecutor(max_workers=self.max_upload_workers) as executor:
@@ -227,7 +230,7 @@ class RAGATraceExporter(SpanExporter):
                         base_url=self.base_url
                     ).upload_traces,
                     additional_metadata_keys=additional_metadata
-                )
+                ) 
                 
                 # Implement retry logic - attempt upload up to 3 times
                 max_retries = 3

{ragaai_catalyst-2.1.7.4b2 → ragaai_catalyst-2.1.7.4b3}/ragaai_catalyst/tracers/tracer.py

@@ -427,12 +427,20 @@ class Tracer(AgenticTracing):
             with open(original_path, 'r') as f:
                 data = json.load(f)
             
-            # Apply masking only to data['data']
-            data['data'] = recursive_mask_values(data['data'])
-            
-            # Create new filename with 'processed_' prefix in /var/tmp/
+            # Apply masking only to data['data'] or in case of langchain rag apply on 'traces' field of each element
+            if 'data' in data:
+                data['data'] = recursive_mask_values(data['data'])
+            elif isinstance(data,list):
+                masked_traces = []
+                for item in data:
+                    if isinstance(item, dict) and 'traces' in item:
+                        item['traces'] = recursive_mask_values(item['traces'])
+                        masked_traces.append(item)
+                data = masked_traces
+            # Create new filename with 'processed_' prefix 
             new_filename = f"processed_{original_path.name}"
-            final_trace_json_path = Path("/var/tmp") / new_filename
+            dir_name, original_filename = os.path.split(original_trace_json_path)
+            final_trace_json_path = Path(dir_name) / new_filename
             
             # Write modified data to the new file
             with open(final_trace_json_path, 'w') as f:

{ragaai_catalyst-2.1.7.4b2 → ragaai_catalyst-2.1.7.4b3/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragaai_catalyst
-Version: 2.1.7.4b2
+Version: 2.1.7.4b3
 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