25 lines
684 B
Rust
25 lines
684 B
Rust
use sea_orm_migration::prelude::*;
|
|
|
|
#[derive(DeriveMigrationName)]
|
|
pub struct Migration;
|
|
|
|
#[async_trait::async_trait]
|
|
impl MigrationTrait for Migration {
|
|
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
// 启用 pgvector 扩展（知识库向量检索需要）
|
|
manager
|
|
.get_connection()
|
|
.execute_unprepared("CREATE EXTENSION IF NOT EXISTS vector")
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
manager
|
|
.get_connection()
|
|
.execute_unprepared("DROP EXTENSION IF EXISTS vector")
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
}
|