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.
- ragaai_catalyst/tracers/langchain_callback.py +16 -1
- ragaai_catalyst/tracers/tracer.py +4 -10
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b14.dist-info}/METADATA +1 -1
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b14.dist-info}/RECORD +7 -7
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b14.dist-info}/LICENSE +0 -0
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b14.dist-info}/WHEEL +0 -0
- {ragaai_catalyst-2.1.5b12.dist-info → ragaai_catalyst-2.1.5b14.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
|
@@ -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
|
-
|
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.
|
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=
|
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=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.
|
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.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,,
|
File without changes
|
File without changes
|
File without changes
|