ragaai-catalyst 2.1.4b1__py3-none-any.whl → 2.1.4b2__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py

@@ -514,6 +514,19 @@ class AgentTracerMixin:
                 kwargs["component_id"], []
             )
         start_time = kwargs["start_time"]
+
+        # Get tags, metrics
+        name = kwargs["name"]
+        # tags
+        tags = []
+        if name in self.span_attributes_dict:
+            tags = self.span_attributes_dict[name].tags or []
+
+        # metrics
+        metrics = []
+        if name in self.span_attributes_dict:
+            metrics = self.span_attributes_dict[name].metrics or []
+
         component = {
             "id": kwargs["component_id"],
             "hash_id": kwargs["hash_id"],
@@ -529,14 +542,14 @@ class AgentTracerMixin:
                 "version": kwargs["version"],
                 "capabilities": kwargs["capabilities"],
                 "memory_used": kwargs["memory_used"],
-                "tags": self.span_attributes_dict[kwargs["name"]].tags or [],
+                "tags": tags,
             },
             "data": {
                 "input": kwargs["input_data"],
                 "output": kwargs["output_data"],
                 "children": kwargs.get("children", []),
             },
-            "metrics": self.span_attributes_dict[kwargs["name"]].metrics or [],
+            "metrics": metrics,
             "network_calls": network_calls,
             "interactions": interactions,
         }

ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py

@@ -354,6 +354,17 @@ class LLMTracerMixin:
             list(parameters_to_display.items())[: self.MAX_PARAMETERS_TO_DISPLAY]
         )
 
+        # Get tags, metrics
+        # tags
+        tags = []
+        if name in self.span_attributes_dict:
+            tags = self.span_attributes_dict[name].tags or []
+
+        # metrics
+        metrics = []
+        if name in self.span_attributes_dict:
+            metrics = self.span_attributes_dict[name].metrics or []
+
         component = {
             "id": component_id,
             "hash_id": hash_id,
@@ -370,7 +381,7 @@ class LLMTracerMixin:
                 "memory_used": memory_used,
                 "cost": cost,
                 "tokens": usage,
-                "tags": self.span_attributes_dict[name].tags or [],
+                "tags": tags,
                 **parameters_to_display,
             },
             "extra_info": parameters,
@@ -381,7 +392,7 @@ class LLMTracerMixin:
                 "output": output_data.output_response if output_data else None,
                 "memory_used": memory_used,
             },
-            "metrics": self.span_attributes_dict[name].metrics or [],
+            "metrics": metrics,
             "network_calls": network_calls,
             "interactions": interactions,
         }

ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -137,11 +137,13 @@ class AgenticTracing(
             self.network_tracer.network_calls.copy()
         )
         self.network_tracer.network_calls = []  # Reset for next component
-        self.component_user_interaction[component_id] = [
-            interaction
-            for interaction in self.user_interaction_tracer.interactions
-            if interaction.get("component_id") == component_id
-        ]
+        # self.component_user_interaction[component_id] = [interaction for interaction in self.user_interaction_tracer.interactions if interaction.get('component_id') == component_id]
+        for interaction in self.user_interaction_tracer.interactions:
+            interaction_component_id = interaction.get("component_id")
+            if interaction_component_id not in self.component_user_interaction:
+                self.component_user_interaction[interaction_component_id] = []
+            if interaction not in self.component_user_interaction[interaction_component_id]:
+                self.component_user_interaction[interaction_component_id].append(interaction)
 
     def start(self):
         """Start tracing"""

ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py

@@ -265,6 +265,18 @@ class ToolTracerMixin:
                 kwargs["component_id"], []
             )
 
+        # Get tags, metrics
+        name = kwargs["name"]
+        # tags
+        tags = []
+        if name in self.span_attributes_dict:
+            tags = self.span_attributes_dict[name].tags or []
+
+        # metrics
+        metrics = []
+        if name in self.span_attributes_dict:
+            metrics = self.span_attributes_dict[name].metrics or []
+
         start_time = kwargs["start_time"]
         component = {
             "id": kwargs["component_id"],
@@ -280,14 +292,14 @@ class ToolTracerMixin:
                 "tool_type": kwargs["tool_type"],
                 "version": kwargs["version"],
                 "memory_used": kwargs["memory_used"],
-                "tags": self.span_attributes_dict[kwargs["name"]].tags or [],
+                "tags": tags,
             },
             "data": {
                 "input": kwargs["input_data"],
                 "output": kwargs["output_data"],
                 "memory_used": kwargs["memory_used"],
             },
-            "metrics": self.span_attributes_dict[kwargs["name"]].metrics or [],
+            "metrics": metrics,
             "network_calls": network_calls,
             "interactions": interactions,
         }

{ragaai_catalyst-2.1.4b1.dist-info → ragaai_catalyst-2.1.4b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.4b1
+Version: 2.1.4b2
 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.4b1.dist-info → ragaai_catalyst-2.1.4b2.dist-info}/RECORD

@@ -26,13 +26,13 @@ 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=AQJ-QvpOJ6mjI3nxmD5TU_4IIvDbmP8Iw7eGU7I6iXc,24692
+ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=unr4AVpXu2IJeNz_oL-fEtJ-tNmbCgdWJN5euCfdC8c,24951
 ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=z2e1kGVAi_oBK__tCs0OiSVMYjqAeAdXd7CeZYzYWJA,34660
 ragaai_catalyst/tracers/agentic_tracing/tracers/custom_tracer.py,sha256=tcOXzWTZlDdlBamCR0g5Sks5WwmKCK93R0ijIbDeSpI,12467
-ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=BFXsFqce33RWsYvHok2pPC3kbY2Ogvb6RCA1ko8psnY,28511
-ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=dIXuXFW0yFWbXBaJqHnFKpKE4G9ZOmynXSDbQJAItGk,14649
+ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=9EYrfkMYNYXYur9msb0wmxJlN_c58mzSj-GTDbLb154,28759
+ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=Tthd1lm_QKeY8E4HLYWW9djujnOO_5X-LIKnv1pNwJI,15092
 ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=Tq9by4DV51UR4BCUc3fnAiNR4SZku-gOaOfjJryowAA,10218
-ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=TP0k7f3-drNTYssgmTEZGd-78_lEmgBAEg60n_HDhmQ,11369
+ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=PD027GHkbEwLUINXYradgblSEN8d2cqkHwpyX3RP_uw,11627
 ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py,sha256=nrRqjHgAPNJjlU-WgLKqfQ92SzT7usV45T4jW4hwhrc,4575
 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
@@ -59,7 +59,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.4b1.dist-info/METADATA,sha256=qmEMerLs90GxnGwc7yyO_CPTsMTpd1EilfKRJ936zjg,12770
-ragaai_catalyst-2.1.4b1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.4b1.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.4b1.dist-info/RECORD,,
+ragaai_catalyst-2.1.4b2.dist-info/METADATA,sha256=hAIHL6OVFHUiGDUdh94meRnU5YUhuBdRcco94cTtA8c,12770
+ragaai_catalyst-2.1.4b2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.4b2.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.4b2.dist-info/RECORD,,