使用SQLAlchemy执行CREATE EXTENSION时无报错但创建失败

以下是可能导致SQLAlchemy执行CREATE EXTENSION无报错但未生效的原因及对应解决办法：

• 事务未提交
  SQLAlchemy 2.0+版本中，engine.connect()会默认开启一个事务，所有操作都在事务内执行，若未显式提交，DDL语句不会生效。而psql默认处于自动提交模式，执行语句后立即生效。
  解决方法有两种：

  1. 手动提交事务：

     from sqlalchemy import create_engine, text
     engine = create_engine(f"postgresql+psycopg2://{cfg.DB.USERNAME}:{cfg.DB.PASSWORD}@{cfg.DB.URL}/{cfg.DB.NAME}")
     with engine.connect() as con:
         con.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
         con.commit()  # 显式提交事务
     

  2. 使用engine.begin()自动管理事务（执行完成后自动提交）：

     from sqlalchemy import create_engine, text
     engine = create_engine(f"postgresql+psycopg2://{cfg.DB.USERNAME}:{cfg.DB.PASSWORD}@{cfg.DB.URL}/{cfg.DB.NAME}")
     with engine.begin() as con:
         con.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
     

• 用户权限差异
  确认SQLAlchemy使用的数据库用户与psql连接的用户是否一致，创建扩展需要CREATE EXTENSION权限。若SQLAlchemy用户无此权限，扩展无法创建（部分场景下权限不足可能静默失败）。
  可执行以下SQL检查用户权限：

  SELECT usesuper, usecreatedb FROM pg_user WHERE usename = '你的用户名';
  

  若usesuper为f，可给用户授权：

  GRANT CREATE EXTENSION TO 你的用户名;
  

• 数据库连接目标不一致
  验证代码中cfg.DB.NAME指定的数据库是否与psql连接的数据库完全一致，若连接到不同数据库，扩展自然不会出现在预期的库中。

• Aurora Serverless V2的特殊处理
  虽然psql能创建扩展，但Aurora Serverless可能对DDL操作有延迟或特殊的事务处理机制，确保事务提交后，等待片刻再检查扩展是否存在。

内容的提问来源于stack exchange，提问作者Rizhiy