22 lines
668 B
Python
22 lines
668 B
Python
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()
|