ragaai-catalyst 2.1.5b9__py3-none-any.whl → 2.1.5b10__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 +10 -3
- {ragaai_catalyst-2.1.5b9.dist-info → ragaai_catalyst-2.1.5b10.dist-info}/METADATA +1 -1
- {ragaai_catalyst-2.1.5b9.dist-info → ragaai_catalyst-2.1.5b10.dist-info}/RECORD +6 -6
- {ragaai_catalyst-2.1.5b9.dist-info → ragaai_catalyst-2.1.5b10.dist-info}/LICENSE +0 -0
- {ragaai_catalyst-2.1.5b9.dist-info → ragaai_catalyst-2.1.5b10.dist-info}/WHEEL +0 -0
- {ragaai_catalyst-2.1.5b9.dist-info → ragaai_catalyst-2.1.5b10.dist-info}/top_level.txt +0 -0
@@ -177,9 +177,10 @@ 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"]:
|
180
|
+
"ChatLiteLLM", "ChatBedrock"]:
|
181
181
|
instance = args[0] if args else None
|
182
|
-
model_name = kwargs.get('model_name') or kwargs.get('model')
|
182
|
+
model_name = kwargs.get('model_name') or kwargs.get('model') or kwargs.get('model_id')
|
183
|
+
|
183
184
|
if instance and model_name:
|
184
185
|
self.model_names[id(instance)] = model_name
|
185
186
|
|
@@ -218,6 +219,12 @@ class LangchainTracer(BaseCallbackHandler):
|
|
218
219
|
components_to_patch["OpenAI"] = (OpenAI, "__init__")
|
219
220
|
except ImportError:
|
220
221
|
logger.debug("OpenAI not available for patching")
|
222
|
+
|
223
|
+
try:
|
224
|
+
from langchain_aws import ChatBedrock
|
225
|
+
components_to_patch["ChatBedrock"] = (ChatBedrock, "__init__")
|
226
|
+
except ImportError:
|
227
|
+
logger.debug("ChatBedrock not available for patching")
|
221
228
|
|
222
229
|
try:
|
223
230
|
from langchain_google_vertexai import ChatVertexAI, VertexAI
|
@@ -509,7 +516,7 @@ class LangchainTracer(BaseCallbackHandler):
|
|
509
516
|
model = list(self.model_names.values())[0]
|
510
517
|
except Exception as e:
|
511
518
|
model=""
|
512
|
-
|
519
|
+
|
513
520
|
self.additional_metadata = {
|
514
521
|
'latency': latency,
|
515
522
|
'model_name': model,
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.2
|
2
2
|
Name: ragaai_catalyst
|
3
|
-
Version: 2.1.
|
3
|
+
Version: 2.1.5b10
|
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,7 +13,7 @@ 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=t5s9JD6HdzJRnAd9Zp8qLppCY6azz11hqHAykkNq9zw,29204
|
17
17
|
ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
|
18
18
|
ragaai_catalyst/tracers/tracer.py,sha256=Yq2HhgT4785t9573kksJ7ngM3qCLPgZbZ0IpgOHdTTo,19223
|
19
19
|
ragaai_catalyst/tracers/upload_traces.py,sha256=2TWdRTN6FMaX-dqDv8BJWQS0xrCGYKkXEYOi2kK3Z3Y,5487
|
@@ -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=cghjCuUe8w-2MZdh9xgtRGe3y219u26GGzpnuY4Wt6Q,3047
|
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.5b10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
69
|
+
ragaai_catalyst-2.1.5b10.dist-info/METADATA,sha256=DVGyoXWxgwks82eTOMYm_eNKxwcIvvbnuiRMA_f5EkU,12765
|
70
|
+
ragaai_catalyst-2.1.5b10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
71
|
+
ragaai_catalyst-2.1.5b10.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
|
72
|
+
ragaai_catalyst-2.1.5b10.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|