Update

agents.py

@@ -5,7 +5,7 @@ import json
 
 logger = logging.getLogger(__name__)
 
-# Try to import Google AI packages with fallback options
+# Try both import methods
 try:
     import google.generativeai as genai
     GENAI_PACKAGE = "generativeai"
@@ -14,7 +14,7 @@ except ImportError:
         import google.genai as genai
         GENAI_PACKAGE = "genai"
     except ImportError:
-        logger.error("Failed to import Google AI package. Please install required dependencies.")
+        logger.error("Failed to import Google AI package")
         raise
 
 from openai import OpenAI
@@ -29,13 +29,11 @@ class BaseAgent:
             
             # Version-specific configuration
             try:
-                if HAS_GENERATIVEAI:
+                if GENAI_PACKAGE == "generativeai":
                     genai.configure(api_key=api_key)
-                elif HAS_GENAI:
-                    # For google.genai package
+                else:
                     os.environ["GOOGLE_API_KEY"] = api_key
-                    # Initialize model directly if needed
-                    self.genai_client = genai.GenerativeModel
+                    
             except Exception as e:
                 logger.warning(f"Using fallback configuration method: {str(e)}")
                 os.environ["GOOGLE_API_KEY"] = api_key
@@ -50,11 +48,7 @@ class BaseAgent:
 
     def _generate_with_gemini(self, prompt: str, system_prompt: str) -> str:
         try:
-            if HAS_GENERATIVEAI:
-                model = genai.GenerativeModel(model_name=self.gemini_model)
-            else:
-                model = self.genai_client(model_name=self.gemini_model)
-                
+            model = genai.GenerativeModel(model_name=self.gemini_model)
             combined_prompt = f"System: {system_prompt}\n\nUser: {prompt}"
             
             try: