瀏覽代碼

fix: rollback when AnalyticDB create zhparser failed (#24260)

Co-authored-by: xiaozeyu <xiaozeyu.xzy@alibaba-inc.com>
tags/1.8.0
8bitpd 2 月之前
父節點
當前提交
a183b2affb
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 3 行新增0 行删除
  1. 3
    0
      api/core/rag/datasource/vdb/analyticdb/analyticdb_vector_sql.py

+ 3
- 0
api/core/rag/datasource/vdb/analyticdb/analyticdb_vector_sql.py 查看文件

@@ -105,9 +105,11 @@ class AnalyticdbVectorBySql:
conn.close()
self.pool = self._create_connection_pool()
with self._get_cursor() as cur:
conn = cur.connection
try:
cur.execute("CREATE EXTENSION IF NOT EXISTS zhparser;")
except Exception as e:
conn.rollback()
raise RuntimeError(
"Failed to create zhparser extension. Please ensure it is available in your AnalyticDB."
) from e
@@ -115,6 +117,7 @@ class AnalyticdbVectorBySql:
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")
except Exception as e:
conn.rollback()
if "already exists" not in str(e):
raise e
cur.execute(

Loading…
取消
儲存