from pathlib import Path

from adapters.postgres import PostgresConversationStore, PostgresTraceRepository
from config import get_settings


def main() -> None:
    dsn = get_settings().postgres_dsn
    migrations_dir = Path(__file__).parent / "migrations"

    trace_migration = migrations_dir / "001_rag_retrieval_trace.sql"
    PostgresTraceRepository(dsn).migrate(trace_migration)
    print(f"Applied {trace_migration.name}")

    conversation_migration = migrations_dir / "002_rag_conversation_turn.sql"
    PostgresConversationStore(dsn).migrate(conversation_migration)
    print(f"Applied {conversation_migration.name}")


if __name__ == "__main__":
    main()