estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: keep idempotent when init AnalyticdbVectorBySql (#24239)

Browse Source 
Co-authored-by: xiaozeyu <xiaozeyu.xzy@alibaba-inc.com>
This commit is contained in:
8bitpd
2025-08-20 23:22:27 +08:00
committed by GitHub
parent f0684c1957
commit 6b1606f4f4
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/rag/datasource/vdb/analyticdb/analyticdb_vector_sql.py
View File

@@ -98,14 +98,19 @@ class AnalyticdbVectorBySql:
try: try:
cur.execute(f"CREATE DATABASE {self.databaseName}") cur.execute(f"CREATE DATABASE {self.databaseName}")
except Exception as e: except Exception as e:
if "already exists" in str(e): if "already exists" not in str(e):
return
raise e raise e
finally: finally:
cur.close() cur.close()
conn.close() conn.close()
self.pool = self._create_connection_pool() self.pool = self._create_connection_pool()
with self._get_cursor() as cur: with self._get_cursor() as cur:
try:
cur.execute("CREATE EXTENSION IF NOT EXISTS zhparser;")
except Exception as e:
raise RuntimeError(
"Failed to create zhparser extension. Please ensure it is available in your AnalyticDB."
) from e
try: try:
cur.execute("CREATE TEXT SEARCH CONFIGURATION zh_cn (PARSER = zhparser)") cur.execute("CREATE TEXT SEARCH CONFIGURATION zh_cn (PARSER = zhparser)")
cur.execute("ALTER TEXT SEARCH CONFIGURATION zh_cn ADD MAPPING FOR n,v,a,i,e,l,x WITH simple") cur.execute("ALTER TEXT SEARCH CONFIGURATION zh_cn ADD MAPPING FOR n,v,a,i,e,l,x WITH simple")