ragaai-catalyst 2.1.5b12__py3-none-any.whl → 2.1.5b13__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/langchain_callback.py +16 -1
- ragaai_catalyst/tracers/tracer.py +4 -4
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b13.dist-info}/METADATA +1 -1
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b13.dist-info}/RECORD +7 -7
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b13.dist-info}/LICENSE +0 -0
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b13.dist-info}/WHEEL +0 -0
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b13.dist-info}/top_level.txt +0 -0
@@ -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
|
@@ -297,13 +297,13 @@ class Tracer(AgenticTracing):
|
|
297
297
|
|
298
298
|
# with open(filepath, 'r') as f:
|
299
299
|
# data = json.load(f)
|
300
|
-
additional_metadata["total_tokens"] = additional_metadata["tokens"]["total"]
|
300
|
+
additional_metadata["total_tokens"] = np.float32(additional_metadata["tokens"]["total"])
|
301
301
|
del additional_metadata["tokens"]
|
302
302
|
if "cost" in additional_metadata:
|
303
|
-
additional_metadata["total_cost"] = additional_metadata["cost"]["total_cost"]
|
303
|
+
additional_metadata["total_cost"] = np.float32(additional_metadata["cost"]["total_cost"])
|
304
304
|
del additional_metadata["cost"]
|
305
305
|
else:
|
306
|
-
additional_metadata["total_cost"] = 0.0
|
306
|
+
additional_metadata["total_cost"] = np.float32(0.0)
|
307
307
|
|
308
308
|
combined_metadata = user_detail['trace_user_detail']['metadata'].copy()
|
309
309
|
combined_metadata.update(additional_metadata)
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.2
|
2
2
|
Name: ragaai_catalyst
|
3
|
-
Version: 2.1.
|
3
|
+
Version: 2.1.5b13
|
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=
|
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=
|
18
|
+
ragaai_catalyst/tracers/tracer.py,sha256=VIg6haG2DBsqkVfyFVa2Nd8XPuboPgSrZnSabYNxQdM,19277
|
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.
|
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.5b13.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
69
|
+
ragaai_catalyst-2.1.5b13.dist-info/METADATA,sha256=ULxrwmhLZr0ME6f4nAQAegCCehNupksHAXH70NKwhUA,12765
|
70
|
+
ragaai_catalyst-2.1.5b13.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
71
|
+
ragaai_catalyst-2.1.5b13.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
|
72
|
+
ragaai_catalyst-2.1.5b13.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|