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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -297,32 +297,50 @@ class Tracer(AgenticTracing):
297
297
  else:
298
298
  logger.debug("Model name not available in additional_metadata, skipping cost calculation")
299
299
 
300
- additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
301
- additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
300
+ # Safely get total tokens and cost
301
+ if 'tokens' in additional_metadata and 'total' in additional_metadata['tokens']:
302
+ additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
303
+ else:
304
+ additional_metadata["total_tokens"] = 0
305
+ logger.warning("Total tokens information not available")
306
+
307
+ if 'cost' in additional_metadata and 'total_cost' in additional_metadata['cost']:
308
+ additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
309
+ else:
310
+ additional_metadata["total_cost"] = 0.0
311
+ logger.warning("Total cost information not available")
302
312
 
303
- del additional_metadata["tokens"]
304
- del additional_metadata["cost"]
313
+ # Safely remove tokens and cost dictionaries if they exist
314
+ additional_metadata.pop("tokens", None)
315
+ additional_metadata.pop("cost", None)
305
316
 
306
- combined_metadata = user_detail['trace_user_detail']['metadata'].copy()
307
- combined_metadata.update(additional_metadata)
308
- combined_metadata
317
+ # Safely merge metadata
318
+ combined_metadata = {}
319
+ if user_detail.get('trace_user_detail', {}).get('metadata'):
320
+ combined_metadata.update(user_detail['trace_user_detail']['metadata'])
321
+ if additional_metadata:
322
+ combined_metadata.update(additional_metadata)
309
323
 
310
324
  langchain_traces = langchain_tracer_extraction(data)
311
325
  final_result = convert_langchain_callbacks_output(langchain_traces)
312
- final_result[0]['project_name'] = user_detail['project_name']
313
- final_result[0]['trace_id'] = str(uuid.uuid4())
314
- final_result[0]['session_id'] = None
315
- final_result[0]['metadata'] = combined_metadata
316
- final_result[0]['pipeline'] = user_detail['trace_user_detail']['pipeline']
317
-
318
- filepath_3 = os.path.join(os.getcwd(), "final_result.json")
319
- with open(filepath_3, 'w') as f:
320
- json.dump(final_result, f, indent=2)
321
326
 
322
-
323
- print(filepath_3)
327
+ # Safely set required fields in final_result
328
+ if final_result and isinstance(final_result, list) and len(final_result) > 0:
329
+ final_result[0]['project_name'] = user_detail.get('project_name', '')
330
+ final_result[0]['trace_id'] = str(uuid.uuid4())
331
+ final_result[0]['session_id'] = None
332
+ final_result[0]['metadata'] = combined_metadata
333
+ final_result[0]['pipeline'] = user_detail.get('trace_user_detail', {}).get('pipeline')
334
+
335
+ filepath_3 = os.path.join(os.getcwd(), "final_result.json")
336
+ with open(filepath_3, 'w') as f:
337
+ json.dump(final_result, f, indent=2)
338
+
339
+ print(filepath_3)
340
+ else:
341
+ logger.warning("No valid langchain traces found in final_result")
324
342
 
325
- additional_metadata_keys = additional_metadata.keys() if additional_metadata else None
343
+ additional_metadata_keys = list(additional_metadata.keys()) if additional_metadata else None
326
344
 
327
345
  UploadTraces(json_file_path=filepath_3,
328
346
  project_name=self.project_name,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: ragaai_catalyst
3
- Version: 2.1.5b5
3
+ Version: 2.1.5b7
4
4
  Summary: RAGA AI CATALYST
5
5
  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>
6
6
  Requires-Python: <3.13,>=3.9
@@ -15,7 +15,7 @@ ragaai_catalyst/tracers/__init__.py,sha256=LfgTes-nHpazssbGKnn8kyLZNr49kIPrlkrqq
15
15
  ragaai_catalyst/tracers/distributed.py,sha256=AIRvS5Ur4jbFDXsUkYuCTmtGoHHx3LOG4n5tWOh610U,10330
16
16
  ragaai_catalyst/tracers/langchain_callback.py,sha256=LvMBhgvAX8ftyBQ9Naeui46EoDa2nHQZq48Ra6nL-Qg,21991
17
17
  ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
18
- ragaai_catalyst/tracers/tracer.py,sha256=spAiNQ85vpMntckHIgn0w7h33gxqWg5WoxOEcvT-e7U,19179
18
+ ragaai_catalyst/tracers/tracer.py,sha256=bLgO3lQmoumo-JtqZFi4DUqhu9itM5GaLFNY7hmsI1g,20267
19
19
  ragaai_catalyst/tracers/upload_traces.py,sha256=2TWdRTN6FMaX-dqDv8BJWQS0xrCGYKkXEYOi2kK3Z3Y,5487
20
20
  ragaai_catalyst/tracers/agentic_tracing/README.md,sha256=X4QwLb7-Jg7GQMIXj-SerZIgDETfw-7VgYlczOR8ZeQ,4508
21
21
  ragaai_catalyst/tracers/agentic_tracing/__init__.py,sha256=yf6SKvOPSpH-9LiKaoLKXwqj5sez8F_5wkOb91yp0oE,260
@@ -65,8 +65,8 @@ ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpa
65
65
  ragaai_catalyst/tracers/utils/convert_langchain_callbacks_output.py,sha256=ofrNrxf2b1hpjDh_zeaxiYq86azn1MF3kW8-ViYPEg0,1641
66
66
  ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=cghjCuUe8w-2MZdh9xgtRGe3y219u26GGzpnuY4Wt6Q,3047
67
67
  ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
68
- ragaai_catalyst-2.1.5b5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
69
- ragaai_catalyst-2.1.5b5.dist-info/METADATA,sha256=ShN_vaYhKMLho8Q49eBzlWYUNEQD3U6w9qgRtiMYsiM,12764
70
- ragaai_catalyst-2.1.5b5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
71
- ragaai_catalyst-2.1.5b5.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
72
- ragaai_catalyst-2.1.5b5.dist-info/RECORD,,
68
+ ragaai_catalyst-2.1.5b7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
69
+ ragaai_catalyst-2.1.5b7.dist-info/METADATA,sha256=cVHEchxtHjkR_9AKHeDlqTpZqYNMSoF5gKR37D9icAY,12764
70
+ ragaai_catalyst-2.1.5b7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
71
+ ragaai_catalyst-2.1.5b7.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
72
+ ragaai_catalyst-2.1.5b7.dist-info/RECORD,,