ragaai-catalyst 2.1.4b1__tar.gz → 2.1.4b2__tar.gz

{ragaai_catalyst-2.1.4b1/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.4b2}/PKG-INFO

@@ -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 → ragaai_catalyst-2.1.4b2}/pyproject.toml

@@ -9,7 +9,7 @@ readme = "README.md"
 requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
 
-version = "2.1.4.beta.1"
+version = "2.1.4.beta.2"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.4b1 → ragaai_catalyst-2.1.4b2}/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-2.1.4b1 → ragaai_catalyst-2.1.4b2}/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-2.1.4b1 → ragaai_catalyst-2.1.4b2}/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-2.1.4b1 → ragaai_catalyst-2.1.4b2}/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 → ragaai_catalyst-2.1.4b2/ragaai_catalyst.egg-info}/PKG-INFO

@@ -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