ragaai-catalyst 2.1.5b39__tar.gz → 2.1.5b40__tar.gz

{ragaai_catalyst-2.1.5b39/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.5b40}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b39
+Version: 2.1.5b40
 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>, Tushar Kumar <tushar.kumar@raga.ai>
-Requires-Python: <3.13,>=3.10
+Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b40}/pyproject.toml

@@ -6,9 +6,9 @@ build-backend = "setuptools.build_meta"
 name = "ragaai_catalyst"
 description = "RAGA AI CATALYST"
 readme = "README.md"
-requires-python = ">=3.10,<3.13"
+requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
-version = "2.1.5.b39"
+version = "2.1.5.b40"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b40}/ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py

@@ -378,7 +378,7 @@ class AgenticTracing(
             super().add_component(component)
 
         # Handle error case
-        if is_error:
+        if is_error and not self.current_agent_id.get():
             self.stop()
 
     def __enter__(self):

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b40}/ragaai_catalyst/tracers/exporters/dynamic_trace_exporter.py

@@ -42,7 +42,7 @@ class DynamicTraceExporter(SpanExporter):
         self._project_id = project_id
         self._dataset_name = dataset_name
         self._user_details = user_details
-        self._base_url = base_url,
+        self._base_url = base_url
         self._custom_model_cost = custom_model_cost
 
     
@@ -57,22 +57,37 @@ class DynamicTraceExporter(SpanExporter):
         Returns:
             SpanExportResult: Result of the export operation
         """
-        # Update the exporter's properties
-        self._update_exporter_properties()
-        
-        # Forward the call to the underlying exporter
-        return self._exporter.export(spans)
-    
+        try:
+            # Update the exporter's properties
+            self._update_exporter_properties()
+        except Exception as e:
+            raise Exception(f"Error updating exporter properties: {e}")
+
+        try:
+            # Forward the call to the underlying exporter
+            result = self._exporter.export(spans)
+            return result
+        except Exception as e:
+            raise Exception(f"Error exporting trace: {e}")
+            
+
+
     def shutdown(self):
         """
         Shutdown the exporter by forwarding to the underlying exporter.
         Before shutting down, update the exporter's properties with the current values.
         """
-        # Update the exporter's properties
-        self._update_exporter_properties()
-        
-        # Forward the call to the underlying exporter
-        return self._exporter.shutdown()
+        try:
+            # Update the exporter's properties
+            self._update_exporter_properties()
+        except Exception as e:
+            raise Exception(f"Error updating exporter properties: {e}")
+
+        try:
+            # Forward the call to the underlying exporter
+            return self._exporter.shutdown()
+        except Exception as e:
+            raise Exception(f"Error shutting down exporter: {e}")
     
     def _update_exporter_properties(self):
         """

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b40}/ragaai_catalyst/tracers/exporters/ragaai_trace_exporter.py

@@ -35,6 +35,8 @@ class RAGATraceExporter(SpanExporter):
         for span in spans:
             span_json = json.loads(span.to_json())
             trace_id = span_json.get("context").get("trace_id")
+            if trace_id is None:
+                raise Exception("Trace ID is None")
 
             if trace_id not in self.trace_spans:
                 self.trace_spans[trace_id] = list()
@@ -43,8 +45,14 @@ class RAGATraceExporter(SpanExporter):
 
             if span_json["parent_id"] is None:
                 trace = self.trace_spans[trace_id]
-                self.process_complete_trace(trace, trace_id)
-                del self.trace_spans[trace_id]
+                try:
+                    self.process_complete_trace(trace, trace_id)
+                except Exception as e:
+                    raise Exception(f"Error processing complete trace: {e}")
+                try:
+                    del self.trace_spans[trace_id]
+                except Exception as e:
+                    raise Exception(f"Error deleting trace: {e}")
 
         return SpanExportResult.SUCCESS
 

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b40}/ragaai_catalyst/tracers/utils/trace_json_converter.py

@@ -223,7 +223,10 @@ def convert_json_format(input_trace, custom_model_cost):
     }
     final_trace["replays"]={"source":None}
     final_trace["data"]=[{}]
-    final_trace["data"][0]["spans"] = get_spans(input_trace, custom_model_cost)
+    try:
+        final_trace["data"][0]["spans"] = get_spans(input_trace, custom_model_cost)
+    except Exception as e:
+        raise Exception(f"Error in get_spans function: {e}")
     final_trace["network_calls"] =[]
     final_trace["interactions"] = []
 
@@ -231,15 +234,15 @@ def convert_json_format(input_trace, custom_model_cost):
         if itr["type"]=="llm":
             if "tokens" in itr["info"]:
                 if "prompt_tokens" in itr["info"]["tokens"]:
-                    final_trace["metadata"]["tokens"]["prompt_tokens"] += itr["info"]["tokens"]['prompt_tokens']
-                    final_trace["metadata"]["cost"]["input_cost"] += itr["info"]["cost"]['input_cost'] 
+                    final_trace["metadata"]["tokens"]["prompt_tokens"] += itr["info"]["tokens"].get('prompt_tokens', 0.0)
+                    final_trace["metadata"]["cost"]["input_cost"] += itr["info"]["cost"].get('input_cost', 0.0) 
                 if "completion_tokens" in itr["info"]["tokens"]:
-                    final_trace["metadata"]["tokens"]["completion_tokens"] += itr["info"]["tokens"]['completion_tokens']
-                    final_trace["metadata"]["cost"]["output_cost"] += itr["info"]["cost"]['output_cost'] 
+                    final_trace["metadata"]["tokens"]["completion_tokens"] += itr["info"]["tokens"].get('completion_tokens', 0.0)
+                    final_trace["metadata"]["cost"]["output_cost"] += itr["info"]["cost"].get('output_cost', 0.0) 
             if "tokens" in itr["info"]:
                 if "total_tokens" in itr["info"]["tokens"]:
-                    final_trace["metadata"]["tokens"]["total_tokens"] += itr["info"]["tokens"]['total_tokens']
-                    final_trace["metadata"]["cost"]["total_cost"] += itr["info"]["cost"]['total_cost'] 
+                    final_trace["metadata"]["tokens"]["total_tokens"] += itr["info"]["tokens"].get('total_tokens', 0.0)
+                    final_trace["metadata"]["cost"]["total_cost"] += itr["info"]["cost"].get('total_cost', 0.0) 
 
     # get the total tokens, cost
     final_trace["metadata"]["total_cost"] = final_trace["metadata"]["cost"]["total_cost"] 

{ragaai_catalyst-2.1.5b39 → ragaai_catalyst-2.1.5b40/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b39
+Version: 2.1.5b40
 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>, Tushar Kumar <tushar.kumar@raga.ai>
-Requires-Python: <3.13,>=3.10
+Requires-Python: <3.13,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.10.2