Verify exact production traces and record rollout

2026-08-11 11:29:59 +07:00
parent 6b8f7584ed
commit 59e6ad2d0d
46 changed files with 2795 additions and 290 deletions
@@ -1,6 +1,6 @@
 from pathlib import Path

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


@@ -8,13 +8,10 @@ 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}")
+    repository = PostgresTraceRepository(dsn)
+    for migration in sorted(migrations_dir.glob("*.sql")):
+        repository.migrate(migration)
+        print(f"Applied {migration.name}")


 if __name__ == "__main__":