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.
- ragaai_catalyst/tracers/tracer.py +37 -19
- {ragaai_catalyst-2.1.5b5.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/METADATA +1 -1
- {ragaai_catalyst-2.1.5b5.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/RECORD +6 -6
- {ragaai_catalyst-2.1.5b5.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/LICENSE +0 -0
- {ragaai_catalyst-2.1.5b5.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/WHEEL +0 -0
- {ragaai_catalyst-2.1.5b5.dist-info → ragaai_catalyst-2.1.5b7.dist-info}/top_level.txt +0 -0
@@ -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
|
-
|
301
|
-
additional_metadata
|
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
|
-
|
304
|
-
|
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
|
-
|
307
|
-
combined_metadata
|
308
|
-
|
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
|
-
|
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.
|
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=
|
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.
|
69
|
-
ragaai_catalyst-2.1.
|
70
|
-
ragaai_catalyst-2.1.
|
71
|
-
ragaai_catalyst-2.1.
|
72
|
-
ragaai_catalyst-2.1.
|
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,,
|
File without changes
|
File without changes
|
File without changes
|