ragaai-catalyst 2.1.5b17__tar.gz → 2.1.5b18__tar.gz

{ragaai_catalyst-2.1.5b17 → ragaai_catalyst-2.1.5b18}/PKG-INFO

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

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

{ragaai_catalyst-2.1.5b17 → ragaai_catalyst-2.1.5b18}/ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py

@@ -10,6 +10,8 @@ from datetime import datetime
 import uuid
 import contextvars
 import traceback
+import importlib
+import sys
 
 from ..utils.llm_utils import (
     extract_model_name,
@@ -63,19 +65,26 @@ class LLMTracerMixin:
         self.auto_instrument_file_io = False
         self.auto_instrument_network = False
 
+    def check_package_available(self, package_name):
+        """Check if a package is available in the environment"""
+        try:
+            importlib.import_module(package_name)
+            return True
+        except ImportError:
+            return False
+
+    def validate_openai_key(self):
+        """Validate if OpenAI API key is available"""
+        return bool(os.getenv("OPENAI_API_KEY"))
+
     def instrument_llm_calls(self):
         """Enable LLM instrumentation"""
         self.auto_instrument_llm = True
 
-        # Handle modules that are already imported
-        import sys
-
+        # Check currently loaded modules
         if "vertexai" in sys.modules:
             self.patch_vertex_ai_methods(sys.modules["vertexai"])
-        if "vertexai.generative_models" in sys.modules:
-            self.patch_vertex_ai_methods(sys.modules["vertexai.generative_models"])
-
-        if "openai" in sys.modules:
+        if "openai" in sys.modules and self.validate_openai_key():
             self.patch_openai_methods(sys.modules["openai"])
             self.patch_openai_beta_methods(sys.modules["openai"])
         if "litellm" in sys.modules:
@@ -85,32 +94,42 @@ class LLMTracerMixin:
         if "google.generativeai" in sys.modules:
             self.patch_google_genai_methods(sys.modules["google.generativeai"])
         if "langchain_google_vertexai" in sys.modules:
-            self.patch_langchain_google_methods(
-                sys.modules["langchain_google_vertexai"]
-            )
+            self.patch_langchain_google_methods(sys.modules["langchain_google_vertexai"])
         if "langchain_google_genai" in sys.modules:
             self.patch_langchain_google_methods(sys.modules["langchain_google_genai"])
 
-        # Register hooks for future imports
-        wrapt.register_post_import_hook(self.patch_vertex_ai_methods, "vertexai")
-        wrapt.register_post_import_hook(
-            self.patch_vertex_ai_methods, "vertexai.generative_models"
-        )
-        wrapt.register_post_import_hook(self.patch_openai_methods, "openai")
-        wrapt.register_post_import_hook(self.patch_openai_beta_methods, "openai")
-        wrapt.register_post_import_hook(self.patch_litellm_methods, "litellm")
-        wrapt.register_post_import_hook(self.patch_anthropic_methods, "anthropic")
-        wrapt.register_post_import_hook(
-            self.patch_google_genai_methods, "google.generativeai"
-        )
+        # Register hooks for future imports with availability checks
+        if self.check_package_available("vertexai"):
+            wrapt.register_post_import_hook(self.patch_vertex_ai_methods, "vertexai")
+            wrapt.register_post_import_hook(
+                self.patch_vertex_ai_methods, "vertexai.generative_models"
+            )
+        
+        if self.check_package_available("openai") and self.validate_openai_key():
+            wrapt.register_post_import_hook(self.patch_openai_methods, "openai")
+            wrapt.register_post_import_hook(self.patch_openai_beta_methods, "openai")
+        
+        if self.check_package_available("litellm"):
+            wrapt.register_post_import_hook(self.patch_litellm_methods, "litellm")
+        
+        if self.check_package_available("anthropic"):
+            wrapt.register_post_import_hook(self.patch_anthropic_methods, "anthropic")
+        
+        if self.check_package_available("google.generativeai"):
+            wrapt.register_post_import_hook(
+                self.patch_google_genai_methods, "google.generativeai"
+            )
 
-        # Add hooks for LangChain integrations
-        wrapt.register_post_import_hook(
-            self.patch_langchain_google_methods, "langchain_google_vertexai"
-        )
-        wrapt.register_post_import_hook(
-            self.patch_langchain_google_methods, "langchain_google_genai"
-        )
+        # Add hooks for LangChain integrations with availability checks
+        if self.check_package_available("langchain_google_vertexai"):
+            wrapt.register_post_import_hook(
+                self.patch_langchain_google_methods, "langchain_google_vertexai"
+            )
+        
+        if self.check_package_available("langchain_google_genai"):
+            wrapt.register_post_import_hook(
+                self.patch_langchain_google_methods, "langchain_google_genai"
+            )
 
     def instrument_user_interaction_calls(self):
         """Enable user interaction instrumentation for LLM calls"""
@@ -256,10 +275,6 @@ class LLMTracerMixin:
                 self.wrap_method(chat_class, "acomplete")
 
     def wrap_openai_client_methods(self, client_class):
-        # Skip if OpenAI is not being used (no API key set)
-        if not os.getenv("OPENAI_API_KEY"):
-            return
-
         original_init = client_class.__init__
 
         @functools.wraps(original_init)

{ragaai_catalyst-2.1.5b17 → ragaai_catalyst-2.1.5b18}/ragaai_catalyst.egg-info/PKG-INFO

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