Fix ai-service Dockerfile: bake in drug_entities.json, override its path

2026-08-10 10:35:13 +07:00
parent a4b8e1c4db
commit 60b4397032
51 changed files with 4302 additions and 2087 deletions
@@ -1,13 +1,20 @@
 from pathlib import Path

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


 def main() -> None:
-    migration = Path(__file__).parent / "migrations/001_rag_retrieval_trace.sql"
-    PostgresTraceRepository(get_settings().postgres_dsn).migrate(migration)
-    print(f"Applied {migration.name}")
+    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__":