ragaai-catalyst 2.1.5b5__tar.gz → 2.1.5b7__tar.gz

{ragaai_catalyst-2.1.5b5/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.5b7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b5
+Version: 2.1.5b7
 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.5b5 → ragaai_catalyst-2.1.5b7}/pyproject.toml

@@ -9,7 +9,7 @@ readme = "README.md"
 requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
 
-version = "2.1.5.beta.5"
+version = "2.1.5.beta.7"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.5b5 → ragaai_catalyst-2.1.5b7}/ragaai_catalyst/tracers/tracer.py

@@ -297,32 +297,50 @@ class Tracer(AgenticTracing):
             else:
                 logger.debug("Model name not available in additional_metadata, skipping cost calculation")
             
-            additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
-            additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
+            # Safely get total tokens and cost
+            if 'tokens' in additional_metadata and 'total' in additional_metadata['tokens']:
+                additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
+            else:
+                additional_metadata["total_tokens"] = 0
+                logger.warning("Total tokens information not available")
+
+            if 'cost' in additional_metadata and 'total_cost' in additional_metadata['cost']:
+                additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
+            else:
+                additional_metadata["total_cost"] = 0.0
+                logger.warning("Total cost information not available")
 
-            del additional_metadata["tokens"]
-            del additional_metadata["cost"]
+            # Safely remove tokens and cost dictionaries if they exist
+            additional_metadata.pop("tokens", None)
+            additional_metadata.pop("cost", None)
             
-            combined_metadata = user_detail['trace_user_detail']['metadata'].copy()
-            combined_metadata.update(additional_metadata)
-            combined_metadata
+            # Safely merge metadata
+            combined_metadata = {}
+            if user_detail.get('trace_user_detail', {}).get('metadata'):
+                combined_metadata.update(user_detail['trace_user_detail']['metadata'])
+            if additional_metadata:
+                combined_metadata.update(additional_metadata)
 
             langchain_traces = langchain_tracer_extraction(data)
             final_result = convert_langchain_callbacks_output(langchain_traces)
-            final_result[0]['project_name'] = user_detail['project_name']
-            final_result[0]['trace_id'] = str(uuid.uuid4())
-            final_result[0]['session_id'] = None
-            final_result[0]['metadata'] = combined_metadata
-            final_result[0]['pipeline'] = user_detail['trace_user_detail']['pipeline']
-
-            filepath_3 = os.path.join(os.getcwd(), "final_result.json")
-            with open(filepath_3, 'w') as f:
-                json.dump(final_result, f, indent=2)
             
-            
-            print(filepath_3)
+            # Safely set required fields in final_result
+            if final_result and isinstance(final_result, list) and len(final_result) > 0:
+                final_result[0]['project_name'] = user_detail.get('project_name', '')
+                final_result[0]['trace_id'] = str(uuid.uuid4())
+                final_result[0]['session_id'] = None
+                final_result[0]['metadata'] = combined_metadata
+                final_result[0]['pipeline'] = user_detail.get('trace_user_detail', {}).get('pipeline')
+
+                filepath_3 = os.path.join(os.getcwd(), "final_result.json")
+                with open(filepath_3, 'w') as f:
+                    json.dump(final_result, f, indent=2)
+                
+                print(filepath_3)
+            else:
+                logger.warning("No valid langchain traces found in final_result")
 
-            additional_metadata_keys = additional_metadata.keys() if additional_metadata else None
+            additional_metadata_keys = list(additional_metadata.keys()) if additional_metadata else None
 
             UploadTraces(json_file_path=filepath_3,
                          project_name=self.project_name,

{ragaai_catalyst-2.1.5b5 → ragaai_catalyst-2.1.5b7/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b5
+Version: 2.1.5b7
 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