95 lines
3.4 KiB
Python
95 lines
3.4 KiB
Python
"""API key validation module."""
|
|
|
|
from typing import Dict, Any, List, Tuple
|
|
from loguru import logger
|
|
import os
|
|
from dotenv import load_dotenv
|
|
from .manager import APIKeyManager
|
|
|
|
def check_all_api_keys(api_key_manager: APIKeyManager) -> bool:
|
|
"""Check if minimum required API keys are present.
|
|
|
|
Args:
|
|
api_key_manager (APIKeyManager): The API key manager instance
|
|
|
|
Returns:
|
|
bool: True if minimum required keys are present (at least one AI provider and one research provider)
|
|
"""
|
|
try:
|
|
# Load environment variables
|
|
logger.info("Starting API key validation process...")
|
|
|
|
# Get the current working directory and .env file path
|
|
current_dir = os.getcwd()
|
|
env_path = os.path.join(current_dir, '.env')
|
|
logger.info(f"Looking for .env file at: {env_path}")
|
|
|
|
# Check if .env file exists
|
|
if not os.path.exists(env_path):
|
|
logger.error(f".env file not found at {env_path}")
|
|
return False
|
|
|
|
# Load environment variables
|
|
load_dotenv(env_path)
|
|
logger.debug("Environment variables loaded")
|
|
|
|
# Log all environment variables (without their values)
|
|
logger.debug("Available environment variables:")
|
|
for key in os.environ.keys():
|
|
if any(provider in key for provider in ['API_KEY', 'SERPAPI', 'TAVILY', 'METAPHOR', 'FIRECRAWL']):
|
|
logger.debug(f"Found environment variable: {key}")
|
|
|
|
# Step 1: Check for at least one AI provider
|
|
logger.info("Checking AI provider API keys...")
|
|
ai_providers = [
|
|
'OPENAI_API_KEY',
|
|
'GEMINI_API_KEY',
|
|
'ANTHROPIC_API_KEY',
|
|
'MISTRAL_API_KEY'
|
|
]
|
|
|
|
# Log which AI providers are found
|
|
for provider in ai_providers:
|
|
value = os.getenv(provider)
|
|
if value:
|
|
logger.info(f"Found {provider} (length: {len(value)})")
|
|
else:
|
|
logger.debug(f"Missing {provider}")
|
|
|
|
has_ai_provider = any(os.getenv(key) for key in ai_providers)
|
|
if not has_ai_provider:
|
|
logger.warning("No AI provider API key found")
|
|
return False
|
|
else:
|
|
logger.success("✓ At least one AI provider key found")
|
|
|
|
# Step 2: Check for at least one research provider
|
|
logger.info("Checking research provider API keys...")
|
|
research_providers = [
|
|
'SERPAPI_KEY',
|
|
'TAVILY_API_KEY',
|
|
'METAPHOR_API_KEY',
|
|
'FIRECRAWL_API_KEY'
|
|
]
|
|
|
|
# Log which research providers are found
|
|
for provider in research_providers:
|
|
value = os.getenv(provider)
|
|
if value:
|
|
logger.info(f"Found {provider} (length: {len(value)})")
|
|
else:
|
|
logger.debug(f"Missing {provider}")
|
|
|
|
has_research_provider = any(os.getenv(key) for key in research_providers)
|
|
if not has_research_provider:
|
|
logger.warning("No research provider API key found")
|
|
return False
|
|
else:
|
|
logger.success("✓ At least one research provider key found")
|
|
|
|
logger.success("All required API keys validated successfully!")
|
|
return True
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error checking API keys: {str(e)}", exc_info=True)
|
|
return False |