ragaai-catalyst 2.1.5b5__py3-none-any.whl → 2.1.5b7__py3-none-any.whl

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.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/METADATA

@@ -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.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/RECORD

@@ -15,7 +15,7 @@ ragaai_catalyst/tracers/__init__.py,sha256=LfgTes-nHpazssbGKnn8kyLZNr49kIPrlkrqq
 ragaai_catalyst/tracers/distributed.py,sha256=AIRvS5Ur4jbFDXsUkYuCTmtGoHHx3LOG4n5tWOh610U,10330
 ragaai_catalyst/tracers/langchain_callback.py,sha256=LvMBhgvAX8ftyBQ9Naeui46EoDa2nHQZq48Ra6nL-Qg,21991
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
-ragaai_catalyst/tracers/tracer.py,sha256=spAiNQ85vpMntckHIgn0w7h33gxqWg5WoxOEcvT-e7U,19179
+ragaai_catalyst/tracers/tracer.py,sha256=bLgO3lQmoumo-JtqZFi4DUqhu9itM5GaLFNY7hmsI1g,20267
 ragaai_catalyst/tracers/upload_traces.py,sha256=2TWdRTN6FMaX-dqDv8BJWQS0xrCGYKkXEYOi2kK3Z3Y,5487
 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
@@ -65,8 +65,8 @@ ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpa
 ragaai_catalyst/tracers/utils/convert_langchain_callbacks_output.py,sha256=ofrNrxf2b1hpjDh_zeaxiYq86azn1MF3kW8-ViYPEg0,1641
 ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=cghjCuUe8w-2MZdh9xgtRGe3y219u26GGzpnuY4Wt6Q,3047
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.5b5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b5.dist-info/METADATA,sha256=ShN_vaYhKMLho8Q49eBzlWYUNEQD3U6w9qgRtiMYsiM,12764
-ragaai_catalyst-2.1.5b5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.5b5.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b5.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b7.dist-info/METADATA,sha256=cVHEchxtHjkR_9AKHeDlqTpZqYNMSoF5gKR37D9icAY,12764
+ragaai_catalyst-2.1.5b7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.5b7.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b7.dist-info/RECORD,,