ragaai-catalyst 2.1.4b4__py3-none-any.whl → 2.1.4b5__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py

@@ -66,8 +66,8 @@ class AgentTracerMixin:
         if metrics:
             if isinstance(metrics, dict):
                 metrics = [metrics]
-            for metric in metrics:
-                try:
+            try:
+                for metric in metrics:
                     self.span(name).add_metrics(
                         name=metric["name"],
                         score=metric["score"],
@@ -77,8 +77,10 @@ class AgentTracerMixin:
                         metadata=metric.get("metadata", {}),
                         config=metric.get("config", {}),
                     )
-                except KeyError as e:
-                    logger.error(f"Error adding metric: {e}")
+            except ValueError as e:
+                logger.error(f"Validation Error: {e}")
+            except Exception as e:
+                logger.error(f"Error adding metric: {e}")
         if feedback:
             self.span(name).add_feedback(feedback)
 
@@ -525,7 +527,14 @@ class AgentTracerMixin:
         # metrics
         metrics = []
         if name in self.span_attributes_dict:
-            metrics = self.span_attributes_dict[name].metrics or []
+            raw_metrics = self.span_attributes_dict[name].metrics or []
+            # Check for duplicates
+            for metric in raw_metrics:
+                metric_name = metric["name"]
+                if self.visited_metrics.count(metric_name) >= 1:
+                    raise ValueError(f"Duplicate metric '{metric_name}' found in trace. Metric names must be unique across all spans.")
+                self.visited_metrics.append(metric_name)
+                metrics.append(metric)
 
         component = {
             "id": kwargs["component_id"],

ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -78,6 +78,7 @@ class BaseTracer:
         self.dataset_name = self.user_details["dataset_name"]  # Access the dataset_name
         self.project_id = self.user_details["project_id"]  # Access the project_id
         self.trace_name = self.user_details["trace_name"]  # Access the trace_name
+        self.visited_metrics = []
 
         # Initialize trace data
         self.trace_id = None
@@ -802,3 +803,4 @@ class BaseTracer:
         if span_name not in self.span_attributes_dict:
             self.span_attributes_dict[span_name] = SpanAttributes(span_name)
         return self.span_attributes_dict[span_name]
+    

ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py

@@ -363,7 +363,14 @@ class LLMTracerMixin:
         # metrics
         metrics = []
         if name in self.span_attributes_dict:
-            metrics = self.span_attributes_dict[name].metrics or []
+            raw_metrics = self.span_attributes_dict[name].metrics or []
+            # Check for duplicates
+            for metric in raw_metrics:
+                metric_name = metric["name"]
+                if self.visited_metrics.count(metric_name) >= 1:
+                    raise ValueError(f"Duplicate metric '{metric_name}' found in trace. Metric names must be unique across all spans.")
+                self.visited_metrics.append(metric_name)
+                metrics.append(metric)
 
         component = {
             "id": component_id,
@@ -621,8 +628,8 @@ class LLMTracerMixin:
         if metrics:
             if isinstance(metrics, dict):
                 metrics = [metrics]
-            for metric in metrics:
-                try:
+            try:
+                for metric in metrics:
                     self.span(name).add_metrics(
                         name=metric["name"],
                         score=metric["score"],
@@ -632,8 +639,11 @@ class LLMTracerMixin:
                         metadata=metric.get("metadata", {}),
                         config=metric.get("config", {}),
                     )
-                except KeyError as e:
-                    logger.error(f"Error adding metric: {e}")
+            except ValueError as e:
+                logger.error(f"Validation Error: {e}")
+            except Exception as e:
+                logger.error(f"Error adding metric: {e}")
+                
         if feedback:
             self.span(name).add_feedback(feedback)
 

ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -209,6 +209,9 @@ class AgenticTracing(
             # Deactivate network tracing
             self.network_tracer.deactivate_patches()
 
+            # Clear visited metrics when stopping trace
+            self.visited_metrics.clear()
+
             # Stop base tracer (includes saving to file)
             super().stop()
 

ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py

@@ -63,8 +63,8 @@ class ToolTracerMixin:
         if metrics:
             if isinstance(metrics, dict):
                 metrics = [metrics]
-            for metric in metrics:
-                try:
+            try:
+                for metric in metrics:
                     self.span(name).add_metrics(
                         name=metric["name"],
                         score=metric["score"],
@@ -74,8 +74,11 @@ class ToolTracerMixin:
                         metadata=metric.get("metadata", {}),
                         config=metric.get("config", {}),
                     )
-                except KeyError as e:
-                    logger.error(f"Error adding metric: {e}")
+            except ValueError as e:
+                    logger.error(f"Validation Error: {e}")
+            except Exception as e:
+                logger.error(f"Error adding metric: {e}")
+            
         if feedback:
             self.span(name).add_feedback(feedback)
 
@@ -274,8 +277,16 @@ class ToolTracerMixin:
 
         # metrics
         metrics = []
+        metrics = []
         if name in self.span_attributes_dict:
-            metrics = self.span_attributes_dict[name].metrics or []
+            raw_metrics = self.span_attributes_dict[name].metrics or []
+            # Check for duplicates
+            for metric in raw_metrics:
+                metric_name = metric["name"]
+                if self.visited_metrics.count(metric_name) >= 1:
+                    raise ValueError(f"Duplicate metric '{metric_name}' found in trace. Metric names must be unique across all spans.")
+                self.visited_metrics.append(metric_name)
+                metrics.append(metric)
 
         start_time = kwargs["start_time"]
         component = {

{ragaai_catalyst-2.1.4b4.dist-info → ragaai_catalyst-2.1.4b5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.4b4
+Version: 2.1.4b5
 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>
 Requires-Python: <3.13,>=3.9

{ragaai_catalyst-2.1.4b4.dist-info → ragaai_catalyst-2.1.4b5.dist-info}/RECORD

@@ -26,14 +26,14 @@ ragaai_catalyst/tracers/agentic_tracing/tests/__init__.py,sha256=47DEQpj8HBSa-_T
 ragaai_catalyst/tracers/agentic_tracing/tests/ai_travel_agent.py,sha256=S4rCcKzU_5SB62BYEbNn_1VbbTdG4396N8rdZ3ZNGcE,5654
 ragaai_catalyst/tracers/agentic_tracing/tests/unique_decorator_test.py,sha256=Xk1cLzs-2A3dgyBwRRnCWs7Eubki40FVonwd433hPN8,4805
 ragaai_catalyst/tracers/agentic_tracing/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=IpcSJG8lxghWmkIpcOvQ_4G0OqK79-Y2_75CEa5dHtg,25027
-ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=xNiWN0xJDm0GbnB0jo8xFnt-BO0-x4CmaHInsWxHREA,31266
+ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=Xe7LcJWfHphezpHNBlKKU_D-AjjU-c9mNpYWCNZbhoQ,25531
+ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=ihB0y54-JmSYLZRH6gmkm73jrqeD0NtuVWswuuq5N1o,31304
 ragaai_catalyst/tracers/agentic_tracing/tracers/custom_tracer.py,sha256=uay8lU7T-CKsVu8KvWX31qfMqufK9S3Ive7XKo2Ksmk,12252
 ragaai_catalyst/tracers/agentic_tracing/tracers/langgraph_tracer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=NzMLIf21foZplOBx2zq32sUHV8geBNQmI2B-sIoSpPI,28810
-ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=5zUfKG87-sJPvDY56uueEL4J8Qk797vl0WDVbqhbx7Y,15117
+ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=VmPxlIQBo58ewf1vyi34gqw7M4YviRmb5KgnSQMdejY,29331
+ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=nxUOQSyWBTnbsZfxmr1lje2OggqNf9fwtGUb-sBo6mI,15215
 ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=CviGiAg0W0krJxORMBDTosQytIoJDQ5RwU6xt_U_mOg,10408
-ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=PD027GHkbEwLUINXYradgblSEN8d2cqkHwpyX3RP_uw,11627
+ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=XUDWn9F07lzrhFTaKcLZyJnLjIbFaucNGSHDgDQqxks,12169
 ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py,sha256=bhSUhNQCuJXKjgJAXhjKEYjnHMpYN90FSZdR84fNIKU,4614
 ragaai_catalyst/tracers/agentic_tracing/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_agentic_traces.py,sha256=1MDKXAAPzOEdxFKWWQrRgrmM3kz--DGXSywGXQmR3lQ,6041
@@ -60,7 +60,7 @@ ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hak
 ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.4b4.dist-info/METADATA,sha256=4xoAwoSRRw5GD6yEeygoXU49jMKM4RXa1_MWLV9EblA,12770
-ragaai_catalyst-2.1.4b4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.4b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.4b4.dist-info/RECORD,,
+ragaai_catalyst-2.1.4b5.dist-info/METADATA,sha256=7pslb4wedeSS3p7J4rw4Lh8IL70dRP_vNKK0_5giJqE,12770
+ragaai_catalyst-2.1.4b5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.4b5.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.4b5.dist-info/RECORD,,