ragaai-catalyst 2.1.5b10__py3-none-any.whl → 2.1.5b12__py3-none-any.whl

ragaai_catalyst/tracers/langchain_callback.py

@@ -177,7 +177,7 @@ class LangchainTracer(BaseCallbackHandler):
                 # Store model name if available
                 if component_name in ["OpenAI", "ChatOpenAI_LangchainOpenAI", "ChatOpenAI_ChatModels",
                                     "ChatVertexAI", "VertexAI", "ChatGoogleGenerativeAI", "ChatAnthropic", 
-                                    "ChatLiteLLM", "ChatBedrock"]:
+                                    "ChatLiteLLM", "ChatBedrock", "AzureChatOpenAI"]:
                     instance = args[0] if args else None
                     model_name = kwargs.get('model_name') or kwargs.get('model') or kwargs.get('model_id')
 
@@ -227,11 +227,16 @@ class LangchainTracer(BaseCallbackHandler):
             logger.debug("ChatBedrock not available for patching")
             
         try:
-            from langchain_google_vertexai import ChatVertexAI, VertexAI
+            from langchain_google_vertexai import ChatVertexAI
             components_to_patch["ChatVertexAI"] = (ChatVertexAI, "__init__")
+        except ImportError:
+            logger.debug("ChatVertexAI not available for patching")
+
+        try:
+            from langchain_google_vertexai import VertexAI
             components_to_patch["VertexAI"] = (VertexAI, "__init__")
         except ImportError:
-            logger.debug("ChatVertexAI/VertexAI not available for patching")
+            logger.debug("VertexAI not available for patching")
             
         try:
             from langchain_google_genai import ChatGoogleGenerativeAI
@@ -255,13 +260,19 @@ class LangchainTracer(BaseCallbackHandler):
             from langchain_openai import ChatOpenAI as ChatOpenAI_LangchainOpenAI
             components_to_patch["ChatOpenAI_LangchainOpenAI"] = (ChatOpenAI_LangchainOpenAI, "__init__")
         except ImportError:
-            logger.debug("ChatOpenAI_LangchainOpenAI not available for patching")
+            logger.debug("ChatOpenAI (from langchain_openai) not available for patching")
+
+        try:
+            from langchain_openai import AzureChatOpenAI
+            components_to_patch["AzureChatOpenAI"] = (AzureChatOpenAI, "__init__")
+        except ImportError:
+            logger.debug("AzureChatOpenAI (from langchain_openai) not available for patching")
             
         try:
             from langchain.chat_models import ChatOpenAI as ChatOpenAI_ChatModels
             components_to_patch["ChatOpenAI_ChatModels"] = (ChatOpenAI_ChatModels, "__init__")
         except ImportError:
-            logger.debug("ChatOpenAI_ChatModels not available for patching")
+            logger.debug("ChatOpenAI (from langchain.chat_models) not available for patching")
             
         try:
             from langchain.chains import create_retrieval_chain, RetrievalQA

ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py

@@ -61,13 +61,13 @@ def langchain_tracer_extraction(data):
                 if item["event"] == "retriever_end":
                     context = item["documents"][0]["page_content"].replace('\n', ' ')
                     return context
-        if "chat_model_calls" in data and data["chat_model_calls"] != []:
-            for item in data["chat_model_calls"]:
-                messages = item["messages"][0]
-                for message in messages:
-                    if message["type"]=="system":
-                        content = message["content"].strip().replace('\n', ' ')
-                        return content
+        # if "chat_model_calls" in data and data["chat_model_calls"] != []:
+        #     for item in data["chat_model_calls"]:
+        #         messages = item["messages"][0]
+        #         for message in messages:
+        #             if message["type"]=="system":
+        #                 content = message["content"].strip().replace('\n', ' ')
+        #                 return content
 
 
     prompt = get_prompt(data)

{ragaai_catalyst-2.1.5b10.dist-info → ragaai_catalyst-2.1.5b12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b10
+Version: 2.1.5b12
 Summary: RAGA AI CATALYST
 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>
 Requires-Python: <3.13,>=3.9

{ragaai_catalyst-2.1.5b10.dist-info → ragaai_catalyst-2.1.5b12.dist-info}/RECORD

@@ -13,7 +13,7 @@ ragaai_catalyst/synthetic_data_generation.py,sha256=uDV9tNwto2xSkWg5XHXUvjErW-4P
 ragaai_catalyst/utils.py,sha256=TlhEFwLyRU690HvANbyoRycR3nQ67lxVUQoUOfTPYQ0,3772
 ragaai_catalyst/tracers/__init__.py,sha256=LfgTes-nHpazssbGKnn8kyLZNr49kIPrlkrqqoTFTfc,301
 ragaai_catalyst/tracers/distributed.py,sha256=AIRvS5Ur4jbFDXsUkYuCTmtGoHHx3LOG4n5tWOh610U,10330
-ragaai_catalyst/tracers/langchain_callback.py,sha256=t5s9JD6HdzJRnAd9Zp8qLppCY6azz11hqHAykkNq9zw,29204
+ragaai_catalyst/tracers/langchain_callback.py,sha256=ubdkiaE7iouE5vUGmTcRQyfVVzRwNOiwr8qBEVHzWUY,29672
 ragaai_catalyst/tracers/llamaindex_callback.py,sha256=ZY0BJrrlz-P9Mg2dX-ZkVKG3gSvzwqBtk7JL_05MiYA,14028
 ragaai_catalyst/tracers/tracer.py,sha256=Yq2HhgT4785t9573kksJ7ngM3qCLPgZbZ0IpgOHdTTo,19223
 ragaai_catalyst/tracers/upload_traces.py,sha256=2TWdRTN6FMaX-dqDv8BJWQS0xrCGYKkXEYOi2kK3Z3Y,5487
@@ -63,10 +63,10 @@ ragaai_catalyst/tracers/instrumentators/llamaindex.py,sha256=SMrRlR4xM7k9HK43hak
 ragaai_catalyst/tracers/instrumentators/openai.py,sha256=14R4KW9wQCR1xysLfsP_nxS7cqXrTPoD8En4MBAaZUU,379
 ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpacDA0U3wg6Ybw,64
 ragaai_catalyst/tracers/utils/convert_langchain_callbacks_output.py,sha256=ofrNrxf2b1hpjDh_zeaxiYq86azn1MF3kW8-ViYPEg0,1641
-ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=cghjCuUe8w-2MZdh9xgtRGe3y219u26GGzpnuY4Wt6Q,3047
+ragaai_catalyst/tracers/utils/langchain_tracer_extraction_logic.py,sha256=qK67fdUBz5Xr99ajqXbYf1ueKS1V3a3_XR0zCcN4iGI,3061
 ragaai_catalyst/tracers/utils/utils.py,sha256=ViygfJ7vZ7U0CTSA1lbxVloHp4NSlmfDzBRNCJuMhis,2374
-ragaai_catalyst-2.1.5b10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b10.dist-info/METADATA,sha256=DVGyoXWxgwks82eTOMYm_eNKxwcIvvbnuiRMA_f5EkU,12765
-ragaai_catalyst-2.1.5b10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.5b10.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b10.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b12.dist-info/METADATA,sha256=yKKeKwBa1Miwi3DdRdMDxgqELklmj6zMHIH8qrz19uA,12765
+ragaai_catalyst-2.1.5b12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.5b12.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b12.dist-info/RECORD,,