ragaai-catalyst 2.1.5b12__py3-none-any.whl → 2.1.5b14__py3-none-any.whl

ragaai_catalyst/tracers/langchain_callback.py

@@ -177,7 +177,7 @@ class LangchainTracer(BaseCallbackHandler):
                 # Store model name if available
                 if component_name in ["OpenAI", "ChatOpenAI_LangchainOpenAI", "ChatOpenAI_ChatModels",
                                     "ChatVertexAI", "VertexAI", "ChatGoogleGenerativeAI", "ChatAnthropic", 
-                                    "ChatLiteLLM", "ChatBedrock", "AzureChatOpenAI"]:
+                                    "ChatLiteLLM", "ChatBedrock", "AzureChatOpenAI", "ChatAnthropicVertex"]:
                     instance = args[0] if args else None
                     model_name = kwargs.get('model_name') or kwargs.get('model') or kwargs.get('model_id')
 
@@ -237,6 +237,12 @@ class LangchainTracer(BaseCallbackHandler):
             components_to_patch["VertexAI"] = (VertexAI, "__init__")
         except ImportError:
             logger.debug("VertexAI not available for patching")
+
+        try:
+            from langchain_google_vertexai.model_garden import ChatAnthropicVertex
+            components_to_patch["ChatAnthropicVertex"] = (ChatAnthropicVertex, "__init__")
+        except ImportError:
+            logger.debug("ChatAnthropicVertex not available for patching")
             
         try:
             from langchain_google_genai import ChatGoogleGenerativeAI
@@ -327,6 +333,15 @@ class LangchainTracer(BaseCallbackHandler):
                     elif name == "ChatAnthropic":
                         from langchain_anthropic import ChatAnthropic
                         imported_components[name] = ChatAnthropic
+                    elif name == "ChatBedrock":
+                        from langchain_aws import ChatBedrock
+                        imported_components[name] = ChatBedrock
+                    elif name == "AzureChatOpenAI":
+                        from langchain_openai import AzureChatOpenAI
+                        imported_components[name] = AzureChatOpenAI
+                    elif name == "ChatAnthropicVertex":
+                        from langchain_google_vertexai.model_garden import ChatAnthropicVertex
+                        imported_components[name] = ChatAnthropicVertex
                     elif name == "ChatLiteLLM":
                         from langchain_community.chat_models import ChatLiteLLM
                         imported_components[name] = ChatLiteLLM

ragaai_catalyst/tracers/tracer.py

@@ -15,7 +15,7 @@ from ragaai_catalyst.tracers.utils.langchain_tracer_extraction_logic import lang
 from ragaai_catalyst.tracers.upload_traces import UploadTraces
 import tempfile
 import json
-
+import numpy as np
 from opentelemetry.sdk import trace as trace_sdk
 from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 from ragaai_catalyst.tracers.exporters.file_span_exporter import FileSpanExporter
@@ -275,15 +275,11 @@ class Tracer(AgenticTracing):
             # self._upload_task = self._run_async(self._upload_traces())
             # self.is_active = False
             # self.dataset_name = None
-
-            # filename = f"langchain_callback_traces.json"
-            # filepath = os.path.join(tempfile.gettempdir(), filename) 
             
             user_detail = self._pass_user_data()
             data, additional_metadata = self.langchain_tracer.stop()
 
             # Add cost if possible
-            # import pdb; pdb.set_trace()
             if additional_metadata['model_name']:
                 try:
                     model_cost_data = self.model_cost_dict[additional_metadata['model_name']]
@@ -295,15 +291,13 @@ class Tracer(AgenticTracing):
                 except Exception as e:
                     logger.warning(f"Error adding cost: {e}")
 
-            # with open(filepath, 'r') as f:
-            #     data = json.load(f)
-            additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
+            additional_metadata["total_tokens"] = float(additional_metadata["tokens"]["total"])
             del additional_metadata["tokens"]
             if "cost" in additional_metadata:
-                additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
+                additional_metadata["total_cost"] = float(additional_metadata["cost"]["total_cost"])
                 del additional_metadata["cost"]
             else:
-                additional_metadata["total_cost"] = 0.0
+                additional_metadata["total_cost"] = float(0.0)
             
             combined_metadata = user_detail['trace_user_detail']['metadata'].copy()
             combined_metadata.update(additional_metadata)

{ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b12
+Version: 2.1.5b14
 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.5b12.dist-info → ragaai_catalyst-2.1.5b14.dist-info}/RECORD

@@ -13,9 +13,9 @@ ragaai_catalyst/synthetic_data_generation.py,sha256=uDV9tNwto2xSkWg5XHXUvjErW-4P
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/tracers/__init__.py,sha256=LfgTes-nHpazssbGKnn8kyLZNr49kIPrlkrqqoTFTfc,301
 ragaai_catalyst/tracers/distributed.py,sha256=AIRvS5Ur4jbFDXsUkYuCTmtGoHHx3LOG4n5tWOh610U,10330
-ragaai_catalyst/tracers/langchain_callback.py,sha256=ubdkiaE7iouE5vUGmTcRQyfVVzRwNOiwr8qBEVHzWUY,29672
+ragaai_catalyst/tracers/langchain_callback.py,sha256=v004nQuim4qgb3k4iWyfhvSsMPoI3ZSS26hzbBQT8CI,30572
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
-ragaai_catalyst/tracers/tracer.py,sha256=Yq2HhgT4785t9573kksJ7ngM3qCLPgZbZ0IpgOHdTTo,19223
+ragaai_catalyst/tracers/tracer.py,sha256=dn2Abz9Or7F2lp3hgUf7AUKzp2Dq-WDtg-_dW6gf1e0,19005
 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=qK67fdUBz5Xr99ajqXbYf1ueKS1V3a3_XR0zCcN4iGI,3061
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.5b12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b12.dist-info/METADATA,sha256=yKKeKwBa1Miwi3DdRdMDxgqELklmj6zMHIH8qrz19uA,12765
-ragaai_catalyst-2.1.5b12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.5b12.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b12.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b14.dist-info/METADATA,sha256=xCMz9BuQQxwXe_FO4ZWN2_q_WYVwk_opJYIQN6GMt6g,12765
+ragaai_catalyst-2.1.5b14.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.5b14.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b14.dist-info/RECORD,,