ragaai-catalyst 2.1.5b10__py3-none-any.whl → 2.1.5b12__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"]:
180
+ "ChatLiteLLM", "ChatBedrock", "AzureChatOpenAI"]:
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
 
@@ -227,11 +227,16 @@ class LangchainTracer(BaseCallbackHandler):
227
227
  logger.debug("ChatBedrock not available for patching")
228
228
 
229
229
  try:
230
- from langchain_google_vertexai import ChatVertexAI, VertexAI
230
+ from langchain_google_vertexai import ChatVertexAI
231
231
  components_to_patch["ChatVertexAI"] = (ChatVertexAI, "__init__")
232
+ except ImportError:
233
+ logger.debug("ChatVertexAI not available for patching")
234
+
235
+ try:
236
+ from langchain_google_vertexai import VertexAI
232
237
  components_to_patch["VertexAI"] = (VertexAI, "__init__")
233
238
  except ImportError:
234
- logger.debug("ChatVertexAI/VertexAI not available for patching")
239
+ logger.debug("VertexAI not available for patching")
235
240
 
236
241
  try:
237
242
  from langchain_google_genai import ChatGoogleGenerativeAI
@@ -255,13 +260,19 @@ class LangchainTracer(BaseCallbackHandler):
255
260
  from langchain_openai import ChatOpenAI as ChatOpenAI_LangchainOpenAI
256
261
  components_to_patch["ChatOpenAI_LangchainOpenAI"] = (ChatOpenAI_LangchainOpenAI, "__init__")
257
262
  except ImportError:
258
- logger.debug("ChatOpenAI_LangchainOpenAI not available for patching")
263
+ logger.debug("ChatOpenAI (from langchain_openai) not available for patching")
264
+
265
+ try:
266
+ from langchain_openai import AzureChatOpenAI
267
+ components_to_patch["AzureChatOpenAI"] = (AzureChatOpenAI, "__init__")
268
+ except ImportError:
269
+ logger.debug("AzureChatOpenAI (from langchain_openai) not available for patching")
259
270
 
260
271
  try:
261
272
  from langchain.chat_models import ChatOpenAI as ChatOpenAI_ChatModels
262
273
  components_to_patch["ChatOpenAI_ChatModels"] = (ChatOpenAI_ChatModels, "__init__")
263
274
  except ImportError:
264
- logger.debug("ChatOpenAI_ChatModels not available for patching")
275
+ logger.debug("ChatOpenAI (from langchain.chat_models) not available for patching")
265
276
 
266
277
  try:
267
278
  from langchain.chains import create_retrieval_chain, RetrievalQA
@@ -61,13 +61,13 @@ def langchain_tracer_extraction(data):
61
61
  if item["event"] == "retriever_end":
62
62
  context = item["documents"][0]["page_content"].replace('\n', ' ')
63
63
  return context
64
- if "chat_model_calls" in data and data["chat_model_calls"] != []:
65
- for item in data["chat_model_calls"]:
66
- messages = item["messages"][0]
67
- for message in messages:
68
- if message["type"]=="system":
69
- content = message["content"].strip().replace('\n', ' ')
70
- return content
64
+ # if "chat_model_calls" in data and data["chat_model_calls"] != []:
65
+ # for item in data["chat_model_calls"]:
66
+ # messages = item["messages"][0]
67
+ # for message in messages:
68
+ # if message["type"]=="system":
69
+ # content = message["content"].strip().replace('\n', ' ')
70
+ # return content
71
71
 
72
72
 
73
73
  prompt = get_prompt(data)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: ragaai_catalyst
3
- Version: 2.1.5b10
3
+ Version: 2.1.5b12
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=t5s9JD6HdzJRnAd9Zp8qLppCY6azz11hqHAykkNq9zw,29204
16
+ ragaai_catalyst/tracers/langchain_callback.py,sha256=ubdkiaE7iouE5vUGmTcRQyfVVzRwNOiwr8qBEVHzWUY,29672
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
@@ -63,10 +63,10 @@ ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hak
63
63
  ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
64
64
  ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
65
65
  ragaai_catalyst/tracers/utils/convert_langchain_callbacks_output.py,sha256=ofrNrxf2b1hpjDh_zeaxiYq86azn1MF3kW8-ViYPEg0,1641
66
- ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=cghjCuUe8w-2MZdh9xgtRGe3y219u26GGzpnuY4Wt6Q,3047
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.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,,
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,,