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

ragaai_catalyst/tracers/agentic_tracing/data/data_structure.py

@@ -218,7 +218,7 @@ class Component:
                             id=interaction.get("id", str(uuid.uuid4())),
                             type=interaction.get("interaction_type", ""),
                             content=str(interaction.get("content", "")),
-                            timestamp=interaction.get("timestamp", datetime.utcnow().isoformat())
+                            timestamp=interaction.get("timestamp", datetime.now().astimezone().isoformat())
                         )
                     )
                 else:

ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py

@@ -117,7 +117,7 @@ class AgentTracerMixin:
                         agent_type=agent_type,
                         version=version,
                         capabilities=capabilities or [],
-                        start_time=datetime.now(),
+                        start_time=datetime.now().astimezone().isoformat(),
                         memory_used=0,
                         input_data=tracer._sanitize_input(args, kwargs),
                         output_data=None,
@@ -167,7 +167,7 @@ class AgentTracerMixin:
                                     children_token = tracer.agent_children.set([])
 
                                     try:
-                                        start_time = datetime.now()
+                                        start_time = datetime.now().astimezone().isoformat()
                                         result = method(self, *args, **kwargs)
 
                                         # Update agent component with method result
@@ -183,7 +183,7 @@ class AgentTracerMixin:
                                                     tracer._sanitize_input(args, kwargs)
                                                 )
                                                 component["start_time"] = (
-                                                    start_time.isoformat()
+                                                    start_time
                                                 )
 
                                                 # Get children accumulated during method execution
@@ -265,7 +265,7 @@ class AgentTracerMixin:
         if not self.auto_instrument_agent:
             return func(*args, **kwargs)
 
-        start_time = datetime.now()
+        start_time = datetime.now().astimezone().isoformat()
         self.start_time = start_time
         self.input_data = self._sanitize_input(args, kwargs)
         start_memory = psutil.Process().memory_info().rss
@@ -390,7 +390,7 @@ class AgentTracerMixin:
         if not self.auto_instrument_agent:
             return await func(*args, **kwargs)
 
-        start_time = datetime.now()
+        start_time = datetime.now().astimezone().isoformat()
         start_memory = psutil.Process().memory_info().rss
         component_id = str(uuid.uuid4())
 
@@ -533,7 +533,7 @@ class AgentTracerMixin:
             "source_hash_id": None,
             "type": "agent",
             "name": kwargs["name"],
-            "start_time": start_time.isoformat(),
+            "start_time": start_time,
             "end_time": datetime.now().astimezone().isoformat(),
             "error": kwargs.get("error"),
             "parent_id": kwargs.get("parent_id"),

ragaai_catalyst/tracers/agentic_tracing/tracers/base.py

@@ -170,11 +170,11 @@ class BaseTracer:
         self.trace_id = str(uuid.uuid4())
 
         # Get the start time
-        self.start_time = datetime.now().isoformat()
+        self.start_time = datetime.now().astimezone().isoformat()
 
         self.data_key = [
             {
-                "start_time": datetime.now().isoformat(),
+                "start_time": datetime.now().astimezone().isoformat(),
                 "end_time": "",
                 "spans": self.components,
             }
@@ -184,7 +184,7 @@ class BaseTracer:
             id=self.trace_id,
             trace_name=self.trace_name,
             project_name=self.project_name,
-            start_time=datetime.now().isoformat(),
+            start_time=datetime.now().astimezone().isoformat(),
             end_time="",  # Will be set when trace is stopped
             metadata=metadata,
             data=self.data_key,
@@ -194,8 +194,8 @@ class BaseTracer:
     def stop(self):
         """Stop the trace and save to JSON file"""
         if hasattr(self, "trace"):
-            self.trace.data[0]["end_time"] = datetime.now().isoformat()
-            self.trace.end_time = datetime.now().isoformat()
+            self.trace.data[0]["end_time"] = datetime.now().astimezone().isoformat()
+            self.trace.end_time = datetime.now().astimezone().isoformat()
 
             # Change span ids to int
             self.trace = self._change_span_ids_to_int(self.trace)
@@ -284,17 +284,25 @@ class BaseTracer:
     def __exit__(self, exc_type, exc_value, traceback):
         self.stop()
 
+    def _process_children(self, children_list, parent_id, current_id):
+        """Helper function to process children recursively."""
+        for child in children_list:
+            child["id"] = current_id
+            child["parent_id"] = parent_id
+            current_id += 1
+            # Recursively process nested children if they exist
+            if "children" in child["data"]:
+                current_id = self._process_children(child["data"]["children"], child["id"], current_id)
+        return current_id
+
     def _change_span_ids_to_int(self, trace):
         id, parent_id = 1, 0
         for span in trace.data[0]["spans"]:
             span.id = id
             span.parent_id = parent_id
             id += 1
-            if span.type == "agent":
-                for children in span.data["children"]:
-                    children["id"] = id
-                    children["parent_id"] = span.id
-                    id += 1
+            if span.type == "agent" and "children" in span.data:
+                id = self._process_children(span.data["children"], span.id, id)
         return trace
 
     def _change_agent_input_output(self, trace):
@@ -445,63 +453,169 @@ class BaseTracer:
     def add_tags(self, tags: List[str]):
         raise NotImplementedError
 
-    # def _add_span_attributes_to_trace(self, trace):
-    #     if not hasattr(trace, 'data'):
-    #         return trace
-    #     for data in trace.data:
-    #         for span in data.get('spans', []):
-    #             if not hasattr(span, 'name'):
-    #                 continue
-    #             span_name = span.name
-    #             if span_name in self.span_attributes_dict:
-    #                 span_attributes = self.span_attributes_dict[span_name]
-    #                 span = self._add_span_attributes_to_span(span_attributes, span)
-    #             if hasattr(span, 'type'):
-    #                 if span.type == 'agent':
-    #                     if hasattr(span, 'data'):
-    #                         if 'children' in span.data:
-    #                             span.data['children'] = self._add_span_attributes_to_children(span_attributes, span.data['children'])
-
-    #     return trace
-
-    # def _add_span_attributes_to_children(self, span_attributes: SpanAttributes, children):
-    #     attributed_children = []
-    #     for child in children:
-    #         if 'name' not in child:
-    #             continue
-    #         child_name = child['name']
-    #         if child_name in self.span_attributes_dict:
-    #             span_attributes = self.span_attributes_dict[child_name]
-    #             child = self._add_span_attributes_to_span(span_attributes, child)
-    #         if 'type' in child:
-    #             if child['type'] == 'agent':
-    #                 if 'data' in child:
-    #                     if 'children' in child['data']:
-    #                         child['data']['children'] = self._add_span_attributes_to_children(span_attributes, child['data']['children'])
-    #         attributed_children.append(child)
-    #     return attributed_children
-
-    # def _add_span_attributes_to_span(self, span_attributes: SpanAttributes, span):
-    #     metadata = {
-    #         'tags': span_attributes.tags,
-    #         'user_metadata': span_attributes.metadata
-    #     }
-    #     metrics = span_attributes.metrics
-    #     feedback = span_attributes.feedback
-    #     if isinstance(span, dict):
-    #         span['metadata'] = metadata
-    #         span['metrics'] = metrics
-    #         span['feedback'] = feedback
-    #     else:
-    #         span.metadata = metadata
-    #         span.metrics = metrics
-    #         span.feedback = feedback
-    #     return span
+    def _process_child_interactions(self, child, interaction_id, interactions):
+        """
+        Helper method to process child interactions recursively.
+        
+        Args:
+            child (dict): The child span to process
+            interaction_id (int): Current interaction ID
+            interactions (list): List of interactions to append to
+            
+        Returns:
+            int: Next interaction ID to use
+        """
+        child_type = child.get("type")
+        
+        if child_type == "tool":
+            # Tool call start
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": "tool_call_start",
+                    "name": child.get("name"),
+                    "content": {
+                        "parameters": [
+                            child.get("data", {}).get("input", {}).get("args"),
+                            child.get("data", {}).get("input", {}).get("kwargs"),
+                        ]
+                    },
+                    "timestamp": child.get("start_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
+
+            # Tool call end
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": "tool_call_end",
+                    "name": child.get("name"),
+                    "content": {
+                        "returns": child.get("data", {}).get("output"),
+                    },
+                    "timestamp": child.get("end_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
+
+        elif child_type == "llm":
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": "llm_call_start",
+                    "name": child.get("name"),
+                    "content": {
+                        "prompt": child.get("data", {}).get("input"),
+                    },
+                    "timestamp": child.get("start_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
+
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": "llm_call_end",
+                    "name": child.get("name"),
+                    "content": {
+                        "response": child.get("data", {}).get("output")
+                    },
+                    "timestamp": child.get("end_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
+
+        elif child_type == "agent":
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": "agent_call_start",
+                    "name": child.get("name"),
+                    "content": None,
+                    "timestamp": child.get("start_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
 
-    def span(self, span_name):
-        if span_name not in self.span_attributes_dict:
-            self.span_attributes_dict[span_name] = SpanAttributes(span_name)
-        return self.span_attributes_dict[span_name]
+            # Process nested children recursively
+            if "children" in child.get("data", {}):
+                for nested_child in child["data"]["children"]:
+                    interaction_id = self._process_child_interactions(
+                        nested_child, interaction_id, interactions
+                    )
+
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": "agent_call_end",
+                    "name": child.get("name"),
+                    "content": child.get("data", {}).get("output"),
+                    "timestamp": child.get("end_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
+
+        else:
+            interactions.append(
+                {
+                    "id": str(interaction_id),
+                    "span_id": child.get("id"),
+                    "interaction_type": child_type,
+                    "name": child.get("name"),
+                    "content": child.get("data", {}),
+                    "timestamp": child.get("start_time"),
+                    "error": child.get("error"),
+                }
+            )
+            interaction_id += 1
+
+        # Process additional interactions and network calls
+        if "interactions" in child:
+            for interaction in child["interactions"]:
+                interaction["id"] = str(interaction_id)
+                interaction["span_id"] = child.get("id")
+                interaction["error"] = None
+                interactions.append(interaction)
+                interaction_id += 1
+
+        if "network_calls" in child:
+            for child_network_call in child["network_calls"]:
+                network_call = {}
+                network_call["id"] = str(interaction_id)
+                network_call["span_id"] = child.get("id")
+                network_call["interaction_type"] = "network_call"
+                network_call["name"] = None
+                network_call["content"] = {
+                    "request": {
+                        "url": child_network_call.get("url"),
+                        "method": child_network_call.get("method"),
+                        "headers": child_network_call.get("headers"),
+                    },
+                    "response": {
+                        "status_code": child_network_call.get("status_code"),
+                        "headers": child_network_call.get("response_headers"),
+                        "body": child_network_call.get("response_body"),
+                    },
+                }
+                network_call["timestamp"] = child_network_call.get("start_time")
+                network_call["error"] = child_network_call.get("error")
+                interactions.append(network_call)
+                interaction_id += 1
+
+        return interaction_id
 
     def format_interactions(self) -> dict:
         """
@@ -513,14 +627,14 @@ class BaseTracer:
         tool_call_end, llm_call, file_read, file_write, network_call.
 
         Returns:
-            dict: A dictionary with "interactions" key containing a list of interactions
+            dict: A dictionary with "workflow" key containing a list of interactions
                   sorted by timestamp.
         """
         interactions = []
         interaction_id = 1
 
         if not hasattr(self, "trace") or not self.trace.data:
-            return {"interactions": []}
+            return {"workflow": []}
 
         for span in self.trace.data[0]["spans"]:
             # Process agent spans
@@ -539,159 +653,12 @@ class BaseTracer:
                 )
                 interaction_id += 1
 
-                # Process children of agent
+                # Process children of agent recursively
                 if "children" in span.data:
                     for child in span.data["children"]:
-                        child_type = child.get("type")
-                        if child_type == "tool":
-                            # Tool call start
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": "tool_call_start",
-                                    "name": child.get("name"),
-                                    "content": {
-                                        "parameters": [
-                                            child.get("data", {})
-                                            .get("input")
-                                            .get("args"),
-                                            child.get("data", {})
-                                            .get("input")
-                                            .get("kwargs"),
-                                        ]
-                                    },
-                                    "timestamp": child.get("start_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                            # Tool call end
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": "tool_call_end",
-                                    "name": child.get("name"),
-                                    "content": {
-                                        "returns": child.get("data", {}).get("output"),
-                                    },
-                                    "timestamp": child.get("end_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                        elif child_type == "llm":
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": "llm_call_start",
-                                    "name": child.get("name"),
-                                    "content": {
-                                        "prompt": child.get("data", {}).get("input"),
-                                    },
-                                    "timestamp": child.get("start_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": "llm_call_end",
-                                    "name": child.get("name"),
-                                    "content": {
-                                        "response": child.get("data", {}).get("output")
-                                    },
-                                    "timestamp": child.get("end_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                        elif child_type == "agent":
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": "agent_call_start",
-                                    "name": child.get("name"),
-                                    "content": None,
-                                    "timestamp": child.get("start_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": "agent_call_end",
-                                    "name": child.get("name"),
-                                    "content": child.get("data", {}).get("output"),
-                                    "timestamp": child.get("end_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                        else:
-                            interactions.append(
-                                {
-                                    "id": str(interaction_id),
-                                    "span_id": child.get("id"),
-                                    "interaction_type": child_type,
-                                    "name": child.get("name"),
-                                    "content": child.get("data", {}),
-                                    "timestamp": child.get("start_time"),
-                                    "error": child.get("error"),
-                                }
-                            )
-                            interaction_id += 1
-
-                        if "interactions" in child:
-                            for interaction in child["interactions"]:
-                                interaction["id"] = str(interaction_id)
-                                interaction["span_id"] = child.get("id")
-                                interaction["error"] = None
-                                interactions.append(interaction)
-                                interaction_id += 1
-
-                        if "network_calls" in child:
-                            for child_network_call in child["network_calls"]:
-                                network_call = {}
-                                network_call["id"] = str(interaction_id)
-                                network_call["span_id"] = child.get("id")
-                                network_call["interaction_type"] = "network_call"
-                                network_call["name"] = None
-                                network_call["content"] = {
-                                    "request": {
-                                        "url": child_network_call.get("url"),
-                                        "method": child_network_call.get("method"),
-                                        "headers": child_network_call.get("headers"),
-                                    },
-                                    "response": {
-                                        "status_code": child_network_call.get(
-                                            "status_code"
-                                        ),
-                                        "headers": child_network_call.get(
-                                            "response_headers"
-                                        ),
-                                        "body": child_network_call.get("response_body"),
-                                    },
-                                }
-                                network_call["timestamp"] = child_network_call[
-                                    "start_time"
-                                ]
-                                network_call["error"] = child_network_call.get("error")
-                                interactions.append(network_call)
-                                interaction_id += 1
+                        interaction_id = self._process_child_interactions(
+                            child, interaction_id, interactions
+                        )
 
                 # Add agent_end interaction
                 interactions.append(
@@ -830,3 +797,8 @@ class BaseTracer:
             interaction["id"] = str(idx)
 
         return {"workflow": sorted_interactions}
+
+    def span(self, span_name):
+        if span_name not in self.span_attributes_dict:
+            self.span_attributes_dict[span_name] = SpanAttributes(span_name)
+        return self.span_attributes_dict[span_name]

ragaai_catalyst/tracers/agentic_tracing/tracers/custom_tracer.py

@@ -69,7 +69,7 @@ class CustomTracerMixin:
         if not self.is_active or not self.auto_instrument_custom:
             return func(*args, **kwargs)
 
-        start_time = datetime.now().astimezone()
+        start_time = datetime.now().astimezone().isoformat()
         start_memory = psutil.Process().memory_info().rss
         component_id = str(uuid.uuid4())
         hash_id = generate_unique_hash_simple(func)
@@ -91,7 +91,6 @@ class CustomTracerMixin:
                         pass
                 return trace_variables_func
 
-            sys.settrace(trace_variables_func)
 
         # Start tracking network calls for this component
         self.start_component(component_id)
@@ -101,7 +100,7 @@ class CustomTracerMixin:
             result = func(*args, **kwargs)
 
             # Calculate resource usage
-            end_time = datetime.now().astimezone()
+            end_time = datetime.now().astimezone().isoformat()
             end_memory = psutil.Process().memory_info().rss
             memory_used = max(0, end_memory - start_memory)
 
@@ -137,7 +136,7 @@ class CustomTracerMixin:
             # End tracking network calls for this component
             self.end_component(component_id)
             
-            end_time = datetime.now().astimezone()
+            end_time = datetime.now().astimezone().isoformat()
             
             custom_component = self.create_custom_component(
                 component_id=component_id,
@@ -156,16 +155,13 @@ class CustomTracerMixin:
 
             self.add_component(custom_component)
             raise
-        finally:
-            if trace_variables:
-                sys.settrace(None)
 
     async def _trace_custom_execution(self, func, name, custom_type, version, trace_variables, *args, **kwargs):
         """Asynchronous version of custom tracing"""
         if not self.is_active or not self.auto_instrument_custom:
             return await func(*args, **kwargs)
 
-        start_time = datetime.now().astimezone()
+        start_time = datetime.now().astimezone().isoformat()
         start_memory = psutil.Process().memory_info().rss
         component_id = str(uuid.uuid4())
         hash_id = generate_unique_hash_simple(func)
@@ -187,14 +183,12 @@ class CustomTracerMixin:
                         pass
                 return trace_variables_func
 
-            sys.settrace(trace_variables_func)
-
         try:
             # Execute the function
             result = await func(*args, **kwargs)
 
             # Calculate resource usage
-            end_time = datetime.now().astimezone()
+            end_time = datetime.now().astimezone().isoformat()
             end_memory = psutil.Process().memory_info().rss
             memory_used = max(0, end_memory - start_memory)
 
@@ -223,7 +217,7 @@ class CustomTracerMixin:
                 "details": {}
             }
             
-            end_time = datetime.now().astimezone()
+            end_time = datetime.now().astimezone().isoformat()
             
             custom_component = self.create_custom_component(
                 component_id=component_id,
@@ -241,9 +235,6 @@ class CustomTracerMixin:
             )
             self.add_component(custom_component)
             raise
-        finally:
-            if trace_variables:
-                sys.settrace(None)
 
     def create_custom_component(self, **kwargs):
         """Create a custom component according to the data structure"""
@@ -263,8 +254,8 @@ class CustomTracerMixin:
             "source_hash_id": None,
             "type": "custom",
             "name": kwargs["name"],
-            "start_time": start_time.isoformat(),
-            "end_time": kwargs["end_time"].isoformat(),
+            "start_time": start_time,
+            "end_time": kwargs["end_time"],
             "error": kwargs.get("error"),
             "parent_id": self.current_agent_id.get() if hasattr(self, 'current_agent_id') else None,
             "info": {

ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py

@@ -665,7 +665,7 @@ class LLMTracerMixin:
                             "type": type(e).__name__,
                             "message": str(e),
                             "traceback": traceback.format_exc(),
-                            "timestamp": datetime.now().isoformat(),
+                            "timestamp": datetime.now().astimezone().isoformat(),
                         }
                     }
                     raise
@@ -708,7 +708,7 @@ class LLMTracerMixin:
                 parent_agent_id = self.current_agent_id.get()
                 self.start_component(component_id)
 
-                start_time = datetime.now()
+                start_time = datetime.now().astimezone().isoformat()
                 error_info = None
                 result = None
 
@@ -721,7 +721,7 @@ class LLMTracerMixin:
                             "type": type(e).__name__,
                             "message": str(e),
                             "traceback": traceback.format_exc(),
-                            "timestamp": datetime.now().isoformat(),
+                            "timestamp": datetime.now().astimezone().isoformat(),
                         }
                     }
                     raise

ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -346,7 +346,7 @@ class AgenticTracing(
                 version=self.version.get(),
                 capabilities=self.capabilities.get(),
                 start_time=self.start_time,
-                end_time=datetime.now(),
+                end_time=datetime.now().astimezone().isoformat(),
                 memory_used=0,
                 input_data=self.input_data,
                 output_data=None,

ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py

@@ -105,10 +105,10 @@ def monkey_patch_urllib(network_tracer):
             method = url.get_method()
             url_str = url.full_url
 
-        start_time = datetime.now()
+        start_time = datetime.now().astimezone()
         try:
             response = original_urlopen(url, data, timeout, *args, **kwargs)
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone()
             network_tracer.record_call(
                 method=method,
                 url=url_str,
@@ -122,7 +122,7 @@ def monkey_patch_urllib(network_tracer):
             )
             return response
         except Exception as e:
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone()
             network_tracer.record_call(
                 method=method,
                 url=url_str,
@@ -144,10 +144,10 @@ def monkey_patch_requests(network_tracer):
     original_request = requests.Session.request
 
     def patched_request(self, method, url, *args, **kwargs):
-        start_time = datetime.now()
+        start_time = datetime.now().astimezone()
         try:
             response = original_request(self, method, url, *args, **kwargs)
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone() 
             network_tracer.record_call(
                 method=method,
                 url=url,
@@ -161,7 +161,7 @@ def monkey_patch_requests(network_tracer):
             )
             return response
         except Exception as e:
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone() 
             network_tracer.record_call(
                 method=method,
                 url=url,
@@ -184,7 +184,7 @@ def monkey_patch_http_client(network_tracer):
     original_https_request = HTTPSConnection.request
 
     def patched_request(self, method, url, body=None, headers=None, *args, **kwargs):
-        start_time = datetime.now()
+        start_time = datetime.now().astimezone() 
         try:
             result = (
                 original_http_request(self, method, url, body, headers, *args, **kwargs)
@@ -194,7 +194,7 @@ def monkey_patch_http_client(network_tracer):
                 )
             )
             response = self.getresponse()
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone() 
             network_tracer.record_call(
                 method=method,
                 url=f"{self._http_vsn_str} {self.host}:{self.port}{url}",
@@ -208,7 +208,7 @@ def monkey_patch_http_client(network_tracer):
             )
             return result
         except Exception as e:
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone() 
             network_tracer.record_call(
                 method=method,
                 url=f"{self._http_vsn_str} {self.host}:{self.port}{url}",
@@ -233,10 +233,10 @@ def monkey_patch_socket(network_tracer):
 
     def patched_create_connection(address, *args, **kwargs):
         host, port = address
-        start_time = datetime.now()
+        start_time = datetime.now().astimezone() 
         try:
             result = original_create_connection(address, *args, **kwargs)
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone()
             network_tracer.record_call(
                 method="CONNECT",
                 url=f"{host}:{port}",
@@ -245,7 +245,7 @@ def monkey_patch_socket(network_tracer):
             )
             return result
         except Exception as e:
-            end_time = datetime.now()
+            end_time = datetime.now().astimezone() 
             network_tracer.record_call(
                 method="CONNECT",
                 url=f"{host}:{port}",
@@ -265,10 +265,10 @@ def restore_socket(original_create_connection):
 
 async def patch_aiohttp_trace_config(network_tracer):
     async def on_request_start(session, trace_config_ctx, params):
-        trace_config_ctx.start = datetime.now()
+        trace_config_ctx.start = datetime.now().astimezone()
 
     async def on_request_end(session, trace_config_ctx, params):
-        end_time = datetime.now()
+        end_time = datetime.now().astimezone() 
         response = params.response
         network_tracer.record_call(
             method=params.method,

ragaai_catalyst/tracers/agentic_tracing/tracers/user_interaction_tracer.py

@@ -58,7 +58,7 @@ class UserInteractionTracer:
             "component_id": self.component_id.get(),
             "interaction_type": "input",
             "content": content,
-            "timestamp": datetime.now().isoformat()
+            "timestamp": datetime.now().astimezone().isoformat()
         })
         return content
 
@@ -70,7 +70,7 @@ class UserInteractionTracer:
             "component_id": self.component_id.get(),
             "interaction_type": "output",
             "content": content,
-            "timestamp": datetime.now().isoformat()
+            "timestamp": datetime.now().astimezone().isoformat()
         })
         return self.original_print(*args, **kwargs)
 
@@ -112,7 +112,7 @@ class UserInteractionTracer:
             "component_id": self.component_id.get(),
             "interaction_type": interaction_type,
             "file_path": file_path,
-            "timestamp": datetime.now().isoformat()
+            "timestamp": datetime.now().astimezone().isoformat()
         }
         interaction.update(kwargs)
         self.interactions.append(interaction)

ragaai_catalyst/tracers/agentic_tracing/utils/model_costs.json

@@ -1959,6 +1959,20 @@
         "litellm_provider": "mistral",
         "mode": "embedding"
     },
+    "deepseek/deepseek-reasoner": {
+        "max_tokens": 8192,
+        "max_input_tokens": 64000,
+        "max_output_tokens": 8192,
+        "input_cost_per_token": 5.5e-07,
+        "input_cost_per_token_cache_hit": 1.4e-07,
+        "output_cost_per_token": 2.19e-06,
+        "litellm_provider": "deepseek",
+        "mode": "chat",
+        "supports_function_calling": true,
+        "supports_assistant_prefill": true,
+        "supports_tool_choice": true,
+        "supports_prompt_caching": true
+    },
     "deepseek/deepseek-chat": {
         "max_tokens": 4096,
         "max_input_tokens": 128000,

ragaai_catalyst/tracers/llamaindex_callback.py

@@ -58,7 +58,7 @@ class LlamaIndexTracer:
             ) -> None:
                 trace = {
                     "event_type": event_type,
-                    "timestamp": datetime.now().isoformat(),
+                    "timestamp": datetime.now().astimezone().isoformat(),
                     "payload": payload,
                     "status": "started",
                     "event_id": event_id,
@@ -82,7 +82,7 @@ class LlamaIndexTracer:
             ) -> None:
                 trace = {
                     "event_type": event_type,
-                    "timestamp": datetime.now().isoformat(),
+                    "timestamp": datetime.now().astimezone().isoformat(),
                     "payload": payload,
                     "status": "completed",
                     "event_id": event_id,
@@ -220,7 +220,7 @@ class LlamaIndexTracer:
         user_detail["trace_id"] = self._generate_trace_id()
         metadata = user_detail["metadata"]
         metadata["log_source"] = "llamaindex_tracer"
-        metadata["recorded_on"] = datetime.utcnow().isoformat().replace('T', ' ')
+        metadata["recorded_on"] = datetime.now().astimezone().replace('T', ' ')
         user_detail["metadata"] = metadata
         return user_detail
     

ragaai_catalyst/tracers/tracer.py

@@ -116,7 +116,7 @@ class Tracer(AgenticTracing):
         self.base_url = f"{RagaAICatalyst.BASE_URL}"
         self.timeout = 30
         self.num_projects = 100
-        self.start_time = datetime.datetime.now(datetime.timezone.utc)
+        self.start_time = datetime.datetime.now().astimezone().isoformat()
 
         if update_llm_cost:
             # First update the model costs file from GitHub

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

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

@@ -12,13 +12,13 @@ ragaai_catalyst/ragaai_catalyst.py,sha256=FdqMzwuQLqS2-3JJDsTQ8uh2itllOxfPrRUjb8
 ragaai_catalyst/synthetic_data_generation.py,sha256=uDV9tNwto2xSkWg5XHXUvjErW-4P34CTrxaJpRfezyA,19250
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/tracers/__init__.py,sha256=yxepo7iVjTNI_wFdk3Z6Ghu64SazVyszCPEHYrX5WQk,50
-ragaai_catalyst/tracers/llamaindex_callback.py,sha256=vPE7MieKjfwLrLUnnPs20Df0xNYqoCCj-Mt2NbiuiKU,14023
-ragaai_catalyst/tracers/tracer.py,sha256=Q-C5lMXh6p5bqnDEQaqoK6C9H-V1odSi_YrYYT0jeW0,15581
+ragaai_catalyst/tracers/llamaindex_callback.py,sha256=HGs0TgtSgn8xXL8CSgdL9ymvYQGxZCNmPY5tRTw-I4s,14047
+ragaai_catalyst/tracers/tracer.py,sha256=4RaLmtAvVA-KT-O4Oi5T-VLVFmKqhbWjHYNxuQ6Fktk,15585
 ragaai_catalyst/tracers/upload_traces.py,sha256=hs0PEmit3n3_uUqrdbwcBdyK5Nbkik3JQVwJMEwYTd4,4796
 ragaai_catalyst/tracers/agentic_tracing/README.md,sha256=X4QwLb7-Jg7GQMIXj-SerZIgDETfw-7VgYlczOR8ZeQ,4508
 ragaai_catalyst/tracers/agentic_tracing/__init__.py,sha256=yf6SKvOPSpH-9LiKaoLKXwqj5sez8F_5wkOb91yp0oE,260
 ragaai_catalyst/tracers/agentic_tracing/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/data/data_structure.py,sha256=yRIWO4Lg2dUQy2IRjNxfrItj8haKVeULYAyEibinji0,9233
+ragaai_catalyst/tracers/agentic_tracing/data/data_structure.py,sha256=nFnwqL1-Uznwndi2ugDnhziUbIASlcBYnM6Dyq7pPt8,9243
 ragaai_catalyst/tracers/agentic_tracing/tests/FinancialAnalysisSystem.ipynb,sha256=0qZxjWqYCTAVvdo3Tsp544D8Am48wfeMQ9RKpKgAL8g,34291
 ragaai_catalyst/tracers/agentic_tracing/tests/GameActivityEventPlanner.ipynb,sha256=QCMFJYbGX0fd9eMW4PqyQLZjyWuTXo7n1nqO_hMLf0s,4225
 ragaai_catalyst/tracers/agentic_tracing/tests/TravelPlanner.ipynb,sha256=fU3inXoemJbdTkGAQl_N1UwVEZ10LrKv4gCEpbQ4ISg,43481
@@ -26,14 +26,15 @@ 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=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=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/agent_tracer.py,sha256=IpcSJG8lxghWmkIpcOvQ_4G0OqK79-Y2_75CEa5dHtg,25027
+ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=xNiWN0xJDm0GbnB0jo8xFnt-BO0-x4CmaHInsWxHREA,31266
+ragaai_catalyst/tracers/agentic_tracing/tracers/custom_tracer.py,sha256=uay8lU7T-CKsVu8KvWX31qfMqufK9S3Ive7XKo2Ksmk,12252
+ragaai_catalyst/tracers/agentic_tracing/tracers/langgraph_tracer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=NzMLIf21foZplOBx2zq32sUHV8geBNQmI2B-sIoSpPI,28810
+ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=5zUfKG87-sJPvDY56uueEL4J8Qk797vl0WDVbqhbx7Y,15117
+ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=CviGiAg0W0krJxORMBDTosQytIoJDQ5RwU6xt_U_mOg,10408
 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/tracers/user_interaction_tracer.py,sha256=bhSUhNQCuJXKjgJAXhjKEYjnHMpYN90FSZdR84fNIKU,4614
 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
 ragaai_catalyst/tracers/agentic_tracing/upload/upload_code.py,sha256=HgpMgI-JTWZrizcM7GGUIaAgaZF4aRT3D0dJXVEkblY,4271
@@ -45,7 +46,7 @@ ragaai_catalyst/tracers/agentic_tracing/utils/file_name_tracker.py,sha256=515NND
 ragaai_catalyst/tracers/agentic_tracing/utils/generic.py,sha256=WwXT01xmp8MSr7KinuDCSK9a1ifpLcT7ajFkvYviG_A,1190
 ragaai_catalyst/tracers/agentic_tracing/utils/get_user_trace_metrics.py,sha256=vPZ4dn4EHFW0kqd1GyRpsYXbfrRrd0DXCmh-pzsDBNE,1109
 ragaai_catalyst/tracers/agentic_tracing/utils/llm_utils.py,sha256=wlXCuaRe81s-7FWdJ_MquXFGRZZfNrZxLIIxl-Ohbqk,15541
-ragaai_catalyst/tracers/agentic_tracing/utils/model_costs.json,sha256=GTeYvV1eclT9ZeJV9KqZRKev7_P0DeIP9ji8KfbCD68,293328
+ragaai_catalyst/tracers/agentic_tracing/utils/model_costs.json,sha256=weznWpRd6N5X-O8lkzVA6nd5LSYJKa62s3GZL51NQpU,293833
 ragaai_catalyst/tracers/agentic_tracing/utils/span_attributes.py,sha256=MqeRNGxzeuh9qTK0NbYMftl9V9Z0V7gMgBoHkrXP56k,1592
 ragaai_catalyst/tracers/agentic_tracing/utils/trace_utils.py,sha256=RciiDdo2riibEoM8X0FKHaXi78y3bWwNkV8U0leqigk,3508
 ragaai_catalyst/tracers/agentic_tracing/utils/unique_decorator.py,sha256=DQHjcEuqEKsNSWaNs7SoOaq50yK4Jsl966S7mBnV-zA,5723
@@ -59,7 +60,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.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,,
+ragaai_catalyst-2.1.4b4.dist-info/METADATA,sha256=4xoAwoSRRw5GD6yEeygoXU49jMKM4RXa1_MWLV9EblA,12770
+ragaai_catalyst-2.1.4b4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.4b4.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.4b4.dist-info/RECORD,,