fix: Add missing validate_gemini_response method and cleanup method

- Added validate_gemini_response method to AISafetyGuard
- Added cleanup method to ChartDataTool to prevent AttributeError
- Improved Gemini tool parsing logic to handle all suggested tools
- Updated Gemini model to use 'gemini-2.0-flash-lite'

src/agent/research_agent.py

@@ -417,6 +417,20 @@ Respond with only the tool names, comma-separated (no explanations)."""
                 'etherscan_data', 'chart_data_provider'
             }]
             
+            # If no valid tools found, extract from response content
+            if not suggested_tools:
+                response_text = str(tool_response).lower()
+                if 'cryptocompare' in response_text:
+                    suggested_tools.append('cryptocompare_data')
+                if 'coingecko' in response_text:
+                    suggested_tools.append('coingecko_data')
+                if 'defillama' in response_text:
+                    suggested_tools.append('defillama_data')
+                if 'etherscan' in response_text:
+                    suggested_tools.append('etherscan_data')
+                if 'chart' in response_text or 'visualization' in response_text:
+                    suggested_tools.append('chart_data_provider')
+            
             logger.info(f"🛠️ Gemini suggested tools: {suggested_tools}")
 
             # Step 2: Execute tools (same logic as Ollama version)

src/tools/chart_data_tool.py

@@ -393,3 +393,8 @@ class ChartDataTool(BaseTool):
             "1d": 1, "7d": 7, "30d": 30, "90d": 90, "365d": 365, "1y": 365
         }
         return timeframe_map.get(timeframe, 30)
+    
+    async def cleanup(self):
+        """Cleanup method for session management"""
+        # ChartDataTool doesn't maintain persistent connections, so nothing to clean up
+        pass

src/utils/ai_safety.py

@@ -130,6 +130,40 @@ class AISafetyGuard:
         
         return cleaned, True, "Response is safe"
     
+    def validate_gemini_response(self, response: str) -> Tuple[str, bool, str]:
+        """
+        Validate Gemini response for safety and quality
+        Returns: (cleaned_response, is_valid, reason)
+        """
+        if not response or not response.strip():
+            return "", False, "Empty response from Gemini"
+        
+        # Check for dangerous content in response
+        dangerous_patterns = [
+            r'(?i)here.*is.*how.*to.*hack',
+            r'(?i)steps.*to.*exploit',
+            r'(?i)bypass.*security.*by',
+            r'(?i)manipulate.*market.*by',
+        ]
+        
+        for pattern in dangerous_patterns:
+            if re.search(pattern, response):
+                logger.warning(f"Blocked unsafe Gemini response: {pattern}")
+                return "", False, "Response contains potentially unsafe content"
+        
+        # Basic response cleaning
+        cleaned = response.strip()
+        
+        # Remove any potential HTML/JavaScript
+        cleaned = re.sub(r'<script.*?</script>', '', cleaned, flags=re.DOTALL | re.IGNORECASE)
+        cleaned = re.sub(r'<[^>]+>', '', cleaned)
+        
+        # Ensure response is within reasonable length
+        if len(cleaned) > 10000:  # 10k character limit
+            cleaned = cleaned[:10000] + "\n\n[Response truncated for safety]"
+        
+        return cleaned, True, "Response is safe"
+    
     def create_safe_prompt(self, user_query: str, tool_context: str) -> str:
         """Create a safety-enhanced prompt for Ollama"""
         safety_instructions = """