ragaai-catalyst 2.1.5b9__tar.gz → 2.1.5b10__tar.gz

{ragaai_catalyst-2.1.5b9/ragaai_catalyst.egg-info → ragaai_catalyst-2.1.5b10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b9
+Version: 2.1.5b10
 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.5b9 → ragaai_catalyst-2.1.5b10}/pyproject.toml

@@ -9,7 +9,7 @@ readme = "README.md"
 requires-python = ">=3.9,<3.13"
 # license = {file = "LICENSE"}
 
-version = "2.1.5.beta.9"
+version = "2.1.5.beta.10"
 authors = [
     {name = "Kiran Scaria", email = "kiran.scaria@raga.ai"},
     {name = "Kedar Gaikwad", email = "kedar.gaikwad@raga.ai"},

{ragaai_catalyst-2.1.5b9 → ragaai_catalyst-2.1.5b10}/ragaai_catalyst/tracers/langchain_callback.py

@@ -177,9 +177,10 @@ class LangchainTracer(BaseCallbackHandler):
                 # Store model name if available
                 if component_name in ["OpenAI", "ChatOpenAI_LangchainOpenAI", "ChatOpenAI_ChatModels",
                                     "ChatVertexAI", "VertexAI", "ChatGoogleGenerativeAI", "ChatAnthropic", 
-                                    "ChatLiteLLM"]:
+                                    "ChatLiteLLM", "ChatBedrock"]:
                     instance = args[0] if args else None
-                    model_name = kwargs.get('model_name') or kwargs.get('model')
+                    model_name = kwargs.get('model_name') or kwargs.get('model') or kwargs.get('model_id')
+
                     if instance and model_name:
                         self.model_names[id(instance)] = model_name
                 
@@ -218,6 +219,12 @@ class LangchainTracer(BaseCallbackHandler):
             components_to_patch["OpenAI"] = (OpenAI, "__init__")
         except ImportError:
             logger.debug("OpenAI not available for patching")
+
+        try:
+            from langchain_aws import ChatBedrock
+            components_to_patch["ChatBedrock"] = (ChatBedrock, "__init__")
+        except ImportError:
+            logger.debug("ChatBedrock not available for patching")
             
         try:
             from langchain_google_vertexai import ChatVertexAI, VertexAI
@@ -509,7 +516,7 @@ class LangchainTracer(BaseCallbackHandler):
                     model = list(self.model_names.values())[0]
             except Exception as e:
                 model=""
-            
+
             self.additional_metadata = {
                 'latency': latency,
                 'model_name': model,

{ragaai_catalyst-2.1.5b9 → ragaai_catalyst-2.1.5b10/ragaai_catalyst.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b9
+Version: 2.1.5b10
 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