base_class.py 336 B

12345678910
  1. from sqlalchemy.ext.declarative import declarative_base
  2. from sqlalchemy import event
  3. from sqlalchemy import DDL
  4. Base = declarative_base()
  5. # 启用pgvector扩展
  6. event.listens_for(Base.metadata, 'before_create')
  7. def enable_vector_extension(target, connection, **kw):
  8. connection.execute(DDL('CREATE EXTENSION IF NOT EXISTS vector'))