ragaai-catalyst 2.1.5b16__py3-none-any.whl → 2.1.5b18__py3-none-any.whl

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,
@@ -45,7 +47,6 @@ class LLMTracerMixin:
             self.model_costs = load_model_costs()
         except Exception as e:
             self.model_costs = {
-                # TODO: Default cost handling needs to be improved
                 "default": {"input_cost_per_token": 0.0, "output_cost_per_token": 0.0}
             }
         self.MAX_PARAMETERS_TO_DISPLAY = 10
@@ -59,25 +60,31 @@ class LLMTracerMixin:
         self.total_cost = 0.0
         self.llm_data = {}
 
-        # Add auto_instrument options
         self.auto_instrument_llm = False
         self.auto_instrument_user_interaction = False
         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:
@@ -87,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"""

{ragaai_catalyst-2.1.5b16.dist-info → ragaai_catalyst-2.1.5b18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragaai_catalyst
-Version: 2.1.5b16
+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.5b16.dist-info → ragaai_catalyst-2.1.5b18.dist-info}/RECORD

@@ -32,7 +32,7 @@ ragaai_catalyst/tracers/agentic_tracing/tracers/agent_tracer.py,sha256=8d6YovuWi
 ragaai_catalyst/tracers/agentic_tracing/tracers/base.py,sha256=88rX7OkOGEyVNECUrc4bYqODyulXve_-99d9ku5hBeQ,37373
 ragaai_catalyst/tracers/agentic_tracing/tracers/custom_tracer.py,sha256=OHet_Cphyrsq2CP4WiooTsWSgg3Rc1n8QsOl1s2vqdY,13480
 ragaai_catalyst/tracers/agentic_tracing/tracers/langgraph_tracer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=NklCHbkG_ITlJ0zxGm-QHdk6it5xeu5gUHnK18YuLGo,34499
+ragaai_catalyst/tracers/agentic_tracing/tracers/llm_tracer.py,sha256=OSRGwftAYpylvziXD50HGjEfEMe8k11-3SRSZKw2tHI,35255
 ragaai_catalyst/tracers/agentic_tracing/tracers/main_tracer.py,sha256=JEwvFV6KdYcg8zt2qUTKmmPvrkEft0YNAyUYK6FWF7c,18335
 ragaai_catalyst/tracers/agentic_tracing/tracers/network_tracer.py,sha256=m8CxYkl7iMiFya_lNwN1ykBc3Pmo-2pR_2HmpptwHWQ,10352
 ragaai_catalyst/tracers/agentic_tracing/tracers/tool_tracer.py,sha256=nmgkE3jRlTpq_uk6FYw0hTv2VgT8ejxvvBRL7gso39I,21037
@@ -66,8 +66,8 @@ ragaai_catalyst/tracers/utils/__init__.py,sha256=KeMaZtYaTojilpLv65qH08QmpYclfpa
 ragaai_catalyst/tracers/utils/convert_langchain_callbacks_output.py,sha256=ofrNrxf2b1hpjDh_zeaxiYq86azn1MF3kW8-ViYPEg0,1641
 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.5b16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ragaai_catalyst-2.1.5b16.dist-info/METADATA,sha256=HtFk47NJx5iIsozMdsDyLdbodnfEwAfksqNyeiqzDWo,12796
-ragaai_catalyst-2.1.5b16.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragaai_catalyst-2.1.5b16.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
-ragaai_catalyst-2.1.5b16.dist-info/RECORD,,
+ragaai_catalyst-2.1.5b18.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ragaai_catalyst-2.1.5b18.dist-info/METADATA,sha256=U0TwkW7NhS1k21IvYfh-g5WtmMcFbeGJYQwl4L18xyc,12796
+ragaai_catalyst-2.1.5b18.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragaai_catalyst-2.1.5b18.dist-info/top_level.txt,sha256=HpgsdRgEJMk8nqrU6qdCYk3di7MJkDL0B19lkc7dLfM,16
+ragaai_catalyst-2.1.5b18.dist-info/RECORD,,