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

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,19 +614,10 @@ 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)
             raise
 
     def trace_llm(self, name: str = None):

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

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: ragaai_catalyst
-Version: 2.1b4
+Version: 2.1b5
 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.9
 Description-Content-Type: text/markdown
 Requires-Dist: aiohttp>=3.10.2
 Requires-Dist: opentelemetry-api==1.25.0

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

@@ -21,7 +21,7 @@ ragaai_catalyst/tracers/agentic_tracing/agentic_tracing.py,sha256=WsddEXn2afdPx8
 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=dI2Pg4cNGTf5k7g7ViVRUu6Pck8In4aqQYGLn_xwNxY,32705
 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.1b5.dist-info/METADATA,sha256=Aqk9psd4uVTwFEtlydBvf79poBz1uPttrh_qLw-p5Bc,1795
+ragaai_catalyst-2.1b5.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+ragaai_catalyst-2.1b5.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1b5.dist-info/RECORD,,