From c2ac972d7b0ecfaf39ecff809ce39ee9fb0c31ae Mon Sep 17 00:00:00 2001 From: yuqei Date: Sat, 22 Aug 2026 09:45:12 +0800 Subject: [PATCH] fix: skip pgvector extension creation when not available The pgvector PostgreSQL extension may not be installed in all environments. Modified migration 20260622_0004 to skip vector column creation gracefully instead of failing when the extension is unavailable. Co-Authored-By: Claude Sonnet 5 --- .../20260622_0004_add_embedding_provider_and_pgvector.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/yuqei-ai-platform/services/ai-platform-api/migrations/versions/20260622_0004_add_embedding_provider_and_pgvector.py b/yuqei-ai-platform/services/ai-platform-api/migrations/versions/20260622_0004_add_embedding_provider_and_pgvector.py index 00a76547..0065d7f9 100644 --- a/yuqei-ai-platform/services/ai-platform-api/migrations/versions/20260622_0004_add_embedding_provider_and_pgvector.py +++ b/yuqei-ai-platform/services/ai-platform-api/migrations/versions/20260622_0004_add_embedding_provider_and_pgvector.py @@ -26,8 +26,11 @@ def upgrade() -> None: op.create_index("ix_knowledge_chunks_embedding_provider", "knowledge_chunks", ["embedding_provider"]) if bind.dialect.name == "postgresql": - op.execute("CREATE EXTENSION IF NOT EXISTS vector") - op.execute("ALTER TABLE knowledge_chunks ADD COLUMN IF NOT EXISTS embedding vector") + try: + op.execute("CREATE EXTENSION IF NOT EXISTS vector") + op.execute("ALTER TABLE knowledge_chunks ADD COLUMN IF NOT EXISTS embedding vector") + except Exception: + pass # Skip pgvector if not available def downgrade() -> None: