72 lines
3.5 KiB
PowerShell
72 lines
3.5 KiB
PowerShell
|
#Requires -Version 5.1
|
|||
|
|
<#
|
|||
|
|
.SYNOPSIS
|
|||
|
|
新机器一键灌库：jinrong_agent（底座+Agent专用）→ jinrong_core（模拟库+归属同步）→ 可选 RAG 知识库
|
|||
|
|
.EXAMPLE
|
|||
|
|
.\scripts\bootstrap.ps1 -MysqlPassword 123456 -SkipNeo4j
|
|||
|
|
.\scripts\bootstrap.ps1 -MysqlPassword 123456 -SkipNeo4j -WithKb # 连同 RAG 知识库一起灌入
|
|||
|
|
.NOTES
|
|||
|
|
前置依赖：
|
|||
|
|
1. Python 3.12 + 项目依赖（pip install -r requirements.txt）
|
|||
|
|
2. MySQL 8.x 本地运行，且 mysql CLI 在 PATH 中
|
|||
|
|
3. Redis 8.x 本地运行（会话短期记忆）
|
|||
|
|
4. Milvus（Lite 本地文件或 standalone）运行中
|
|||
|
|
5. .env 已按 .env.example 配置（DEEPSEEK_API_KEY、KB_ROOT_DIR 等）
|
|||
|
|
灌库完成后：agent 库 18 张表（底座 11 + Agent 专用 7），core 模拟库种子数据为相对跑库日日期。
|
|||
|
|
#>
|
|||
|
|
param(
|
|||
|
|
[string]$MysqlUser = "root",
|
|||
|
|
[string]$MysqlHost = "127.0.0.1",
|
|||
|
|
[string]$MysqlPassword = "",
|
|||
|
|
[switch]$SkipNeo4j,
|
|||
|
|
[switch]$WithKb
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
$ErrorActionPreference = "Stop"
|
|||
|
|
$Root = Split-Path $PSScriptRoot -Parent
|
|||
|
|
Set-Location $Root
|
|||
|
|
|
|||
|
|
# 密码经 MYSQL_PWD 环境变量传递：Python 脚本与 mysql CLI 共用，避免命令行明文
|
|||
|
|
if ($MysqlPassword -ne "") { $env:MYSQL_PWD = $MysqlPassword }
|
|||
|
|
if (-not $env:MYSQL_PWD) {
|
|||
|
|
Write-Error "缺少密码：请用 -MysqlPassword 传参，或预设 MYSQL_PWD 环境变量"
|
|||
|
|
exit 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# --- 前置检查 -------------------------------------------------------------
|
|||
|
|
Write-Host "=== 前置检查 ===" -ForegroundColor Cyan
|
|||
|
|
foreach ($cmd in @("python", "mysql")) {
|
|||
|
|
if (-not (Get-Command $cmd -ErrorAction SilentlyContinue)) {
|
|||
|
|
Write-Error "未找到命令 '$cmd'，请先安装并加入 PATH"
|
|||
|
|
exit 1
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (-not (Test-Path (Join-Path $Root ".env"))) {
|
|||
|
|
Write-Host "[WARN] 未找到 .env，请复制 .env.example 为 .env 并填写 DEEPSEEK_API_KEY 等配置" -ForegroundColor Yellow
|
|||
|
|
}
|
|||
|
|
Write-Host "python / mysql 均可用" -ForegroundColor Green
|
|||
|
|
|
|||
|
|
# --- 步骤 1：jinrong_agent（01 底座 11 张 + 02 Agent 专用 7 张）-----------
|
|||
|
|
Write-Host "`n=== [1/3] 初始化 jinrong_agent（底座 + Agent 专用表）===" -ForegroundColor Cyan
|
|||
|
|
python scripts/init_agent_db.py --mysql-host $MysqlHost --mysql-user $MysqlUser
|
|||
|
|
if ($LASTEXITCODE -ne 0) { Write-Error "agent 库初始化失败" }
|
|||
|
|
|
|||
|
|
# --- 步骤 2：jinrong_core 模拟库（DROP 重建 + 种子 + 归属同步）------------
|
|||
|
|
Write-Host "`n=== [2/3] 重建 jinrong_core 模拟库并同步归属 ===" -ForegroundColor Cyan
|
|||
|
|
$resetArgs = @("-MysqlPassword", $MysqlPassword, "-MysqlHost", $MysqlHost, "-MysqlUser", $MysqlUser)
|
|||
|
|
if ($SkipNeo4j) { $resetArgs += "-SkipNeo4j" }
|
|||
|
|
& powershell -ExecutionPolicy Bypass -File (Join-Path $Root "scripts/core/reset.ps1") @resetArgs
|
|||
|
|
if ($LASTEXITCODE -ne 0) { Write-Error "core 模拟库重建失败" }
|
|||
|
|
|
|||
|
|
# --- 步骤 3（可选）：RAG 知识库灌入 Milvus --------------------------------
|
|||
|
|
if ($WithKb) {
|
|||
|
|
Write-Host "`n=== [3/3] RAG 知识库灌入 Milvus ===" -ForegroundColor Cyan
|
|||
|
|
& powershell -ExecutionPolicy Bypass -File (Join-Path $Root "scripts/kb/reset.ps1") -Rebuild
|
|||
|
|
if ($LASTEXITCODE -ne 0) { Write-Host "[WARN] 知识库灌入失败，请检查 Milvus / DEEPSEEK_API_KEY / KB_ROOT_DIR" -ForegroundColor Yellow }
|
|||
|
|
} else {
|
|||
|
|
Write-Host "`n=== [3/3] 跳过 RAG 知识库（如需灌入加 -WithKb；需自备知识库源文档并配置 KB_ROOT_DIR）===" -ForegroundColor DarkGray
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Write-Host "`n=== 全部完成 ===" -ForegroundColor Green
|
|||
|
|
Write-Host "后续：确认 Redis 8.x 与 Milvus 已启动，.env 配置就绪后即可启动服务（uvicorn app.main:app --reload）" -ForegroundColor Green
|