MCP Chat Support System

from sqlalchemy import create_engine from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker import os from dotenv import load_dotenv load_dotenv() # Database URL from environment variable DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./mcp_server.db") # Create SQLAlchemy engine engine = create_engine( DATABASE_URL, # SQLite specific setting connect_args={"check_same_thread": False} if "sqlite" in DATABASE_URL else {} ) # Create SessionLocal class SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) # Create Base class Base = declarative_base() # Dependency to get database session def get_db(): db = SessionLocal() try: yield db finally: db.close()