dannybhatt/AI_fast_api

1"""2Test configuration for the Document Chatbot.3"""4 5import os6 7# Set test environment variables8os.environ["OPENAI_API_KEY"] = "sk-test-key-for-demo"9os.environ["OPENAI_MODEL"] = "gpt-4o"10os.environ["EMBEDDING_MODEL"] = "text-embedding-3-small"11# Add other providers for testing12os.environ["GROQ_API_KEY"] = "groq-test-key"13os.environ["GROQ_MODEL"] = "llama3-70b-8192"14os.environ["COHERE_API_KEY"] = "cohere-test-key"15os.environ["COHERE_MODEL"] = "command-r-plus"16os.environ["DEEPSEEK_API_KEY"] = "deepseek-test-key"17os.environ["DEEPSEEK_MODEL"] = "deepseek-chat"18os.environ["CLAUDE_API_KEY"] = "claude-test-key"19os.environ["CLAUDE_MODEL"] = "claude-3-opus-20240229"20os.environ["DEBUG"] = "True"21os.environ["SECRET_KEY"] = "test_secret_key_for_demo"22 23# Create necessary directories24os.makedirs("data/documents", exist_ok=True)25os.makedirs("data/vector_store", exist_ok=True)26os.makedirs("logs", exist_ok=True)27 28print("✅ Test configuration loaded")29print("📁 Directories created")30print("🔧 Environment variables set")31