add migration script

tembo-io · Jan 24, 2024 · 5d3d6e3 · 5d3d6e3
1 parent 493d147
commit 5d3d6e3
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/sql/vectorize--0.7.1--0.8.0.sql b/sql/vectorize--0.7.1--0.8.0.sql
@@ -0,0 +1,22 @@
+DROP function vectorize."table";
+
+-- src/api.rs:15
+-- vectorize::api::table
+CREATE  FUNCTION vectorize."table"(
+        "table" TEXT, /* &str */
+        "columns" TEXT[], /* alloc::vec::Vec<alloc::string::String> */
+        "job_name" TEXT, /* alloc::string::String */
+        "primary_key" TEXT, /* alloc::string::String */
+        "args" json DEFAULT '{}', /* pgrx::datum::json::Json */
+        "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 */
+        "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> */
+STRICT
+LANGUAGE c /* Rust */
+AS 'MODULE_PATHNAME', 'table_wrapper';
+
+