Skip to content

Commit

Permalink
add new migration file
Browse files Browse the repository at this point in the history
  • Loading branch information
@Bhavyajain21
Bhavyajain21 committed Oct 18, 2024
1 parent ece2e18 commit d29cfbb
Show file tree
Hide file tree
Showing 10 changed files with 49 additions and 10 deletions.
4 changes: 2 additions & 2 deletions extension/sql/vectorize--0.11.1--0.12.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* &str */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"transformer" TEXT DEFAULT 'text-embedding-ada-002', /* &str */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'join', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* &str */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand All @@ -29,7 +29,7 @@ CREATE FUNCTION vectorize."init_rag"(
"column" TEXT, /* &str */
"schema" TEXT DEFAULT 'public', /* &str */
"transformer" TEXT DEFAULT 'text-embedding-ada-002', /* &str */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'join', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* &str */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
6 changes: 4 additions & 2 deletions extension/sql/vectorize--0.13.0--0.14.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,9 @@ CREATE FUNCTION vectorize."init_rag"(
"unique_record_id" TEXT, /* &str */
"column" TEXT, /* &str */
"schema" TEXT DEFAULT 'public', /* &str */
"index_dist_type" vectorize.IndexDist DEFAULT, /* vectorize::types::IndexDist */
"index_dist_type" TEXT DEFAULT 'pgv_hsnw_cosine', /* vectorize::types::IndexDist */
"transformer" TEXT DEFAULT 'openai/text-embedding-ada-002', /* &str */
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'join', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* &str */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand All @@ -66,7 +67,8 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* &str */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"index_dist_type" TEXT DEFAULT 'pgv_hsnw_cosine', /* vectorize::types::IndexDist */
"transformer" TEXT DEFAULT 'openai/text-embedding-ada-002', /* &str */
"transformer" TEXT DEFAULT 'openai/text-embedding-ada-002', /* &str */
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'join', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* &str */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
2 changes: 2 additions & 0 deletions extension/sql/vectorize--0.17.1--0.18.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ CREATE FUNCTION vectorize."table"(
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine', /* vectorize::types::IndexDist */
"transformer" TEXT DEFAULT 'sentence-transformers/all-MiniLM-L6-v2', /* &str */
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'join', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* &str */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand All @@ -71,6 +72,7 @@ CREATE FUNCTION vectorize."init_rag"(
"schema" TEXT DEFAULT 'public', /* &str */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine', /* vectorize::types::IndexDist */
"transformer" TEXT DEFAULT 'sentence-transformers/all-MiniLM-L6-v2', /* &str */
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'join', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* &str */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
35 changes: 35 additions & 0 deletions extension/sql/vectorize--0.19.0--0.20.0.sql
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
DROP FUNCTION IF EXISTS vectorize."table";
DROP FUNCTION IF EXISTS vectorize."init_rag";
ALTER TABLE vectorize.job DROP COLUMN search_alg;

-- Recreate functions with updated signatures
CREATE FUNCTION vectorize."table"(
"table" TEXT,
"columns" TEXT[],
"job_name" TEXT,
"primary_key" TEXT,
"schema" TEXT DEFAULT 'public',
"update_col" TEXT DEFAULT 'last_updated_at',
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"transformer" TEXT DEFAULT 'sentence-transformers/all-MiniLM-L6-v2',
"table_method" vectorize.TableMethod DEFAULT 'join',
"schedule" TEXT DEFAULT '* * * * *'
) RETURNS TEXT
STRICT
LANGUAGE c
AS 'MODULE_PATHNAME', 'table_wrapper';

CREATE FUNCTION vectorize."init_rag"(
"agent_name" TEXT,
"table_name" TEXT,
"unique_record_id" TEXT,
"column" TEXT,
"schema" TEXT DEFAULT 'public',
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"transformer" TEXT DEFAULT 'sentence-transformers/all-MiniLM-L6-v2',
"table_method" vectorize.TableMethod DEFAULT 'join',
"schedule" TEXT DEFAULT '* * * * *'
) RETURNS TEXT
STRICT
LANGUAGE c
AS 'MODULE_PATHNAME', 'init_rag_wrapper';
2 changes: 1 addition & 1 deletion extension/sql/vectorize--0.2.0--0.3.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* alloc::string::String */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"transformer" vectorize.Transformer DEFAULT 'openai', /* vectorize::types::Transformer */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'append', /* vectorize::init::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* alloc::string::String */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
2 changes: 1 addition & 1 deletion extension/sql/vectorize--0.4.0--0.5.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* alloc::string::String */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"transformer" vectorize.Transformer DEFAULT 'openai', /* vectorize::types::Transformer */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'append', /* vectorize::init::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* alloc::string::String */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
2 changes: 1 addition & 1 deletion extension/sql/vectorize--0.5.0--0.6.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* alloc::string::String */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"transformer" vectorize.Transformer DEFAULT 'openai', /* vectorize::types::Transformer */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'append', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* alloc::string::String */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
2 changes: 1 addition & 1 deletion extension/sql/vectorize--0.7.0--0.8.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* alloc::string::String */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"transformer" TEXT DEFAULT 'text-embedding-ada-002', /* alloc::string::String */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'append', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT '* * * * *' /* alloc::string::String */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
2 changes: 1 addition & 1 deletion extension/sql/vectorize--0.8.1--0.9.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ CREATE FUNCTION vectorize."table"(
"schema" TEXT DEFAULT 'public', /* alloc::string::String */
"update_col" TEXT DEFAULT 'last_updated_at', /* alloc::string::String */
"transformer" TEXT DEFAULT 'text-embedding-ada-002', /* alloc::string::String */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'append', /* vectorize::types::TableMethod */
"schedule" TEXT DEFAULT 'realtime' /* alloc::string::String */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
Expand Down
2 changes: 1 addition & 1 deletion extension/sql/vectorize--0.9.0--0.10.0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ CREATE FUNCTION vectorize."init_rag"(
"column" TEXT, /* &str */
"schema" TEXT DEFAULT 'public', /* &str */
"transformer" TEXT DEFAULT 'text-embedding-ada-002', /* &str */
"index_dist_type" vectorize.IndexDist DEFAULT 'pgv_hnsw_cosine',
"search_alg" vectorize.SimilarityAlg DEFAULT 'pgv_cosine_similarity', /* vectorize::types::SimilarityAlg */
"table_method" vectorize.TableMethod DEFAULT 'append' /* vectorize::types::TableMethod */
) RETURNS TEXT /* core::result::Result<alloc::string::String, anyhow::Error> */
STRICT
Expand Down

0 comments on commit d29cfbb

Please sign in to comment.