from sqlmodel import SQLModel, create_engine, Session
from sqlalchemy.dialects import registry
from src.core.config import settings

# Register the libsql dialect manually to bypass uvicorn/uv entry_points cache issues on Hugging Face
registry.register("sqlite.libsql", "sqlalchemy_libsql.libsql", "SQLiteDialect_libsql")

from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor

engine = create_engine(
    settings.get_sqlalchemy_url, 
    echo=True, 
    connect_args=settings.get_sqlalchemy_connect_args
)

# Instrument the database engine for tracing
SQLAlchemyInstrumentor().instrument(
    engine=engine,
)

def get_session():
    with Session(engine) as session:
        yield session