Merge pull request #624 from AppFlowy-IO/embedding-migration-optional

chore: make vector extension optional for migrations

migrations/20240521092310_collab_embeddings.sql

@@ -1,17 +0,0 @@
--- Add migration script here
-CREATE EXTENSION IF NOT EXISTS vector;
-
--- create table to store collab embeddings
-CREATE TABLE IF NOT EXISTS af_collab_embeddings
-(
-    fragment_id TEXT NOT NULL PRIMARY KEY,
-    oid TEXT NOT NULL,
-    partition_key INTEGER NOT NULL,
-    content_type INTEGER NOT NULL,
-    indexed_at TIMESTAMP WITHOUT TIME ZONE NOT NULL DEFAULT (NOW()),
-    content TEXT,
-    embedding VECTOR(1536),
-    FOREIGN KEY (oid, partition_key) REFERENCES af_collab (oid, partition_key) ON DELETE CASCADE
-);
-
-CREATE INDEX IF NOT EXISTS af_collab_embeddings_similarity_idx ON af_collab_embeddings USING hnsw (embedding vector_cosine_ops);

migrations/20240614171931_collab_embeddings.sql

@@ -0,0 +1,24 @@
+DO $$
+BEGIN
+    -- Add migration script here
+    CREATE EXTENSION IF NOT EXISTS vector;
+
+    -- create table to store collab embeddings
+    CREATE TABLE IF NOT EXISTS af_collab_embeddings
+    (
+        fragment_id TEXT NOT NULL PRIMARY KEY,
+        oid TEXT NOT NULL,
+        partition_key INTEGER NOT NULL,
+        content_type INTEGER NOT NULL,
+        indexed_at TIMESTAMP WITHOUT TIME ZONE NOT NULL DEFAULT (NOW()),
+        content TEXT,
+        embedding VECTOR(1536),
+        FOREIGN KEY (oid, partition_key) REFERENCES af_collab (oid, partition_key) ON DELETE CASCADE
+    );
+
+    CREATE INDEX IF NOT EXISTS af_collab_embeddings_similarity_idx ON af_collab_embeddings USING hnsw (embedding vector_cosine_ops);
+
+EXCEPTION WHEN OTHERS THEN
+    RAISE NOTICE 'could not create "vector" extension, ignoring this migration';
+END;
+$$ LANGUAGE plpgsql;