diff --git a/migrations/20240521092310_collab_embeddings.sql b/migrations/20240521092310_collab_embeddings.sql deleted file mode 100644 index 74836a3f..00000000 --- a/migrations/20240521092310_collab_embeddings.sql +++ /dev/null @@ -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); \ No newline at end of file diff --git a/migrations/20240614171931_collab_embeddings.sql b/migrations/20240614171931_collab_embeddings.sql new file mode 100644 index 00000000..7377990f --- /dev/null +++ b/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; \ No newline at end of file