ragaai-catalyst 2.1.5b12__py3-none-any.whl → 2.1.5b14__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.
@@ -177,7 +177,7 @@ class LangchainTracer(BaseCallbackHandler):
177
177
  # Store model name if available
178
178
  if component_name in ["OpenAI", "ChatOpenAI_LangchainOpenAI", "ChatOpenAI_ChatModels",
179
179
  "ChatVertexAI", "VertexAI", "ChatGoogleGenerativeAI", "ChatAnthropic",
180
- "ChatLiteLLM", "ChatBedrock", "AzureChatOpenAI"]:
180
+ "ChatLiteLLM", "ChatBedrock", "AzureChatOpenAI", "ChatAnthropicVertex"]:
181
181
  instance = args[0] if args else None
182
182
  model_name = kwargs.get('model_name') or kwargs.get('model') or kwargs.get('model_id')
183
183
 
@@ -237,6 +237,12 @@ class LangchainTracer(BaseCallbackHandler):
237
237
  components_to_patch["VertexAI"] = (VertexAI, "__init__")
238
238
  except ImportError:
239
239
  logger.debug("VertexAI not available for patching")
240
+
241
+ try:
242
+ from langchain_google_vertexai.model_garden import ChatAnthropicVertex
243
+ components_to_patch["ChatAnthropicVertex"] = (ChatAnthropicVertex, "__init__")
244
+ except ImportError:
245
+ logger.debug("ChatAnthropicVertex not available for patching")
240
246
 
241
247
  try:
242
248
  from langchain_google_genai import ChatGoogleGenerativeAI
@@ -327,6 +333,15 @@ class LangchainTracer(BaseCallbackHandler):
327
333
  elif name == "ChatAnthropic":
328
334
  from langchain_anthropic import ChatAnthropic
329
335
  imported_components[name] = ChatAnthropic
336
+ elif name == "ChatBedrock":
337
+ from langchain_aws import ChatBedrock
338
+ imported_components[name] = ChatBedrock
339
+ elif name == "AzureChatOpenAI":
340
+ from langchain_openai import AzureChatOpenAI
341
+ imported_components[name] = AzureChatOpenAI
342
+ elif name == "ChatAnthropicVertex":
343
+ from langchain_google_vertexai.model_garden import ChatAnthropicVertex
344
+ imported_components[name] = ChatAnthropicVertex
330
345
  elif name == "ChatLiteLLM":
331
346
  from langchain_community.chat_models import ChatLiteLLM
332
347
  imported_components[name] = ChatLiteLLM
@@ -15,7 +15,7 @@ from ragaai_catalyst.tracers.utils.langchain_tracer_extraction_logic import lang
15
15
  from ragaai_catalyst.tracers.upload_traces import UploadTraces
16
16
  import tempfile
17
17
  import json
18
-
18
+ import numpy as np
19
19
  from opentelemetry.sdk import trace as trace_sdk
20
20
  from opentelemetry.sdk.trace.export import SimpleSpanProcessor
21
21
  from ragaai_catalyst.tracers.exporters.file_span_exporter import FileSpanExporter
@@ -275,15 +275,11 @@ class Tracer(AgenticTracing):
275
275
  # self._upload_task = self._run_async(self._upload_traces())
276
276
  # self.is_active = False
277
277
  # self.dataset_name = None
278
-
279
- # filename = f"langchain_callback_traces.json"
280
- # filepath = os.path.join(tempfile.gettempdir(), filename)
281
278
 
282
279
  user_detail = self._pass_user_data()
283
280
  data, additional_metadata = self.langchain_tracer.stop()
284
281
 
285
282
  # Add cost if possible
286
- # import pdb; pdb.set_trace()
287
283
  if additional_metadata['model_name']:
288
284
  try:
289
285
  model_cost_data = self.model_cost_dict[additional_metadata['model_name']]
@@ -295,15 +291,13 @@ class Tracer(AgenticTracing):
295
291
  except Exception as e:
296
292
  logger.warning(f"Error adding cost: {e}")
297
293
 
298
- # with open(filepath, 'r') as f:
299
- # data = json.load(f)
300
- additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
294
+ additional_metadata["total_tokens"] = float(additional_metadata["tokens"]["total"])
301
295
  del additional_metadata["tokens"]
302
296
  if "cost" in additional_metadata:
303
- additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
297
+ additional_metadata["total_cost"] = float(additional_metadata["cost"]["total_cost"])
304
298
  del additional_metadata["cost"]
305
299
  else:
306
- additional_metadata["total_cost"] = 0.0
300
+ additional_metadata["total_cost"] = float(0.0)
307
301
 
308
302
  combined_metadata = user_detail['trace_user_detail']['metadata'].copy()
309
303
  combined_metadata.update(additional_metadata)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: ragaai_catalyst
3
- Version: 2.1.5b12
3
+ Version: 2.1.5b14
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
@@ -13,9 +13,9 @@ ragaai_catalyst/synthetic_data_generation.py,sha256=uDV9tNwto2xSkWg5XHXUvjErW-4P
13
13
  ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
14
14
  ragaai_catalyst/tracers/__init__.py,sha256=LfgTes-nHpazssbGKnn8kyLZNr49kIPrlkrqqoTFTfc,301
15
15
  ragaai_catalyst/tracers/distributed.py,sha256=AIRvS5Ur4jbFDXsUkYuCTmtGoHHx3LOG4n5tWOh610U,10330
16
- ragaai_catalyst/tracers/langchain_callback.py,sha256=ubdkiaE7iouE5vUGmTcRQyfVVzRwNOiwr8qBEVHzWUY,29672
16
+ ragaai_catalyst/tracers/langchain_callback.py,sha256=v004nQuim4qgb3k4iWyfhvSsMPoI3ZSS26hzbBQT8CI,30572
17
17
  ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
18
- ragaai_catalyst/tracers/tracer.py,sha256=Yq2HhgT4785t9573kksJ7ngM3qCLPgZbZ0IpgOHdTTo,19223
18
+ ragaai_catalyst/tracers/tracer.py,sha256=dn2Abz9Or7F2lp3hgUf7AUKzp2Dq-WDtg-_dW6gf1e0,19005
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=qK67fdUBz5Xr99ajqXbYf1ueKS1V3a3_XR0zCcN4iGI,3061
67
67
  ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
68
- ragaai_catalyst-2.1.5b12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
69
- ragaai_catalyst-2.1.5b12.dist-info/METADATA,sha256=yKKeKwBa1Miwi3DdRdMDxgqELklmj6zMHIH8qrz19uA,12765
70
- ragaai_catalyst-2.1.5b12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
71
- ragaai_catalyst-2.1.5b12.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
72
- ragaai_catalyst-2.1.5b12.dist-info/RECORD,,
68
+ ragaai_catalyst-2.1.5b14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
69
+ ragaai_catalyst-2.1.5b14.dist-info/METADATA,sha256=xCMz9BuQQxwXe_FO4ZWN2_q_WYVwk_opJYIQN6GMt6g,12765
70
+ ragaai_catalyst-2.1.5b14.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
71
+ ragaai_catalyst-2.1.5b14.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
72
+ ragaai_catalyst-2.1.5b14.dist-info/RECORD,,