ragaai-catalyst 2.0.7.2b1__py3-none-any.whl → 2.1b1__py3-none-any.whl

ragaai_catalyst/tracers/agentic_tracing/user_interaction_tracer.py

@@ -1,67 +1,43 @@
 import builtins
-from contextlib import contextmanager, asynccontextmanager
 from datetime import datetime
-
-from ..data import UserInteractionModel
-
+import contextvars
+import inspect
+import uuid
 
 class UserInteractionTracer:
-    def __init__(self, tracer):
-        self.tracer = tracer
+    def __init__(self, *args, **kwargs):
+        self.project_id = contextvars.ContextVar("project_id", default=None)
+        self.trace_id = contextvars.ContextVar("trace_id", default=None)
+        self.tracer = contextvars.ContextVar("tracer", default=None)
+        self.component_id = contextvars.ContextVar("component_id", default=None)
         self.original_input = builtins.input
         self.original_print = builtins.print
+        self.interactions = []
 
-    def input(self, prompt=""):
-        content = prompt
-        user_input = self.original_input(prompt)
-        self._log_interaction("input", user_input)
-        return user_input
-
-    def print(self, *args, **kwargs):
-        content = " ".join(str(arg) for arg in args)
-        self._log_interaction("output", content)
-        self.original_print(*args, **kwargs)
-
-    def _log_interaction(self, interaction_type, content):
-        agent_id = self.tracer.current_agent_id.get()
-        user_interaction = UserInteractionModel(
-            project_id=self.tracer.project_id,
-            trace_id=self.tracer.trace_id,
-            agent_id=agent_id,
-            interaction_type=interaction_type,
-            content=content,
-            timestamp=datetime.now(),
-        )
-        with self.tracer.Session() as session:
-            session.add(user_interaction)
-            session.commit()
-
-        # Also add to trace data
-        self.tracer.trace_data.setdefault("user_interactions", []).append(
-            {
-                "interaction_type": interaction_type,
-                "content": content,
-                "timestamp": datetime.now(),
-                "agent_id": agent_id,
-            }
-        )
-
-    @contextmanager
-    def capture(self):
-        builtins.input = self.input
-        builtins.print = self.print
+    def traced_input(self, prompt=""):
+        # Get caller information
+        if prompt:
+            self.traced_print(prompt, end="")
         try:
-            yield
-        finally:
-            builtins.input = self.original_input
-            builtins.print = self.original_print
-
-    @asynccontextmanager
-    async def async_capture(self):
-        builtins.input = self.input
-        builtins.print = self.print
-        try:
-            yield
-        finally:
-            builtins.input = self.original_input
-            builtins.print = self.original_print
+            content = self.original_input()
+        except EOFError:
+            content = ""  # Return empty string on EOF
+            
+        self.interactions.append({
+            "id": str(uuid.uuid4()),
+            "interaction_type": "input",
+            "content": content,
+            "timestamp": datetime.now().isoformat()
+        })
+        return content
+
+    def traced_print(self, *args, **kwargs):
+        content = " ".join(str(arg) for arg in args)
+        
+        self.interactions.append({
+            "id": str(uuid.uuid4()),
+            "interaction_type": "output",
+            "content": content,
+            "timestamp": datetime.now().isoformat()
+        })
+        return self.original_print(*args, **kwargs)

ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py

@@ -23,10 +23,6 @@ except FileNotFoundError:
 
 
 def extract_llm_output(result):
-
-    # import pdb
-
-    # pdb.set_trace()
     # Initialize variables
     model_name = None
     output_response = ""