Browse Source

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

Co-authored-by: xiaozeyu <xiaozeyu.xzy@alibaba-inc.com>
tags/1.8.0
8bitpd 2 months ago
parent
commit
a183b2affb
No account linked to committer's email address
1 changed files with 3 additions and 0 deletions
  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 View File

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:
conn = cur.connection
try: try:
cur.execute("CREATE EXTENSION IF NOT EXISTS zhparser;") cur.execute("CREATE EXTENSION IF NOT EXISTS zhparser;")
except Exception as e: except Exception as e:
conn.rollback()
raise RuntimeError( raise RuntimeError(
"Failed to create zhparser extension. Please ensure it is available in your AnalyticDB." "Failed to create zhparser extension. Please ensure it is available in your AnalyticDB."
) from e ) from e
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")
except Exception as e: except Exception as e:
conn.rollback()
if "already exists" not in str(e): if "already exists" not in str(e):
raise e raise e
cur.execute( cur.execute(

Loading…
Cancel
Save