rag / app /config.py
Tahasaif3's picture
'changes'
fe10c91
import os
from pydantic_settings import BaseSettings
class Settings(BaseSettings):
OPENAI_API_KEY: str
GEMINI_API_KEY: str
# Set to "openai" or "gemini" to choose the AI provider
AI_PROVIDER: str = "openai"
DATABASE_URL: str = os.getenv("DATABASE_URL", "")
NEON_DATABASE_URL: str = os.getenv("NEON_DATABASE_URL", "")
QDRANT_URL: str = os.getenv("QDRANT_URL", "http://localhost:6333")
QDRANT_API_KEY: str = os.getenv("QDRANT_API_KEY", "")
OPENAI_MODEL_CHAT: str = "gpt-4o-mini"
OPENAI_MODEL_EMBEDDING: str = "text-embedding-3-small"
GEMINI_MODEL_CHAT: str = "gemini-2.0-flash"
GEMINI_MODEL_EMBEDDING: str = "embedding-001"
class Config:
env_file = ".env"
extra = "ignore"
settings = Settings()