Merge pull request #1037 from julep-ai/x/openai-embed

fix: Add safety clause to text seaerch altering query
julep-ai · Jan 9, 2025 · 30bfff9 · 30bfff9
2 parents ad24128 + 3d919fc
commit 30bfff9
Showing 1 changed file with 21 additions and 16 deletions.
diff --git a/memory-store/migrations/000006_docs.up.sql b/memory-store/migrations/000006_docs.up.sql
@@ -122,22 +122,27 @@ DO $$
 DECLARE
     lang text;
 BEGIN
-    FOR lang IN (SELECT cfgname FROM pg_ts_config WHERE cfgname IN (
-        'arabic', 'danish', 'dutch', 'english', 'finnish', 'french',
-        'german', 'greek', 'hungarian', 'indonesian', 'irish', 'italian',
-        'lithuanian', 'nepali', 'norwegian', 'portuguese', 'romanian',
-        'russian', 'spanish', 'swedish', 'tamil', 'turkish'
-    ))
-    LOOP
-        -- Configure integer dictionary
-        EXECUTE format('ALTER TEXT SEARCH CONFIGURATION %I
-            ALTER MAPPING FOR int, uint WITH intdict', lang);
-
-        -- Configure synonym and stemming
-        EXECUTE format('ALTER TEXT SEARCH CONFIGURATION %I
-            ALTER MAPPING FOR asciihword, hword_asciipart, hword, hword_part, word, asciiword
-            WITH xsyn, %I_stem', lang, lang);
-    END LOOP;
+    BEGIN
+        FOR lang IN (SELECT cfgname FROM pg_ts_config WHERE cfgname IN (
+            'arabic', 'danish', 'dutch', 'english', 'finnish', 'french',
+            'german', 'greek', 'hungarian', 'indonesian', 'irish', 'italian',
+            'lithuanian', 'nepali', 'norwegian', 'portuguese', 'romanian',
+            'russian', 'spanish', 'swedish', 'tamil', 'turkish'
+        ))
+        LOOP
+            -- Configure integer dictionary
+            EXECUTE format('ALTER TEXT SEARCH CONFIGURATION %I
+                ALTER MAPPING FOR int, uint WITH intdict', lang);
+
+            -- Configure synonym and stemming
+            EXECUTE format('ALTER TEXT SEARCH CONFIGURATION %I
+                ALTER MAPPING FOR asciihword, hword_asciipart, hword, hword_part, word, asciiword
+                WITH xsyn, %I_stem', lang, lang);
+        END LOOP;
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during altering search configurations: %, %', SQLSTATE, SQLERRM;
+    END;
 END
 $$;