ragaai-catalyst 2.1b4__py3-none-any.whl → 2.1.1b0__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/agent_tracer.py

@@ -20,16 +20,23 @@ class AgentTracerMixin:
         self.agent_children = contextvars.ContextVar("agent_children", default=[])
         self.component_network_calls = contextvars.ContextVar("component_network_calls", default={})
         self.component_user_interaction = contextvars.ContextVar("component_user_interaction", default={})
+        self.version = contextvars.ContextVar("version", default="1.0.0")
+        self.agent_type = contextvars.ContextVar("agent_type", default="generic")
+        self.capabilities = contextvars.ContextVar("capabilities", default=[])
+        self.start_time = contextvars.ContextVar("start_time", default=None)
+        self.input_data = contextvars.ContextVar("input_data", default=None)
         self.gt = None
 
 
-    def trace_agent(self, name: str, agent_type: str = "generic", version: str = "1.0.0", capabilities: List[str] = None):
+    def trace_agent(self, name: str, agent_type: str = None, version: str = None, capabilities: List[str] = None):
         def decorator(target):
             # Check if target is a class
             is_class = isinstance(target, type)
             tracer = self  # Store reference to tracer instance
             top_level_hash_id = generate_unique_hash_simple(target)   # Generate hash based on the decorated target code
-
+            self.version.set(version)
+            self.agent_type.set(agent_type)
+            self.capabilities.set(capabilities)
             
             if is_class:
                 # Store original __init__
@@ -160,6 +167,8 @@ class AgentTracerMixin:
             return func(*args, **kwargs)
 
         start_time = datetime.now()
+        self.start_time = start_time
+        self.input_data = self._sanitize_input(args, kwargs)
         start_memory = psutil.Process().memory_info().rss
         component_id = str(uuid.uuid4())
 
@@ -206,12 +215,11 @@ class AgentTracerMixin:
                 start_time=start_time,
                 end_time=end_time,
                 memory_used=memory_used,
-                input_data=self._sanitize_input(args, kwargs),
+                input_data=self.input_data,
                 output_data=self._sanitize_output(result),
                 children=children,
                 parent_id=parent_agent_id
             )
-
             # Add ground truth to component data if present
             if ground_truth is not None:
                 agent_component["data"]["gt"] = ground_truth
@@ -253,13 +261,12 @@ class AgentTracerMixin:
                 start_time=start_time,
                 end_time=datetime.now(),
                 memory_used=0,
-                input_data=self._sanitize_input(args, kwargs),
+                input_data=self.input_data,
                 output_data=None,
                 error=error_component,
                 children=children,
                 parent_id=parent_agent_id  # Add parent ID if exists
             )
-            
             # If this is a nested agent, add it to parent's children
             if parent_agent_id:
                 parent_component = self._agent_components.get(parent_agent_id)

ragaai_catalyst/tracers/agentic_tracing/agentic_tracing.py

@@ -164,12 +164,11 @@ class AgenticTracing(BaseTracer, LLMTracerMixin, ToolTracerMixin, AgentTracerMix
                 self.trace.metadata.total_cost = total_cost
                 self.trace.metadata.total_tokens = total_tokens
 
-    def add_component(self, component_data: dict):
+    def add_component(self, component_data: dict, is_error: bool = False):
         """Add a component to the trace data"""
         # Convert dict to appropriate Component type
         filtered_data = {k: v for k, v in component_data.items() if k in ["id", "hash_id", "type", "name", "start_time", "end_time", "parent_id", "info", "data", "network_calls", "interactions", "error"]}
 
-               
         if component_data["type"] == "llm":
             component = LLMComponent(**filtered_data)
         elif component_data["type"] == "agent": 
@@ -179,8 +178,6 @@ class AgenticTracing(BaseTracer, LLMTracerMixin, ToolTracerMixin, AgentTracerMix
         else:
             component = Component(**component_data)
 
-        # import pdb; pdb.set_trace()
-        
         # Check if there's an active agent context
         current_agent_id = self.current_agent_id.get()
         if current_agent_id and component_data["type"] in ["llm", "tool"]:
@@ -191,6 +188,43 @@ class AgenticTracing(BaseTracer, LLMTracerMixin, ToolTracerMixin, AgentTracerMix
         else:
             # Add component to the main trace
             super().add_component(component)
+            
+        # Handle error case
+        if is_error:
+            # Get the parent component if it exists
+            parent_id = component_data.get("parent_id")
+            children = self.agent_children.get()
+            
+            # Set parent_id for all children
+            for child in children:
+                child["parent_id"] = parent_id
+            
+            agent_tracer_mixin = AgentTracerMixin()
+            agent_tracer_mixin.component_network_calls = self.component_network_calls
+            agent_tracer_mixin.component_user_interaction = self.component_user_interaction
+            
+            # Create parent component with error info
+            parent_component = agent_tracer_mixin.create_agent_component(
+                component_id=parent_id,
+                hash_id=str(uuid.uuid4()),
+                name=self.current_agent_name.get(),
+                agent_type=self.agent_type.get(),
+                version=self.version.get(),
+                capabilities=self.capabilities.get(),
+                start_time=self.start_time,
+                end_time=datetime.now(),
+                memory_used=0,
+                input_data=self.input_data,
+                output_data=None,
+                children=children,
+                parent_id=None  # Add parent ID if exists
+            )
+
+            filtered_data = {k: v for k, v in parent_component.items() if k in ["id", "hash_id", "type", "name", "start_time", "end_time", "parent_id", "info", "data", "network_calls", "interactions", "error"]}
+            parent_agent_component = AgentComponent(**filtered_data)
+            # Add the parent component to trace and stop tracing
+            super().add_component(parent_agent_component)
+            self.stop()
 
     def __enter__(self):
         """Context manager entry"""

ragaai_catalyst/tracers/agentic_tracing/llm_tracer.py

@@ -501,12 +501,6 @@ class LLMTracerMixin:
             name = self.current_llm_call_name.get()
             if name is None:
                 name = original_func.__name__
-
-            end_memory = psutil.Process().memory_info().rss
-            memory_used = max(0, end_memory - start_memory)
-            
-            # Get parent agent ID if exists
-            parent_agent_id = self.current_agent_id.get() if hasattr(self, 'current_agent_id') else None
             
             llm_component = self.create_llm_component(
                 component_id=component_id,
@@ -514,24 +508,15 @@ class LLMTracerMixin:
                 name=name,
                 llm_type="unknown",
                 version="1.0.0",
-                memory_used=memory_used,
+                memory_used=0,
                 start_time=start_time,
                 end_time=end_time,
                 input_data=self._extract_input_data(args, kwargs, None),
                 output_data=None,
-                error=error_component,
-                parent_id=parent_agent_id
+                error=error_component
             )
     
-            # If this is part of an agent trace, add it to agent's children
-            if parent_agent_id and hasattr(self, 'agent_children'):
-                parent_children = self.agent_children.get()
-                parent_children.append(llm_component)
-                self.agent_children.set(parent_children)
-            else:
-                # Only add to root components if no parent
-                self.add_component(llm_component)
-                
+            self.add_component(llm_component)
             raise
 
     def trace_llm_call_sync(self, original_func, *args, **kwargs):
@@ -618,9 +603,6 @@ class LLMTracerMixin:
             end_memory = psutil.Process().memory_info().rss
             memory_used = max(0, end_memory - start_memory)
             
-            # Get parent agent ID if exists
-            parent_agent_id = self.current_agent_id.get() if hasattr(self, 'current_agent_id') else None
-            
             llm_component = self.create_llm_component(
                 component_id=component_id,
                 hash_id=hash_id,
@@ -632,22 +614,14 @@ class LLMTracerMixin:
                 end_time=end_time,
                 input_data=self._extract_input_data(args, kwargs, None),
                 output_data=None,
-                error=error_component,
-                parent_id=parent_agent_id
+                error=error_component
             )
     
-            # If this is part of an agent trace, add it to agent's children
-            if parent_agent_id and hasattr(self, 'agent_children'):
-                parent_children = self.agent_children.get()
-                parent_children.append(llm_component)
-                self.agent_children.set(parent_children)
-            else:
-                # Only add to root components if no parent
-                self.add_component(llm_component)
-                
+            self.add_component(llm_component, is_error=True)
             raise
 
     def trace_llm(self, name: str = None):
+        self.current_llm_call_name.set(name)
         def decorator(func):
             @self.file_tracker.trace_decorator
             @functools.wraps(func)
@@ -661,7 +635,6 @@ class LLMTracerMixin:
                 parent_agent_id = self.current_agent_id.get()
                 self.start_component(component_id)
                 
-                start_time = datetime.now()
                 error_info = None
                 result = None
                 

{ragaai_catalyst-2.1b4.dist-info → ragaai_catalyst-2.1.1b0.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.1b4
+Version: 2.1.1b0
 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.11
+Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: aiohttp>=3.10.2
 Requires-Dist: opentelemetry-api==1.25.0
@@ -27,7 +27,7 @@ Requires-Dist: Markdown>=3.7
 Requires-Dist: litellm==1.51.1
 Requires-Dist: tenacity==8.3.0
 Requires-Dist: tqdm>=4.66.5
-Requires-Dist: llama-index>=0.10.0
+Requires-Dist: llama-index~=0.10
 Requires-Dist: pyopenssl>=24.2.1
 Requires-Dist: psutil~=6.0.0
 Requires-Dist: py-cpuinfo~=9.0.0

{ragaai_catalyst-2.1b4.dist-info → ragaai_catalyst-2.1.1b0.dist-info}/RECORD

@@ -16,12 +16,12 @@ ragaai_catalyst/tracers/llamaindex_callback.py,sha256=vPE7MieKjfwLrLUnnPs20Df0xN
 ragaai_catalyst/tracers/tracer.py,sha256=dGYQfo0RXms6w-sAJf04gqTo1zOXTqreGXaJlnvK48A,12463
 ragaai_catalyst/tracers/upload_traces.py,sha256=hs0PEmit3n3_uUqrdbwcBdyK5Nbkik3JQVwJMEwYTd4,4796
 ragaai_catalyst/tracers/agentic_tracing/__init__.py,sha256=6QyQI8P7aNFHTantNJOP1ZdSNrDKBLhlg_gGNj7tm1c,73
-ragaai_catalyst/tracers/agentic_tracing/agent_tracer.py,sha256=M5RBNzcuSCEKv-rYwmzBMN-BxvthM3y6d_279UYERZQ,20832
-ragaai_catalyst/tracers/agentic_tracing/agentic_tracing.py,sha256=WsddEXn2afdPx8PfignQoLIwZfn9350XsqtkqtZdMxw,8502
+ragaai_catalyst/tracers/agentic_tracing/agent_tracer.py,sha256=oLe5QtrndaO-sBAHvxUPfj9GB_EwQbi0T3JwjcqmvtY,21388
+ragaai_catalyst/tracers/agentic_tracing/agentic_tracing.py,sha256=yqWu_xnKfLE6nVYM0TqaAYLI_tzy5Z2wZwgA9-Gjppo,10218
 ragaai_catalyst/tracers/agentic_tracing/base.py,sha256=jRf_-5EIfCzGbaSQtkqgiDQAH4ymoKUrg9A8YqB08jk,14008
 ragaai_catalyst/tracers/agentic_tracing/data_structure.py,sha256=wkZctMTUQUViqKrHhdZiMERSBVfwURAJ-lFlQUou638,7395
 ragaai_catalyst/tracers/agentic_tracing/file_name_tracker.py,sha256=515NNDQJTyy3O-2rdlUYUoWL9qSwLIfvV3sMB9BtHp8,1366
-ragaai_catalyst/tracers/agentic_tracing/llm_tracer.py,sha256=oxziex6mT4RcZY1zWTVgsPXVewiTWt16CQAfWi9_YD0,34072
+ragaai_catalyst/tracers/agentic_tracing/llm_tracer.py,sha256=BpazvaHy9YBsnizQMIEZoRlan7DiM7pCUO9u-RSYiXo,32721
 ragaai_catalyst/tracers/agentic_tracing/network_tracer.py,sha256=6FTA15xMnum9omM_0Jd9cMIuWdKu1gR5Tc8fOXAkP8E,10068
 ragaai_catalyst/tracers/agentic_tracing/sample.py,sha256=S4rCcKzU_5SB62BYEbNn_1VbbTdG4396N8rdZ3ZNGcE,5654
 ragaai_catalyst/tracers/agentic_tracing/tool_tracer.py,sha256=Yc4x82rk0hCANwXUt4M66Qv_4OdpsXsjlq6OIOef1io,8763
@@ -50,7 +50,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.1b4.dist-info/METADATA,sha256=A9MMHw5G0oZq-8lh5mwo57hNwgYsOhM80wOXe418FDQ,1796
-ragaai_catalyst-2.1b4.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
-ragaai_catalyst-2.1b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1b4.dist-info/RECORD,,
+ragaai_catalyst-2.1.1b0.dist-info/METADATA,sha256=qAtdLG3fZX-NLrkgejCFS-aIna4O_suzsle3OKByqew,1801
+ragaai_catalyst-2.1.1b0.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+ragaai_catalyst-2.1.1b0.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.1b0.dist-info/RECORD,,