ragaai-catalyst 2.1.5b37__py3-none-any.whl → 2.1.5b39__py3-none-any.whl

ragaai_catalyst/tracers/utils/trace_json_converter.py

@@ -158,7 +158,7 @@ def get_spans(input_trace, custom_model_cost):
                     final_span["data"]["output"] = span["attributes"]["output.value"]
 
         final_span["info"]["cost"] = {}
-        final_span["info"]["token"] = {}
+        final_span["info"]["tokens"] = {}
 
         if "model" in final_span["info"]:
             model_name = final_span["info"]["model"] 
@@ -174,19 +174,19 @@ def get_spans(input_trace, custom_model_cost):
         if "resource" in span:
             final_span["info"].update(span["resource"])
         if "llm.token_count.prompt" in span['attributes']:
-            final_span["info"]["token"]["prompt_tokens"] = span['attributes']['llm.token_count.prompt']
+            final_span["info"]["tokens"]["prompt_tokens"] = span['attributes']['llm.token_count.prompt']
         if "llm.token_count.completion" in span['attributes']:
-            final_span["info"]["token"]["completion_tokens"] = span['attributes']['llm.token_count.completion']
+            final_span["info"]["tokens"]["completion_tokens"] = span['attributes']['llm.token_count.completion']
         if "llm.token_count.total" in span['attributes']:
-            final_span["info"]["token"]["total_tokens"] = span['attributes']['llm.token_count.total']
+            final_span["info"]["tokens"]["total_tokens"] = span['attributes']['llm.token_count.total']
         
         if "info" in final_span:
-            if "token" in final_span["info"]:
-                if "prompt_tokens" in final_span["info"]["token"]:
+            if "tokens" in final_span["info"]:
+                if "prompt_tokens" in final_span["info"]["tokens"]:
                     token_usage = {
-                        "prompt_tokens": final_span["info"]["token"]["prompt_tokens"],
-                        "completion_tokens": final_span["info"]["token"]["completion_tokens"],
-                        "total_tokens": final_span["info"]["token"]["total_tokens"]
+                        "prompt_tokens": final_span["info"]["tokens"]["prompt_tokens"],
+                        "completion_tokens": final_span["info"]["tokens"]["completion_tokens"],
+                        "total_tokens": final_span["info"]["tokens"]["total_tokens"]
                     }
                     final_span["info"]["cost"] = calculate_llm_cost(token_usage=token_usage, model_name=model_name, model_costs=model_costs, model_custom_cost=custom_model_cost) 
         data.append(final_span)
@@ -229,16 +229,16 @@ def convert_json_format(input_trace, custom_model_cost):
 
     for itr in final_trace["data"][0]["spans"]:
         if itr["type"]=="llm":
-            if "token" in itr["info"]:
-                if "prompt_tokens" in itr["info"]["token"]:
-                    final_trace["metadata"]["tokens"]["prompt_tokens"] += itr["info"]["token"]['prompt_tokens']
+            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'] 
-                if "completion_tokens" in itr["info"]["token"]:
-                    final_trace["metadata"]["tokens"]["completion_tokens"] += itr["info"]["token"]['completion_tokens']
+                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'] 
-            if "token" in itr["info"]:
-                if "total_tokens" in itr["info"]["token"]:
-                    final_trace["metadata"]["tokens"]["total_tokens"] += itr["info"]["token"]['total_tokens']
+            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'] 
 
     # get the total tokens, cost

{ragaai_catalyst-2.1.5b37.dist-info → ragaai_catalyst-2.1.5b39.dist-info}/METADATA

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

{ragaai_catalyst-2.1.5b37.dist-info → ragaai_catalyst-2.1.5b39.dist-info}/RECORD

@@ -88,10 +88,10 @@ ragaai_catalyst/tracers/utils/convert_llama_instru_callback.py,sha256=8qLo7x4Zsn
 ragaai_catalyst/tracers/utils/extraction_logic_llama_index.py,sha256=ZhPs0YhVtB82-Pq9o1BvCinKE_WPvVxPTEcZjlJbFYM,2371
 ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=XS2_x2qneqEx9oAighLg-LRiueWcESLwIC2r7eJT-Ww,3117
 ragaai_catalyst/tracers/utils/model_prices_and_context_window_backup.json,sha256=C3uwkibJ08C9sOX-54kulZYmJlIpZ-SQpfE6HNGrjbM,343502
-ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=5zeJRD3gY_arfXWGZR4cqAvRhfC_wQUabm_7peiNMdY,12268
+ragaai_catalyst/tracers/utils/trace_json_converter.py,sha256=9I3cidegQuedauvUF2NoZiNCPSjCaaskkaXxCcUJ9Po,12285
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.5b37.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b37.dist-info/METADATA,sha256=B7m2Fu7d9nCZWJmZlM8uXUB7mo3VahUb8s4QuwLbNqw,22060
-ragaai_catalyst-2.1.5b37.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-ragaai_catalyst-2.1.5b37.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b37.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b39.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b39.dist-info/METADATA,sha256=bIq4JlzWIo7te22u94rhmkGGxCtBNsFoGeiJv5IVMDo,22061
+ragaai_catalyst-2.1.5b39.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+ragaai_catalyst-2.1.5b39.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b39.dist-info/RECORD,,