Initial commit: iConsulting 香港移民咨询智能客服系统
项目架构: - Monorepo (pnpm + Turborepo) - 后端: NestJS 微服务 + Claude Agent SDK - 前端: React + Vite + Ant Design 包含服务: - conversation-service: 对话服务 (Claude AI) - user-service: 用户认证服务 - payment-service: 支付服务 (支付宝/微信/Stripe) - knowledge-service: 知识库服务 (RAG + Neo4j) - evolution-service: 自我进化服务 - web-client: 用户前端 - admin-client: 管理后台 基础设施: - PostgreSQL + Redis + Neo4j - Kong API Gateway - Nginx 反向代理 - Docker Compose 部署配置 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
commit
a7add8ff90
|
|
@ -0,0 +1,13 @@
|
||||||
|
{
|
||||||
|
"permissions": {
|
||||||
|
"allow": [
|
||||||
|
"Bash(pnpm install:*)",
|
||||||
|
"Bash(npm install:*)",
|
||||||
|
"Bash(npx tsc:*)",
|
||||||
|
"Bash(pnpm add:*)",
|
||||||
|
"Bash(git init:*)",
|
||||||
|
"Bash(git checkout:*)",
|
||||||
|
"Bash(git add:*)"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,72 @@
|
||||||
|
# ===========================================
|
||||||
|
# Docker 忽略文件
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
# 依赖
|
||||||
|
node_modules
|
||||||
|
**/node_modules
|
||||||
|
|
||||||
|
# 构建产物
|
||||||
|
dist
|
||||||
|
**/dist
|
||||||
|
.next
|
||||||
|
.nuxt
|
||||||
|
|
||||||
|
# 日志
|
||||||
|
logs
|
||||||
|
*.log
|
||||||
|
npm-debug.log*
|
||||||
|
pnpm-debug.log*
|
||||||
|
|
||||||
|
# 环境配置
|
||||||
|
.env
|
||||||
|
.env.*
|
||||||
|
!.env.example
|
||||||
|
|
||||||
|
# IDE
|
||||||
|
.idea
|
||||||
|
.vscode
|
||||||
|
*.swp
|
||||||
|
*.swo
|
||||||
|
|
||||||
|
# 系统文件
|
||||||
|
.DS_Store
|
||||||
|
Thumbs.db
|
||||||
|
|
||||||
|
# Git
|
||||||
|
.git
|
||||||
|
.gitignore
|
||||||
|
|
||||||
|
# 测试
|
||||||
|
coverage
|
||||||
|
.nyc_output
|
||||||
|
test
|
||||||
|
tests
|
||||||
|
**/*.test.ts
|
||||||
|
**/*.spec.ts
|
||||||
|
|
||||||
|
# 文档
|
||||||
|
docs
|
||||||
|
*.md
|
||||||
|
!README.md
|
||||||
|
|
||||||
|
# 临时文件
|
||||||
|
tmp
|
||||||
|
temp
|
||||||
|
.tmp
|
||||||
|
.temp
|
||||||
|
|
||||||
|
# 备份
|
||||||
|
backups
|
||||||
|
*.bak
|
||||||
|
|
||||||
|
# 证书 (敏感)
|
||||||
|
*.pem
|
||||||
|
*.key
|
||||||
|
*.crt
|
||||||
|
nginx/ssl/*
|
||||||
|
|
||||||
|
# 其他
|
||||||
|
.turbo
|
||||||
|
.cache
|
||||||
|
.parcel-cache
|
||||||
|
|
@ -0,0 +1,151 @@
|
||||||
|
# ===========================================
|
||||||
|
# iConsulting - Environment Configuration
|
||||||
|
# ===========================================
|
||||||
|
# 使用方法: 复制此文件为 .env 并填入实际值
|
||||||
|
# cp .env.example .env
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
# Application
|
||||||
|
NODE_ENV=production
|
||||||
|
APP_NAME=iConsulting
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# 服务器网络配置
|
||||||
|
# ===========================================
|
||||||
|
# 对外服务 IP (用户访问)
|
||||||
|
SERVER_PUBLIC_IP=14.215.128.96
|
||||||
|
# Claude API 出口 IP
|
||||||
|
CLAUDE_API_OUTBOUND_IP=154.84.135.121
|
||||||
|
# Claude API 服务器
|
||||||
|
CLAUDE_API_SERVER=67.223.119.33
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Anthropic Claude API
|
||||||
|
# ===========================================
|
||||||
|
ANTHROPIC_API_KEY=sk-ant-api03-xxx
|
||||||
|
ANTHROPIC_BASE_URL=https://api.anthropic.com
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# OpenAI API (用于 Embedding)
|
||||||
|
# ===========================================
|
||||||
|
OPENAI_API_KEY=sk-xxx
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# PostgreSQL Database
|
||||||
|
# ===========================================
|
||||||
|
POSTGRES_HOST=localhost
|
||||||
|
POSTGRES_PORT=5432
|
||||||
|
POSTGRES_USER=postgres
|
||||||
|
POSTGRES_PASSWORD=your_secure_password
|
||||||
|
POSTGRES_DB=iconsulting
|
||||||
|
DATABASE_URL=postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:${POSTGRES_PORT}/${POSTGRES_DB}
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Neo4j Graph Database
|
||||||
|
# ===========================================
|
||||||
|
NEO4J_URI=bolt://localhost:7687
|
||||||
|
NEO4J_USER=neo4j
|
||||||
|
NEO4J_PASSWORD=your_secure_password
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Redis Cache
|
||||||
|
# ===========================================
|
||||||
|
REDIS_HOST=localhost
|
||||||
|
REDIS_PORT=6379
|
||||||
|
REDIS_PASSWORD=your_redis_password
|
||||||
|
REDIS_URL=redis://:${REDIS_PASSWORD}@${REDIS_HOST}:${REDIS_PORT}
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Payment - Alipay
|
||||||
|
# ===========================================
|
||||||
|
ALIPAY_APP_ID=your_app_id
|
||||||
|
ALIPAY_PRIVATE_KEY=your_private_key
|
||||||
|
ALIPAY_PUBLIC_KEY=alipay_public_key
|
||||||
|
ALIPAY_GATEWAY=https://openapi.alipay.com/gateway.do
|
||||||
|
ALIPAY_NOTIFY_URL=https://your-domain.com/api/v1/payments/alipay/notify
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Payment - WeChat Pay
|
||||||
|
# ===========================================
|
||||||
|
WECHAT_APP_ID=your_app_id
|
||||||
|
WECHAT_MCH_ID=your_merchant_id
|
||||||
|
WECHAT_API_KEY=your_api_key
|
||||||
|
WECHAT_CERT_PATH=/path/to/wechat/cert
|
||||||
|
WECHAT_NOTIFY_URL=https://your-domain.com/api/v1/payments/wechat/notify
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Payment - Stripe (Credit Card)
|
||||||
|
# ===========================================
|
||||||
|
STRIPE_SECRET_KEY=sk_test_xxx
|
||||||
|
STRIPE_PUBLISHABLE_KEY=pk_test_xxx
|
||||||
|
STRIPE_WEBHOOK_SECRET=whsec_xxx
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Payment Callback URLs
|
||||||
|
# ===========================================
|
||||||
|
PAYMENT_CALLBACK_BASE_URL=https://your-domain.com
|
||||||
|
PAYMENT_SUCCESS_REDIRECT_URL=https://your-domain.com/payment/success
|
||||||
|
PAYMENT_CANCEL_REDIRECT_URL=https://your-domain.com/payment/cancel
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# JWT Authentication
|
||||||
|
# ===========================================
|
||||||
|
JWT_SECRET=your_super_secret_jwt_key_change_in_production
|
||||||
|
JWT_EXPIRES_IN=7d
|
||||||
|
JWT_REFRESH_EXPIRES_IN=30d
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# SMS Service (for phone verification)
|
||||||
|
# ===========================================
|
||||||
|
SMS_PROVIDER=aliyun
|
||||||
|
ALIYUN_SMS_ACCESS_KEY_ID=your_access_key_id
|
||||||
|
ALIYUN_SMS_ACCESS_KEY_SECRET=your_access_key_secret
|
||||||
|
ALIYUN_SMS_SIGN_NAME=iConsulting
|
||||||
|
ALIYUN_SMS_TEMPLATE_CODE=SMS_xxx
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Service Ports
|
||||||
|
# ===========================================
|
||||||
|
USER_SERVICE_PORT=3001
|
||||||
|
PAYMENT_SERVICE_PORT=3002
|
||||||
|
KNOWLEDGE_SERVICE_PORT=3003
|
||||||
|
CONVERSATION_SERVICE_PORT=3004
|
||||||
|
EVOLUTION_SERVICE_PORT=3005
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# 服务间通信 URL
|
||||||
|
# ===========================================
|
||||||
|
USER_SERVICE_URL=http://localhost:3001
|
||||||
|
PAYMENT_SERVICE_URL=http://localhost:3002
|
||||||
|
KNOWLEDGE_SERVICE_URL=http://localhost:3003
|
||||||
|
CONVERSATION_SERVICE_URL=http://localhost:3004
|
||||||
|
EVOLUTION_SERVICE_URL=http://localhost:3005
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Kong API Gateway
|
||||||
|
# ===========================================
|
||||||
|
KONG_PROXY_PORT=8000
|
||||||
|
KONG_ADMIN_PORT=8001
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Frontend URLs
|
||||||
|
# ===========================================
|
||||||
|
WEB_CLIENT_URL=http://localhost
|
||||||
|
ADMIN_CLIENT_URL=http://localhost/admin
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# CORS
|
||||||
|
# ===========================================
|
||||||
|
CORS_ORIGINS=http://localhost,http://14.215.128.96
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Rate Limiting
|
||||||
|
# ===========================================
|
||||||
|
RATE_LIMIT_TTL=60
|
||||||
|
RATE_LIMIT_MAX=100
|
||||||
|
|
||||||
|
# ===========================================
|
||||||
|
# Logging
|
||||||
|
# ===========================================
|
||||||
|
LOG_LEVEL=info
|
||||||
|
LOG_FORMAT=json
|
||||||
|
|
@ -0,0 +1,97 @@
|
||||||
|
# ===========================================
|
||||||
|
# iConsulting Git 忽略配置
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
# 依赖目录
|
||||||
|
node_modules/
|
||||||
|
**/node_modules/
|
||||||
|
.pnpm-store/
|
||||||
|
|
||||||
|
# 构建产物
|
||||||
|
dist/
|
||||||
|
**/dist/
|
||||||
|
build/
|
||||||
|
.next/
|
||||||
|
.nuxt/
|
||||||
|
.output/
|
||||||
|
out/
|
||||||
|
|
||||||
|
# 环境变量 (包含敏感信息)
|
||||||
|
.env
|
||||||
|
.env.local
|
||||||
|
.env.*.local
|
||||||
|
!.env.example
|
||||||
|
|
||||||
|
# 日志文件
|
||||||
|
logs/
|
||||||
|
*.log
|
||||||
|
npm-debug.log*
|
||||||
|
pnpm-debug.log*
|
||||||
|
yarn-debug.log*
|
||||||
|
yarn-error.log*
|
||||||
|
|
||||||
|
# 运行时数据
|
||||||
|
pids/
|
||||||
|
*.pid
|
||||||
|
*.seed
|
||||||
|
*.pid.lock
|
||||||
|
|
||||||
|
# IDE 和编辑器
|
||||||
|
.idea/
|
||||||
|
.vscode/
|
||||||
|
*.swp
|
||||||
|
*.swo
|
||||||
|
*.sublime-workspace
|
||||||
|
*.sublime-project
|
||||||
|
|
||||||
|
# 系统文件
|
||||||
|
.DS_Store
|
||||||
|
Thumbs.db
|
||||||
|
desktop.ini
|
||||||
|
|
||||||
|
# 测试覆盖率
|
||||||
|
coverage/
|
||||||
|
.nyc_output/
|
||||||
|
*.lcov
|
||||||
|
|
||||||
|
# 缓存
|
||||||
|
.cache/
|
||||||
|
.parcel-cache/
|
||||||
|
.turbo/
|
||||||
|
.eslintcache
|
||||||
|
.stylelintcache
|
||||||
|
*.tsbuildinfo
|
||||||
|
|
||||||
|
# 临时文件
|
||||||
|
tmp/
|
||||||
|
temp/
|
||||||
|
.tmp/
|
||||||
|
.temp/
|
||||||
|
*.tmp
|
||||||
|
|
||||||
|
# 备份文件
|
||||||
|
backups/
|
||||||
|
*.bak
|
||||||
|
*~
|
||||||
|
|
||||||
|
# SSL 证书 (敏感)
|
||||||
|
*.pem
|
||||||
|
*.key
|
||||||
|
*.crt
|
||||||
|
*.p12
|
||||||
|
nginx/ssl/*
|
||||||
|
!nginx/ssl/.gitkeep
|
||||||
|
|
||||||
|
# 数据库文件
|
||||||
|
*.sqlite
|
||||||
|
*.db
|
||||||
|
|
||||||
|
# Docker
|
||||||
|
docker-compose.override.yml
|
||||||
|
|
||||||
|
# PM2
|
||||||
|
.pm2/
|
||||||
|
|
||||||
|
# 打包文件
|
||||||
|
*.tar.gz
|
||||||
|
*.zip
|
||||||
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,176 @@
|
||||||
|
# iConsulting - 香港移民在线咨询系统
|
||||||
|
|
||||||
|
基于 Claude Agent SDK 的智能在线客服系统，专注于提供香港移民咨询服务。
|
||||||
|
|
||||||
|
## 功能特性
|
||||||
|
|
||||||
|
- **智能咨询**: 基于 Claude Agent SDK 的自然语言对话
|
||||||
|
- **付费评估**: 移民资格评估服务，支持支付宝/微信/信用卡
|
||||||
|
- **知识增强**: RAG + Neo4j 知识图谱
|
||||||
|
- **自我进化**: 从对话中学习，根据管理员指令调整
|
||||||
|
- **长期记忆**: 基于时间线的知识图谱记录
|
||||||
|
- **多端支持**: PC Web / H5 响应式设计
|
||||||
|
|
||||||
|
## 支持的移民类别
|
||||||
|
|
||||||
|
1. **优才计划 (QMAS)** - 行业翘楚、精英人士
|
||||||
|
2. **专才计划 (GEP)** - 专业人才
|
||||||
|
3. **留学IANG** - 非本地毕业生
|
||||||
|
4. **高才通 (TTPS)** - 高端人才
|
||||||
|
5. **投资移民 (CIES)** - 投资者
|
||||||
|
6. **科技人才 (TechTAS)** - 科技领域人才
|
||||||
|
|
||||||
|
## 技术架构
|
||||||
|
|
||||||
|
```
|
||||||
|
├── 前端 (Clean Architecture)
|
||||||
|
│ ├── React 18 + TypeScript
|
||||||
|
│ ├── TailwindCSS + Radix UI
|
||||||
|
│ └── Zustand + React Query
|
||||||
|
│
|
||||||
|
├── 后端 (DDD + Hexagonal + 微服务)
|
||||||
|
│ ├── NestJS
|
||||||
|
│ ├── Claude Agent SDK
|
||||||
|
│ └── TypeORM
|
||||||
|
│
|
||||||
|
└── 基础设施
|
||||||
|
├── PostgreSQL + pgvector (RAG)
|
||||||
|
├── Neo4j (知识图谱)
|
||||||
|
├── Redis (缓存)
|
||||||
|
└── Kafka (消息队列)
|
||||||
|
```
|
||||||
|
|
||||||
|
## 项目结构
|
||||||
|
|
||||||
|
```
|
||||||
|
iconsulting/
|
||||||
|
├── packages/
|
||||||
|
│ ├── shared/ # 共享类型、常量、工具
|
||||||
|
│ ├── web-client/ # 用户端 Web 应用
|
||||||
|
│ ├── admin-client/ # 管理端 Web 应用
|
||||||
|
│ └── services/ # 后端微服务
|
||||||
|
│ ├── api-gateway/ # API 网关
|
||||||
|
│ ├── user-service/ # 用户服务
|
||||||
|
│ ├── conversation-service/ # 对话服务 (核心)
|
||||||
|
│ ├── knowledge-service/ # 知识服务
|
||||||
|
│ ├── payment-service/ # 支付服务
|
||||||
|
│ ├── admin-service/ # 管理服务
|
||||||
|
│ └── evolution-service/ # 进化服务
|
||||||
|
│
|
||||||
|
├── infrastructure/
|
||||||
|
│ └── docker/ # Docker 配置
|
||||||
|
│
|
||||||
|
├── DEVELOPMENT_GUIDE.md # 详细开发指导
|
||||||
|
└── README.md
|
||||||
|
```
|
||||||
|
|
||||||
|
## 快速开始
|
||||||
|
|
||||||
|
### 1. 安装依赖
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 安装 pnpm (如果没有)
|
||||||
|
npm install -g pnpm
|
||||||
|
|
||||||
|
# 安装项目依赖
|
||||||
|
pnpm install
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. 启动基础设施
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 启动 Docker 容器 (PostgreSQL, Neo4j, Redis, Kafka)
|
||||||
|
pnpm docker:dev
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. 配置环境变量
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 复制环境变量示例文件
|
||||||
|
cp .env.example .env
|
||||||
|
|
||||||
|
# 编辑 .env 文件，填入必要的配置
|
||||||
|
# 特别是 ANTHROPIC_API_KEY
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4. 运行数据库迁移
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pnpm db:migrate
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5. 启动开发服务器
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 启动所有服务
|
||||||
|
pnpm dev
|
||||||
|
```
|
||||||
|
|
||||||
|
访问:
|
||||||
|
- 用户端: http://localhost:5173
|
||||||
|
- 管理端: http://localhost:5174
|
||||||
|
- API: http://localhost:3000
|
||||||
|
|
||||||
|
## 环境变量
|
||||||
|
|
||||||
|
关键配置项:
|
||||||
|
|
||||||
|
```env
|
||||||
|
# Claude API
|
||||||
|
ANTHROPIC_API_KEY=sk-ant-xxx
|
||||||
|
|
||||||
|
# 数据库
|
||||||
|
POSTGRES_HOST=localhost
|
||||||
|
POSTGRES_PORT=5432
|
||||||
|
POSTGRES_USER=iconsulting
|
||||||
|
POSTGRES_PASSWORD=your_password
|
||||||
|
POSTGRES_DB=iconsulting
|
||||||
|
|
||||||
|
# Neo4j
|
||||||
|
NEO4J_URI=bolt://localhost:7687
|
||||||
|
NEO4J_USER=neo4j
|
||||||
|
NEO4J_PASSWORD=your_password
|
||||||
|
|
||||||
|
# 支付 (支付宝/微信)
|
||||||
|
ALIPAY_APP_ID=xxx
|
||||||
|
WECHAT_APP_ID=xxx
|
||||||
|
```
|
||||||
|
|
||||||
|
完整配置请参考 `.env.example`
|
||||||
|
|
||||||
|
## 开发指南
|
||||||
|
|
||||||
|
详细的开发指导请参考 [DEVELOPMENT_GUIDE.md](./DEVELOPMENT_GUIDE.md)
|
||||||
|
|
||||||
|
## 开发进度
|
||||||
|
|
||||||
|
### 已完成
|
||||||
|
|
||||||
|
- [x] 项目架构设计
|
||||||
|
- [x] 开发指导文档
|
||||||
|
- [x] Monorepo 配置
|
||||||
|
- [x] 共享类型定义
|
||||||
|
- [x] Docker 基础设施配置
|
||||||
|
- [x] 数据库 Schema
|
||||||
|
- [x] 对话服务 (Claude Agent SDK 集成)
|
||||||
|
- [x] 用户端前端基础框架
|
||||||
|
|
||||||
|
### 进行中
|
||||||
|
|
||||||
|
- [ ] 用户服务
|
||||||
|
- [ ] 知识服务 (RAG + Neo4j)
|
||||||
|
- [ ] 支付服务
|
||||||
|
- [ ] 管理服务 (自我进化)
|
||||||
|
- [ ] 管理后台前端
|
||||||
|
|
||||||
|
## 贡献指南
|
||||||
|
|
||||||
|
1. Fork 本仓库
|
||||||
|
2. 创建特性分支 (`git checkout -b feature/amazing-feature`)
|
||||||
|
3. 提交更改 (`git commit -m 'Add some amazing feature'`)
|
||||||
|
4. 推送到分支 (`git push origin feature/amazing-feature`)
|
||||||
|
5. 创建 Pull Request
|
||||||
|
|
||||||
|
## 许可证
|
||||||
|
|
||||||
|
私有项目，保留所有权利。
|
||||||
|
|
@ -0,0 +1,903 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# iConsulting 部署管理脚本
|
||||||
|
#
|
||||||
|
# 用法: ./deploy.sh <command> [service] [options]
|
||||||
|
#
|
||||||
|
# 命令:
|
||||||
|
# build - 编译构建
|
||||||
|
# start - 启动服务
|
||||||
|
# stop - 停止服务
|
||||||
|
# restart - 重启服务
|
||||||
|
# status - 查看状态
|
||||||
|
# logs - 查看日志
|
||||||
|
# clean - 清理构建产物
|
||||||
|
# deploy - 完整部署（构建+启动）
|
||||||
|
# db - 数据库操作
|
||||||
|
# help - 显示帮助
|
||||||
|
#
|
||||||
|
# 服务:
|
||||||
|
# all - 所有服务
|
||||||
|
# web-client - 用户前端
|
||||||
|
# admin-client - 管理后台前端
|
||||||
|
# conversation - 对话服务
|
||||||
|
# user - 用户服务
|
||||||
|
# payment - 支付服务
|
||||||
|
# knowledge - 知识库服务
|
||||||
|
# evolution - 进化服务
|
||||||
|
# kong - API网关
|
||||||
|
# postgres - PostgreSQL数据库
|
||||||
|
# redis - Redis缓存
|
||||||
|
# neo4j - Neo4j图数据库
|
||||||
|
# nginx - Nginx静态服务
|
||||||
|
#
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
# 颜色定义
|
||||||
|
RED='\033[0;31m'
|
||||||
|
GREEN='\033[0;32m'
|
||||||
|
YELLOW='\033[1;33m'
|
||||||
|
BLUE='\033[0;34m'
|
||||||
|
PURPLE='\033[0;35m'
|
||||||
|
CYAN='\033[0;36m'
|
||||||
|
NC='\033[0m' # No Color
|
||||||
|
|
||||||
|
# 项目根目录
|
||||||
|
PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
cd "$PROJECT_ROOT"
|
||||||
|
|
||||||
|
# 配置
|
||||||
|
COMPOSE_FILE="docker-compose.yml"
|
||||||
|
ENV_FILE=".env"
|
||||||
|
|
||||||
|
# 服务端口配置
|
||||||
|
declare -A SERVICE_PORTS=(
|
||||||
|
["conversation"]=3004
|
||||||
|
["user"]=3001
|
||||||
|
["payment"]=3002
|
||||||
|
["knowledge"]=3003
|
||||||
|
["evolution"]=3005
|
||||||
|
["kong"]=8000
|
||||||
|
["postgres"]=5432
|
||||||
|
["redis"]=6379
|
||||||
|
["neo4j"]=7474
|
||||||
|
["nginx"]=80
|
||||||
|
)
|
||||||
|
|
||||||
|
# 服务目录映射
|
||||||
|
declare -A SERVICE_DIRS=(
|
||||||
|
["conversation"]="packages/services/conversation-service"
|
||||||
|
["user"]="packages/services/user-service"
|
||||||
|
["payment"]="packages/services/payment-service"
|
||||||
|
["knowledge"]="packages/services/knowledge-service"
|
||||||
|
["evolution"]="packages/services/evolution-service"
|
||||||
|
["web-client"]="packages/web-client"
|
||||||
|
["admin-client"]="packages/admin-client"
|
||||||
|
["shared"]="packages/shared"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Docker服务名映射
|
||||||
|
declare -A DOCKER_SERVICES=(
|
||||||
|
["conversation"]="conversation-service"
|
||||||
|
["user"]="user-service"
|
||||||
|
["payment"]="payment-service"
|
||||||
|
["knowledge"]="knowledge-service"
|
||||||
|
["evolution"]="evolution-service"
|
||||||
|
["web-client"]="web-client"
|
||||||
|
["admin-client"]="admin-client"
|
||||||
|
["kong"]="kong"
|
||||||
|
["postgres"]="postgres"
|
||||||
|
["redis"]="redis"
|
||||||
|
["neo4j"]="neo4j"
|
||||||
|
["nginx"]="nginx"
|
||||||
|
)
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 工具函数
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
log_info() {
|
||||||
|
echo -e "${BLUE}[INFO]${NC} $1"
|
||||||
|
}
|
||||||
|
|
||||||
|
log_success() {
|
||||||
|
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||||||
|
}
|
||||||
|
|
||||||
|
log_warning() {
|
||||||
|
echo -e "${YELLOW}[WARNING]${NC} $1"
|
||||||
|
}
|
||||||
|
|
||||||
|
log_error() {
|
||||||
|
echo -e "${RED}[ERROR]${NC} $1"
|
||||||
|
}
|
||||||
|
|
||||||
|
log_step() {
|
||||||
|
echo -e "${PURPLE}[STEP]${NC} $1"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 检查命令是否存在
|
||||||
|
check_command() {
|
||||||
|
if ! command -v "$1" &> /dev/null; then
|
||||||
|
log_error "$1 未安装，请先安装"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 检查环境
|
||||||
|
check_environment() {
|
||||||
|
log_step "检查运行环境..."
|
||||||
|
|
||||||
|
check_command "node"
|
||||||
|
check_command "pnpm"
|
||||||
|
check_command "docker"
|
||||||
|
check_command "docker-compose"
|
||||||
|
|
||||||
|
# 检查 Node 版本
|
||||||
|
NODE_VERSION=$(node -v | cut -d'v' -f2 | cut -d'.' -f1)
|
||||||
|
if [ "$NODE_VERSION" -lt 18 ]; then
|
||||||
|
log_error "Node.js 版本需要 >= 18，当前版本: $(node -v)"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_success "环境检查通过"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 加载环境变量
|
||||||
|
load_env() {
|
||||||
|
if [ -f "$ENV_FILE" ]; then
|
||||||
|
export $(grep -v '^#' "$ENV_FILE" | xargs)
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 等待服务就绪
|
||||||
|
wait_for_service() {
|
||||||
|
local host=$1
|
||||||
|
local port=$2
|
||||||
|
local service=$3
|
||||||
|
local max_attempts=${4:-30}
|
||||||
|
local attempt=1
|
||||||
|
|
||||||
|
log_info "等待 $service ($host:$port) 就绪..."
|
||||||
|
|
||||||
|
while [ $attempt -le $max_attempts ]; do
|
||||||
|
if nc -z "$host" "$port" 2>/dev/null; then
|
||||||
|
log_success "$service 已就绪"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
echo -n "."
|
||||||
|
sleep 2
|
||||||
|
attempt=$((attempt + 1))
|
||||||
|
done
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
log_error "$service 启动超时"
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 构建函数
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
# 安装依赖
|
||||||
|
install_deps() {
|
||||||
|
log_step "安装项目依赖..."
|
||||||
|
pnpm install
|
||||||
|
log_success "依赖安装完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建共享包
|
||||||
|
build_shared() {
|
||||||
|
log_step "构建 shared 包..."
|
||||||
|
cd "$PROJECT_ROOT/${SERVICE_DIRS[shared]}"
|
||||||
|
pnpm run build
|
||||||
|
cd "$PROJECT_ROOT"
|
||||||
|
log_success "shared 构建完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建单个后端服务
|
||||||
|
build_backend_service() {
|
||||||
|
local service=$1
|
||||||
|
local dir="${SERVICE_DIRS[$service]}"
|
||||||
|
|
||||||
|
if [ -z "$dir" ]; then
|
||||||
|
log_error "未知服务: $service"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_step "构建 $service..."
|
||||||
|
cd "$PROJECT_ROOT/$dir"
|
||||||
|
|
||||||
|
# 清理旧构建
|
||||||
|
rm -rf dist
|
||||||
|
|
||||||
|
# TypeScript 编译
|
||||||
|
pnpm run build
|
||||||
|
|
||||||
|
cd "$PROJECT_ROOT"
|
||||||
|
log_success "$service 构建完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建单个前端
|
||||||
|
build_frontend() {
|
||||||
|
local service=$1
|
||||||
|
local dir="${SERVICE_DIRS[$service]}"
|
||||||
|
|
||||||
|
if [ -z "$dir" ]; then
|
||||||
|
log_error "未知服务: $service"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_step "构建 $service..."
|
||||||
|
cd "$PROJECT_ROOT/$dir"
|
||||||
|
|
||||||
|
# 清理旧构建
|
||||||
|
rm -rf dist
|
||||||
|
|
||||||
|
# Vite 构建
|
||||||
|
pnpm run build
|
||||||
|
|
||||||
|
cd "$PROJECT_ROOT"
|
||||||
|
log_success "$service 构建完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建所有后端服务
|
||||||
|
build_all_backend() {
|
||||||
|
build_shared
|
||||||
|
|
||||||
|
for service in conversation user payment knowledge evolution; do
|
||||||
|
build_backend_service "$service"
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建所有前端
|
||||||
|
build_all_frontend() {
|
||||||
|
for service in web-client admin-client; do
|
||||||
|
build_frontend "$service"
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建所有
|
||||||
|
build_all() {
|
||||||
|
log_info "开始构建所有服务..."
|
||||||
|
install_deps
|
||||||
|
build_all_backend
|
||||||
|
build_all_frontend
|
||||||
|
log_success "所有服务构建完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 构建入口
|
||||||
|
do_build() {
|
||||||
|
local target=${1:-all}
|
||||||
|
|
||||||
|
case $target in
|
||||||
|
all)
|
||||||
|
build_all
|
||||||
|
;;
|
||||||
|
shared)
|
||||||
|
build_shared
|
||||||
|
;;
|
||||||
|
backend)
|
||||||
|
build_all_backend
|
||||||
|
;;
|
||||||
|
frontend)
|
||||||
|
build_all_frontend
|
||||||
|
;;
|
||||||
|
web-client|admin-client)
|
||||||
|
build_frontend "$target"
|
||||||
|
;;
|
||||||
|
conversation|user|payment|knowledge|evolution)
|
||||||
|
build_backend_service "$target"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_error "未知构建目标: $target"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# Docker 操作函数
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
# 构建 Docker 镜像
|
||||||
|
build_docker_images() {
|
||||||
|
local service=${1:-}
|
||||||
|
|
||||||
|
log_step "构建 Docker 镜像..."
|
||||||
|
|
||||||
|
if [ -n "$service" ] && [ "$service" != "all" ]; then
|
||||||
|
local docker_service="${DOCKER_SERVICES[$service]}"
|
||||||
|
if [ -n "$docker_service" ]; then
|
||||||
|
docker-compose build "$docker_service"
|
||||||
|
else
|
||||||
|
log_error "未知服务: $service"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
docker-compose build
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_success "Docker 镜像构建完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动基础设施
|
||||||
|
start_infrastructure() {
|
||||||
|
log_step "启动基础设施服务..."
|
||||||
|
|
||||||
|
docker-compose up -d postgres redis neo4j
|
||||||
|
|
||||||
|
# 等待数据库就绪
|
||||||
|
wait_for_service localhost 5432 "PostgreSQL"
|
||||||
|
wait_for_service localhost 6379 "Redis"
|
||||||
|
wait_for_service localhost 7474 "Neo4j"
|
||||||
|
|
||||||
|
log_success "基础设施启动完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动 Kong 网关
|
||||||
|
start_kong() {
|
||||||
|
log_step "启动 Kong API 网关..."
|
||||||
|
|
||||||
|
docker-compose up -d kong-database
|
||||||
|
sleep 5
|
||||||
|
|
||||||
|
# Kong 数据库迁移
|
||||||
|
docker-compose run --rm kong kong migrations bootstrap || true
|
||||||
|
|
||||||
|
docker-compose up -d kong
|
||||||
|
wait_for_service localhost 8000 "Kong"
|
||||||
|
|
||||||
|
log_success "Kong 启动完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动后端服务 (非 Docker 模式)
|
||||||
|
start_backend_service_local() {
|
||||||
|
local service=$1
|
||||||
|
local dir="${SERVICE_DIRS[$service]}"
|
||||||
|
local port="${SERVICE_PORTS[$service]}"
|
||||||
|
|
||||||
|
if [ -z "$dir" ]; then
|
||||||
|
log_error "未知服务: $service"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_step "启动 $service (端口: $port)..."
|
||||||
|
|
||||||
|
cd "$PROJECT_ROOT/$dir"
|
||||||
|
|
||||||
|
# 检查是否已构建
|
||||||
|
if [ ! -d "dist" ]; then
|
||||||
|
log_warning "$service 未构建，先进行构建..."
|
||||||
|
pnpm run build
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 使用 PM2 或直接启动
|
||||||
|
if command -v pm2 &> /dev/null; then
|
||||||
|
pm2 start dist/main.js --name "iconsulting-$service" --cwd "$PROJECT_ROOT/$dir"
|
||||||
|
else
|
||||||
|
# 后台启动
|
||||||
|
nohup node dist/main.js > "$PROJECT_ROOT/logs/$service.log" 2>&1 &
|
||||||
|
echo $! > "$PROJECT_ROOT/pids/$service.pid"
|
||||||
|
fi
|
||||||
|
|
||||||
|
cd "$PROJECT_ROOT"
|
||||||
|
|
||||||
|
sleep 2
|
||||||
|
wait_for_service localhost "$port" "$service" 15
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动后端服务 (Docker 模式)
|
||||||
|
start_backend_service_docker() {
|
||||||
|
local service=$1
|
||||||
|
local docker_service="${DOCKER_SERVICES[$service]}"
|
||||||
|
|
||||||
|
log_step "启动 $service (Docker)..."
|
||||||
|
docker-compose up -d "$docker_service"
|
||||||
|
|
||||||
|
local port="${SERVICE_PORTS[$service]}"
|
||||||
|
wait_for_service localhost "$port" "$service"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动所有后端服务
|
||||||
|
start_all_backend() {
|
||||||
|
local mode=${1:-docker}
|
||||||
|
|
||||||
|
for service in user payment knowledge conversation evolution; do
|
||||||
|
if [ "$mode" = "docker" ]; then
|
||||||
|
start_backend_service_docker "$service"
|
||||||
|
else
|
||||||
|
start_backend_service_local "$service"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动 Nginx (静态文件服务)
|
||||||
|
start_nginx() {
|
||||||
|
log_step "启动 Nginx..."
|
||||||
|
docker-compose up -d nginx
|
||||||
|
wait_for_service localhost 80 "Nginx"
|
||||||
|
log_success "Nginx 启动完成"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动所有服务
|
||||||
|
start_all() {
|
||||||
|
local mode=${1:-docker}
|
||||||
|
|
||||||
|
log_info "开始启动所有服务 (模式: $mode)..."
|
||||||
|
|
||||||
|
# 创建必要目录
|
||||||
|
mkdir -p "$PROJECT_ROOT/logs"
|
||||||
|
mkdir -p "$PROJECT_ROOT/pids"
|
||||||
|
|
||||||
|
start_infrastructure
|
||||||
|
start_kong
|
||||||
|
start_all_backend "$mode"
|
||||||
|
start_nginx
|
||||||
|
|
||||||
|
log_success "所有服务启动完成"
|
||||||
|
do_status
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动入口
|
||||||
|
do_start() {
|
||||||
|
local target=${1:-all}
|
||||||
|
local mode=${2:-docker}
|
||||||
|
|
||||||
|
load_env
|
||||||
|
|
||||||
|
case $target in
|
||||||
|
all)
|
||||||
|
start_all "$mode"
|
||||||
|
;;
|
||||||
|
infra|infrastructure)
|
||||||
|
start_infrastructure
|
||||||
|
;;
|
||||||
|
kong)
|
||||||
|
start_kong
|
||||||
|
;;
|
||||||
|
nginx)
|
||||||
|
start_nginx
|
||||||
|
;;
|
||||||
|
postgres|redis|neo4j)
|
||||||
|
docker-compose up -d "$target"
|
||||||
|
;;
|
||||||
|
conversation|user|payment|knowledge|evolution)
|
||||||
|
if [ "$mode" = "docker" ]; then
|
||||||
|
start_backend_service_docker "$target"
|
||||||
|
else
|
||||||
|
start_backend_service_local "$target"
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
backend)
|
||||||
|
start_all_backend "$mode"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_error "未知启动目标: $target"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 停止函数
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
# 停止单个服务 (本地模式)
|
||||||
|
stop_service_local() {
|
||||||
|
local service=$1
|
||||||
|
|
||||||
|
log_step "停止 $service..."
|
||||||
|
|
||||||
|
if command -v pm2 &> /dev/null; then
|
||||||
|
pm2 stop "iconsulting-$service" 2>/dev/null || true
|
||||||
|
pm2 delete "iconsulting-$service" 2>/dev/null || true
|
||||||
|
else
|
||||||
|
local pid_file="$PROJECT_ROOT/pids/$service.pid"
|
||||||
|
if [ -f "$pid_file" ]; then
|
||||||
|
kill $(cat "$pid_file") 2>/dev/null || true
|
||||||
|
rm -f "$pid_file"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_success "$service 已停止"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 停止单个服务 (Docker 模式)
|
||||||
|
stop_service_docker() {
|
||||||
|
local service=$1
|
||||||
|
local docker_service="${DOCKER_SERVICES[$service]}"
|
||||||
|
|
||||||
|
if [ -n "$docker_service" ]; then
|
||||||
|
log_step "停止 $service..."
|
||||||
|
docker-compose stop "$docker_service"
|
||||||
|
log_success "$service 已停止"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 停止所有服务
|
||||||
|
stop_all() {
|
||||||
|
local mode=${1:-docker}
|
||||||
|
|
||||||
|
log_info "停止所有服务..."
|
||||||
|
|
||||||
|
if [ "$mode" = "docker" ]; then
|
||||||
|
docker-compose down
|
||||||
|
else
|
||||||
|
for service in conversation user payment knowledge evolution; do
|
||||||
|
stop_service_local "$service"
|
||||||
|
done
|
||||||
|
docker-compose down
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_success "所有服务已停止"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 停止入口
|
||||||
|
do_stop() {
|
||||||
|
local target=${1:-all}
|
||||||
|
local mode=${2:-docker}
|
||||||
|
|
||||||
|
case $target in
|
||||||
|
all)
|
||||||
|
stop_all "$mode"
|
||||||
|
;;
|
||||||
|
infra|infrastructure)
|
||||||
|
docker-compose stop postgres redis neo4j
|
||||||
|
;;
|
||||||
|
conversation|user|payment|knowledge|evolution)
|
||||||
|
if [ "$mode" = "docker" ]; then
|
||||||
|
stop_service_docker "$target"
|
||||||
|
else
|
||||||
|
stop_service_local "$target"
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
kong|postgres|redis|neo4j|nginx)
|
||||||
|
docker-compose stop "$target"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_error "未知停止目标: $target"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 重启函数
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
do_restart() {
|
||||||
|
local target=${1:-all}
|
||||||
|
local mode=${2:-docker}
|
||||||
|
|
||||||
|
log_info "重启 $target..."
|
||||||
|
do_stop "$target" "$mode"
|
||||||
|
sleep 2
|
||||||
|
do_start "$target" "$mode"
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 状态查看
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
do_status() {
|
||||||
|
echo ""
|
||||||
|
echo -e "${CYAN}═══════════════════════════════════════════════════════════════${NC}"
|
||||||
|
echo -e "${CYAN} iConsulting 服务状态 ${NC}"
|
||||||
|
echo -e "${CYAN}═══════════════════════════════════════════════════════════════${NC}"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# Docker 服务状态
|
||||||
|
echo -e "${PURPLE}Docker 容器状态:${NC}"
|
||||||
|
docker-compose ps
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 端口检查
|
||||||
|
echo -e "${PURPLE}服务端口检查:${NC}"
|
||||||
|
printf "%-20s %-10s %-10s\n" "服务" "端口" "状态"
|
||||||
|
echo "----------------------------------------"
|
||||||
|
|
||||||
|
for service in "${!SERVICE_PORTS[@]}"; do
|
||||||
|
local port="${SERVICE_PORTS[$service]}"
|
||||||
|
if nc -z localhost "$port" 2>/dev/null; then
|
||||||
|
printf "%-20s %-10s ${GREEN}%-10s${NC}\n" "$service" "$port" "运行中"
|
||||||
|
else
|
||||||
|
printf "%-20s %-10s ${RED}%-10s${NC}\n" "$service" "$port" "未运行"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# PM2 状态 (如果使用)
|
||||||
|
if command -v pm2 &> /dev/null; then
|
||||||
|
echo -e "${PURPLE}PM2 进程状态:${NC}"
|
||||||
|
pm2 list 2>/dev/null | grep iconsulting || echo "无 PM2 管理的服务"
|
||||||
|
echo ""
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 日志查看
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
do_logs() {
|
||||||
|
local service=${1:-all}
|
||||||
|
local lines=${2:-100}
|
||||||
|
|
||||||
|
if [ "$service" = "all" ]; then
|
||||||
|
docker-compose logs -f --tail="$lines"
|
||||||
|
else
|
||||||
|
local docker_service="${DOCKER_SERVICES[$service]}"
|
||||||
|
if [ -n "$docker_service" ]; then
|
||||||
|
docker-compose logs -f --tail="$lines" "$docker_service"
|
||||||
|
else
|
||||||
|
# 本地日志
|
||||||
|
local log_file="$PROJECT_ROOT/logs/$service.log"
|
||||||
|
if [ -f "$log_file" ]; then
|
||||||
|
tail -f -n "$lines" "$log_file"
|
||||||
|
else
|
||||||
|
log_error "日志文件不存在: $log_file"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 清理函数
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
do_clean() {
|
||||||
|
local target=${1:-build}
|
||||||
|
|
||||||
|
case $target in
|
||||||
|
build)
|
||||||
|
log_step "清理构建产物..."
|
||||||
|
for dir in "${SERVICE_DIRS[@]}"; do
|
||||||
|
rm -rf "$PROJECT_ROOT/$dir/dist"
|
||||||
|
done
|
||||||
|
log_success "构建产物已清理"
|
||||||
|
;;
|
||||||
|
deps)
|
||||||
|
log_step "清理依赖..."
|
||||||
|
rm -rf node_modules
|
||||||
|
for dir in "${SERVICE_DIRS[@]}"; do
|
||||||
|
rm -rf "$PROJECT_ROOT/$dir/node_modules"
|
||||||
|
done
|
||||||
|
log_success "依赖已清理"
|
||||||
|
;;
|
||||||
|
docker)
|
||||||
|
log_step "清理 Docker 资源..."
|
||||||
|
docker-compose down -v --rmi local
|
||||||
|
docker system prune -f
|
||||||
|
log_success "Docker 资源已清理"
|
||||||
|
;;
|
||||||
|
logs)
|
||||||
|
log_step "清理日志..."
|
||||||
|
rm -rf "$PROJECT_ROOT/logs/*"
|
||||||
|
log_success "日志已清理"
|
||||||
|
;;
|
||||||
|
all)
|
||||||
|
do_clean build
|
||||||
|
do_clean deps
|
||||||
|
do_clean docker
|
||||||
|
do_clean logs
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_error "未知清理目标: $target (可选: build, deps, docker, logs, all)"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 完整部署
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
do_deploy() {
|
||||||
|
local mode=${1:-docker}
|
||||||
|
|
||||||
|
log_info "开始完整部署 (模式: $mode)..."
|
||||||
|
|
||||||
|
check_environment
|
||||||
|
|
||||||
|
# 构建
|
||||||
|
do_build all
|
||||||
|
|
||||||
|
# 如果是 Docker 模式，构建镜像
|
||||||
|
if [ "$mode" = "docker" ]; then
|
||||||
|
build_docker_images
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 启动
|
||||||
|
do_start all "$mode"
|
||||||
|
|
||||||
|
log_success "部署完成！"
|
||||||
|
echo ""
|
||||||
|
echo -e "${CYAN}访问地址:${NC}"
|
||||||
|
echo " 用户前端: http://localhost"
|
||||||
|
echo " 管理后台: http://localhost/admin"
|
||||||
|
echo " API 网关: http://localhost:8000"
|
||||||
|
echo " Kong 管理: http://localhost:8001"
|
||||||
|
echo ""
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 数据库操作
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
do_db() {
|
||||||
|
local action=${1:-status}
|
||||||
|
|
||||||
|
case $action in
|
||||||
|
migrate)
|
||||||
|
log_step "执行数据库迁移..."
|
||||||
|
# 可以添加 TypeORM 迁移命令
|
||||||
|
for service in user payment knowledge conversation evolution; do
|
||||||
|
local dir="${SERVICE_DIRS[$service]}"
|
||||||
|
cd "$PROJECT_ROOT/$dir"
|
||||||
|
pnpm run migration:run 2>/dev/null || log_warning "$service 无迁移或迁移失败"
|
||||||
|
cd "$PROJECT_ROOT"
|
||||||
|
done
|
||||||
|
log_success "数据库迁移完成"
|
||||||
|
;;
|
||||||
|
seed)
|
||||||
|
log_step "初始化种子数据..."
|
||||||
|
# 添加种子数据脚本
|
||||||
|
log_success "种子数据初始化完成"
|
||||||
|
;;
|
||||||
|
backup)
|
||||||
|
local backup_dir="$PROJECT_ROOT/backups/$(date +%Y%m%d_%H%M%S)"
|
||||||
|
mkdir -p "$backup_dir"
|
||||||
|
|
||||||
|
log_step "备份数据库..."
|
||||||
|
docker-compose exec -T postgres pg_dump -U postgres iconsulting > "$backup_dir/postgres.sql"
|
||||||
|
log_success "数据库备份到: $backup_dir"
|
||||||
|
;;
|
||||||
|
restore)
|
||||||
|
local backup_file=$2
|
||||||
|
if [ -z "$backup_file" ]; then
|
||||||
|
log_error "请指定备份文件: ./deploy.sh db restore <backup_file>"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log_step "恢复数据库..."
|
||||||
|
docker-compose exec -T postgres psql -U postgres iconsulting < "$backup_file"
|
||||||
|
log_success "数据库恢复完成"
|
||||||
|
;;
|
||||||
|
reset)
|
||||||
|
log_warning "这将删除所有数据！"
|
||||||
|
read -p "确认继续? (y/N) " confirm
|
||||||
|
if [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]; then
|
||||||
|
docker-compose down -v
|
||||||
|
docker-compose up -d postgres redis neo4j
|
||||||
|
wait_for_service localhost 5432 "PostgreSQL"
|
||||||
|
do_db migrate
|
||||||
|
log_success "数据库已重置"
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
status)
|
||||||
|
echo -e "${PURPLE}数据库状态:${NC}"
|
||||||
|
docker-compose exec postgres psql -U postgres -c "SELECT version();" 2>/dev/null || echo "PostgreSQL 未运行"
|
||||||
|
docker-compose exec redis redis-cli ping 2>/dev/null || echo "Redis 未运行"
|
||||||
|
curl -s http://localhost:7474 > /dev/null && echo "Neo4j 运行中" || echo "Neo4j 未运行"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_error "未知数据库操作: $action (可选: migrate, seed, backup, restore, reset, status)"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 帮助信息
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
show_help() {
|
||||||
|
cat << 'EOF'
|
||||||
|
|
||||||
|
╔═══════════════════════════════════════════════════════════════════════════════╗
|
||||||
|
║ iConsulting 部署管理脚本 ║
|
||||||
|
╚═══════════════════════════════════════════════════════════════════════════════╝
|
||||||
|
|
||||||
|
用法: ./deploy.sh <command> [target] [options]
|
||||||
|
|
||||||
|
命令:
|
||||||
|
build [target] 编译构建
|
||||||
|
target: all, shared, backend, frontend,
|
||||||
|
conversation, user, payment, knowledge, evolution,
|
||||||
|
web-client, admin-client
|
||||||
|
|
||||||
|
start [target] [mode] 启动服务
|
||||||
|
target: all, infra, kong, nginx, backend,
|
||||||
|
conversation, user, payment, knowledge, evolution,
|
||||||
|
postgres, redis, neo4j
|
||||||
|
mode: docker (默认), local
|
||||||
|
|
||||||
|
stop [target] [mode] 停止服务
|
||||||
|
(target 同上)
|
||||||
|
|
||||||
|
restart [target] [mode] 重启服务
|
||||||
|
(target 同上)
|
||||||
|
|
||||||
|
status 查看所有服务状态
|
||||||
|
|
||||||
|
logs [service] [lines] 查看日志
|
||||||
|
service: 服务名或 all
|
||||||
|
lines: 显示行数 (默认 100)
|
||||||
|
|
||||||
|
clean [target] 清理
|
||||||
|
target: build, deps, docker, logs, all
|
||||||
|
|
||||||
|
deploy [mode] 完整部署 (构建 + 启动)
|
||||||
|
mode: docker (默认), local
|
||||||
|
|
||||||
|
db <action> 数据库操作
|
||||||
|
action: migrate, seed, backup, restore, reset, status
|
||||||
|
|
||||||
|
help 显示此帮助信息
|
||||||
|
|
||||||
|
示例:
|
||||||
|
./deploy.sh deploy # 完整部署
|
||||||
|
./deploy.sh build conversation # 只构建对话服务
|
||||||
|
./deploy.sh start backend local # 本地模式启动所有后端
|
||||||
|
./deploy.sh restart user docker # 重启用户服务 (Docker)
|
||||||
|
./deploy.sh logs conversation 200 # 查看对话服务最近200行日志
|
||||||
|
./deploy.sh clean all # 清理所有构建产物和依赖
|
||||||
|
./deploy.sh db backup # 备份数据库
|
||||||
|
./deploy.sh db migrate # 执行数据库迁移
|
||||||
|
|
||||||
|
EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 主入口
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
main() {
|
||||||
|
local command=${1:-help}
|
||||||
|
shift || true
|
||||||
|
|
||||||
|
case $command in
|
||||||
|
build)
|
||||||
|
do_build "$@"
|
||||||
|
;;
|
||||||
|
start)
|
||||||
|
do_start "$@"
|
||||||
|
;;
|
||||||
|
stop)
|
||||||
|
do_stop "$@"
|
||||||
|
;;
|
||||||
|
restart)
|
||||||
|
do_restart "$@"
|
||||||
|
;;
|
||||||
|
status)
|
||||||
|
do_status
|
||||||
|
;;
|
||||||
|
logs)
|
||||||
|
do_logs "$@"
|
||||||
|
;;
|
||||||
|
clean)
|
||||||
|
do_clean "$@"
|
||||||
|
;;
|
||||||
|
deploy)
|
||||||
|
do_deploy "$@"
|
||||||
|
;;
|
||||||
|
db)
|
||||||
|
do_db "$@"
|
||||||
|
;;
|
||||||
|
help|--help|-h)
|
||||||
|
show_help
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_error "未知命令: $command"
|
||||||
|
show_help
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
|
# 执行主函数
|
||||||
|
main "$@"
|
||||||
|
|
@ -0,0 +1,319 @@
|
||||||
|
#===============================================================================
|
||||||
|
# iConsulting Docker Compose 配置
|
||||||
|
#
|
||||||
|
# 服务架构:
|
||||||
|
# - 基础设施: PostgreSQL, Redis, Neo4j
|
||||||
|
# - API网关: Kong
|
||||||
|
# - 后端服务: conversation, user, payment, knowledge, evolution
|
||||||
|
# - 前端服务: nginx (托管 web-client 和 admin-client)
|
||||||
|
#
|
||||||
|
# 网络配置:
|
||||||
|
# - 对外网卡: 14.215.128.96 (用户访问)
|
||||||
|
# - 出口网卡: 154.84.135.121 (Claude API 调用)
|
||||||
|
#
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
version: '3.8'
|
||||||
|
|
||||||
|
services:
|
||||||
|
#=============================================================================
|
||||||
|
# 基础设施服务
|
||||||
|
#=============================================================================
|
||||||
|
|
||||||
|
postgres:
|
||||||
|
image: postgres:15-alpine
|
||||||
|
container_name: iconsulting-postgres
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
POSTGRES_USER: ${POSTGRES_USER:-postgres}
|
||||||
|
POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-postgres}
|
||||||
|
POSTGRES_DB: ${POSTGRES_DB:-iconsulting}
|
||||||
|
ports:
|
||||||
|
- "5432:5432"
|
||||||
|
volumes:
|
||||||
|
- postgres_data:/var/lib/postgresql/data
|
||||||
|
- ./scripts/init-db.sql:/docker-entrypoint-initdb.d/init.sql:ro
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD-SHELL", "pg_isready -U postgres"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
redis:
|
||||||
|
image: redis:7-alpine
|
||||||
|
container_name: iconsulting-redis
|
||||||
|
restart: unless-stopped
|
||||||
|
command: redis-server --appendonly yes --requirepass ${REDIS_PASSWORD:-redis123}
|
||||||
|
ports:
|
||||||
|
- "6379:6379"
|
||||||
|
volumes:
|
||||||
|
- redis_data:/data
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "redis-cli", "ping"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
neo4j:
|
||||||
|
image: neo4j:5-community
|
||||||
|
container_name: iconsulting-neo4j
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
NEO4J_AUTH: ${NEO4J_USER:-neo4j}/${NEO4J_PASSWORD:-neo4j123}
|
||||||
|
NEO4J_PLUGINS: '["apoc"]'
|
||||||
|
NEO4J_dbms_memory_heap_max__size: 1G
|
||||||
|
ports:
|
||||||
|
- "7474:7474" # HTTP
|
||||||
|
- "7687:7687" # Bolt
|
||||||
|
volumes:
|
||||||
|
- neo4j_data:/data
|
||||||
|
- neo4j_logs:/logs
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD-SHELL", "wget --no-verbose --tries=1 --spider http://localhost:7474 || exit 1"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
#=============================================================================
|
||||||
|
# Kong API 网关
|
||||||
|
#=============================================================================
|
||||||
|
|
||||||
|
kong-database:
|
||||||
|
image: postgres:15-alpine
|
||||||
|
container_name: iconsulting-kong-db
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
POSTGRES_USER: kong
|
||||||
|
POSTGRES_PASSWORD: kong
|
||||||
|
POSTGRES_DB: kong
|
||||||
|
volumes:
|
||||||
|
- kong_data:/var/lib/postgresql/data
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD-SHELL", "pg_isready -U kong"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
kong:
|
||||||
|
image: kong:3.4-alpine
|
||||||
|
container_name: iconsulting-kong
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
kong-database:
|
||||||
|
condition: service_healthy
|
||||||
|
environment:
|
||||||
|
KONG_DATABASE: postgres
|
||||||
|
KONG_PG_HOST: kong-database
|
||||||
|
KONG_PG_USER: kong
|
||||||
|
KONG_PG_PASSWORD: kong
|
||||||
|
KONG_PG_DATABASE: kong
|
||||||
|
KONG_PROXY_ACCESS_LOG: /dev/stdout
|
||||||
|
KONG_ADMIN_ACCESS_LOG: /dev/stdout
|
||||||
|
KONG_PROXY_ERROR_LOG: /dev/stderr
|
||||||
|
KONG_ADMIN_ERROR_LOG: /dev/stderr
|
||||||
|
KONG_ADMIN_LISTEN: 0.0.0.0:8001
|
||||||
|
KONG_PROXY_LISTEN: 0.0.0.0:8000, 0.0.0.0:8443 ssl
|
||||||
|
ports:
|
||||||
|
- "8000:8000" # Proxy
|
||||||
|
- "8443:8443" # Proxy SSL
|
||||||
|
- "8001:8001" # Admin API
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "kong", "health"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
#=============================================================================
|
||||||
|
# 后端微服务
|
||||||
|
#=============================================================================
|
||||||
|
|
||||||
|
user-service:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: packages/services/user-service/Dockerfile
|
||||||
|
container_name: iconsulting-user
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
postgres:
|
||||||
|
condition: service_healthy
|
||||||
|
redis:
|
||||||
|
condition: service_healthy
|
||||||
|
environment:
|
||||||
|
NODE_ENV: production
|
||||||
|
PORT: 3001
|
||||||
|
DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:5432/${POSTGRES_DB:-iconsulting}
|
||||||
|
REDIS_URL: redis://:${REDIS_PASSWORD:-redis123}@redis:6379
|
||||||
|
JWT_SECRET: ${JWT_SECRET:-your-jwt-secret-key}
|
||||||
|
JWT_EXPIRES_IN: ${JWT_EXPIRES_IN:-7d}
|
||||||
|
ports:
|
||||||
|
- "3001:3001"
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
payment-service:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: packages/services/payment-service/Dockerfile
|
||||||
|
container_name: iconsulting-payment
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
postgres:
|
||||||
|
condition: service_healthy
|
||||||
|
redis:
|
||||||
|
condition: service_healthy
|
||||||
|
environment:
|
||||||
|
NODE_ENV: production
|
||||||
|
PORT: 3002
|
||||||
|
DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:5432/${POSTGRES_DB:-iconsulting}
|
||||||
|
REDIS_URL: redis://:${REDIS_PASSWORD:-redis123}@redis:6379
|
||||||
|
ALIPAY_APP_ID: ${ALIPAY_APP_ID}
|
||||||
|
ALIPAY_PRIVATE_KEY: ${ALIPAY_PRIVATE_KEY}
|
||||||
|
WECHAT_APP_ID: ${WECHAT_APP_ID}
|
||||||
|
WECHAT_MCH_ID: ${WECHAT_MCH_ID}
|
||||||
|
WECHAT_API_KEY: ${WECHAT_API_KEY}
|
||||||
|
STRIPE_SECRET_KEY: ${STRIPE_SECRET_KEY}
|
||||||
|
ports:
|
||||||
|
- "3002:3002"
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
knowledge-service:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: packages/services/knowledge-service/Dockerfile
|
||||||
|
container_name: iconsulting-knowledge
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
postgres:
|
||||||
|
condition: service_healthy
|
||||||
|
redis:
|
||||||
|
condition: service_healthy
|
||||||
|
neo4j:
|
||||||
|
condition: service_healthy
|
||||||
|
environment:
|
||||||
|
NODE_ENV: production
|
||||||
|
PORT: 3003
|
||||||
|
DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:5432/${POSTGRES_DB:-iconsulting}
|
||||||
|
REDIS_URL: redis://:${REDIS_PASSWORD:-redis123}@redis:6379
|
||||||
|
NEO4J_URI: bolt://neo4j:7687
|
||||||
|
NEO4J_USER: ${NEO4J_USER:-neo4j}
|
||||||
|
NEO4J_PASSWORD: ${NEO4J_PASSWORD:-neo4j123}
|
||||||
|
OPENAI_API_KEY: ${OPENAI_API_KEY}
|
||||||
|
ports:
|
||||||
|
- "3003:3003"
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
conversation-service:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: packages/services/conversation-service/Dockerfile
|
||||||
|
container_name: iconsulting-conversation
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
postgres:
|
||||||
|
condition: service_healthy
|
||||||
|
redis:
|
||||||
|
condition: service_healthy
|
||||||
|
knowledge-service:
|
||||||
|
condition: service_started
|
||||||
|
environment:
|
||||||
|
NODE_ENV: production
|
||||||
|
PORT: 3004
|
||||||
|
DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:5432/${POSTGRES_DB:-iconsulting}
|
||||||
|
REDIS_URL: redis://:${REDIS_PASSWORD:-redis123}@redis:6379
|
||||||
|
ANTHROPIC_API_KEY: ${ANTHROPIC_API_KEY}
|
||||||
|
ANTHROPIC_BASE_URL: ${ANTHROPIC_BASE_URL:-https://api.anthropic.com}
|
||||||
|
KNOWLEDGE_SERVICE_URL: http://knowledge-service:3003
|
||||||
|
# Claude API 出口配置 (如需指定出口IP，在宿主机配置路由)
|
||||||
|
ports:
|
||||||
|
- "3004:3004"
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
evolution-service:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: packages/services/evolution-service/Dockerfile
|
||||||
|
container_name: iconsulting-evolution
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
postgres:
|
||||||
|
condition: service_healthy
|
||||||
|
redis:
|
||||||
|
condition: service_healthy
|
||||||
|
environment:
|
||||||
|
NODE_ENV: production
|
||||||
|
PORT: 3005
|
||||||
|
DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:5432/${POSTGRES_DB:-iconsulting}
|
||||||
|
REDIS_URL: redis://:${REDIS_PASSWORD:-redis123}@redis:6379
|
||||||
|
ANTHROPIC_API_KEY: ${ANTHROPIC_API_KEY}
|
||||||
|
ANTHROPIC_BASE_URL: ${ANTHROPIC_BASE_URL:-https://api.anthropic.com}
|
||||||
|
ports:
|
||||||
|
- "3005:3005"
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
#=============================================================================
|
||||||
|
# 前端 Nginx
|
||||||
|
#=============================================================================
|
||||||
|
|
||||||
|
nginx:
|
||||||
|
image: nginx:alpine
|
||||||
|
container_name: iconsulting-nginx
|
||||||
|
restart: unless-stopped
|
||||||
|
depends_on:
|
||||||
|
- kong
|
||||||
|
ports:
|
||||||
|
- "80:80"
|
||||||
|
- "443:443"
|
||||||
|
volumes:
|
||||||
|
- ./packages/web-client/dist:/usr/share/nginx/html/web:ro
|
||||||
|
- ./packages/admin-client/dist:/usr/share/nginx/html/admin:ro
|
||||||
|
- ./nginx/nginx.conf:/etc/nginx/nginx.conf:ro
|
||||||
|
- ./nginx/conf.d:/etc/nginx/conf.d:ro
|
||||||
|
- ./nginx/ssl:/etc/nginx/ssl:ro
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://localhost/health"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 3
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 网络配置
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
networks:
|
||||||
|
iconsulting-network:
|
||||||
|
driver: bridge
|
||||||
|
ipam:
|
||||||
|
config:
|
||||||
|
- subnet: 172.20.0.0/16
|
||||||
|
|
||||||
|
#===============================================================================
|
||||||
|
# 数据卷
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
postgres_data:
|
||||||
|
driver: local
|
||||||
|
redis_data:
|
||||||
|
driver: local
|
||||||
|
neo4j_data:
|
||||||
|
driver: local
|
||||||
|
neo4j_logs:
|
||||||
|
driver: local
|
||||||
|
kong_data:
|
||||||
|
driver: local
|
||||||
Binary file not shown.
|
After Width: | Height: | Size: 155 KiB |
|
|
@ -0,0 +1,134 @@
|
||||||
|
version: '3.8'
|
||||||
|
|
||||||
|
services:
|
||||||
|
# PostgreSQL with pgvector extension
|
||||||
|
postgres:
|
||||||
|
image: pgvector/pgvector:pg15
|
||||||
|
container_name: iconsulting-postgres
|
||||||
|
environment:
|
||||||
|
POSTGRES_USER: iconsulting
|
||||||
|
POSTGRES_PASSWORD: dev_password_123
|
||||||
|
POSTGRES_DB: iconsulting
|
||||||
|
ports:
|
||||||
|
- "5432:5432"
|
||||||
|
volumes:
|
||||||
|
- postgres_data:/var/lib/postgresql/data
|
||||||
|
- ./services/postgres/init.sql:/docker-entrypoint-initdb.d/init.sql
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD-SHELL", "pg_isready -U iconsulting"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
# Neo4j Graph Database
|
||||||
|
neo4j:
|
||||||
|
image: neo4j:5
|
||||||
|
container_name: iconsulting-neo4j
|
||||||
|
environment:
|
||||||
|
NEO4J_AUTH: neo4j/dev_password_123
|
||||||
|
NEO4J_PLUGINS: '["apoc"]'
|
||||||
|
NEO4J_dbms_security_procedures_unrestricted: apoc.*
|
||||||
|
ports:
|
||||||
|
- "7474:7474" # HTTP
|
||||||
|
- "7687:7687" # Bolt
|
||||||
|
volumes:
|
||||||
|
- neo4j_data:/data
|
||||||
|
- neo4j_logs:/logs
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD-SHELL", "wget --no-verbose --tries=1 --spider http://localhost:7474 || exit 1"]
|
||||||
|
interval: 15s
|
||||||
|
timeout: 10s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
# Redis Cache
|
||||||
|
redis:
|
||||||
|
image: redis:7-alpine
|
||||||
|
container_name: iconsulting-redis
|
||||||
|
ports:
|
||||||
|
- "6379:6379"
|
||||||
|
volumes:
|
||||||
|
- redis_data:/data
|
||||||
|
command: redis-server --appendonly yes
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "redis-cli", "ping"]
|
||||||
|
interval: 10s
|
||||||
|
timeout: 5s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
# Zookeeper for Kafka
|
||||||
|
zookeeper:
|
||||||
|
image: confluentinc/cp-zookeeper:7.4.0
|
||||||
|
container_name: iconsulting-zookeeper
|
||||||
|
environment:
|
||||||
|
ZOOKEEPER_CLIENT_PORT: 2181
|
||||||
|
ZOOKEEPER_TICK_TIME: 2000
|
||||||
|
ports:
|
||||||
|
- "2181:2181"
|
||||||
|
volumes:
|
||||||
|
- zookeeper_data:/var/lib/zookeeper/data
|
||||||
|
- zookeeper_logs:/var/lib/zookeeper/log
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
# Kafka Message Broker
|
||||||
|
kafka:
|
||||||
|
image: confluentinc/cp-kafka:7.4.0
|
||||||
|
container_name: iconsulting-kafka
|
||||||
|
depends_on:
|
||||||
|
- zookeeper
|
||||||
|
ports:
|
||||||
|
- "9092:9092"
|
||||||
|
- "29092:29092"
|
||||||
|
environment:
|
||||||
|
KAFKA_BROKER_ID: 1
|
||||||
|
KAFKA_ZOOKEEPER_CONNECT: zookeeper:2181
|
||||||
|
KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: PLAINTEXT:PLAINTEXT,PLAINTEXT_HOST:PLAINTEXT
|
||||||
|
KAFKA_ADVERTISED_LISTENERS: PLAINTEXT://kafka:29092,PLAINTEXT_HOST://localhost:9092
|
||||||
|
KAFKA_OFFSETS_TOPIC_REPLICATION_FACTOR: 1
|
||||||
|
KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: 1
|
||||||
|
KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: 1
|
||||||
|
KAFKA_GROUP_INITIAL_REBALANCE_DELAY_MS: 0
|
||||||
|
KAFKA_AUTO_CREATE_TOPICS_ENABLE: 'true'
|
||||||
|
volumes:
|
||||||
|
- kafka_data:/var/lib/kafka/data
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD-SHELL", "kafka-broker-api-versions --bootstrap-server localhost:9092"]
|
||||||
|
interval: 30s
|
||||||
|
timeout: 10s
|
||||||
|
retries: 5
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
# Kafka UI (optional, for debugging)
|
||||||
|
kafka-ui:
|
||||||
|
image: provectuslabs/kafka-ui:latest
|
||||||
|
container_name: iconsulting-kafka-ui
|
||||||
|
depends_on:
|
||||||
|
- kafka
|
||||||
|
ports:
|
||||||
|
- "8080:8080"
|
||||||
|
environment:
|
||||||
|
KAFKA_CLUSTERS_0_NAME: iconsulting
|
||||||
|
KAFKA_CLUSTERS_0_BOOTSTRAPSERVERS: kafka:29092
|
||||||
|
KAFKA_CLUSTERS_0_ZOOKEEPER: zookeeper:2181
|
||||||
|
networks:
|
||||||
|
- iconsulting-network
|
||||||
|
|
||||||
|
networks:
|
||||||
|
iconsulting-network:
|
||||||
|
driver: bridge
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
postgres_data:
|
||||||
|
neo4j_data:
|
||||||
|
neo4j_logs:
|
||||||
|
redis_data:
|
||||||
|
zookeeper_data:
|
||||||
|
zookeeper_logs:
|
||||||
|
kafka_data:
|
||||||
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,130 @@
|
||||||
|
#===============================================================================
|
||||||
|
# iConsulting Nginx 配置
|
||||||
|
#
|
||||||
|
# 路由规则:
|
||||||
|
# / -> web-client (用户前端)
|
||||||
|
# /admin -> admin-client (管理后台)
|
||||||
|
# /api/v1/* -> Kong API Gateway
|
||||||
|
# /ws/* -> WebSocket (conversation-service)
|
||||||
|
#
|
||||||
|
#===============================================================================
|
||||||
|
|
||||||
|
server {
|
||||||
|
listen 80;
|
||||||
|
server_name _;
|
||||||
|
|
||||||
|
# 健康检查端点
|
||||||
|
location /health {
|
||||||
|
access_log off;
|
||||||
|
return 200 'OK';
|
||||||
|
add_header Content-Type text/plain;
|
||||||
|
}
|
||||||
|
|
||||||
|
# 用户前端 (web-client)
|
||||||
|
location / {
|
||||||
|
root /usr/share/nginx/html/web;
|
||||||
|
index index.html;
|
||||||
|
try_files $uri $uri/ /index.html;
|
||||||
|
|
||||||
|
# 缓存静态资源
|
||||||
|
location ~* \.(js|css|png|jpg|jpeg|gif|ico|svg|woff|woff2|ttf|eot)$ {
|
||||||
|
expires 1y;
|
||||||
|
add_header Cache-Control "public, immutable";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
# 管理后台 (admin-client)
|
||||||
|
location /admin {
|
||||||
|
alias /usr/share/nginx/html/admin;
|
||||||
|
index index.html;
|
||||||
|
try_files $uri $uri/ /admin/index.html;
|
||||||
|
|
||||||
|
# 缓存静态资源
|
||||||
|
location ~* \.(js|css|png|jpg|jpeg|gif|ico|svg|woff|woff2|ttf|eot)$ {
|
||||||
|
expires 1y;
|
||||||
|
add_header Cache-Control "public, immutable";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
# API 请求代理到 Kong
|
||||||
|
location /api/ {
|
||||||
|
proxy_pass http://kong_upstream/;
|
||||||
|
proxy_http_version 1.1;
|
||||||
|
proxy_set_header Host $host;
|
||||||
|
proxy_set_header X-Real-IP $remote_addr;
|
||||||
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
|
||||||
|
# 超时设置
|
||||||
|
proxy_connect_timeout 60s;
|
||||||
|
proxy_send_timeout 60s;
|
||||||
|
proxy_read_timeout 60s;
|
||||||
|
|
||||||
|
# 缓冲设置
|
||||||
|
proxy_buffering on;
|
||||||
|
proxy_buffer_size 4k;
|
||||||
|
proxy_buffers 8 4k;
|
||||||
|
}
|
||||||
|
|
||||||
|
# WebSocket 代理
|
||||||
|
location /ws/ {
|
||||||
|
proxy_pass http://websocket_upstream/;
|
||||||
|
proxy_http_version 1.1;
|
||||||
|
proxy_set_header Upgrade $http_upgrade;
|
||||||
|
proxy_set_header Connection "upgrade";
|
||||||
|
proxy_set_header Host $host;
|
||||||
|
proxy_set_header X-Real-IP $remote_addr;
|
||||||
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
|
||||||
|
# WebSocket 超时 (保持长连接)
|
||||||
|
proxy_connect_timeout 7d;
|
||||||
|
proxy_send_timeout 7d;
|
||||||
|
proxy_read_timeout 7d;
|
||||||
|
}
|
||||||
|
|
||||||
|
# Socket.IO 专用路径
|
||||||
|
location /socket.io/ {
|
||||||
|
proxy_pass http://websocket_upstream/socket.io/;
|
||||||
|
proxy_http_version 1.1;
|
||||||
|
proxy_set_header Upgrade $http_upgrade;
|
||||||
|
proxy_set_header Connection "upgrade";
|
||||||
|
proxy_set_header Host $host;
|
||||||
|
proxy_set_header X-Real-IP $remote_addr;
|
||||||
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
|
||||||
|
proxy_connect_timeout 7d;
|
||||||
|
proxy_send_timeout 7d;
|
||||||
|
proxy_read_timeout 7d;
|
||||||
|
}
|
||||||
|
|
||||||
|
# 禁止访问隐藏文件
|
||||||
|
location ~ /\. {
|
||||||
|
deny all;
|
||||||
|
access_log off;
|
||||||
|
log_not_found off;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
# HTTPS 配置 (如有SSL证书，取消注释)
|
||||||
|
# server {
|
||||||
|
# listen 443 ssl http2;
|
||||||
|
# server_name _;
|
||||||
|
#
|
||||||
|
# ssl_certificate /etc/nginx/ssl/fullchain.pem;
|
||||||
|
# ssl_certificate_key /etc/nginx/ssl/privkey.pem;
|
||||||
|
# ssl_session_timeout 1d;
|
||||||
|
# ssl_session_cache shared:SSL:50m;
|
||||||
|
# ssl_session_tickets off;
|
||||||
|
#
|
||||||
|
# ssl_protocols TLSv1.2 TLSv1.3;
|
||||||
|
# ssl_ciphers ECDHE-ECDSA-AES128-GCM-SHA256:ECDHE-RSA-AES128-GCM-SHA256;
|
||||||
|
# ssl_prefer_server_ciphers off;
|
||||||
|
#
|
||||||
|
# # HSTS
|
||||||
|
# add_header Strict-Transport-Security "max-age=63072000" always;
|
||||||
|
#
|
||||||
|
# # 其他配置同上...
|
||||||
|
# include /etc/nginx/conf.d/locations.conf;
|
||||||
|
# }
|
||||||
|
|
@ -0,0 +1,51 @@
|
||||||
|
user nginx;
|
||||||
|
worker_processes auto;
|
||||||
|
|
||||||
|
error_log /var/log/nginx/error.log notice;
|
||||||
|
pid /var/run/nginx.pid;
|
||||||
|
|
||||||
|
events {
|
||||||
|
worker_connections 1024;
|
||||||
|
use epoll;
|
||||||
|
multi_accept on;
|
||||||
|
}
|
||||||
|
|
||||||
|
http {
|
||||||
|
include /etc/nginx/mime.types;
|
||||||
|
default_type application/octet-stream;
|
||||||
|
|
||||||
|
log_format main '$remote_addr - $remote_user [$time_local] "$request" '
|
||||||
|
'$status $body_bytes_sent "$http_referer" '
|
||||||
|
'"$http_user_agent" "$http_x_forwarded_for"';
|
||||||
|
|
||||||
|
access_log /var/log/nginx/access.log main;
|
||||||
|
|
||||||
|
sendfile on;
|
||||||
|
tcp_nopush on;
|
||||||
|
tcp_nodelay on;
|
||||||
|
|
||||||
|
keepalive_timeout 65;
|
||||||
|
|
||||||
|
# Gzip 压缩
|
||||||
|
gzip on;
|
||||||
|
gzip_vary on;
|
||||||
|
gzip_proxied any;
|
||||||
|
gzip_comp_level 6;
|
||||||
|
gzip_types text/plain text/css text/xml application/json application/javascript application/rss+xml application/atom+xml image/svg+xml;
|
||||||
|
|
||||||
|
# 请求大小限制
|
||||||
|
client_max_body_size 50M;
|
||||||
|
|
||||||
|
# 上游服务定义
|
||||||
|
upstream kong_upstream {
|
||||||
|
server kong:8000;
|
||||||
|
keepalive 32;
|
||||||
|
}
|
||||||
|
|
||||||
|
upstream websocket_upstream {
|
||||||
|
server conversation-service:3004;
|
||||||
|
keepalive 32;
|
||||||
|
}
|
||||||
|
|
||||||
|
include /etc/nginx/conf.d/*.conf;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,2 @@
|
||||||
|
# 此目录用于存放 SSL 证书
|
||||||
|
# 请将 fullchain.pem 和 privkey.pem 放置于此
|
||||||
|
|
@ -0,0 +1,32 @@
|
||||||
|
{
|
||||||
|
"name": "iconsulting",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "Hong Kong Immigration Consulting System based on Claude Agent SDK",
|
||||||
|
"private": true,
|
||||||
|
"workspaces": [
|
||||||
|
"packages/*",
|
||||||
|
"packages/services/*"
|
||||||
|
],
|
||||||
|
"scripts": {
|
||||||
|
"dev": "turbo run dev",
|
||||||
|
"build": "turbo run build",
|
||||||
|
"lint": "turbo run lint",
|
||||||
|
"test": "turbo run test",
|
||||||
|
"clean": "turbo run clean && rm -rf node_modules",
|
||||||
|
"db:migrate": "turbo run db:migrate",
|
||||||
|
"docker:dev": "docker-compose -f infrastructure/docker/docker-compose.dev.yml up -d",
|
||||||
|
"docker:down": "docker-compose -f infrastructure/docker/docker-compose.dev.yml down"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"turbo": "^2.0.0",
|
||||||
|
"typescript": "^5.3.0",
|
||||||
|
"@types/node": "^20.10.0",
|
||||||
|
"prettier": "^3.1.0",
|
||||||
|
"eslint": "^8.55.0"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=18.0.0",
|
||||||
|
"pnpm": ">=8.0.0"
|
||||||
|
},
|
||||||
|
"packageManager": "pnpm@8.15.0"
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,13 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8" />
|
||||||
|
<link rel="icon" type="image/svg+xml" href="/vite.svg" />
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||||
|
<title>iConsulting 管理后台</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div id="root"></div>
|
||||||
|
<script type="module" src="/src/main.tsx"></script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
|
@ -0,0 +1,40 @@
|
||||||
|
{
|
||||||
|
"name": "@iconsulting/admin-client",
|
||||||
|
"version": "0.1.0",
|
||||||
|
"description": "iConsulting 管理后台",
|
||||||
|
"private": true,
|
||||||
|
"type": "module",
|
||||||
|
"scripts": {
|
||||||
|
"dev": "vite",
|
||||||
|
"build": "tsc && vite build",
|
||||||
|
"preview": "vite preview",
|
||||||
|
"lint": "eslint . --ext ts,tsx --report-unused-disable-directives --max-warnings 0"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@tanstack/react-query": "^5.17.0",
|
||||||
|
"antd": "^5.12.8",
|
||||||
|
"@ant-design/icons": "^5.2.6",
|
||||||
|
"axios": "^1.6.5",
|
||||||
|
"dayjs": "^1.11.10",
|
||||||
|
"react": "^18.2.0",
|
||||||
|
"react-dom": "^18.2.0",
|
||||||
|
"react-router-dom": "^6.21.1",
|
||||||
|
"zustand": "^4.4.7",
|
||||||
|
"recharts": "^2.10.4"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@types/react": "^18.2.47",
|
||||||
|
"@types/react-dom": "^18.2.18",
|
||||||
|
"@typescript-eslint/eslint-plugin": "^6.18.1",
|
||||||
|
"@typescript-eslint/parser": "^6.18.1",
|
||||||
|
"@vitejs/plugin-react": "^4.2.1",
|
||||||
|
"autoprefixer": "^10.4.16",
|
||||||
|
"eslint": "^8.56.0",
|
||||||
|
"eslint-plugin-react-hooks": "^4.6.0",
|
||||||
|
"eslint-plugin-react-refresh": "^0.4.5",
|
||||||
|
"postcss": "^8.4.33",
|
||||||
|
"tailwindcss": "^3.4.1",
|
||||||
|
"typescript": "^5.3.3",
|
||||||
|
"vite": "^5.0.11"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,6 @@
|
||||||
|
export default {
|
||||||
|
plugins: {
|
||||||
|
tailwindcss: {},
|
||||||
|
autoprefixer: {},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
import { Routes, Route, Navigate } from 'react-router-dom';
|
||||||
|
import { MainLayout } from './shared/components/MainLayout';
|
||||||
|
import { ProtectedRoute } from './shared/components/ProtectedRoute';
|
||||||
|
import { LoginPage } from './features/auth/presentation/pages/LoginPage';
|
||||||
|
import { DashboardPage } from './features/dashboard/presentation/pages/DashboardPage';
|
||||||
|
import { KnowledgePage } from './features/knowledge/presentation/pages/KnowledgePage';
|
||||||
|
import { ExperiencePage } from './features/experience/presentation/pages/ExperiencePage';
|
||||||
|
|
||||||
|
function App() {
|
||||||
|
return (
|
||||||
|
<Routes>
|
||||||
|
{/* 登录页 */}
|
||||||
|
<Route path="/login" element={<LoginPage />} />
|
||||||
|
|
||||||
|
{/* 需要认证的路由 */}
|
||||||
|
<Route
|
||||||
|
path="/"
|
||||||
|
element={
|
||||||
|
<ProtectedRoute>
|
||||||
|
<MainLayout />
|
||||||
|
</ProtectedRoute>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<Route index element={<DashboardPage />} />
|
||||||
|
<Route path="knowledge" element={<KnowledgePage />} />
|
||||||
|
<Route path="experience" element={<ExperiencePage />} />
|
||||||
|
<Route path="users" element={<div className="p-6">用户管理（开发中）</div>} />
|
||||||
|
<Route path="settings" element={<div className="p-6">系统设置（开发中）</div>} />
|
||||||
|
</Route>
|
||||||
|
|
||||||
|
{/* 未匹配路由重定向 */}
|
||||||
|
<Route path="*" element={<Navigate to="/" replace />} />
|
||||||
|
</Routes>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export default App;
|
||||||
|
|
@ -0,0 +1,82 @@
|
||||||
|
import { useState } from 'react';
|
||||||
|
import { useNavigate } from 'react-router-dom';
|
||||||
|
import { Form, Input, Button, Card, message } from 'antd';
|
||||||
|
import { UserOutlined, LockOutlined } from '@ant-design/icons';
|
||||||
|
import { useAuth } from '../../../../shared/hooks/useAuth';
|
||||||
|
|
||||||
|
interface LoginFormValues {
|
||||||
|
username: string;
|
||||||
|
password: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function LoginPage() {
|
||||||
|
const [loading, setLoading] = useState(false);
|
||||||
|
const navigate = useNavigate();
|
||||||
|
const login = useAuth((state) => state.login);
|
||||||
|
|
||||||
|
const onFinish = async (values: LoginFormValues) => {
|
||||||
|
setLoading(true);
|
||||||
|
try {
|
||||||
|
await login(values.username, values.password);
|
||||||
|
message.success('登录成功');
|
||||||
|
navigate('/');
|
||||||
|
} catch (error) {
|
||||||
|
message.error('用户名或密码错误');
|
||||||
|
} finally {
|
||||||
|
setLoading(false);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="min-h-screen flex items-center justify-center bg-gray-100">
|
||||||
|
<Card className="w-96 shadow-lg">
|
||||||
|
<div className="text-center mb-8">
|
||||||
|
<h1 className="text-2xl font-bold text-gray-800">iConsulting</h1>
|
||||||
|
<p className="text-gray-500">管理后台</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<Form
|
||||||
|
name="login"
|
||||||
|
onFinish={onFinish}
|
||||||
|
autoComplete="off"
|
||||||
|
size="large"
|
||||||
|
>
|
||||||
|
<Form.Item
|
||||||
|
name="username"
|
||||||
|
rules={[{ required: true, message: '请输入用户名' }]}
|
||||||
|
>
|
||||||
|
<Input
|
||||||
|
prefix={<UserOutlined />}
|
||||||
|
placeholder="用户名"
|
||||||
|
/>
|
||||||
|
</Form.Item>
|
||||||
|
|
||||||
|
<Form.Item
|
||||||
|
name="password"
|
||||||
|
rules={[{ required: true, message: '请输入密码' }]}
|
||||||
|
>
|
||||||
|
<Input.Password
|
||||||
|
prefix={<LockOutlined />}
|
||||||
|
placeholder="密码"
|
||||||
|
/>
|
||||||
|
</Form.Item>
|
||||||
|
|
||||||
|
<Form.Item>
|
||||||
|
<Button
|
||||||
|
type="primary"
|
||||||
|
htmlType="submit"
|
||||||
|
loading={loading}
|
||||||
|
block
|
||||||
|
>
|
||||||
|
登录
|
||||||
|
</Button>
|
||||||
|
</Form.Item>
|
||||||
|
</Form>
|
||||||
|
|
||||||
|
<div className="text-center text-gray-400 text-sm">
|
||||||
|
默认账号: admin / admin123
|
||||||
|
</div>
|
||||||
|
</Card>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,298 @@
|
||||||
|
import { useQuery } from '@tanstack/react-query';
|
||||||
|
import { Card, Row, Col, Statistic, Tag, Progress, List, Typography } from 'antd';
|
||||||
|
import {
|
||||||
|
UserOutlined,
|
||||||
|
MessageOutlined,
|
||||||
|
DollarOutlined,
|
||||||
|
RobotOutlined,
|
||||||
|
CheckCircleOutlined,
|
||||||
|
ClockCircleOutlined,
|
||||||
|
} from '@ant-design/icons';
|
||||||
|
import {
|
||||||
|
LineChart,
|
||||||
|
Line,
|
||||||
|
XAxis,
|
||||||
|
YAxis,
|
||||||
|
CartesianGrid,
|
||||||
|
Tooltip,
|
||||||
|
ResponsiveContainer,
|
||||||
|
PieChart,
|
||||||
|
Pie,
|
||||||
|
Cell,
|
||||||
|
} from 'recharts';
|
||||||
|
import api from '../../../../shared/utils/api';
|
||||||
|
|
||||||
|
const { Title, Text } = Typography;
|
||||||
|
|
||||||
|
// Mock数据 - 实际应该从API获取
|
||||||
|
const mockTrendData = [
|
||||||
|
{ date: '01-01', conversations: 120, users: 45 },
|
||||||
|
{ date: '01-02', conversations: 150, users: 52 },
|
||||||
|
{ date: '01-03', conversations: 180, users: 68 },
|
||||||
|
{ date: '01-04', conversations: 145, users: 55 },
|
||||||
|
{ date: '01-05', conversations: 200, users: 75 },
|
||||||
|
{ date: '01-06', conversations: 230, users: 88 },
|
||||||
|
{ date: '01-07', conversations: 210, users: 82 },
|
||||||
|
];
|
||||||
|
|
||||||
|
const mockCategoryData = [
|
||||||
|
{ name: 'QMAS', value: 35, color: '#1890ff' },
|
||||||
|
{ name: 'GEP', value: 25, color: '#52c41a' },
|
||||||
|
{ name: 'IANG', value: 20, color: '#faad14' },
|
||||||
|
{ name: 'TTPS', value: 10, color: '#722ed1' },
|
||||||
|
{ name: 'CIES', value: 7, color: '#eb2f96' },
|
||||||
|
{ name: 'TechTAS', value: 3, color: '#13c2c2' },
|
||||||
|
];
|
||||||
|
|
||||||
|
export function DashboardPage() {
|
||||||
|
const { data: evolutionStats } = useQuery({
|
||||||
|
queryKey: ['evolution-stats'],
|
||||||
|
queryFn: async () => {
|
||||||
|
const response = await api.get('/evolution/statistics');
|
||||||
|
return response.data.data;
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const { data: healthReport } = useQuery({
|
||||||
|
queryKey: ['system-health'],
|
||||||
|
queryFn: async () => {
|
||||||
|
const response = await api.get('/evolution/health');
|
||||||
|
return response.data.data;
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const getHealthColor = (status: string) => {
|
||||||
|
switch (status) {
|
||||||
|
case 'healthy':
|
||||||
|
return 'success';
|
||||||
|
case 'warning':
|
||||||
|
return 'warning';
|
||||||
|
case 'critical':
|
||||||
|
return 'error';
|
||||||
|
default:
|
||||||
|
return 'default';
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="p-6">
|
||||||
|
<Title level={4} className="mb-6">仪表盘</Title>
|
||||||
|
|
||||||
|
{/* 核心指标 */}
|
||||||
|
<Row gutter={[16, 16]}>
|
||||||
|
<Col xs={24} sm={12} lg={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="今日用户"
|
||||||
|
value={156}
|
||||||
|
prefix={<UserOutlined />}
|
||||||
|
valueStyle={{ color: '#1890ff' }}
|
||||||
|
/>
|
||||||
|
<div className="mt-2 text-gray-500 text-sm">
|
||||||
|
较昨日 <span className="text-green-500">+12%</span>
|
||||||
|
</div>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={24} sm={12} lg={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="今日对话"
|
||||||
|
value={428}
|
||||||
|
prefix={<MessageOutlined />}
|
||||||
|
valueStyle={{ color: '#52c41a' }}
|
||||||
|
/>
|
||||||
|
<div className="mt-2 text-gray-500 text-sm">
|
||||||
|
较昨日 <span className="text-green-500">+8%</span>
|
||||||
|
</div>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={24} sm={12} lg={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="今日收入"
|
||||||
|
value={3580}
|
||||||
|
prefix={<DollarOutlined />}
|
||||||
|
suffix="元"
|
||||||
|
valueStyle={{ color: '#faad14' }}
|
||||||
|
/>
|
||||||
|
<div className="mt-2 text-gray-500 text-sm">
|
||||||
|
较昨日 <span className="text-green-500">+15%</span>
|
||||||
|
</div>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={24} sm={12} lg={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="转化率"
|
||||||
|
value={8.5}
|
||||||
|
suffix="%"
|
||||||
|
prefix={<RobotOutlined />}
|
||||||
|
valueStyle={{ color: '#722ed1' }}
|
||||||
|
/>
|
||||||
|
<div className="mt-2 text-gray-500 text-sm">
|
||||||
|
较昨日 <span className="text-red-500">-2%</span>
|
||||||
|
</div>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
</Row>
|
||||||
|
|
||||||
|
{/* 趋势图表 */}
|
||||||
|
<Row gutter={[16, 16]} className="mt-4">
|
||||||
|
<Col xs={24} lg={16}>
|
||||||
|
<Card title="对话趋势">
|
||||||
|
<ResponsiveContainer width="100%" height={300}>
|
||||||
|
<LineChart data={mockTrendData}>
|
||||||
|
<CartesianGrid strokeDasharray="3 3" />
|
||||||
|
<XAxis dataKey="date" />
|
||||||
|
<YAxis />
|
||||||
|
<Tooltip />
|
||||||
|
<Line
|
||||||
|
type="monotone"
|
||||||
|
dataKey="conversations"
|
||||||
|
stroke="#1890ff"
|
||||||
|
name="对话数"
|
||||||
|
/>
|
||||||
|
<Line
|
||||||
|
type="monotone"
|
||||||
|
dataKey="users"
|
||||||
|
stroke="#52c41a"
|
||||||
|
name="用户数"
|
||||||
|
/>
|
||||||
|
</LineChart>
|
||||||
|
</ResponsiveContainer>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={24} lg={8}>
|
||||||
|
<Card title="类别分布">
|
||||||
|
<ResponsiveContainer width="100%" height={300}>
|
||||||
|
<PieChart>
|
||||||
|
<Pie
|
||||||
|
data={mockCategoryData}
|
||||||
|
dataKey="value"
|
||||||
|
nameKey="name"
|
||||||
|
cx="50%"
|
||||||
|
cy="50%"
|
||||||
|
outerRadius={80}
|
||||||
|
label={({ name, percent }) =>
|
||||||
|
`${name} ${(percent * 100).toFixed(0)}%`
|
||||||
|
}
|
||||||
|
>
|
||||||
|
{mockCategoryData.map((entry, index) => (
|
||||||
|
<Cell key={`cell-${index}`} fill={entry.color} />
|
||||||
|
))}
|
||||||
|
</Pie>
|
||||||
|
<Tooltip />
|
||||||
|
</PieChart>
|
||||||
|
</ResponsiveContainer>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
</Row>
|
||||||
|
|
||||||
|
{/* 系统状态 */}
|
||||||
|
<Row gutter={[16, 16]} className="mt-4">
|
||||||
|
<Col xs={24} lg={12}>
|
||||||
|
<Card
|
||||||
|
title={
|
||||||
|
<div className="flex items-center justify-between">
|
||||||
|
<span>系统健康</span>
|
||||||
|
<Tag color={getHealthColor(healthReport?.overall || 'healthy')}>
|
||||||
|
{healthReport?.overall === 'healthy'
|
||||||
|
? '健康'
|
||||||
|
: healthReport?.overall === 'warning'
|
||||||
|
? '警告'
|
||||||
|
: '异常'}
|
||||||
|
</Tag>
|
||||||
|
</div>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<List
|
||||||
|
dataSource={healthReport?.metrics || []}
|
||||||
|
renderItem={(item: { name: string; value: number; threshold: number; status: string }) => (
|
||||||
|
<List.Item>
|
||||||
|
<div className="w-full">
|
||||||
|
<div className="flex justify-between mb-1">
|
||||||
|
<Text>{item.name}</Text>
|
||||||
|
<Text type="secondary">
|
||||||
|
{item.value} / {item.threshold}
|
||||||
|
</Text>
|
||||||
|
</div>
|
||||||
|
<Progress
|
||||||
|
percent={Math.min(100, (item.value / item.threshold) * 100)}
|
||||||
|
status={
|
||||||
|
item.status === 'good'
|
||||||
|
? 'success'
|
||||||
|
: item.status === 'warning'
|
||||||
|
? 'exception'
|
||||||
|
: 'active'
|
||||||
|
}
|
||||||
|
showInfo={false}
|
||||||
|
size="small"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</List.Item>
|
||||||
|
)}
|
||||||
|
/>
|
||||||
|
{healthReport?.recommendations?.length > 0 && (
|
||||||
|
<div className="mt-4 p-3 bg-yellow-50 rounded">
|
||||||
|
<Text type="warning">建议：</Text>
|
||||||
|
<ul className="mt-2 ml-4 text-sm text-gray-600">
|
||||||
|
{healthReport.recommendations.map((rec: string, i: number) => (
|
||||||
|
<li key={i}>{rec}</li>
|
||||||
|
))}
|
||||||
|
</ul>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={24} lg={12}>
|
||||||
|
<Card title="经验学习进度">
|
||||||
|
<Row gutter={[16, 16]}>
|
||||||
|
<Col span={12}>
|
||||||
|
<Statistic
|
||||||
|
title="总经验"
|
||||||
|
value={evolutionStats?.totalExperiences || 0}
|
||||||
|
prefix={<RobotOutlined />}
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
<Col span={12}>
|
||||||
|
<Statistic
|
||||||
|
title="活跃经验"
|
||||||
|
value={evolutionStats?.activeExperiences || 0}
|
||||||
|
prefix={<CheckCircleOutlined />}
|
||||||
|
valueStyle={{ color: '#52c41a' }}
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
<Col span={12}>
|
||||||
|
<Statistic
|
||||||
|
title="待审核"
|
||||||
|
value={evolutionStats?.pendingExperiences || 0}
|
||||||
|
prefix={<ClockCircleOutlined />}
|
||||||
|
valueStyle={{ color: '#faad14' }}
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
<Col span={12}>
|
||||||
|
<Statistic
|
||||||
|
title="已审核"
|
||||||
|
value={evolutionStats?.approvedExperiences || 0}
|
||||||
|
prefix={<CheckCircleOutlined />}
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
</Row>
|
||||||
|
<div className="mt-4">
|
||||||
|
<Text type="secondary">经验类型分布</Text>
|
||||||
|
<div className="mt-2">
|
||||||
|
{evolutionStats?.topExperienceTypes?.map(
|
||||||
|
(item: { type: string; count: number }) => (
|
||||||
|
<Tag key={item.type} className="mb-1">
|
||||||
|
{item.type}: {item.count}
|
||||||
|
</Tag>
|
||||||
|
)
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
</Row>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,393 @@
|
||||||
|
import { useState } from 'react';
|
||||||
|
import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
|
||||||
|
import {
|
||||||
|
Card,
|
||||||
|
Table,
|
||||||
|
Button,
|
||||||
|
Select,
|
||||||
|
Tag,
|
||||||
|
Space,
|
||||||
|
Modal,
|
||||||
|
message,
|
||||||
|
Tabs,
|
||||||
|
Typography,
|
||||||
|
Statistic,
|
||||||
|
Row,
|
||||||
|
Col,
|
||||||
|
} from 'antd';
|
||||||
|
import {
|
||||||
|
CheckOutlined,
|
||||||
|
CloseOutlined,
|
||||||
|
EyeOutlined,
|
||||||
|
PlayCircleOutlined,
|
||||||
|
} from '@ant-design/icons';
|
||||||
|
import api from '../../../../shared/utils/api';
|
||||||
|
import { useAuth } from '../../../../shared/hooks/useAuth';
|
||||||
|
|
||||||
|
const { Title, Text, Paragraph } = Typography;
|
||||||
|
|
||||||
|
const EXPERIENCE_TYPES = [
|
||||||
|
{ value: 'COMMON_QUESTION', label: '常见问题' },
|
||||||
|
{ value: 'ANSWER_TEMPLATE', label: '回答模板' },
|
||||||
|
{ value: 'CLARIFICATION', label: '澄清方式' },
|
||||||
|
{ value: 'USER_PATTERN', label: '用户模式' },
|
||||||
|
{ value: 'CONVERSION_TRIGGER', label: '转化触发' },
|
||||||
|
{ value: 'KNOWLEDGE_GAP', label: '知识缺口' },
|
||||||
|
{ value: 'CONVERSATION_SKILL', label: '对话技巧' },
|
||||||
|
{ value: 'OBJECTION_HANDLING', label: '异议处理' },
|
||||||
|
];
|
||||||
|
|
||||||
|
interface Experience {
|
||||||
|
id: string;
|
||||||
|
experienceType: string;
|
||||||
|
content: string;
|
||||||
|
scenario: string;
|
||||||
|
confidence: number;
|
||||||
|
relatedCategory: string;
|
||||||
|
sourceConversationIds: string[];
|
||||||
|
verificationStatus: string;
|
||||||
|
usageCount: number;
|
||||||
|
positiveCount: number;
|
||||||
|
negativeCount: number;
|
||||||
|
isActive: boolean;
|
||||||
|
createdAt: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function ExperiencePage() {
|
||||||
|
const [activeTab, setActiveTab] = useState('pending');
|
||||||
|
const [typeFilter, setTypeFilter] = useState<string>();
|
||||||
|
const [selectedExperience, setSelectedExperience] = useState<Experience | null>(null);
|
||||||
|
const [isModalOpen, setIsModalOpen] = useState(false);
|
||||||
|
const queryClient = useQueryClient();
|
||||||
|
const admin = useAuth((state) => state.admin);
|
||||||
|
|
||||||
|
const { data: pendingData, isLoading: pendingLoading } = useQuery({
|
||||||
|
queryKey: ['pending-experiences', typeFilter],
|
||||||
|
queryFn: async () => {
|
||||||
|
const params = new URLSearchParams();
|
||||||
|
if (typeFilter) params.append('type', typeFilter);
|
||||||
|
const response = await api.get(`/memory/experience/pending?${params}`);
|
||||||
|
return response.data.data;
|
||||||
|
},
|
||||||
|
enabled: activeTab === 'pending',
|
||||||
|
});
|
||||||
|
|
||||||
|
const { data: stats } = useQuery({
|
||||||
|
queryKey: ['experience-stats'],
|
||||||
|
queryFn: async () => {
|
||||||
|
const response = await api.get('/memory/experience/statistics');
|
||||||
|
return response.data.data;
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const approveMutation = useMutation({
|
||||||
|
mutationFn: (id: string) =>
|
||||||
|
api.post(`/memory/experience/${id}/approve`, { adminId: admin?.id }),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('经验已批准');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['pending-experiences'] });
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['experience-stats'] });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const rejectMutation = useMutation({
|
||||||
|
mutationFn: (id: string) =>
|
||||||
|
api.post(`/memory/experience/${id}/reject`, { adminId: admin?.id }),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('经验已拒绝');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['pending-experiences'] });
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['experience-stats'] });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const runEvolutionMutation = useMutation({
|
||||||
|
mutationFn: () => api.post('/evolution/run', { hoursBack: 24, limit: 50 }),
|
||||||
|
onSuccess: (response) => {
|
||||||
|
const result = response.data.data;
|
||||||
|
message.success(
|
||||||
|
`进化任务完成：分析了${result.conversationsAnalyzed}个对话，提取了${result.experiencesExtracted}条经验`
|
||||||
|
);
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['pending-experiences'] });
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['experience-stats'] });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleView = (exp: Experience) => {
|
||||||
|
setSelectedExperience(exp);
|
||||||
|
setIsModalOpen(true);
|
||||||
|
};
|
||||||
|
|
||||||
|
const getTypeLabel = (type: string) => {
|
||||||
|
return EXPERIENCE_TYPES.find((t) => t.value === type)?.label || type;
|
||||||
|
};
|
||||||
|
|
||||||
|
const getStatusTag = (status: string) => {
|
||||||
|
const statusMap: Record<string, { color: string; label: string }> = {
|
||||||
|
PENDING: { color: 'orange', label: '待审核' },
|
||||||
|
APPROVED: { color: 'green', label: '已通过' },
|
||||||
|
REJECTED: { color: 'red', label: '已拒绝' },
|
||||||
|
DEPRECATED: { color: 'default', label: '已弃用' },
|
||||||
|
};
|
||||||
|
const s = statusMap[status] || { color: 'default', label: status };
|
||||||
|
return <Tag color={s.color}>{s.label}</Tag>;
|
||||||
|
};
|
||||||
|
|
||||||
|
const columns = [
|
||||||
|
{
|
||||||
|
title: '类型',
|
||||||
|
dataIndex: 'experienceType',
|
||||||
|
key: 'experienceType',
|
||||||
|
render: (type: string) => <Tag>{getTypeLabel(type)}</Tag>,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '场景',
|
||||||
|
dataIndex: 'scenario',
|
||||||
|
key: 'scenario',
|
||||||
|
ellipsis: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '内容',
|
||||||
|
dataIndex: 'content',
|
||||||
|
key: 'content',
|
||||||
|
ellipsis: true,
|
||||||
|
render: (text: string) => (
|
||||||
|
<Text ellipsis style={{ maxWidth: 200 }}>
|
||||||
|
{text}
|
||||||
|
</Text>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '置信度',
|
||||||
|
dataIndex: 'confidence',
|
||||||
|
key: 'confidence',
|
||||||
|
render: (confidence: number) => (
|
||||||
|
<span
|
||||||
|
className={
|
||||||
|
confidence >= 70
|
||||||
|
? 'text-green-600'
|
||||||
|
: confidence >= 40
|
||||||
|
? 'text-yellow-600'
|
||||||
|
: 'text-red-600'
|
||||||
|
}
|
||||||
|
>
|
||||||
|
{confidence}%
|
||||||
|
</span>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '来源对话',
|
||||||
|
dataIndex: 'sourceConversationIds',
|
||||||
|
key: 'sources',
|
||||||
|
render: (ids: string[]) => <span>{ids?.length || 0}个</span>,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '状态',
|
||||||
|
dataIndex: 'verificationStatus',
|
||||||
|
key: 'status',
|
||||||
|
render: getStatusTag,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '操作',
|
||||||
|
key: 'action',
|
||||||
|
render: (_: unknown, record: Experience) => (
|
||||||
|
<Space size="small">
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
icon={<EyeOutlined />}
|
||||||
|
onClick={() => handleView(record)}
|
||||||
|
/>
|
||||||
|
{record.verificationStatus === 'PENDING' && (
|
||||||
|
<>
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
icon={<CheckOutlined />}
|
||||||
|
className="text-green-600"
|
||||||
|
onClick={() => approveMutation.mutate(record.id)}
|
||||||
|
loading={approveMutation.isPending}
|
||||||
|
/>
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
icon={<CloseOutlined />}
|
||||||
|
danger
|
||||||
|
onClick={() => rejectMutation.mutate(record.id)}
|
||||||
|
loading={rejectMutation.isPending}
|
||||||
|
/>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</Space>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="p-6">
|
||||||
|
<div className="flex justify-between items-center mb-6">
|
||||||
|
<Title level={4} className="mb-0">系统经验管理</Title>
|
||||||
|
<Button
|
||||||
|
type="primary"
|
||||||
|
icon={<PlayCircleOutlined />}
|
||||||
|
onClick={() => runEvolutionMutation.mutate()}
|
||||||
|
loading={runEvolutionMutation.isPending}
|
||||||
|
>
|
||||||
|
运行进化任务
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 统计卡片 */}
|
||||||
|
<Row gutter={[16, 16]} className="mb-4">
|
||||||
|
<Col xs={12} sm={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic title="总经验" value={stats?.total || 0} />
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={12} sm={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="待审核"
|
||||||
|
value={stats?.byStatus?.PENDING || 0}
|
||||||
|
valueStyle={{ color: '#faad14' }}
|
||||||
|
/>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={12} sm={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="已通过"
|
||||||
|
value={stats?.byStatus?.APPROVED || 0}
|
||||||
|
valueStyle={{ color: '#52c41a' }}
|
||||||
|
/>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
<Col xs={12} sm={6}>
|
||||||
|
<Card>
|
||||||
|
<Statistic
|
||||||
|
title="已拒绝"
|
||||||
|
value={stats?.byStatus?.REJECTED || 0}
|
||||||
|
valueStyle={{ color: '#ff4d4f' }}
|
||||||
|
/>
|
||||||
|
</Card>
|
||||||
|
</Col>
|
||||||
|
</Row>
|
||||||
|
|
||||||
|
<Card>
|
||||||
|
<Tabs
|
||||||
|
activeKey={activeTab}
|
||||||
|
onChange={setActiveTab}
|
||||||
|
items={[
|
||||||
|
{ key: 'pending', label: '待审核' },
|
||||||
|
{ key: 'approved', label: '已通过' },
|
||||||
|
{ key: 'rejected', label: '已拒绝' },
|
||||||
|
]}
|
||||||
|
/>
|
||||||
|
|
||||||
|
<div className="mb-4">
|
||||||
|
<Select
|
||||||
|
placeholder="筛选类型"
|
||||||
|
allowClear
|
||||||
|
style={{ width: 180 }}
|
||||||
|
value={typeFilter}
|
||||||
|
onChange={setTypeFilter}
|
||||||
|
options={EXPERIENCE_TYPES}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<Table
|
||||||
|
columns={columns}
|
||||||
|
dataSource={pendingData?.items || []}
|
||||||
|
rowKey="id"
|
||||||
|
loading={pendingLoading}
|
||||||
|
pagination={{
|
||||||
|
total: pendingData?.total || 0,
|
||||||
|
pageSize: 20,
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
</Card>
|
||||||
|
|
||||||
|
{/* 详情弹窗 */}
|
||||||
|
<Modal
|
||||||
|
title="经验详情"
|
||||||
|
open={isModalOpen}
|
||||||
|
onCancel={() => {
|
||||||
|
setIsModalOpen(false);
|
||||||
|
setSelectedExperience(null);
|
||||||
|
}}
|
||||||
|
footer={
|
||||||
|
selectedExperience?.verificationStatus === 'PENDING'
|
||||||
|
? [
|
||||||
|
<Button
|
||||||
|
key="reject"
|
||||||
|
danger
|
||||||
|
onClick={() => {
|
||||||
|
rejectMutation.mutate(selectedExperience.id);
|
||||||
|
setIsModalOpen(false);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
拒绝
|
||||||
|
</Button>,
|
||||||
|
<Button
|
||||||
|
key="approve"
|
||||||
|
type="primary"
|
||||||
|
onClick={() => {
|
||||||
|
approveMutation.mutate(selectedExperience.id);
|
||||||
|
setIsModalOpen(false);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
批准
|
||||||
|
</Button>,
|
||||||
|
]
|
||||||
|
: null
|
||||||
|
}
|
||||||
|
width={600}
|
||||||
|
>
|
||||||
|
{selectedExperience && (
|
||||||
|
<div>
|
||||||
|
<div className="mb-4">
|
||||||
|
<Tag>{getTypeLabel(selectedExperience.experienceType)}</Tag>
|
||||||
|
{getStatusTag(selectedExperience.verificationStatus)}
|
||||||
|
{selectedExperience.isActive && <Tag color="blue">激活中</Tag>}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="mb-4">
|
||||||
|
<Text type="secondary">场景</Text>
|
||||||
|
<Paragraph>{selectedExperience.scenario}</Paragraph>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="mb-4">
|
||||||
|
<Text type="secondary">内容</Text>
|
||||||
|
<Paragraph>{selectedExperience.content}</Paragraph>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<Row gutter={16}>
|
||||||
|
<Col span={8}>
|
||||||
|
<Statistic
|
||||||
|
title="置信度"
|
||||||
|
value={selectedExperience.confidence}
|
||||||
|
suffix="%"
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
<Col span={8}>
|
||||||
|
<Statistic
|
||||||
|
title="使用次数"
|
||||||
|
value={selectedExperience.usageCount}
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
<Col span={8}>
|
||||||
|
<Statistic
|
||||||
|
title="来源对话"
|
||||||
|
value={selectedExperience.sourceConversationIds?.length || 0}
|
||||||
|
/>
|
||||||
|
</Col>
|
||||||
|
</Row>
|
||||||
|
|
||||||
|
{selectedExperience.relatedCategory && (
|
||||||
|
<div className="mt-4">
|
||||||
|
<Text type="secondary">相关类别: </Text>
|
||||||
|
<Tag color="blue">{selectedExperience.relatedCategory}</Tag>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</Modal>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,404 @@
|
||||||
|
import { useState } from 'react';
|
||||||
|
import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
|
||||||
|
import {
|
||||||
|
Card,
|
||||||
|
Table,
|
||||||
|
Button,
|
||||||
|
Input,
|
||||||
|
Select,
|
||||||
|
Tag,
|
||||||
|
Space,
|
||||||
|
Modal,
|
||||||
|
Form,
|
||||||
|
message,
|
||||||
|
Popconfirm,
|
||||||
|
Typography,
|
||||||
|
Drawer,
|
||||||
|
} from 'antd';
|
||||||
|
import {
|
||||||
|
PlusOutlined,
|
||||||
|
SearchOutlined,
|
||||||
|
EditOutlined,
|
||||||
|
DeleteOutlined,
|
||||||
|
EyeOutlined,
|
||||||
|
CheckOutlined,
|
||||||
|
StopOutlined,
|
||||||
|
} from '@ant-design/icons';
|
||||||
|
import api from '../../../../shared/utils/api';
|
||||||
|
|
||||||
|
const { Title, Paragraph } = Typography;
|
||||||
|
const { TextArea } = Input;
|
||||||
|
|
||||||
|
const CATEGORIES = [
|
||||||
|
{ value: 'QMAS', label: '优秀人才入境计划' },
|
||||||
|
{ value: 'GEP', label: '一般就业政策' },
|
||||||
|
{ value: 'IANG', label: '非本地毕业生留港/回港就业安排' },
|
||||||
|
{ value: 'TTPS', label: '科技人才入境计划' },
|
||||||
|
{ value: 'CIES', label: '资本投资者入境计划' },
|
||||||
|
{ value: 'TechTAS', label: '顶尖人才通行证计划' },
|
||||||
|
{ value: 'GENERAL', label: '通用知识' },
|
||||||
|
];
|
||||||
|
|
||||||
|
interface Article {
|
||||||
|
id: string;
|
||||||
|
title: string;
|
||||||
|
content: string;
|
||||||
|
summary: string;
|
||||||
|
category: string;
|
||||||
|
tags: string[];
|
||||||
|
source: string;
|
||||||
|
isPublished: boolean;
|
||||||
|
citationCount: number;
|
||||||
|
qualityScore: number;
|
||||||
|
createdAt: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function KnowledgePage() {
|
||||||
|
const [searchText, setSearchText] = useState('');
|
||||||
|
const [categoryFilter, setCategoryFilter] = useState<string>();
|
||||||
|
const [isModalOpen, setIsModalOpen] = useState(false);
|
||||||
|
const [isDrawerOpen, setIsDrawerOpen] = useState(false);
|
||||||
|
const [selectedArticle, setSelectedArticle] = useState<Article | null>(null);
|
||||||
|
const [form] = Form.useForm();
|
||||||
|
const queryClient = useQueryClient();
|
||||||
|
|
||||||
|
const { data, isLoading } = useQuery({
|
||||||
|
queryKey: ['knowledge-articles', categoryFilter],
|
||||||
|
queryFn: async () => {
|
||||||
|
const params = new URLSearchParams();
|
||||||
|
if (categoryFilter) params.append('category', categoryFilter);
|
||||||
|
const response = await api.get(`/knowledge/articles?${params}`);
|
||||||
|
return response.data.data;
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const createMutation = useMutation({
|
||||||
|
mutationFn: (values: Partial<Article>) =>
|
||||||
|
api.post('/knowledge/articles', values),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('文章创建成功');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['knowledge-articles'] });
|
||||||
|
setIsModalOpen(false);
|
||||||
|
form.resetFields();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const updateMutation = useMutation({
|
||||||
|
mutationFn: ({ id, ...values }: { id: string } & Partial<Article>) =>
|
||||||
|
api.put(`/knowledge/articles/${id}`, values),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('文章更新成功');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['knowledge-articles'] });
|
||||||
|
setIsModalOpen(false);
|
||||||
|
form.resetFields();
|
||||||
|
setSelectedArticle(null);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const deleteMutation = useMutation({
|
||||||
|
mutationFn: (id: string) => api.delete(`/knowledge/articles/${id}`),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('文章已删除');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['knowledge-articles'] });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const publishMutation = useMutation({
|
||||||
|
mutationFn: (id: string) => api.post(`/knowledge/articles/${id}/publish`),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('文章已发布');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['knowledge-articles'] });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const unpublishMutation = useMutation({
|
||||||
|
mutationFn: (id: string) => api.post(`/knowledge/articles/${id}/unpublish`),
|
||||||
|
onSuccess: () => {
|
||||||
|
message.success('文章已取消发布');
|
||||||
|
queryClient.invalidateQueries({ queryKey: ['knowledge-articles'] });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const handleEdit = (article: Article) => {
|
||||||
|
setSelectedArticle(article);
|
||||||
|
form.setFieldsValue({
|
||||||
|
title: article.title,
|
||||||
|
content: article.content,
|
||||||
|
category: article.category,
|
||||||
|
tags: article.tags,
|
||||||
|
});
|
||||||
|
setIsModalOpen(true);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleView = (article: Article) => {
|
||||||
|
setSelectedArticle(article);
|
||||||
|
setIsDrawerOpen(true);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleSubmit = (values: Partial<Article>) => {
|
||||||
|
if (selectedArticle) {
|
||||||
|
updateMutation.mutate({ id: selectedArticle.id, ...values });
|
||||||
|
} else {
|
||||||
|
createMutation.mutate(values);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const columns = [
|
||||||
|
{
|
||||||
|
title: '标题',
|
||||||
|
dataIndex: 'title',
|
||||||
|
key: 'title',
|
||||||
|
render: (text: string, record: Article) => (
|
||||||
|
<a onClick={() => handleView(record)}>{text}</a>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '类别',
|
||||||
|
dataIndex: 'category',
|
||||||
|
key: 'category',
|
||||||
|
render: (category: string) => {
|
||||||
|
const cat = CATEGORIES.find((c) => c.value === category);
|
||||||
|
return <Tag color="blue">{cat?.label || category}</Tag>;
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '来源',
|
||||||
|
dataIndex: 'source',
|
||||||
|
key: 'source',
|
||||||
|
render: (source: string) => {
|
||||||
|
const sourceMap: Record<string, { color: string; label: string }> = {
|
||||||
|
MANUAL: { color: 'green', label: '手动' },
|
||||||
|
CRAWL: { color: 'orange', label: '爬取' },
|
||||||
|
EXTRACT: { color: 'purple', label: '提取' },
|
||||||
|
IMPORT: { color: 'cyan', label: '导入' },
|
||||||
|
};
|
||||||
|
const s = sourceMap[source] || { color: 'default', label: source };
|
||||||
|
return <Tag color={s.color}>{s.label}</Tag>;
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '状态',
|
||||||
|
dataIndex: 'isPublished',
|
||||||
|
key: 'isPublished',
|
||||||
|
render: (isPublished: boolean) =>
|
||||||
|
isPublished ? (
|
||||||
|
<Tag color="success">已发布</Tag>
|
||||||
|
) : (
|
||||||
|
<Tag color="default">草稿</Tag>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '质量分',
|
||||||
|
dataIndex: 'qualityScore',
|
||||||
|
key: 'qualityScore',
|
||||||
|
render: (score: number) => (
|
||||||
|
<span
|
||||||
|
className={
|
||||||
|
score >= 70
|
||||||
|
? 'text-green-600'
|
||||||
|
: score >= 40
|
||||||
|
? 'text-yellow-600'
|
||||||
|
: 'text-red-600'
|
||||||
|
}
|
||||||
|
>
|
||||||
|
{score}
|
||||||
|
</span>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '引用次数',
|
||||||
|
dataIndex: 'citationCount',
|
||||||
|
key: 'citationCount',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: '操作',
|
||||||
|
key: 'action',
|
||||||
|
render: (_: unknown, record: Article) => (
|
||||||
|
<Space size="small">
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
icon={<EyeOutlined />}
|
||||||
|
onClick={() => handleView(record)}
|
||||||
|
/>
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
icon={<EditOutlined />}
|
||||||
|
onClick={() => handleEdit(record)}
|
||||||
|
/>
|
||||||
|
{record.isPublished ? (
|
||||||
|
<Popconfirm
|
||||||
|
title="确定取消发布？"
|
||||||
|
onConfirm={() => unpublishMutation.mutate(record.id)}
|
||||||
|
>
|
||||||
|
<Button type="text" icon={<StopOutlined />} />
|
||||||
|
</Popconfirm>
|
||||||
|
) : (
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
icon={<CheckOutlined />}
|
||||||
|
onClick={() => publishMutation.mutate(record.id)}
|
||||||
|
/>
|
||||||
|
)}
|
||||||
|
<Popconfirm
|
||||||
|
title="确定删除此文章？"
|
||||||
|
onConfirm={() => deleteMutation.mutate(record.id)}
|
||||||
|
>
|
||||||
|
<Button type="text" danger icon={<DeleteOutlined />} />
|
||||||
|
</Popconfirm>
|
||||||
|
</Space>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="p-6">
|
||||||
|
<Title level={4} className="mb-6">知识库管理</Title>
|
||||||
|
|
||||||
|
<Card>
|
||||||
|
<div className="flex justify-between mb-4">
|
||||||
|
<Space>
|
||||||
|
<Input
|
||||||
|
placeholder="搜索文章"
|
||||||
|
prefix={<SearchOutlined />}
|
||||||
|
value={searchText}
|
||||||
|
onChange={(e) => setSearchText(e.target.value)}
|
||||||
|
style={{ width: 200 }}
|
||||||
|
/>
|
||||||
|
<Select
|
||||||
|
placeholder="选择类别"
|
||||||
|
allowClear
|
||||||
|
style={{ width: 180 }}
|
||||||
|
value={categoryFilter}
|
||||||
|
onChange={setCategoryFilter}
|
||||||
|
options={CATEGORIES}
|
||||||
|
/>
|
||||||
|
</Space>
|
||||||
|
<Button
|
||||||
|
type="primary"
|
||||||
|
icon={<PlusOutlined />}
|
||||||
|
onClick={() => {
|
||||||
|
setSelectedArticle(null);
|
||||||
|
form.resetFields();
|
||||||
|
setIsModalOpen(true);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
新建文章
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<Table
|
||||||
|
columns={columns}
|
||||||
|
dataSource={data?.items || []}
|
||||||
|
rowKey="id"
|
||||||
|
loading={isLoading}
|
||||||
|
pagination={{
|
||||||
|
total: data?.total || 0,
|
||||||
|
pageSize: 20,
|
||||||
|
showSizeChanger: false,
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
</Card>
|
||||||
|
|
||||||
|
{/* 编辑/新建弹窗 */}
|
||||||
|
<Modal
|
||||||
|
title={selectedArticle ? '编辑文章' : '新建文章'}
|
||||||
|
open={isModalOpen}
|
||||||
|
onCancel={() => {
|
||||||
|
setIsModalOpen(false);
|
||||||
|
setSelectedArticle(null);
|
||||||
|
form.resetFields();
|
||||||
|
}}
|
||||||
|
footer={null}
|
||||||
|
width={800}
|
||||||
|
>
|
||||||
|
<Form form={form} layout="vertical" onFinish={handleSubmit}>
|
||||||
|
<Form.Item
|
||||||
|
name="title"
|
||||||
|
label="标题"
|
||||||
|
rules={[{ required: true, message: '请输入标题' }]}
|
||||||
|
>
|
||||||
|
<Input placeholder="文章标题" />
|
||||||
|
</Form.Item>
|
||||||
|
|
||||||
|
<Form.Item
|
||||||
|
name="category"
|
||||||
|
label="类别"
|
||||||
|
rules={[{ required: true, message: '请选择类别' }]}
|
||||||
|
>
|
||||||
|
<Select options={CATEGORIES} placeholder="选择移民类别" />
|
||||||
|
</Form.Item>
|
||||||
|
|
||||||
|
<Form.Item name="tags" label="标签">
|
||||||
|
<Select mode="tags" placeholder="输入标签后回车" />
|
||||||
|
</Form.Item>
|
||||||
|
|
||||||
|
<Form.Item
|
||||||
|
name="content"
|
||||||
|
label="内容"
|
||||||
|
rules={[{ required: true, message: '请输入内容' }]}
|
||||||
|
>
|
||||||
|
<TextArea rows={12} placeholder="支持Markdown格式" />
|
||||||
|
</Form.Item>
|
||||||
|
|
||||||
|
<Form.Item className="mb-0 text-right">
|
||||||
|
<Space>
|
||||||
|
<Button onClick={() => setIsModalOpen(false)}>取消</Button>
|
||||||
|
<Button
|
||||||
|
type="primary"
|
||||||
|
htmlType="submit"
|
||||||
|
loading={createMutation.isPending || updateMutation.isPending}
|
||||||
|
>
|
||||||
|
保存
|
||||||
|
</Button>
|
||||||
|
</Space>
|
||||||
|
</Form.Item>
|
||||||
|
</Form>
|
||||||
|
</Modal>
|
||||||
|
|
||||||
|
{/* 预览抽屉 */}
|
||||||
|
<Drawer
|
||||||
|
title={selectedArticle?.title}
|
||||||
|
open={isDrawerOpen}
|
||||||
|
onClose={() => {
|
||||||
|
setIsDrawerOpen(false);
|
||||||
|
setSelectedArticle(null);
|
||||||
|
}}
|
||||||
|
width={600}
|
||||||
|
>
|
||||||
|
{selectedArticle && (
|
||||||
|
<div>
|
||||||
|
<div className="mb-4">
|
||||||
|
<Tag color="blue">
|
||||||
|
{CATEGORIES.find((c) => c.value === selectedArticle.category)
|
||||||
|
?.label || selectedArticle.category}
|
||||||
|
</Tag>
|
||||||
|
{selectedArticle.isPublished ? (
|
||||||
|
<Tag color="success">已发布</Tag>
|
||||||
|
) : (
|
||||||
|
<Tag>草稿</Tag>
|
||||||
|
)}
|
||||||
|
<span className="ml-2 text-gray-500 text-sm">
|
||||||
|
质量分: {selectedArticle.qualityScore} | 引用:{' '}
|
||||||
|
{selectedArticle.citationCount}次
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{selectedArticle.tags?.length > 0 && (
|
||||||
|
<div className="mb-4">
|
||||||
|
{selectedArticle.tags.map((tag) => (
|
||||||
|
<Tag key={tag}>{tag}</Tag>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<Paragraph className="text-gray-600 mb-4">
|
||||||
|
{selectedArticle.summary}
|
||||||
|
</Paragraph>
|
||||||
|
|
||||||
|
<div className="whitespace-pre-wrap">{selectedArticle.content}</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</Drawer>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,13 @@
|
||||||
|
@tailwind base;
|
||||||
|
@tailwind components;
|
||||||
|
@tailwind utilities;
|
||||||
|
|
||||||
|
body {
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
|
||||||
|
}
|
||||||
|
|
||||||
|
#root {
|
||||||
|
min-height: 100vh;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,36 @@
|
||||||
|
import React from 'react';
|
||||||
|
import ReactDOM from 'react-dom/client';
|
||||||
|
import { BrowserRouter } from 'react-router-dom';
|
||||||
|
import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
|
||||||
|
import { ConfigProvider } from 'antd';
|
||||||
|
import zhCN from 'antd/locale/zh_CN';
|
||||||
|
import App from './App';
|
||||||
|
import './index.css';
|
||||||
|
|
||||||
|
const queryClient = new QueryClient({
|
||||||
|
defaultOptions: {
|
||||||
|
queries: {
|
||||||
|
staleTime: 5 * 60 * 1000,
|
||||||
|
retry: 1,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
ReactDOM.createRoot(document.getElementById('root')!).render(
|
||||||
|
<React.StrictMode>
|
||||||
|
<QueryClientProvider client={queryClient}>
|
||||||
|
<ConfigProvider
|
||||||
|
locale={zhCN}
|
||||||
|
theme={{
|
||||||
|
token: {
|
||||||
|
colorPrimary: '#1890ff',
|
||||||
|
},
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<BrowserRouter>
|
||||||
|
<App />
|
||||||
|
</BrowserRouter>
|
||||||
|
</ConfigProvider>
|
||||||
|
</QueryClientProvider>
|
||||||
|
</React.StrictMode>,
|
||||||
|
);
|
||||||
|
|
@ -0,0 +1,128 @@
|
||||||
|
import { useState } from 'react';
|
||||||
|
import { useNavigate, useLocation, Outlet } from 'react-router-dom';
|
||||||
|
import { Layout, Menu, Avatar, Dropdown, Typography } from 'antd';
|
||||||
|
import type { MenuProps } from 'antd';
|
||||||
|
import {
|
||||||
|
DashboardOutlined,
|
||||||
|
BookOutlined,
|
||||||
|
RobotOutlined,
|
||||||
|
UserOutlined,
|
||||||
|
SettingOutlined,
|
||||||
|
LogoutOutlined,
|
||||||
|
MenuFoldOutlined,
|
||||||
|
MenuUnfoldOutlined,
|
||||||
|
} from '@ant-design/icons';
|
||||||
|
import { useAuth } from '../hooks/useAuth';
|
||||||
|
|
||||||
|
const { Header, Sider, Content } = Layout;
|
||||||
|
const { Text } = Typography;
|
||||||
|
|
||||||
|
const menuItems: MenuProps['items'] = [
|
||||||
|
{
|
||||||
|
key: '/',
|
||||||
|
icon: <DashboardOutlined />,
|
||||||
|
label: '仪表盘',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: '/knowledge',
|
||||||
|
icon: <BookOutlined />,
|
||||||
|
label: '知识库',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: '/experience',
|
||||||
|
icon: <RobotOutlined />,
|
||||||
|
label: '系统经验',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: '/users',
|
||||||
|
icon: <UserOutlined />,
|
||||||
|
label: '用户管理',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: '/settings',
|
||||||
|
icon: <SettingOutlined />,
|
||||||
|
label: '系统设置',
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
export function MainLayout() {
|
||||||
|
const [collapsed, setCollapsed] = useState(false);
|
||||||
|
const navigate = useNavigate();
|
||||||
|
const location = useLocation();
|
||||||
|
const { admin, logout } = useAuth();
|
||||||
|
|
||||||
|
const handleMenuClick = (e: { key: string }) => {
|
||||||
|
navigate(e.key);
|
||||||
|
};
|
||||||
|
|
||||||
|
const userMenuItems: MenuProps['items'] = [
|
||||||
|
{
|
||||||
|
key: 'profile',
|
||||||
|
icon: <UserOutlined />,
|
||||||
|
label: '个人设置',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'divider',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
key: 'logout',
|
||||||
|
icon: <LogoutOutlined />,
|
||||||
|
label: '退出登录',
|
||||||
|
danger: true,
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
const handleUserMenuClick: MenuProps['onClick'] = ({ key }) => {
|
||||||
|
if (key === 'logout') {
|
||||||
|
logout();
|
||||||
|
navigate('/login');
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Layout className="min-h-screen">
|
||||||
|
<Sider
|
||||||
|
trigger={null}
|
||||||
|
collapsible
|
||||||
|
collapsed={collapsed}
|
||||||
|
theme="light"
|
||||||
|
className="shadow-sm"
|
||||||
|
>
|
||||||
|
<div className="h-16 flex items-center justify-center border-b">
|
||||||
|
<Text strong className="text-lg">
|
||||||
|
{collapsed ? 'iC' : 'iConsulting'}
|
||||||
|
</Text>
|
||||||
|
</div>
|
||||||
|
<Menu
|
||||||
|
mode="inline"
|
||||||
|
selectedKeys={[location.pathname]}
|
||||||
|
items={menuItems}
|
||||||
|
onClick={handleMenuClick}
|
||||||
|
className="border-none"
|
||||||
|
/>
|
||||||
|
</Sider>
|
||||||
|
<Layout>
|
||||||
|
<Header className="bg-white px-4 flex items-center justify-between shadow-sm">
|
||||||
|
<div
|
||||||
|
className="cursor-pointer text-lg"
|
||||||
|
onClick={() => setCollapsed(!collapsed)}
|
||||||
|
>
|
||||||
|
{collapsed ? <MenuUnfoldOutlined /> : <MenuFoldOutlined />}
|
||||||
|
</div>
|
||||||
|
<Dropdown
|
||||||
|
menu={{ items: userMenuItems, onClick: handleUserMenuClick }}
|
||||||
|
placement="bottomRight"
|
||||||
|
>
|
||||||
|
<div className="flex items-center cursor-pointer">
|
||||||
|
<Avatar icon={<UserOutlined />} className="mr-2" />
|
||||||
|
<Text>{admin?.name || 'Admin'}</Text>
|
||||||
|
</div>
|
||||||
|
</Dropdown>
|
||||||
|
</Header>
|
||||||
|
<Content className="bg-gray-100 min-h-0 overflow-auto">
|
||||||
|
<Outlet />
|
||||||
|
</Content>
|
||||||
|
</Layout>
|
||||||
|
</Layout>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,36 @@
|
||||||
|
import { useEffect, useState } from 'react';
|
||||||
|
import { Navigate, useLocation } from 'react-router-dom';
|
||||||
|
import { Spin } from 'antd';
|
||||||
|
import { useAuth } from '../hooks/useAuth';
|
||||||
|
|
||||||
|
interface ProtectedRouteProps {
|
||||||
|
children: React.ReactNode;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function ProtectedRoute({ children }: ProtectedRouteProps) {
|
||||||
|
const [checking, setChecking] = useState(true);
|
||||||
|
const location = useLocation();
|
||||||
|
const { isAuthenticated, checkAuth } = useAuth();
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const verify = async () => {
|
||||||
|
await checkAuth();
|
||||||
|
setChecking(false);
|
||||||
|
};
|
||||||
|
verify();
|
||||||
|
}, [checkAuth]);
|
||||||
|
|
||||||
|
if (checking) {
|
||||||
|
return (
|
||||||
|
<div className="min-h-screen flex items-center justify-center">
|
||||||
|
<Spin size="large" />
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!isAuthenticated) {
|
||||||
|
return <Navigate to="/login" state={{ from: location }} replace />;
|
||||||
|
}
|
||||||
|
|
||||||
|
return <>{children}</>;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,105 @@
|
||||||
|
import { create } from 'zustand';
|
||||||
|
import { persist } from 'zustand/middleware';
|
||||||
|
import api from '../utils/api';
|
||||||
|
|
||||||
|
interface AdminInfo {
|
||||||
|
id: string;
|
||||||
|
username: string;
|
||||||
|
name: string;
|
||||||
|
role: string;
|
||||||
|
permissions: string[];
|
||||||
|
}
|
||||||
|
|
||||||
|
interface AuthState {
|
||||||
|
admin: AdminInfo | null;
|
||||||
|
token: string | null;
|
||||||
|
isAuthenticated: boolean;
|
||||||
|
login: (username: string, password: string) => Promise<void>;
|
||||||
|
logout: () => void;
|
||||||
|
checkAuth: () => Promise<boolean>;
|
||||||
|
hasPermission: (permission: string) => boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const useAuth = create<AuthState>()(
|
||||||
|
persist(
|
||||||
|
(set, get) => ({
|
||||||
|
admin: null,
|
||||||
|
token: null,
|
||||||
|
isAuthenticated: false,
|
||||||
|
|
||||||
|
login: async (username: string, password: string) => {
|
||||||
|
const response = await api.post('/admin/login', { username, password });
|
||||||
|
const { data } = response.data;
|
||||||
|
|
||||||
|
localStorage.setItem('admin_token', data.token);
|
||||||
|
|
||||||
|
set({
|
||||||
|
admin: data.admin,
|
||||||
|
token: data.token,
|
||||||
|
isAuthenticated: true,
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
logout: () => {
|
||||||
|
localStorage.removeItem('admin_token');
|
||||||
|
set({
|
||||||
|
admin: null,
|
||||||
|
token: null,
|
||||||
|
isAuthenticated: false,
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
checkAuth: async () => {
|
||||||
|
const token = localStorage.getItem('admin_token');
|
||||||
|
if (!token) {
|
||||||
|
set({ isAuthenticated: false, admin: null, token: null });
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await api.get('/admin/verify');
|
||||||
|
if (response.data.success) {
|
||||||
|
set({
|
||||||
|
admin: response.data.data,
|
||||||
|
token,
|
||||||
|
isAuthenticated: true,
|
||||||
|
});
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
localStorage.removeItem('admin_token');
|
||||||
|
}
|
||||||
|
|
||||||
|
set({ isAuthenticated: false, admin: null, token: null });
|
||||||
|
return false;
|
||||||
|
},
|
||||||
|
|
||||||
|
hasPermission: (permission: string) => {
|
||||||
|
const { admin } = get();
|
||||||
|
if (!admin) return false;
|
||||||
|
|
||||||
|
const permissions = admin.permissions || [];
|
||||||
|
|
||||||
|
// 超管拥有所有权限
|
||||||
|
if (permissions.includes('*')) return true;
|
||||||
|
|
||||||
|
// 完全匹配
|
||||||
|
if (permissions.includes(permission)) return true;
|
||||||
|
|
||||||
|
// 通配符匹配
|
||||||
|
const [resource] = permission.split(':');
|
||||||
|
if (permissions.includes(`${resource}:*`)) return true;
|
||||||
|
|
||||||
|
return false;
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
{
|
||||||
|
name: 'auth-storage',
|
||||||
|
partialize: (state) => ({
|
||||||
|
admin: state.admin,
|
||||||
|
token: state.token,
|
||||||
|
isAuthenticated: state.isAuthenticated,
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
);
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
import axios from 'axios';
|
||||||
|
|
||||||
|
const api = axios.create({
|
||||||
|
baseURL: '/api/v1',
|
||||||
|
timeout: 30000,
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
// 请求拦截器 - 添加Token
|
||||||
|
api.interceptors.request.use(
|
||||||
|
(config) => {
|
||||||
|
const token = localStorage.getItem('admin_token');
|
||||||
|
if (token) {
|
||||||
|
config.headers.Authorization = `Bearer ${token}`;
|
||||||
|
}
|
||||||
|
return config;
|
||||||
|
},
|
||||||
|
(error) => {
|
||||||
|
return Promise.reject(error);
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
// 响应拦截器 - 处理认证错误
|
||||||
|
api.interceptors.response.use(
|
||||||
|
(response) => response,
|
||||||
|
(error) => {
|
||||||
|
if (error.response?.status === 401) {
|
||||||
|
localStorage.removeItem('admin_token');
|
||||||
|
window.location.href = '/login';
|
||||||
|
}
|
||||||
|
return Promise.reject(error);
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
export default api;
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
/// <reference types="vite/client" />
|
||||||
|
|
@ -0,0 +1,14 @@
|
||||||
|
/** @type {import('tailwindcss').Config} */
|
||||||
|
export default {
|
||||||
|
content: [
|
||||||
|
"./index.html",
|
||||||
|
"./src/**/*.{js,ts,jsx,tsx}",
|
||||||
|
],
|
||||||
|
theme: {
|
||||||
|
extend: {},
|
||||||
|
},
|
||||||
|
plugins: [],
|
||||||
|
corePlugins: {
|
||||||
|
preflight: false, // 禁用Tailwind的reset，避免与Antd冲突
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,25 @@
|
||||||
|
{
|
||||||
|
"compilerOptions": {
|
||||||
|
"target": "ES2020",
|
||||||
|
"useDefineForClassFields": true,
|
||||||
|
"lib": ["ES2020", "DOM", "DOM.Iterable"],
|
||||||
|
"module": "ESNext",
|
||||||
|
"skipLibCheck": true,
|
||||||
|
"moduleResolution": "bundler",
|
||||||
|
"allowImportingTsExtensions": true,
|
||||||
|
"resolveJsonModule": true,
|
||||||
|
"isolatedModules": true,
|
||||||
|
"noEmit": true,
|
||||||
|
"jsx": "react-jsx",
|
||||||
|
"strict": true,
|
||||||
|
"noUnusedLocals": true,
|
||||||
|
"noUnusedParameters": true,
|
||||||
|
"noFallthroughCasesInSwitch": true,
|
||||||
|
"baseUrl": ".",
|
||||||
|
"paths": {
|
||||||
|
"@/*": ["src/*"]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"include": ["src"],
|
||||||
|
"references": [{ "path": "./tsconfig.node.json" }]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,10 @@
|
||||||
|
{
|
||||||
|
"compilerOptions": {
|
||||||
|
"composite": true,
|
||||||
|
"skipLibCheck": true,
|
||||||
|
"module": "ESNext",
|
||||||
|
"moduleResolution": "bundler",
|
||||||
|
"allowSyntheticDefaultImports": true
|
||||||
|
},
|
||||||
|
"include": ["vite.config.ts"]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,19 @@
|
||||||
|
import { defineConfig } from 'vite';
|
||||||
|
import react from '@vitejs/plugin-react';
|
||||||
|
|
||||||
|
export default defineConfig({
|
||||||
|
plugins: [react()],
|
||||||
|
server: {
|
||||||
|
port: 5174,
|
||||||
|
proxy: {
|
||||||
|
'/api': {
|
||||||
|
target: 'http://localhost:3005',
|
||||||
|
changeOrigin: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
build: {
|
||||||
|
outDir: 'dist',
|
||||||
|
sourcemap: true,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
@ -0,0 +1,45 @@
|
||||||
|
# ===========================================
|
||||||
|
# iConsulting Conversation Service Dockerfile
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
FROM node:20-alpine AS builder
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
|
||||||
|
COPY pnpm-workspace.yaml package.json pnpm-lock.yaml ./
|
||||||
|
COPY packages/shared/package.json ./packages/shared/
|
||||||
|
COPY packages/services/conversation-service/package.json ./packages/services/conversation-service/
|
||||||
|
|
||||||
|
RUN pnpm install --frozen-lockfile
|
||||||
|
|
||||||
|
COPY packages/shared ./packages/shared
|
||||||
|
COPY packages/services/conversation-service ./packages/services/conversation-service
|
||||||
|
COPY tsconfig.base.json ./
|
||||||
|
|
||||||
|
RUN pnpm --filter @iconsulting/shared build
|
||||||
|
RUN pnpm --filter @iconsulting/conversation-service build
|
||||||
|
|
||||||
|
FROM node:20-alpine AS runner
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
RUN addgroup --system --gid 1001 nodejs && \
|
||||||
|
adduser --system --uid 1001 nestjs
|
||||||
|
|
||||||
|
COPY --from=builder /app/packages/services/conversation-service/dist ./dist
|
||||||
|
COPY --from=builder /app/packages/services/conversation-service/package.json ./
|
||||||
|
COPY --from=builder /app/node_modules ./node_modules
|
||||||
|
|
||||||
|
ENV NODE_ENV=production
|
||||||
|
ENV PORT=3004
|
||||||
|
|
||||||
|
USER nestjs
|
||||||
|
EXPOSE 3004
|
||||||
|
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||||
|
CMD wget --no-verbose --tries=1 --spider http://localhost:3004/health || exit 1
|
||||||
|
|
||||||
|
CMD ["node", "dist/main.js"]
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
{
|
||||||
|
"$schema": "https://json.schemastore.org/nest-cli",
|
||||||
|
"collection": "@nestjs/schematics",
|
||||||
|
"sourceRoot": "src",
|
||||||
|
"compilerOptions": {
|
||||||
|
"deleteOutDir": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,48 @@
|
||||||
|
{
|
||||||
|
"name": "@iconsulting/conversation-service",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "Conversation service with Claude Agent SDK integration",
|
||||||
|
"main": "dist/main.js",
|
||||||
|
"scripts": {
|
||||||
|
"build": "nest build",
|
||||||
|
"dev": "nest start --watch",
|
||||||
|
"start": "node dist/main.js",
|
||||||
|
"start:debug": "nest start --debug --watch",
|
||||||
|
"lint": "eslint \"{src,apps,libs,test}/**/*.ts\"",
|
||||||
|
"test": "jest",
|
||||||
|
"test:watch": "jest --watch",
|
||||||
|
"test:cov": "jest --coverage"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@anthropic-ai/sdk": "^0.52.0",
|
||||||
|
"@iconsulting/shared": "workspace:*",
|
||||||
|
"@nestjs/common": "^10.0.0",
|
||||||
|
"@nestjs/config": "^3.2.0",
|
||||||
|
"@nestjs/core": "^10.0.0",
|
||||||
|
"@nestjs/platform-express": "^10.0.0",
|
||||||
|
"@nestjs/platform-socket.io": "^10.0.0",
|
||||||
|
"@nestjs/typeorm": "^10.0.0",
|
||||||
|
"@nestjs/websockets": "^10.0.0",
|
||||||
|
"class-transformer": "^0.5.1",
|
||||||
|
"class-validator": "^0.14.0",
|
||||||
|
"ioredis": "^5.3.0",
|
||||||
|
"kafkajs": "^2.2.4",
|
||||||
|
"pg": "^8.11.0",
|
||||||
|
"rxjs": "^7.8.0",
|
||||||
|
"socket.io": "^4.8.3",
|
||||||
|
"typeorm": "^0.3.19",
|
||||||
|
"uuid": "^9.0.0"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@nestjs/cli": "^10.0.0",
|
||||||
|
"@nestjs/testing": "^10.0.0",
|
||||||
|
"@types/express": "^4.17.21",
|
||||||
|
"@types/jest": "^29.5.0",
|
||||||
|
"@types/node": "^20.10.0",
|
||||||
|
"@types/socket.io": "^3.0.2",
|
||||||
|
"@types/uuid": "^9.0.0",
|
||||||
|
"jest": "^29.7.0",
|
||||||
|
"ts-jest": "^29.1.0",
|
||||||
|
"typescript": "^5.3.0"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { ConfigModule, ConfigService } from '@nestjs/config';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { ConversationModule } from './conversation/conversation.module';
|
||||||
|
import { ClaudeModule } from './infrastructure/claude/claude.module';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
// Configuration
|
||||||
|
ConfigModule.forRoot({
|
||||||
|
isGlobal: true,
|
||||||
|
envFilePath: ['.env.local', '.env'],
|
||||||
|
}),
|
||||||
|
|
||||||
|
// Database
|
||||||
|
TypeOrmModule.forRootAsync({
|
||||||
|
imports: [ConfigModule],
|
||||||
|
inject: [ConfigService],
|
||||||
|
useFactory: (configService: ConfigService) => ({
|
||||||
|
type: 'postgres',
|
||||||
|
host: configService.get('POSTGRES_HOST', 'localhost'),
|
||||||
|
port: configService.get<number>('POSTGRES_PORT', 5432),
|
||||||
|
username: configService.get('POSTGRES_USER', 'iconsulting'),
|
||||||
|
password: configService.get('POSTGRES_PASSWORD'),
|
||||||
|
database: configService.get('POSTGRES_DB', 'iconsulting'),
|
||||||
|
entities: [__dirname + '/**/*.entity{.ts,.js}'],
|
||||||
|
synchronize: configService.get('NODE_ENV') === 'development',
|
||||||
|
logging: configService.get('NODE_ENV') === 'development',
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
|
||||||
|
// Feature modules
|
||||||
|
ConversationModule,
|
||||||
|
ClaudeModule,
|
||||||
|
],
|
||||||
|
})
|
||||||
|
export class AppModule {}
|
||||||
|
|
@ -0,0 +1,142 @@
|
||||||
|
import {
|
||||||
|
Controller,
|
||||||
|
Get,
|
||||||
|
Post,
|
||||||
|
Param,
|
||||||
|
Body,
|
||||||
|
Headers,
|
||||||
|
HttpCode,
|
||||||
|
HttpStatus,
|
||||||
|
} from '@nestjs/common';
|
||||||
|
import { ConversationService } from './conversation.service';
|
||||||
|
|
||||||
|
class CreateConversationDto {
|
||||||
|
title?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
class SendMessageDto {
|
||||||
|
content: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Controller('conversations')
|
||||||
|
export class ConversationController {
|
||||||
|
constructor(private conversationService: ConversationService) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a new conversation
|
||||||
|
*/
|
||||||
|
@Post()
|
||||||
|
@HttpCode(HttpStatus.CREATED)
|
||||||
|
async createConversation(
|
||||||
|
@Headers('x-user-id') userId: string,
|
||||||
|
@Body() dto: CreateConversationDto,
|
||||||
|
) {
|
||||||
|
const conversation = await this.conversationService.createConversation({
|
||||||
|
userId,
|
||||||
|
title: dto.title,
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: conversation,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get user's conversations
|
||||||
|
*/
|
||||||
|
@Get()
|
||||||
|
async getConversations(@Headers('x-user-id') userId: string) {
|
||||||
|
const conversations = await this.conversationService.getUserConversations(userId);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: conversations,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get a specific conversation
|
||||||
|
*/
|
||||||
|
@Get(':id')
|
||||||
|
async getConversation(
|
||||||
|
@Headers('x-user-id') userId: string,
|
||||||
|
@Param('id') conversationId: string,
|
||||||
|
) {
|
||||||
|
const conversation = await this.conversationService.getConversation(
|
||||||
|
conversationId,
|
||||||
|
userId,
|
||||||
|
);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: conversation,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get conversation messages
|
||||||
|
*/
|
||||||
|
@Get(':id/messages')
|
||||||
|
async getMessages(
|
||||||
|
@Headers('x-user-id') userId: string,
|
||||||
|
@Param('id') conversationId: string,
|
||||||
|
) {
|
||||||
|
const messages = await this.conversationService.getMessages(
|
||||||
|
conversationId,
|
||||||
|
userId,
|
||||||
|
);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: messages,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Send a message (non-streaming HTTP endpoint)
|
||||||
|
* For streaming, use WebSocket
|
||||||
|
*/
|
||||||
|
@Post(':id/messages')
|
||||||
|
async sendMessage(
|
||||||
|
@Headers('x-user-id') userId: string,
|
||||||
|
@Param('id') conversationId: string,
|
||||||
|
@Body() dto: SendMessageDto,
|
||||||
|
) {
|
||||||
|
let fullResponse = '';
|
||||||
|
|
||||||
|
for await (const chunk of this.conversationService.sendMessage({
|
||||||
|
conversationId,
|
||||||
|
userId,
|
||||||
|
content: dto.content,
|
||||||
|
})) {
|
||||||
|
if (chunk.type === 'text' && chunk.content) {
|
||||||
|
fullResponse += chunk.content;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
response: fullResponse,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* End a conversation
|
||||||
|
*/
|
||||||
|
@Post(':id/end')
|
||||||
|
@HttpCode(HttpStatus.OK)
|
||||||
|
async endConversation(
|
||||||
|
@Headers('x-user-id') userId: string,
|
||||||
|
@Param('id') conversationId: string,
|
||||||
|
) {
|
||||||
|
await this.conversationService.endConversation(conversationId, userId);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Conversation ended',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,149 @@
|
||||||
|
import {
|
||||||
|
WebSocketGateway,
|
||||||
|
WebSocketServer,
|
||||||
|
SubscribeMessage,
|
||||||
|
OnGatewayConnection,
|
||||||
|
OnGatewayDisconnect,
|
||||||
|
ConnectedSocket,
|
||||||
|
MessageBody,
|
||||||
|
} from '@nestjs/websockets';
|
||||||
|
import { Server, Socket } from 'socket.io';
|
||||||
|
import { ConversationService } from './conversation.service';
|
||||||
|
|
||||||
|
interface SendMessagePayload {
|
||||||
|
conversationId: string;
|
||||||
|
content: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
@WebSocketGateway({
|
||||||
|
cors: {
|
||||||
|
origin: process.env.CORS_ORIGINS?.split(',') || ['http://localhost:5173'],
|
||||||
|
credentials: true,
|
||||||
|
},
|
||||||
|
namespace: '/ws/conversation',
|
||||||
|
})
|
||||||
|
export class ConversationGateway
|
||||||
|
implements OnGatewayConnection, OnGatewayDisconnect
|
||||||
|
{
|
||||||
|
@WebSocketServer()
|
||||||
|
server: Server;
|
||||||
|
|
||||||
|
// Map socket ID to user ID
|
||||||
|
private connections = new Map<string, string>();
|
||||||
|
|
||||||
|
constructor(private conversationService: ConversationService) {}
|
||||||
|
|
||||||
|
async handleConnection(client: Socket) {
|
||||||
|
// Extract user ID from query or headers
|
||||||
|
const userId =
|
||||||
|
(client.handshake.query.userId as string) ||
|
||||||
|
(client.handshake.headers['x-user-id'] as string);
|
||||||
|
|
||||||
|
if (!userId) {
|
||||||
|
console.log(`Client ${client.id} connected without user ID`);
|
||||||
|
client.emit('error', { message: 'User ID is required' });
|
||||||
|
client.disconnect();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.connections.set(client.id, userId);
|
||||||
|
console.log(`Client ${client.id} connected as user ${userId}`);
|
||||||
|
|
||||||
|
client.emit('connected', { userId, socketId: client.id });
|
||||||
|
}
|
||||||
|
|
||||||
|
handleDisconnect(client: Socket) {
|
||||||
|
const userId = this.connections.get(client.id);
|
||||||
|
this.connections.delete(client.id);
|
||||||
|
console.log(`Client ${client.id} (user ${userId}) disconnected`);
|
||||||
|
}
|
||||||
|
|
||||||
|
@SubscribeMessage('message')
|
||||||
|
async handleMessage(
|
||||||
|
@ConnectedSocket() client: Socket,
|
||||||
|
@MessageBody() payload: SendMessagePayload,
|
||||||
|
) {
|
||||||
|
const userId = this.connections.get(client.id);
|
||||||
|
|
||||||
|
if (!userId) {
|
||||||
|
client.emit('error', { message: 'Not authenticated' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { conversationId, content } = payload;
|
||||||
|
|
||||||
|
if (!conversationId || !content) {
|
||||||
|
client.emit('error', { message: 'conversationId and content are required' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Generate unique message ID for this response
|
||||||
|
const messageId = `msg_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
|
||||||
|
|
||||||
|
// Emit stream start
|
||||||
|
client.emit('stream_start', { messageId, conversationId });
|
||||||
|
|
||||||
|
let chunkIndex = 0;
|
||||||
|
|
||||||
|
// Stream the response
|
||||||
|
for await (const chunk of this.conversationService.sendMessage({
|
||||||
|
conversationId,
|
||||||
|
userId,
|
||||||
|
content,
|
||||||
|
})) {
|
||||||
|
if (chunk.type === 'text' && chunk.content) {
|
||||||
|
client.emit('stream_chunk', {
|
||||||
|
messageId,
|
||||||
|
conversationId,
|
||||||
|
content: chunk.content,
|
||||||
|
index: chunkIndex++,
|
||||||
|
});
|
||||||
|
} else if (chunk.type === 'tool_use') {
|
||||||
|
client.emit('tool_call', {
|
||||||
|
messageId,
|
||||||
|
conversationId,
|
||||||
|
tool: chunk.toolName,
|
||||||
|
input: chunk.toolInput,
|
||||||
|
});
|
||||||
|
} else if (chunk.type === 'tool_result') {
|
||||||
|
client.emit('tool_result', {
|
||||||
|
messageId,
|
||||||
|
conversationId,
|
||||||
|
tool: chunk.toolName,
|
||||||
|
result: chunk.toolResult,
|
||||||
|
});
|
||||||
|
} else if (chunk.type === 'end') {
|
||||||
|
client.emit('stream_end', {
|
||||||
|
messageId,
|
||||||
|
conversationId,
|
||||||
|
isComplete: true,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error processing message:', error);
|
||||||
|
client.emit('error', {
|
||||||
|
message: error instanceof Error ? error.message : 'Failed to process message',
|
||||||
|
conversationId,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@SubscribeMessage('typing_start')
|
||||||
|
handleTypingStart(
|
||||||
|
@ConnectedSocket() client: Socket,
|
||||||
|
@MessageBody() payload: { conversationId: string },
|
||||||
|
) {
|
||||||
|
// Could be used to show typing indicator to admin monitoring
|
||||||
|
console.log(`User typing in conversation ${payload.conversationId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
@SubscribeMessage('typing_end')
|
||||||
|
handleTypingEnd(
|
||||||
|
@ConnectedSocket() client: Socket,
|
||||||
|
@MessageBody() payload: { conversationId: string },
|
||||||
|
) {
|
||||||
|
console.log(`User stopped typing in conversation ${payload.conversationId}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,15 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { ConversationEntity } from '../domain/entities/conversation.entity';
|
||||||
|
import { MessageEntity } from '../domain/entities/message.entity';
|
||||||
|
import { ConversationService } from './conversation.service';
|
||||||
|
import { ConversationController } from './conversation.controller';
|
||||||
|
import { ConversationGateway } from './conversation.gateway';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [TypeOrmModule.forFeature([ConversationEntity, MessageEntity])],
|
||||||
|
controllers: [ConversationController],
|
||||||
|
providers: [ConversationService, ConversationGateway],
|
||||||
|
exports: [ConversationService],
|
||||||
|
})
|
||||||
|
export class ConversationModule {}
|
||||||
|
|
@ -0,0 +1,198 @@
|
||||||
|
import { Injectable, NotFoundException } from '@nestjs/common';
|
||||||
|
import { InjectRepository } from '@nestjs/typeorm';
|
||||||
|
import { Repository } from 'typeorm';
|
||||||
|
import {
|
||||||
|
ConversationEntity,
|
||||||
|
ConversationStatus,
|
||||||
|
} from '../domain/entities/conversation.entity';
|
||||||
|
import {
|
||||||
|
MessageEntity,
|
||||||
|
MessageRole,
|
||||||
|
MessageType,
|
||||||
|
} from '../domain/entities/message.entity';
|
||||||
|
import {
|
||||||
|
ClaudeAgentService,
|
||||||
|
ConversationContext,
|
||||||
|
StreamChunk,
|
||||||
|
} from '../infrastructure/claude/claude-agent.service';
|
||||||
|
|
||||||
|
export interface CreateConversationDto {
|
||||||
|
userId: string;
|
||||||
|
title?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SendMessageDto {
|
||||||
|
conversationId: string;
|
||||||
|
userId: string;
|
||||||
|
content: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Injectable()
|
||||||
|
export class ConversationService {
|
||||||
|
constructor(
|
||||||
|
@InjectRepository(ConversationEntity)
|
||||||
|
private conversationRepo: Repository<ConversationEntity>,
|
||||||
|
@InjectRepository(MessageEntity)
|
||||||
|
private messageRepo: Repository<MessageEntity>,
|
||||||
|
private claudeAgentService: ClaudeAgentService,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a new conversation
|
||||||
|
*/
|
||||||
|
async createConversation(dto: CreateConversationDto): Promise<ConversationEntity> {
|
||||||
|
const conversation = this.conversationRepo.create({
|
||||||
|
userId: dto.userId,
|
||||||
|
title: dto.title || '新对话',
|
||||||
|
status: ConversationStatus.ACTIVE,
|
||||||
|
});
|
||||||
|
|
||||||
|
return this.conversationRepo.save(conversation);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get conversation by ID
|
||||||
|
*/
|
||||||
|
async getConversation(
|
||||||
|
conversationId: string,
|
||||||
|
userId: string,
|
||||||
|
): Promise<ConversationEntity> {
|
||||||
|
const conversation = await this.conversationRepo.findOne({
|
||||||
|
where: { id: conversationId, userId },
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!conversation) {
|
||||||
|
throw new NotFoundException('Conversation not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
return conversation;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get user's conversations
|
||||||
|
*/
|
||||||
|
async getUserConversations(userId: string): Promise<ConversationEntity[]> {
|
||||||
|
return this.conversationRepo.find({
|
||||||
|
where: { userId },
|
||||||
|
order: { updatedAt: 'DESC' },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get conversation messages
|
||||||
|
*/
|
||||||
|
async getMessages(
|
||||||
|
conversationId: string,
|
||||||
|
userId: string,
|
||||||
|
): Promise<MessageEntity[]> {
|
||||||
|
// Verify user owns the conversation
|
||||||
|
await this.getConversation(conversationId, userId);
|
||||||
|
|
||||||
|
return this.messageRepo.find({
|
||||||
|
where: { conversationId },
|
||||||
|
order: { createdAt: 'ASC' },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Send a message and get streaming response
|
||||||
|
*/
|
||||||
|
async *sendMessage(dto: SendMessageDto): AsyncGenerator<StreamChunk> {
|
||||||
|
// Verify conversation exists and belongs to user
|
||||||
|
const conversation = await this.getConversation(dto.conversationId, dto.userId);
|
||||||
|
|
||||||
|
if (conversation.status !== ConversationStatus.ACTIVE) {
|
||||||
|
throw new Error('Conversation is not active');
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save user message
|
||||||
|
const userMessage = this.messageRepo.create({
|
||||||
|
conversationId: dto.conversationId,
|
||||||
|
role: MessageRole.USER,
|
||||||
|
type: MessageType.TEXT,
|
||||||
|
content: dto.content,
|
||||||
|
});
|
||||||
|
await this.messageRepo.save(userMessage);
|
||||||
|
|
||||||
|
// Get previous messages for context
|
||||||
|
const previousMessages = await this.messageRepo.find({
|
||||||
|
where: { conversationId: dto.conversationId },
|
||||||
|
order: { createdAt: 'ASC' },
|
||||||
|
take: 20, // Last 20 messages for context
|
||||||
|
});
|
||||||
|
|
||||||
|
// Build context
|
||||||
|
const context: ConversationContext = {
|
||||||
|
userId: dto.userId,
|
||||||
|
conversationId: dto.conversationId,
|
||||||
|
previousMessages: previousMessages.map((m) => ({
|
||||||
|
role: m.role as 'user' | 'assistant',
|
||||||
|
content: m.content,
|
||||||
|
})),
|
||||||
|
};
|
||||||
|
|
||||||
|
// Collect full response for saving
|
||||||
|
let fullResponse = '';
|
||||||
|
const toolCalls: Array<{ name: string; input: unknown; result: unknown }> = [];
|
||||||
|
|
||||||
|
// Stream response from Claude
|
||||||
|
for await (const chunk of this.claudeAgentService.sendMessage(
|
||||||
|
dto.content,
|
||||||
|
context,
|
||||||
|
)) {
|
||||||
|
if (chunk.type === 'text' && chunk.content) {
|
||||||
|
fullResponse += chunk.content;
|
||||||
|
} else if (chunk.type === 'tool_use') {
|
||||||
|
toolCalls.push({
|
||||||
|
name: chunk.toolName!,
|
||||||
|
input: chunk.toolInput!,
|
||||||
|
result: null,
|
||||||
|
});
|
||||||
|
} else if (chunk.type === 'tool_result') {
|
||||||
|
const lastToolCall = toolCalls[toolCalls.length - 1];
|
||||||
|
if (lastToolCall) {
|
||||||
|
lastToolCall.result = chunk.toolResult;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
yield chunk;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save assistant response
|
||||||
|
const assistantMessage = this.messageRepo.create({
|
||||||
|
conversationId: dto.conversationId,
|
||||||
|
role: MessageRole.ASSISTANT,
|
||||||
|
type: MessageType.TEXT,
|
||||||
|
content: fullResponse,
|
||||||
|
metadata: toolCalls.length > 0 ? { toolCalls } : undefined,
|
||||||
|
});
|
||||||
|
await this.messageRepo.save(assistantMessage);
|
||||||
|
|
||||||
|
// Update conversation title if first message
|
||||||
|
if (conversation.messageCount === 0) {
|
||||||
|
const title = await this.generateTitle(dto.content);
|
||||||
|
await this.conversationRepo.update(conversation.id, { title });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* End a conversation
|
||||||
|
*/
|
||||||
|
async endConversation(conversationId: string, userId: string): Promise<void> {
|
||||||
|
const conversation = await this.getConversation(conversationId, userId);
|
||||||
|
|
||||||
|
await this.conversationRepo.update(conversation.id, {
|
||||||
|
status: ConversationStatus.ENDED,
|
||||||
|
endedAt: new Date(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Generate a title from the first message
|
||||||
|
*/
|
||||||
|
private async generateTitle(firstMessage: string): Promise<string> {
|
||||||
|
// Simple title generation - take first 50 chars
|
||||||
|
const title = firstMessage.substring(0, 50);
|
||||||
|
return title.length < firstMessage.length ? `${title}...` : title;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,55 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
PrimaryGeneratedColumn,
|
||||||
|
Column,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
OneToMany,
|
||||||
|
} from 'typeorm';
|
||||||
|
import { MessageEntity } from './message.entity';
|
||||||
|
|
||||||
|
export enum ConversationStatus {
|
||||||
|
ACTIVE = 'ACTIVE',
|
||||||
|
ENDED = 'ENDED',
|
||||||
|
ARCHIVED = 'ARCHIVED',
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity('conversations')
|
||||||
|
export class ConversationEntity {
|
||||||
|
@PrimaryGeneratedColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'user_id', type: 'uuid' })
|
||||||
|
userId: string;
|
||||||
|
|
||||||
|
@Column({
|
||||||
|
type: 'enum',
|
||||||
|
enum: ConversationStatus,
|
||||||
|
default: ConversationStatus.ACTIVE,
|
||||||
|
})
|
||||||
|
status: ConversationStatus;
|
||||||
|
|
||||||
|
@Column({ nullable: true })
|
||||||
|
title: string;
|
||||||
|
|
||||||
|
@Column({ type: 'text', nullable: true })
|
||||||
|
summary: string;
|
||||||
|
|
||||||
|
@Column({ nullable: true })
|
||||||
|
category: string;
|
||||||
|
|
||||||
|
@Column({ name: 'message_count', default: 0 })
|
||||||
|
messageCount: number;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'ended_at', nullable: true })
|
||||||
|
endedAt: Date;
|
||||||
|
|
||||||
|
@OneToMany(() => MessageEntity, (message) => message.conversation)
|
||||||
|
messages: MessageEntity[];
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,59 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
PrimaryGeneratedColumn,
|
||||||
|
Column,
|
||||||
|
CreateDateColumn,
|
||||||
|
ManyToOne,
|
||||||
|
JoinColumn,
|
||||||
|
} from 'typeorm';
|
||||||
|
import { ConversationEntity } from './conversation.entity';
|
||||||
|
|
||||||
|
export enum MessageRole {
|
||||||
|
USER = 'user',
|
||||||
|
ASSISTANT = 'assistant',
|
||||||
|
SYSTEM = 'system',
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum MessageType {
|
||||||
|
TEXT = 'TEXT',
|
||||||
|
TOOL_CALL = 'TOOL_CALL',
|
||||||
|
TOOL_RESULT = 'TOOL_RESULT',
|
||||||
|
PAYMENT_REQUEST = 'PAYMENT_REQUEST',
|
||||||
|
ASSESSMENT_START = 'ASSESSMENT_START',
|
||||||
|
ASSESSMENT_RESULT = 'ASSESSMENT_RESULT',
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity('messages')
|
||||||
|
export class MessageEntity {
|
||||||
|
@PrimaryGeneratedColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'conversation_id', type: 'uuid' })
|
||||||
|
conversationId: string;
|
||||||
|
|
||||||
|
@Column({
|
||||||
|
type: 'enum',
|
||||||
|
enum: MessageRole,
|
||||||
|
})
|
||||||
|
role: MessageRole;
|
||||||
|
|
||||||
|
@Column({
|
||||||
|
type: 'enum',
|
||||||
|
enum: MessageType,
|
||||||
|
default: MessageType.TEXT,
|
||||||
|
})
|
||||||
|
type: MessageType;
|
||||||
|
|
||||||
|
@Column({ type: 'text' })
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column({ type: 'jsonb', nullable: true })
|
||||||
|
metadata: Record<string, unknown>;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@ManyToOne(() => ConversationEntity, (conversation) => conversation.messages)
|
||||||
|
@JoinColumn({ name: 'conversation_id' })
|
||||||
|
conversation: ConversationEntity;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,231 @@
|
||||||
|
import { Injectable, OnModuleInit } from '@nestjs/common';
|
||||||
|
import { ConfigService } from '@nestjs/config';
|
||||||
|
import Anthropic from '@anthropic-ai/sdk';
|
||||||
|
import { ImmigrationToolsService } from './tools/immigration-tools.service';
|
||||||
|
import { buildSystemPrompt, SystemPromptConfig } from './prompts/system-prompt';
|
||||||
|
|
||||||
|
export interface ConversationContext {
|
||||||
|
userId: string;
|
||||||
|
conversationId: string;
|
||||||
|
userMemory?: string[];
|
||||||
|
previousMessages?: Array<{
|
||||||
|
role: 'user' | 'assistant';
|
||||||
|
content: string;
|
||||||
|
}>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface StreamChunk {
|
||||||
|
type: 'text' | 'tool_use' | 'tool_result' | 'end';
|
||||||
|
content?: string;
|
||||||
|
toolName?: string;
|
||||||
|
toolInput?: Record<string, unknown>;
|
||||||
|
toolResult?: unknown;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Injectable()
|
||||||
|
export class ClaudeAgentService implements OnModuleInit {
|
||||||
|
private client: Anthropic;
|
||||||
|
private systemPromptConfig: SystemPromptConfig;
|
||||||
|
|
||||||
|
constructor(
|
||||||
|
private configService: ConfigService,
|
||||||
|
private immigrationToolsService: ImmigrationToolsService,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
onModuleInit() {
|
||||||
|
this.client = new Anthropic({
|
||||||
|
apiKey: this.configService.get<string>('ANTHROPIC_API_KEY'),
|
||||||
|
});
|
||||||
|
|
||||||
|
// Initialize with default config
|
||||||
|
this.systemPromptConfig = {
|
||||||
|
identity: '专业、友善、耐心的香港移民顾问',
|
||||||
|
conversationStyle: '专业但不生硬，用简洁明了的语言解答',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Update system prompt configuration (for evolution)
|
||||||
|
*/
|
||||||
|
updateSystemPromptConfig(config: Partial<SystemPromptConfig>) {
|
||||||
|
this.systemPromptConfig = {
|
||||||
|
...this.systemPromptConfig,
|
||||||
|
...config,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Send a message and get streaming response
|
||||||
|
*/
|
||||||
|
async *sendMessage(
|
||||||
|
message: string,
|
||||||
|
context: ConversationContext,
|
||||||
|
): AsyncGenerator<StreamChunk> {
|
||||||
|
const tools = this.immigrationToolsService.getTools();
|
||||||
|
const systemPrompt = buildSystemPrompt(this.systemPromptConfig);
|
||||||
|
|
||||||
|
// Build messages array
|
||||||
|
const messages: Anthropic.MessageParam[] = [];
|
||||||
|
|
||||||
|
// Add previous messages if any
|
||||||
|
if (context.previousMessages) {
|
||||||
|
for (const msg of context.previousMessages) {
|
||||||
|
messages.push({
|
||||||
|
role: msg.role,
|
||||||
|
content: msg.content,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add current message
|
||||||
|
messages.push({
|
||||||
|
role: 'user',
|
||||||
|
content: message,
|
||||||
|
});
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Create streaming message
|
||||||
|
const stream = await this.client.messages.stream({
|
||||||
|
model: 'claude-sonnet-4-20250514',
|
||||||
|
max_tokens: 4096,
|
||||||
|
system: systemPrompt,
|
||||||
|
messages,
|
||||||
|
tools: tools as Anthropic.Tool[],
|
||||||
|
});
|
||||||
|
|
||||||
|
let currentToolUse: {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
input: Record<string, unknown>;
|
||||||
|
} | null = null;
|
||||||
|
|
||||||
|
for await (const event of stream) {
|
||||||
|
if (event.type === 'content_block_start') {
|
||||||
|
if (event.content_block.type === 'tool_use') {
|
||||||
|
currentToolUse = {
|
||||||
|
id: event.content_block.id,
|
||||||
|
name: event.content_block.name,
|
||||||
|
input: {},
|
||||||
|
};
|
||||||
|
}
|
||||||
|
} else if (event.type === 'content_block_delta') {
|
||||||
|
if (event.delta.type === 'text_delta') {
|
||||||
|
yield {
|
||||||
|
type: 'text',
|
||||||
|
content: event.delta.text,
|
||||||
|
};
|
||||||
|
} else if (event.delta.type === 'input_json_delta' && currentToolUse) {
|
||||||
|
// Accumulate tool input
|
||||||
|
try {
|
||||||
|
const partialInput = JSON.parse(event.delta.partial_json || '{}');
|
||||||
|
currentToolUse.input = {
|
||||||
|
...currentToolUse.input,
|
||||||
|
...partialInput,
|
||||||
|
};
|
||||||
|
} catch {
|
||||||
|
// Ignore parse errors for partial JSON
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if (event.type === 'content_block_stop') {
|
||||||
|
if (currentToolUse) {
|
||||||
|
yield {
|
||||||
|
type: 'tool_use',
|
||||||
|
toolName: currentToolUse.name,
|
||||||
|
toolInput: currentToolUse.input,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Execute the tool
|
||||||
|
const toolResult = await this.immigrationToolsService.executeTool(
|
||||||
|
currentToolUse.name,
|
||||||
|
currentToolUse.input,
|
||||||
|
context,
|
||||||
|
);
|
||||||
|
|
||||||
|
yield {
|
||||||
|
type: 'tool_result',
|
||||||
|
toolName: currentToolUse.name,
|
||||||
|
toolResult,
|
||||||
|
};
|
||||||
|
|
||||||
|
currentToolUse = null;
|
||||||
|
}
|
||||||
|
} else if (event.type === 'message_stop') {
|
||||||
|
yield { type: 'end' };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Claude API error:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Non-streaming message for simple queries
|
||||||
|
*/
|
||||||
|
async sendMessageSync(
|
||||||
|
message: string,
|
||||||
|
context: ConversationContext,
|
||||||
|
): Promise<string> {
|
||||||
|
const tools = this.immigrationToolsService.getTools();
|
||||||
|
const systemPrompt = buildSystemPrompt(this.systemPromptConfig);
|
||||||
|
|
||||||
|
const messages: Anthropic.MessageParam[] = [];
|
||||||
|
|
||||||
|
if (context.previousMessages) {
|
||||||
|
for (const msg of context.previousMessages) {
|
||||||
|
messages.push({
|
||||||
|
role: msg.role,
|
||||||
|
content: msg.content,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
messages.push({
|
||||||
|
role: 'user',
|
||||||
|
content: message,
|
||||||
|
});
|
||||||
|
|
||||||
|
const response = await this.client.messages.create({
|
||||||
|
model: 'claude-sonnet-4-20250514',
|
||||||
|
max_tokens: 4096,
|
||||||
|
system: systemPrompt,
|
||||||
|
messages,
|
||||||
|
tools: tools as Anthropic.Tool[],
|
||||||
|
});
|
||||||
|
|
||||||
|
// Extract text response
|
||||||
|
let result = '';
|
||||||
|
for (const block of response.content) {
|
||||||
|
if (block.type === 'text') {
|
||||||
|
result += block.text;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Analyze content (for evolution service)
|
||||||
|
*/
|
||||||
|
async analyze(prompt: string): Promise<string> {
|
||||||
|
const response = await this.client.messages.create({
|
||||||
|
model: 'claude-sonnet-4-20250514',
|
||||||
|
max_tokens: 8192,
|
||||||
|
messages: [
|
||||||
|
{
|
||||||
|
role: 'user',
|
||||||
|
content: prompt,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
});
|
||||||
|
|
||||||
|
let result = '';
|
||||||
|
for (const block of response.content) {
|
||||||
|
if (block.type === 'text') {
|
||||||
|
result += block.text;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,12 @@
|
||||||
|
import { Module, Global } from '@nestjs/common';
|
||||||
|
import { ConfigModule } from '@nestjs/config';
|
||||||
|
import { ClaudeAgentService } from './claude-agent.service';
|
||||||
|
import { ImmigrationToolsService } from './tools/immigration-tools.service';
|
||||||
|
|
||||||
|
@Global()
|
||||||
|
@Module({
|
||||||
|
imports: [ConfigModule],
|
||||||
|
providers: [ClaudeAgentService, ImmigrationToolsService],
|
||||||
|
exports: [ClaudeAgentService, ImmigrationToolsService],
|
||||||
|
})
|
||||||
|
export class ClaudeModule {}
|
||||||
|
|
@ -0,0 +1,94 @@
|
||||||
|
/**
|
||||||
|
* System prompt builder for the immigration consultant agent
|
||||||
|
*/
|
||||||
|
|
||||||
|
export interface SystemPromptConfig {
|
||||||
|
identity?: string;
|
||||||
|
conversationStyle?: string;
|
||||||
|
accumulatedExperience?: string;
|
||||||
|
adminInstructions?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const buildSystemPrompt = (config: SystemPromptConfig): string => `
|
||||||
|
你是 iConsulting 的香港移民咨询顾问，专门为用户提供香港各类移民政策的咨询服务。
|
||||||
|
|
||||||
|
## 身份定位
|
||||||
|
${config.identity || '专业、友善、耐心的移民顾问'}
|
||||||
|
|
||||||
|
## 服务范围
|
||||||
|
你只回答与香港移民相关的问题，包括以下6大类别：
|
||||||
|
|
||||||
|
### 1. 优才计划 (QMAS - Quality Migrant Admission Scheme)
|
||||||
|
- **目的**: 吸纳优秀人才来港定居
|
||||||
|
- **审核标准**: 年龄、教育经历、工作经验、语言能力、年收入证明、良好品格、是否持有公司等
|
||||||
|
- **人才类型**: 行业翘楚、精英人士
|
||||||
|
- **名额**: 暂无名额限制
|
||||||
|
- **申请要点**: 年龄、教育经历、工作经验、年收入证明、良好品格等方面的要求（共12项），满足其中6项为基本门槛，择优批准
|
||||||
|
|
||||||
|
### 2. 专才计划 (GEP - General Employment Policy)
|
||||||
|
- **目的**: 吸引专业人才来港就业
|
||||||
|
- **审核标准**: 雇佣企业的行业背景、行业内的地位、政策扶持力度、所属行业经济地位
|
||||||
|
- **人才类型**: 专业人士（≥200万年薪，属于人才清单）
|
||||||
|
- **名额**: 无配额限制（不限行业）
|
||||||
|
- **申请要点**: 必须有雇主担保、要保持工作连续性、工作和居住在非粤港澳、申请前至少有一年半社保记录
|
||||||
|
|
||||||
|
### 3. 留学IANG (Immigration Arrangements for Non-local Graduates)
|
||||||
|
- **目的**: 吸引优秀学生人才来港续读
|
||||||
|
- **审核标准**: 有无完成学业、是否有雇主担保
|
||||||
|
- **人才类型**: 成绩优异、拥有国际化视野的学生
|
||||||
|
- **名额**: 扩展到本港大学大湾区校区毕业生，为期2年、1年后续签
|
||||||
|
- **申请要点**: 必须有雇主担保、要保持工作连续性
|
||||||
|
|
||||||
|
### 4. 高才通计划 (TTPS - Top Talent Pass Scheme)
|
||||||
|
- **目的**: 吸引高端人才来港
|
||||||
|
- **审核标准**:
|
||||||
|
- A类: 年薪≥250万港币
|
||||||
|
- B类: 毕业于世界前100名大学，5年内有3年以上工作经验
|
||||||
|
- C类: 毕业于世界前100名大学，5年内有少于3年工作经验（年度限额10,000名）
|
||||||
|
- **人才类型**: 顶尖人才
|
||||||
|
- **名额**: 无限期计划
|
||||||
|
|
||||||
|
### 5. 投资移民 (新资本投资者入境计划 CIES)
|
||||||
|
- **目的**: 吸引投资者
|
||||||
|
- **审核标准**: 投资不少于3,000万港币（或等值外币）净资产于其指定对实益拥有者的许可投资
|
||||||
|
- **人才类型**: 投资者
|
||||||
|
- **名额**: 无名额限制
|
||||||
|
|
||||||
|
### 6. 科技人才入境计划 (TechTAS)
|
||||||
|
- **目的**: 吸引科技人才来港就业
|
||||||
|
- **审核标准**: 从事先进通讯技术、人工智能、生物科技、网络安全、数据分析、金融科技等领域研究工作，薪酬不低于香港市场水平
|
||||||
|
- **人才类型**: 科技人才
|
||||||
|
- **名额**: 需用公司高新创新科技营配出有效配额
|
||||||
|
|
||||||
|
## 行为准则
|
||||||
|
1. 如果用户询问与移民无关的问题，礼貌地说明你专注于香港移民咨询，并引导回移民话题
|
||||||
|
2. 对于复杂的评估需求，主动建议使用付费评估服务
|
||||||
|
3. 提供信息时注明信息来源和更新时间（如适用）
|
||||||
|
4. 不做任何法律承诺或申请成功率的保证
|
||||||
|
5. 对于敏感问题（如政治、法律纠纷），建议用户咨询专业律师
|
||||||
|
|
||||||
|
## 付费服务
|
||||||
|
当用户需要个性化移民评估时，介绍付费评估服务：
|
||||||
|
- 先了解用户的基本情况（年龄、学历、工作经验、收入等）
|
||||||
|
- 说明评估服务的内容和价值
|
||||||
|
- 确认用户意愿后，使用工具生成支付码
|
||||||
|
|
||||||
|
## 对话风格
|
||||||
|
${config.conversationStyle || '专业但不生硬，用简洁明了的语言解答'}
|
||||||
|
|
||||||
|
## 工具使用说明
|
||||||
|
你有以下工具可以使用：
|
||||||
|
1. **search_knowledge**: 搜索知识库获取最新的移民政策信息
|
||||||
|
2. **check_off_topic**: 检查用户问题是否与移民相关
|
||||||
|
3. **collect_assessment_info**: 收集用户信息用于评估
|
||||||
|
4. **generate_payment**: 为付费服务生成支付二维码
|
||||||
|
5. **save_user_memory**: 保存用户的重要信息以便后续对话记忆
|
||||||
|
|
||||||
|
## 已积累的经验
|
||||||
|
${config.accumulatedExperience || '暂无'}
|
||||||
|
|
||||||
|
## 管理员特别指示
|
||||||
|
${config.adminInstructions || '暂无'}
|
||||||
|
|
||||||
|
请始终保持专业、热情的态度，帮助用户了解香港移民政策，并在适当时机引导用户使用付费评估服务。
|
||||||
|
`;
|
||||||
|
|
@ -0,0 +1,331 @@
|
||||||
|
import { Injectable } from '@nestjs/common';
|
||||||
|
import { ConversationContext } from '../claude-agent.service';
|
||||||
|
|
||||||
|
export interface Tool {
|
||||||
|
name: string;
|
||||||
|
description: string;
|
||||||
|
input_schema: {
|
||||||
|
type: string;
|
||||||
|
properties: Record<string, unknown>;
|
||||||
|
required?: string[];
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
@Injectable()
|
||||||
|
export class ImmigrationToolsService {
|
||||||
|
/**
|
||||||
|
* Get all available tools for the agent
|
||||||
|
*/
|
||||||
|
getTools(): Tool[] {
|
||||||
|
return [
|
||||||
|
{
|
||||||
|
name: 'search_knowledge',
|
||||||
|
description: '搜索香港移民相关知识库，获取最新的政策信息和常见问题解答',
|
||||||
|
input_schema: {
|
||||||
|
type: 'object',
|
||||||
|
properties: {
|
||||||
|
query: {
|
||||||
|
type: 'string',
|
||||||
|
description: '搜索查询内容',
|
||||||
|
},
|
||||||
|
category: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['QMAS', 'GEP', 'IANG', 'TTPS', 'CIES', 'TECHTAS'],
|
||||||
|
description: '移民类别代码（可选）',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
required: ['query'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'check_off_topic',
|
||||||
|
description: '检查用户的问题是否与香港移民相关，用于判断是否需要拒绝回答',
|
||||||
|
input_schema: {
|
||||||
|
type: 'object',
|
||||||
|
properties: {
|
||||||
|
question: {
|
||||||
|
type: 'string',
|
||||||
|
description: '用户的问题',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
required: ['question'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'collect_assessment_info',
|
||||||
|
description: '收集用户的个人信息用于移民资格评估',
|
||||||
|
input_schema: {
|
||||||
|
type: 'object',
|
||||||
|
properties: {
|
||||||
|
category: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['QMAS', 'GEP', 'IANG', 'TTPS', 'CIES', 'TECHTAS'],
|
||||||
|
description: '用户感兴趣的移民类别',
|
||||||
|
},
|
||||||
|
age: {
|
||||||
|
type: 'number',
|
||||||
|
description: '用户年龄',
|
||||||
|
},
|
||||||
|
education: {
|
||||||
|
type: 'string',
|
||||||
|
description: '最高学历',
|
||||||
|
},
|
||||||
|
university: {
|
||||||
|
type: 'string',
|
||||||
|
description: '毕业院校',
|
||||||
|
},
|
||||||
|
yearsOfExperience: {
|
||||||
|
type: 'number',
|
||||||
|
description: '工作年限',
|
||||||
|
},
|
||||||
|
currentJobTitle: {
|
||||||
|
type: 'string',
|
||||||
|
description: '当前职位',
|
||||||
|
},
|
||||||
|
industry: {
|
||||||
|
type: 'string',
|
||||||
|
description: '所属行业',
|
||||||
|
},
|
||||||
|
annualIncome: {
|
||||||
|
type: 'number',
|
||||||
|
description: '年收入（人民币）',
|
||||||
|
},
|
||||||
|
hasHKEmployer: {
|
||||||
|
type: 'boolean',
|
||||||
|
description: '是否有香港雇主',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
required: ['category'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'generate_payment',
|
||||||
|
description: '为付费评估服务生成支付二维码',
|
||||||
|
input_schema: {
|
||||||
|
type: 'object',
|
||||||
|
properties: {
|
||||||
|
serviceType: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['ASSESSMENT'],
|
||||||
|
description: '服务类型',
|
||||||
|
},
|
||||||
|
category: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['QMAS', 'GEP', 'IANG', 'TTPS', 'CIES', 'TECHTAS'],
|
||||||
|
description: '移民类别',
|
||||||
|
},
|
||||||
|
paymentMethod: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['ALIPAY', 'WECHAT', 'CREDIT_CARD'],
|
||||||
|
description: '支付方式',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
required: ['serviceType', 'category', 'paymentMethod'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'save_user_memory',
|
||||||
|
description: '保存用户的重要信息到长期记忆，以便后续对话中记住用户情况',
|
||||||
|
input_schema: {
|
||||||
|
type: 'object',
|
||||||
|
properties: {
|
||||||
|
memoryType: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['FACT', 'PREFERENCE', 'INTENT'],
|
||||||
|
description: '记忆类型：FACT-用户陈述的事实，PREFERENCE-用户偏好，INTENT-用户意图',
|
||||||
|
},
|
||||||
|
content: {
|
||||||
|
type: 'string',
|
||||||
|
description: '要记住的内容',
|
||||||
|
},
|
||||||
|
category: {
|
||||||
|
type: 'string',
|
||||||
|
enum: ['QMAS', 'GEP', 'IANG', 'TTPS', 'CIES', 'TECHTAS'],
|
||||||
|
description: '相关的移民类别（可选）',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
required: ['memoryType', 'content'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Execute a tool and return the result
|
||||||
|
*/
|
||||||
|
async executeTool(
|
||||||
|
toolName: string,
|
||||||
|
input: Record<string, unknown>,
|
||||||
|
context: ConversationContext,
|
||||||
|
): Promise<unknown> {
|
||||||
|
switch (toolName) {
|
||||||
|
case 'search_knowledge':
|
||||||
|
return this.searchKnowledge(input);
|
||||||
|
|
||||||
|
case 'check_off_topic':
|
||||||
|
return this.checkOffTopic(input);
|
||||||
|
|
||||||
|
case 'collect_assessment_info':
|
||||||
|
return this.collectAssessmentInfo(input, context);
|
||||||
|
|
||||||
|
case 'generate_payment':
|
||||||
|
return this.generatePayment(input, context);
|
||||||
|
|
||||||
|
case 'save_user_memory':
|
||||||
|
return this.saveUserMemory(input, context);
|
||||||
|
|
||||||
|
default:
|
||||||
|
return { error: `Unknown tool: ${toolName}` };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Search knowledge base
|
||||||
|
*/
|
||||||
|
private async searchKnowledge(input: Record<string, unknown>): Promise<unknown> {
|
||||||
|
const { query, category } = input as { query: string; category?: string };
|
||||||
|
|
||||||
|
// TODO: Implement actual RAG search via Knowledge Service
|
||||||
|
// For now, return a placeholder response
|
||||||
|
console.log(`[Knowledge Search] Query: ${query}, Category: ${category || 'all'}`);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
results: [
|
||||||
|
{
|
||||||
|
content: '这里将返回从知识库检索到的相关信息',
|
||||||
|
source: '香港入境事务处官网',
|
||||||
|
relevance: 0.95,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
message: '知识库搜索功能即将上线，目前请基于内置知识回答',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Check if question is off-topic
|
||||||
|
*/
|
||||||
|
private async checkOffTopic(input: Record<string, unknown>): Promise<unknown> {
|
||||||
|
const { question } = input as { question: string };
|
||||||
|
|
||||||
|
// Simple keyword-based check
|
||||||
|
const immigrationKeywords = [
|
||||||
|
'移民', '签证', '香港', '优才', '专才', '高才通', '投资', '留学',
|
||||||
|
'IANG', 'QMAS', 'GEP', 'TTPS', '入境', '定居', '永居', '身份',
|
||||||
|
'申请', '条件', '资格', '评估', '审核', '批准', '拒签',
|
||||||
|
];
|
||||||
|
|
||||||
|
const isRelated = immigrationKeywords.some((keyword) =>
|
||||||
|
question.toLowerCase().includes(keyword.toLowerCase()),
|
||||||
|
);
|
||||||
|
|
||||||
|
return {
|
||||||
|
isOffTopic: !isRelated,
|
||||||
|
confidence: isRelated ? 0.9 : 0.7,
|
||||||
|
suggestion: isRelated
|
||||||
|
? null
|
||||||
|
: '这个问题似乎与香港移民无关。作为移民咨询顾问，我专注于香港各类移民政策的咨询。请问您有香港移民相关的问题吗？',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Collect assessment info
|
||||||
|
*/
|
||||||
|
private async collectAssessmentInfo(
|
||||||
|
input: Record<string, unknown>,
|
||||||
|
context: ConversationContext,
|
||||||
|
): Promise<unknown> {
|
||||||
|
const info = input as {
|
||||||
|
category: string;
|
||||||
|
age?: number;
|
||||||
|
education?: string;
|
||||||
|
university?: string;
|
||||||
|
yearsOfExperience?: number;
|
||||||
|
currentJobTitle?: string;
|
||||||
|
industry?: string;
|
||||||
|
annualIncome?: number;
|
||||||
|
hasHKEmployer?: boolean;
|
||||||
|
};
|
||||||
|
|
||||||
|
console.log(`[Assessment Info] User ${context.userId} - Category: ${info.category}`);
|
||||||
|
|
||||||
|
// Store the collected info for later use
|
||||||
|
// TODO: Save to database via User Service
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
collectedInfo: info,
|
||||||
|
message: '已记录您的信息。如需完整评估，请选择付费评估服务。',
|
||||||
|
nextStep: 'payment',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Generate payment QR code
|
||||||
|
*/
|
||||||
|
private async generatePayment(
|
||||||
|
input: Record<string, unknown>,
|
||||||
|
context: ConversationContext,
|
||||||
|
): Promise<unknown> {
|
||||||
|
const { serviceType, category, paymentMethod } = input as {
|
||||||
|
serviceType: string;
|
||||||
|
category: string;
|
||||||
|
paymentMethod: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
console.log(
|
||||||
|
`[Payment] User ${context.userId} - ${serviceType} for ${category} via ${paymentMethod}`,
|
||||||
|
);
|
||||||
|
|
||||||
|
// TODO: Call Payment Service to generate actual payment
|
||||||
|
// For now, return a placeholder
|
||||||
|
|
||||||
|
const priceMap: Record<string, number> = {
|
||||||
|
QMAS: 99,
|
||||||
|
GEP: 99,
|
||||||
|
IANG: 79,
|
||||||
|
TTPS: 99,
|
||||||
|
CIES: 199,
|
||||||
|
TECHTAS: 99,
|
||||||
|
};
|
||||||
|
|
||||||
|
const price = priceMap[category] || 99;
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
orderId: `ORD_${Date.now()}`,
|
||||||
|
amount: price,
|
||||||
|
currency: 'CNY',
|
||||||
|
paymentMethod,
|
||||||
|
qrCodeUrl: `https://placeholder-payment-qr.com/${paymentMethod.toLowerCase()}/${Date.now()}`,
|
||||||
|
expiresAt: new Date(Date.now() + 30 * 60 * 1000).toISOString(), // 30 minutes
|
||||||
|
message: `请扫描二维码支付 ¥${price} 完成${category}类别的移民资格评估服务`,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Save user memory
|
||||||
|
*/
|
||||||
|
private async saveUserMemory(
|
||||||
|
input: Record<string, unknown>,
|
||||||
|
context: ConversationContext,
|
||||||
|
): Promise<unknown> {
|
||||||
|
const { memoryType, content, category } = input as {
|
||||||
|
memoryType: string;
|
||||||
|
content: string;
|
||||||
|
category?: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
console.log(
|
||||||
|
`[Memory] User ${context.userId} - Type: ${memoryType}, Content: ${content}`,
|
||||||
|
);
|
||||||
|
|
||||||
|
// TODO: Save to Neo4j via Knowledge Service
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
memoryId: `MEM_${Date.now()}`,
|
||||||
|
message: '已记住您的信息',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,34 @@
|
||||||
|
import { NestFactory } from '@nestjs/core';
|
||||||
|
import { ValidationPipe } from '@nestjs/common';
|
||||||
|
import { ConfigService } from '@nestjs/config';
|
||||||
|
import { AppModule } from './app.module';
|
||||||
|
|
||||||
|
async function bootstrap() {
|
||||||
|
const app = await NestFactory.create(AppModule);
|
||||||
|
|
||||||
|
// Global validation pipe
|
||||||
|
app.useGlobalPipes(
|
||||||
|
new ValidationPipe({
|
||||||
|
whitelist: true,
|
||||||
|
forbidNonWhitelisted: true,
|
||||||
|
transform: true,
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
|
||||||
|
// Enable CORS
|
||||||
|
app.enableCors({
|
||||||
|
origin: process.env.CORS_ORIGINS?.split(',') || ['http://localhost:5173'],
|
||||||
|
credentials: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
// API prefix
|
||||||
|
app.setGlobalPrefix('api/v1');
|
||||||
|
|
||||||
|
const configService = app.get(ConfigService);
|
||||||
|
const port = configService.get<number>('CONVERSATION_SERVICE_PORT') || 3002;
|
||||||
|
|
||||||
|
await app.listen(port);
|
||||||
|
console.log(`Conversation Service is running on port ${port}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
bootstrap();
|
||||||
|
|
@ -0,0 +1,18 @@
|
||||||
|
{
|
||||||
|
"extends": "../../../tsconfig.base.json",
|
||||||
|
"compilerOptions": {
|
||||||
|
"module": "CommonJS",
|
||||||
|
"moduleResolution": "Node",
|
||||||
|
"outDir": "./dist",
|
||||||
|
"rootDir": "./src",
|
||||||
|
"baseUrl": "./",
|
||||||
|
"strictPropertyInitialization": false,
|
||||||
|
"noUnusedLocals": false,
|
||||||
|
"noUnusedParameters": false,
|
||||||
|
"paths": {
|
||||||
|
"@/*": ["src/*"]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"include": ["src/**/*"],
|
||||||
|
"exclude": ["node_modules", "dist", "test"]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,45 @@
|
||||||
|
# ===========================================
|
||||||
|
# iConsulting Evolution Service Dockerfile
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
FROM node:20-alpine AS builder
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
|
||||||
|
COPY pnpm-workspace.yaml package.json pnpm-lock.yaml ./
|
||||||
|
COPY packages/shared/package.json ./packages/shared/
|
||||||
|
COPY packages/services/evolution-service/package.json ./packages/services/evolution-service/
|
||||||
|
|
||||||
|
RUN pnpm install --frozen-lockfile
|
||||||
|
|
||||||
|
COPY packages/shared ./packages/shared
|
||||||
|
COPY packages/services/evolution-service ./packages/services/evolution-service
|
||||||
|
COPY tsconfig.base.json ./
|
||||||
|
|
||||||
|
RUN pnpm --filter @iconsulting/shared build
|
||||||
|
RUN pnpm --filter @iconsulting/evolution-service build
|
||||||
|
|
||||||
|
FROM node:20-alpine AS runner
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
RUN addgroup --system --gid 1001 nodejs && \
|
||||||
|
adduser --system --uid 1001 nestjs
|
||||||
|
|
||||||
|
COPY --from=builder /app/packages/services/evolution-service/dist ./dist
|
||||||
|
COPY --from=builder /app/packages/services/evolution-service/package.json ./
|
||||||
|
COPY --from=builder /app/node_modules ./node_modules
|
||||||
|
|
||||||
|
ENV NODE_ENV=production
|
||||||
|
ENV PORT=3005
|
||||||
|
|
||||||
|
USER nestjs
|
||||||
|
EXPOSE 3005
|
||||||
|
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||||
|
CMD wget --no-verbose --tries=1 --spider http://localhost:3005/health || exit 1
|
||||||
|
|
||||||
|
CMD ["node", "dist/main.js"]
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
{
|
||||||
|
"$schema": "https://json.schemastore.org/nest-cli",
|
||||||
|
"collection": "@nestjs/schematics",
|
||||||
|
"sourceRoot": "src",
|
||||||
|
"compilerOptions": {
|
||||||
|
"deleteOutDir": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,48 @@
|
||||||
|
{
|
||||||
|
"name": "@iconsulting/evolution-service",
|
||||||
|
"version": "0.1.0",
|
||||||
|
"description": "进化服务 - 系统自我学习与进化引擎",
|
||||||
|
"main": "dist/main.js",
|
||||||
|
"scripts": {
|
||||||
|
"build": "nest build",
|
||||||
|
"start": "nest start",
|
||||||
|
"start:dev": "nest start --watch",
|
||||||
|
"start:debug": "nest start --debug --watch",
|
||||||
|
"start:prod": "node dist/main",
|
||||||
|
"lint": "eslint \"{src,apps,libs,test}/**/*.ts\" --fix",
|
||||||
|
"test": "jest",
|
||||||
|
"test:watch": "jest --watch",
|
||||||
|
"test:cov": "jest --coverage"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@anthropic-ai/sdk": "^0.52.0",
|
||||||
|
"@nestjs/common": "^10.0.0",
|
||||||
|
"@nestjs/config": "^3.2.0",
|
||||||
|
"@nestjs/core": "^10.0.0",
|
||||||
|
"@nestjs/platform-express": "^10.0.0",
|
||||||
|
"@nestjs/typeorm": "^10.0.0",
|
||||||
|
"bcrypt": "^5.1.1",
|
||||||
|
"jsonwebtoken": "^9.0.2",
|
||||||
|
"pg": "^8.11.3",
|
||||||
|
"reflect-metadata": "^0.2.1",
|
||||||
|
"rxjs": "^7.8.1",
|
||||||
|
"typeorm": "^0.3.19",
|
||||||
|
"uuid": "^9.0.1"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@nestjs/cli": "^10.0.0",
|
||||||
|
"@nestjs/schematics": "^10.0.0",
|
||||||
|
"@nestjs/testing": "^10.0.0",
|
||||||
|
"@types/bcrypt": "^5.0.2",
|
||||||
|
"@types/express": "^4.17.21",
|
||||||
|
"@types/jest": "^29.5.11",
|
||||||
|
"@types/jsonwebtoken": "^9.0.5",
|
||||||
|
"@types/node": "^20.10.6",
|
||||||
|
"@types/uuid": "^9.0.7",
|
||||||
|
"jest": "^29.7.0",
|
||||||
|
"ts-jest": "^29.1.1",
|
||||||
|
"ts-loader": "^9.5.1",
|
||||||
|
"ts-node": "^10.9.2",
|
||||||
|
"typescript": "^5.3.3"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,298 @@
|
||||||
|
import {
|
||||||
|
Controller,
|
||||||
|
Get,
|
||||||
|
Post,
|
||||||
|
Put,
|
||||||
|
Body,
|
||||||
|
Param,
|
||||||
|
Query,
|
||||||
|
Headers,
|
||||||
|
HttpCode,
|
||||||
|
HttpStatus,
|
||||||
|
UnauthorizedException,
|
||||||
|
ForbiddenException,
|
||||||
|
} from '@nestjs/common';
|
||||||
|
import { AdminService, AdminRole, LoginResult } from './admin.service';
|
||||||
|
|
||||||
|
// ========== DTOs ==========
|
||||||
|
|
||||||
|
class LoginDto {
|
||||||
|
username: string;
|
||||||
|
password: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
class CreateAdminDto {
|
||||||
|
username: string;
|
||||||
|
password: string;
|
||||||
|
name: string;
|
||||||
|
email?: string;
|
||||||
|
phone?: string;
|
||||||
|
role: AdminRole;
|
||||||
|
}
|
||||||
|
|
||||||
|
class UpdateAdminDto {
|
||||||
|
name?: string;
|
||||||
|
email?: string;
|
||||||
|
phone?: string;
|
||||||
|
role?: AdminRole;
|
||||||
|
isActive?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
class ChangePasswordDto {
|
||||||
|
oldPassword: string;
|
||||||
|
newPassword: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
class ResetPasswordDto {
|
||||||
|
newPassword: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== Controller ==========
|
||||||
|
|
||||||
|
@Controller('admin')
|
||||||
|
export class AdminController {
|
||||||
|
constructor(private adminService: AdminService) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 管理员登录
|
||||||
|
*/
|
||||||
|
@Post('login')
|
||||||
|
@HttpCode(HttpStatus.OK)
|
||||||
|
async login(@Body() dto: LoginDto): Promise<{ success: boolean; data: LoginResult }> {
|
||||||
|
try {
|
||||||
|
const result = await this.adminService.login(dto.username, dto.password);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
throw new UnauthorizedException((error as Error).message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 验证Token
|
||||||
|
*/
|
||||||
|
@Get('verify')
|
||||||
|
async verifyToken(@Headers('authorization') auth: string) {
|
||||||
|
const token = auth?.replace('Bearer ', '');
|
||||||
|
if (!token) {
|
||||||
|
throw new UnauthorizedException('Missing token');
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await this.adminService.verifyToken(token);
|
||||||
|
if (!result.valid) {
|
||||||
|
throw new UnauthorizedException('Invalid token');
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result.admin,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取当前管理员信息
|
||||||
|
*/
|
||||||
|
@Get('me')
|
||||||
|
async getCurrentAdmin(@Headers('authorization') auth: string) {
|
||||||
|
const token = auth?.replace('Bearer ', '');
|
||||||
|
const result = await this.adminService.verifyToken(token);
|
||||||
|
|
||||||
|
if (!result.valid) {
|
||||||
|
throw new UnauthorizedException('Invalid token');
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result.admin,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建管理员（需要ADMIN权限）
|
||||||
|
*/
|
||||||
|
@Post('admins')
|
||||||
|
@HttpCode(HttpStatus.CREATED)
|
||||||
|
async createAdmin(
|
||||||
|
@Headers('authorization') auth: string,
|
||||||
|
@Body() dto: CreateAdminDto,
|
||||||
|
) {
|
||||||
|
await this.checkPermission(auth, 'admin:create');
|
||||||
|
|
||||||
|
try {
|
||||||
|
const admin = await this.adminService.createAdmin(dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
id: admin.id,
|
||||||
|
username: admin.username,
|
||||||
|
name: admin.name,
|
||||||
|
role: admin.role,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
success: false,
|
||||||
|
error: (error as Error).message,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取管理员列表
|
||||||
|
*/
|
||||||
|
@Get('admins')
|
||||||
|
async listAdmins(
|
||||||
|
@Headers('authorization') auth: string,
|
||||||
|
@Query('role') role?: AdminRole,
|
||||||
|
@Query('isActive') isActive?: string,
|
||||||
|
@Query('page') page?: string,
|
||||||
|
@Query('pageSize') pageSize?: string,
|
||||||
|
) {
|
||||||
|
await this.checkPermission(auth, 'admin:read');
|
||||||
|
|
||||||
|
const result = await this.adminService.listAdmins({
|
||||||
|
role,
|
||||||
|
isActive: isActive === undefined ? undefined : isActive === 'true',
|
||||||
|
page: page ? parseInt(page) : 1,
|
||||||
|
pageSize: pageSize ? parseInt(pageSize) : 20,
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
items: result.items.map(a => ({
|
||||||
|
id: a.id,
|
||||||
|
username: a.username,
|
||||||
|
name: a.name,
|
||||||
|
email: a.email,
|
||||||
|
phone: a.phone,
|
||||||
|
role: a.role,
|
||||||
|
isActive: a.isActive,
|
||||||
|
lastLoginAt: a.lastLoginAt,
|
||||||
|
createdAt: a.createdAt,
|
||||||
|
})),
|
||||||
|
total: result.total,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新管理员
|
||||||
|
*/
|
||||||
|
@Put('admins/:id')
|
||||||
|
async updateAdmin(
|
||||||
|
@Headers('authorization') auth: string,
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() dto: UpdateAdminDto,
|
||||||
|
) {
|
||||||
|
await this.checkPermission(auth, 'admin:update');
|
||||||
|
|
||||||
|
try {
|
||||||
|
const admin = await this.adminService.updateAdmin(id, dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
id: admin.id,
|
||||||
|
username: admin.username,
|
||||||
|
name: admin.name,
|
||||||
|
role: admin.role,
|
||||||
|
isActive: admin.isActive,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
success: false,
|
||||||
|
error: (error as Error).message,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 修改密码（当前用户）
|
||||||
|
*/
|
||||||
|
@Post('change-password')
|
||||||
|
async changePassword(
|
||||||
|
@Headers('authorization') auth: string,
|
||||||
|
@Body() dto: ChangePasswordDto,
|
||||||
|
) {
|
||||||
|
const token = auth?.replace('Bearer ', '');
|
||||||
|
const result = await this.adminService.verifyToken(token);
|
||||||
|
|
||||||
|
if (!result.valid || !result.admin) {
|
||||||
|
throw new UnauthorizedException('Invalid token');
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
await this.adminService.changePassword(
|
||||||
|
result.admin.id,
|
||||||
|
dto.oldPassword,
|
||||||
|
dto.newPassword,
|
||||||
|
);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: '密码修改成功',
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
success: false,
|
||||||
|
error: (error as Error).message,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 重置密码（超管功能）
|
||||||
|
*/
|
||||||
|
@Post('admins/:id/reset-password')
|
||||||
|
async resetPassword(
|
||||||
|
@Headers('authorization') auth: string,
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() dto: ResetPasswordDto,
|
||||||
|
) {
|
||||||
|
// 只有超管可以重置密码
|
||||||
|
const token = auth?.replace('Bearer ', '');
|
||||||
|
const result = await this.adminService.verifyToken(token);
|
||||||
|
|
||||||
|
if (!result.valid || !result.admin) {
|
||||||
|
throw new UnauthorizedException('Invalid token');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (result.admin.role !== AdminRole.SUPER_ADMIN) {
|
||||||
|
throw new ForbiddenException('Only super admin can reset passwords');
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
await this.adminService.resetPassword(id, dto.newPassword);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: '密码重置成功',
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
success: false,
|
||||||
|
error: (error as Error).message,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查权限
|
||||||
|
*/
|
||||||
|
private async checkPermission(auth: string, permission: string): Promise<void> {
|
||||||
|
const token = auth?.replace('Bearer ', '');
|
||||||
|
if (!token) {
|
||||||
|
throw new UnauthorizedException('Missing token');
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = await this.adminService.verifyToken(token);
|
||||||
|
if (!result.valid || !result.admin) {
|
||||||
|
throw new UnauthorizedException('Invalid token');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!this.adminService.hasPermission(result.admin.permissions, permission)) {
|
||||||
|
throw new ForbiddenException(`Permission denied: ${permission}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,15 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { AdminController } from './admin.controller';
|
||||||
|
import { AdminService } from './admin.service';
|
||||||
|
import { AdminORM } from '../infrastructure/database/entities/admin.orm';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
TypeOrmModule.forFeature([AdminORM]),
|
||||||
|
],
|
||||||
|
controllers: [AdminController],
|
||||||
|
providers: [AdminService],
|
||||||
|
exports: [AdminService],
|
||||||
|
})
|
||||||
|
export class AdminModule {}
|
||||||
|
|
@ -0,0 +1,344 @@
|
||||||
|
import { Injectable } from '@nestjs/common';
|
||||||
|
import { InjectRepository } from '@nestjs/typeorm';
|
||||||
|
import { Repository } from 'typeorm';
|
||||||
|
import * as bcrypt from 'bcrypt';
|
||||||
|
import * as jwt from 'jsonwebtoken';
|
||||||
|
import { ConfigService } from '@nestjs/config';
|
||||||
|
import { AdminORM } from '../infrastructure/database/entities/admin.orm';
|
||||||
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 管理员角色
|
||||||
|
*/
|
||||||
|
export enum AdminRole {
|
||||||
|
SUPER_ADMIN = 'SUPER_ADMIN',
|
||||||
|
ADMIN = 'ADMIN',
|
||||||
|
OPERATOR = 'OPERATOR',
|
||||||
|
VIEWER = 'VIEWER',
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 角色权限映射
|
||||||
|
*/
|
||||||
|
const ROLE_PERMISSIONS: Record<AdminRole, string[]> = {
|
||||||
|
[AdminRole.SUPER_ADMIN]: ['*'],
|
||||||
|
[AdminRole.ADMIN]: [
|
||||||
|
'knowledge:*',
|
||||||
|
'experience:*',
|
||||||
|
'user:read',
|
||||||
|
'conversation:read',
|
||||||
|
'statistics:*',
|
||||||
|
'admin:read',
|
||||||
|
],
|
||||||
|
[AdminRole.OPERATOR]: [
|
||||||
|
'knowledge:read',
|
||||||
|
'knowledge:create',
|
||||||
|
'knowledge:update',
|
||||||
|
'experience:read',
|
||||||
|
'experience:approve',
|
||||||
|
'user:read',
|
||||||
|
'conversation:read',
|
||||||
|
'statistics:read',
|
||||||
|
],
|
||||||
|
[AdminRole.VIEWER]: [
|
||||||
|
'knowledge:read',
|
||||||
|
'experience:read',
|
||||||
|
'user:read',
|
||||||
|
'conversation:read',
|
||||||
|
'statistics:read',
|
||||||
|
],
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 登录结果
|
||||||
|
*/
|
||||||
|
export interface LoginResult {
|
||||||
|
admin: {
|
||||||
|
id: string;
|
||||||
|
username: string;
|
||||||
|
name: string;
|
||||||
|
role: string;
|
||||||
|
permissions: string[];
|
||||||
|
};
|
||||||
|
token: string;
|
||||||
|
expiresIn: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 管理员服务
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class AdminService {
|
||||||
|
private readonly jwtSecret: string;
|
||||||
|
private readonly jwtExpiresIn: number = 24 * 60 * 60; // 24小时
|
||||||
|
|
||||||
|
constructor(
|
||||||
|
@InjectRepository(AdminORM)
|
||||||
|
private adminRepo: Repository<AdminORM>,
|
||||||
|
private configService: ConfigService,
|
||||||
|
) {
|
||||||
|
this.jwtSecret = this.configService.get('JWT_SECRET') || 'iconsulting-secret-key';
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 管理员登录
|
||||||
|
*/
|
||||||
|
async login(username: string, password: string, ip?: string): Promise<LoginResult> {
|
||||||
|
const admin = await this.adminRepo.findOne({ where: { username } });
|
||||||
|
|
||||||
|
if (!admin || !admin.isActive) {
|
||||||
|
throw new Error('用户名或密码错误');
|
||||||
|
}
|
||||||
|
|
||||||
|
const isPasswordValid = await bcrypt.compare(password, admin.passwordHash);
|
||||||
|
if (!isPasswordValid) {
|
||||||
|
throw new Error('用户名或密码错误');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新登录信息
|
||||||
|
admin.lastLoginAt = new Date();
|
||||||
|
admin.lastLoginIp = ip;
|
||||||
|
await this.adminRepo.save(admin);
|
||||||
|
|
||||||
|
// 生成Token
|
||||||
|
const token = jwt.sign(
|
||||||
|
{
|
||||||
|
sub: admin.id,
|
||||||
|
username: admin.username,
|
||||||
|
role: admin.role,
|
||||||
|
},
|
||||||
|
this.jwtSecret,
|
||||||
|
{ expiresIn: this.jwtExpiresIn },
|
||||||
|
);
|
||||||
|
|
||||||
|
// 获取权限
|
||||||
|
const permissions = this.getPermissions(admin.role as AdminRole, admin.permissions);
|
||||||
|
|
||||||
|
return {
|
||||||
|
admin: {
|
||||||
|
id: admin.id,
|
||||||
|
username: admin.username,
|
||||||
|
name: admin.name,
|
||||||
|
role: admin.role,
|
||||||
|
permissions,
|
||||||
|
},
|
||||||
|
token,
|
||||||
|
expiresIn: this.jwtExpiresIn,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 验证Token
|
||||||
|
*/
|
||||||
|
async verifyToken(token: string): Promise<{
|
||||||
|
valid: boolean;
|
||||||
|
admin?: {
|
||||||
|
id: string;
|
||||||
|
username: string;
|
||||||
|
role: string;
|
||||||
|
permissions: string[];
|
||||||
|
};
|
||||||
|
}> {
|
||||||
|
try {
|
||||||
|
const decoded = jwt.verify(token, this.jwtSecret) as {
|
||||||
|
sub: string;
|
||||||
|
username: string;
|
||||||
|
role: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
const admin = await this.adminRepo.findOne({ where: { id: decoded.sub } });
|
||||||
|
if (!admin || !admin.isActive) {
|
||||||
|
return { valid: false };
|
||||||
|
}
|
||||||
|
|
||||||
|
const permissions = this.getPermissions(admin.role as AdminRole, admin.permissions);
|
||||||
|
|
||||||
|
return {
|
||||||
|
valid: true,
|
||||||
|
admin: {
|
||||||
|
id: admin.id,
|
||||||
|
username: admin.username,
|
||||||
|
role: admin.role,
|
||||||
|
permissions,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
} catch {
|
||||||
|
return { valid: false };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建管理员
|
||||||
|
*/
|
||||||
|
async createAdmin(params: {
|
||||||
|
username: string;
|
||||||
|
password: string;
|
||||||
|
name: string;
|
||||||
|
email?: string;
|
||||||
|
phone?: string;
|
||||||
|
role: AdminRole;
|
||||||
|
}): Promise<AdminORM> {
|
||||||
|
// 检查用户名是否存在
|
||||||
|
const existing = await this.adminRepo.findOne({
|
||||||
|
where: { username: params.username },
|
||||||
|
});
|
||||||
|
if (existing) {
|
||||||
|
throw new Error('用户名已存在');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 加密密码
|
||||||
|
const passwordHash = await bcrypt.hash(params.password, 10);
|
||||||
|
|
||||||
|
const admin = this.adminRepo.create({
|
||||||
|
id: uuidv4(),
|
||||||
|
username: params.username,
|
||||||
|
passwordHash,
|
||||||
|
name: params.name,
|
||||||
|
email: params.email,
|
||||||
|
phone: params.phone,
|
||||||
|
role: params.role,
|
||||||
|
permissions: ROLE_PERMISSIONS[params.role],
|
||||||
|
isActive: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
await this.adminRepo.save(admin);
|
||||||
|
|
||||||
|
return admin;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取管理员列表
|
||||||
|
*/
|
||||||
|
async listAdmins(options?: {
|
||||||
|
role?: AdminRole;
|
||||||
|
isActive?: boolean;
|
||||||
|
page?: number;
|
||||||
|
pageSize?: number;
|
||||||
|
}): Promise<{
|
||||||
|
items: AdminORM[];
|
||||||
|
total: number;
|
||||||
|
}> {
|
||||||
|
const page = options?.page || 1;
|
||||||
|
const pageSize = options?.pageSize || 20;
|
||||||
|
|
||||||
|
const query = this.adminRepo.createQueryBuilder('admin');
|
||||||
|
|
||||||
|
if (options?.role) {
|
||||||
|
query.andWhere('admin.role = :role', { role: options.role });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.isActive !== undefined) {
|
||||||
|
query.andWhere('admin.isActive = :active', { active: options.isActive });
|
||||||
|
}
|
||||||
|
|
||||||
|
query.orderBy('admin.createdAt', 'DESC');
|
||||||
|
|
||||||
|
const [items, total] = await query
|
||||||
|
.skip((page - 1) * pageSize)
|
||||||
|
.take(pageSize)
|
||||||
|
.getManyAndCount();
|
||||||
|
|
||||||
|
return { items, total };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新管理员
|
||||||
|
*/
|
||||||
|
async updateAdmin(
|
||||||
|
adminId: string,
|
||||||
|
params: {
|
||||||
|
name?: string;
|
||||||
|
email?: string;
|
||||||
|
phone?: string;
|
||||||
|
role?: AdminRole;
|
||||||
|
isActive?: boolean;
|
||||||
|
},
|
||||||
|
): Promise<AdminORM> {
|
||||||
|
const admin = await this.adminRepo.findOne({ where: { id: adminId } });
|
||||||
|
if (!admin) {
|
||||||
|
throw new Error('管理员不存在');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (params.name) admin.name = params.name;
|
||||||
|
if (params.email !== undefined) admin.email = params.email;
|
||||||
|
if (params.phone !== undefined) admin.phone = params.phone;
|
||||||
|
if (params.role) {
|
||||||
|
admin.role = params.role;
|
||||||
|
admin.permissions = ROLE_PERMISSIONS[params.role];
|
||||||
|
}
|
||||||
|
if (params.isActive !== undefined) admin.isActive = params.isActive;
|
||||||
|
|
||||||
|
await this.adminRepo.save(admin);
|
||||||
|
|
||||||
|
return admin;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 修改密码
|
||||||
|
*/
|
||||||
|
async changePassword(
|
||||||
|
adminId: string,
|
||||||
|
oldPassword: string,
|
||||||
|
newPassword: string,
|
||||||
|
): Promise<void> {
|
||||||
|
const admin = await this.adminRepo.findOne({ where: { id: adminId } });
|
||||||
|
if (!admin) {
|
||||||
|
throw new Error('管理员不存在');
|
||||||
|
}
|
||||||
|
|
||||||
|
const isOldPasswordValid = await bcrypt.compare(oldPassword, admin.passwordHash);
|
||||||
|
if (!isOldPasswordValid) {
|
||||||
|
throw new Error('原密码错误');
|
||||||
|
}
|
||||||
|
|
||||||
|
admin.passwordHash = await bcrypt.hash(newPassword, 10);
|
||||||
|
await this.adminRepo.save(admin);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 重置密码（超管功能）
|
||||||
|
*/
|
||||||
|
async resetPassword(adminId: string, newPassword: string): Promise<void> {
|
||||||
|
const admin = await this.adminRepo.findOne({ where: { id: adminId } });
|
||||||
|
if (!admin) {
|
||||||
|
throw new Error('管理员不存在');
|
||||||
|
}
|
||||||
|
|
||||||
|
admin.passwordHash = await bcrypt.hash(newPassword, 10);
|
||||||
|
await this.adminRepo.save(admin);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查权限
|
||||||
|
*/
|
||||||
|
hasPermission(adminPermissions: string[], requiredPermission: string): boolean {
|
||||||
|
// 超管拥有所有权限
|
||||||
|
if (adminPermissions.includes('*')) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 完全匹配
|
||||||
|
if (adminPermissions.includes(requiredPermission)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 通配符匹配 (如 knowledge:* 匹配 knowledge:read)
|
||||||
|
const [resource, action] = requiredPermission.split(':');
|
||||||
|
if (adminPermissions.includes(`${resource}:*`)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取管理员权限列表
|
||||||
|
*/
|
||||||
|
private getPermissions(role: AdminRole, customPermissions?: string[]): string[] {
|
||||||
|
const rolePermissions = ROLE_PERMISSIONS[role] || [];
|
||||||
|
if (customPermissions && customPermissions.length > 0) {
|
||||||
|
return [...new Set([...rolePermissions, ...customPermissions])];
|
||||||
|
}
|
||||||
|
return rolePermissions;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { ConfigModule, ConfigService } from '@nestjs/config';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { EvolutionModule } from './evolution/evolution.module';
|
||||||
|
import { AdminModule } from './admin/admin.module';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
// 配置模块
|
||||||
|
ConfigModule.forRoot({
|
||||||
|
isGlobal: true,
|
||||||
|
envFilePath: ['.env.local', '.env'],
|
||||||
|
}),
|
||||||
|
|
||||||
|
// 数据库连接
|
||||||
|
TypeOrmModule.forRootAsync({
|
||||||
|
imports: [ConfigModule],
|
||||||
|
inject: [ConfigService],
|
||||||
|
useFactory: (config: ConfigService) => ({
|
||||||
|
type: 'postgres',
|
||||||
|
host: config.get('DB_HOST', 'localhost'),
|
||||||
|
port: config.get('DB_PORT', 5432),
|
||||||
|
username: config.get('DB_USER', 'iconsulting'),
|
||||||
|
password: config.get('DB_PASSWORD', 'iconsulting_dev'),
|
||||||
|
database: config.get('DB_NAME', 'iconsulting'),
|
||||||
|
autoLoadEntities: true,
|
||||||
|
synchronize: config.get('NODE_ENV') !== 'production',
|
||||||
|
logging: config.get('NODE_ENV') === 'development',
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
|
||||||
|
// 功能模块
|
||||||
|
EvolutionModule,
|
||||||
|
AdminModule,
|
||||||
|
],
|
||||||
|
})
|
||||||
|
export class AppModule {}
|
||||||
|
|
@ -0,0 +1,62 @@
|
||||||
|
import {
|
||||||
|
Controller,
|
||||||
|
Get,
|
||||||
|
Post,
|
||||||
|
Body,
|
||||||
|
Query,
|
||||||
|
HttpCode,
|
||||||
|
HttpStatus,
|
||||||
|
} from '@nestjs/common';
|
||||||
|
import { EvolutionService } from './evolution.service';
|
||||||
|
|
||||||
|
// ========== DTOs ==========
|
||||||
|
|
||||||
|
class RunEvolutionTaskDto {
|
||||||
|
hoursBack?: number;
|
||||||
|
limit?: number;
|
||||||
|
minMessageCount?: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== Controller ==========
|
||||||
|
|
||||||
|
@Controller('evolution')
|
||||||
|
export class EvolutionController {
|
||||||
|
constructor(private evolutionService: EvolutionService) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 手动触发进化任务
|
||||||
|
*/
|
||||||
|
@Post('run')
|
||||||
|
@HttpCode(HttpStatus.OK)
|
||||||
|
async runEvolutionTask(@Body() dto: RunEvolutionTaskDto) {
|
||||||
|
const result = await this.evolutionService.runEvolutionTask(dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取进化统计信息
|
||||||
|
*/
|
||||||
|
@Get('statistics')
|
||||||
|
async getStatistics() {
|
||||||
|
const stats = await this.evolutionService.getEvolutionStatistics();
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: stats,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取系统健康报告
|
||||||
|
*/
|
||||||
|
@Get('health')
|
||||||
|
async getHealthReport() {
|
||||||
|
const report = await this.evolutionService.getSystemHealthReport();
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: report,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,25 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { EvolutionController } from './evolution.controller';
|
||||||
|
import { EvolutionService } from './evolution.service';
|
||||||
|
import { ExperienceExtractorService } from '../infrastructure/claude/experience-extractor.service';
|
||||||
|
import { ConversationORM } from '../infrastructure/database/entities/conversation.orm';
|
||||||
|
import { MessageORM } from '../infrastructure/database/entities/message.orm';
|
||||||
|
import { SystemExperienceORM } from '../infrastructure/database/entities/system-experience.orm';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
TypeOrmModule.forFeature([
|
||||||
|
ConversationORM,
|
||||||
|
MessageORM,
|
||||||
|
SystemExperienceORM,
|
||||||
|
]),
|
||||||
|
],
|
||||||
|
controllers: [EvolutionController],
|
||||||
|
providers: [
|
||||||
|
EvolutionService,
|
||||||
|
ExperienceExtractorService,
|
||||||
|
],
|
||||||
|
exports: [EvolutionService],
|
||||||
|
})
|
||||||
|
export class EvolutionModule {}
|
||||||
|
|
@ -0,0 +1,321 @@
|
||||||
|
import { Injectable } from '@nestjs/common';
|
||||||
|
import { InjectRepository } from '@nestjs/typeorm';
|
||||||
|
import { Repository, MoreThan, LessThan } from 'typeorm';
|
||||||
|
import { ExperienceExtractorService } from '../infrastructure/claude/experience-extractor.service';
|
||||||
|
import { ConversationORM } from '../infrastructure/database/entities/conversation.orm';
|
||||||
|
import { MessageORM } from '../infrastructure/database/entities/message.orm';
|
||||||
|
import { SystemExperienceORM } from '../infrastructure/database/entities/system-experience.orm';
|
||||||
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 进化任务结果
|
||||||
|
*/
|
||||||
|
export interface EvolutionTaskResult {
|
||||||
|
taskId: string;
|
||||||
|
status: 'success' | 'partial' | 'failed';
|
||||||
|
conversationsAnalyzed: number;
|
||||||
|
experiencesExtracted: number;
|
||||||
|
knowledgeGapsFound: number;
|
||||||
|
errors: string[];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 进化服务
|
||||||
|
* 负责系统的自我学习和进化
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class EvolutionService {
|
||||||
|
constructor(
|
||||||
|
@InjectRepository(ConversationORM)
|
||||||
|
private conversationRepo: Repository<ConversationORM>,
|
||||||
|
@InjectRepository(MessageORM)
|
||||||
|
private messageRepo: Repository<MessageORM>,
|
||||||
|
@InjectRepository(SystemExperienceORM)
|
||||||
|
private experienceRepo: Repository<SystemExperienceORM>,
|
||||||
|
private experienceExtractor: ExperienceExtractorService,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行进化任务 - 分析最近的对话并提取经验
|
||||||
|
*/
|
||||||
|
async runEvolutionTask(options?: {
|
||||||
|
hoursBack?: number;
|
||||||
|
limit?: number;
|
||||||
|
minMessageCount?: number;
|
||||||
|
}): Promise<EvolutionTaskResult> {
|
||||||
|
const taskId = uuidv4();
|
||||||
|
const hoursBack = options?.hoursBack || 24;
|
||||||
|
const limit = options?.limit || 50;
|
||||||
|
const minMessageCount = options?.minMessageCount || 4;
|
||||||
|
|
||||||
|
const result: EvolutionTaskResult = {
|
||||||
|
taskId,
|
||||||
|
status: 'success',
|
||||||
|
conversationsAnalyzed: 0,
|
||||||
|
experiencesExtracted: 0,
|
||||||
|
knowledgeGapsFound: 0,
|
||||||
|
errors: [],
|
||||||
|
};
|
||||||
|
|
||||||
|
console.log(`[Evolution] Starting task ${taskId}`);
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 1. 获取待分析的对话
|
||||||
|
const cutoffTime = new Date();
|
||||||
|
cutoffTime.setHours(cutoffTime.getHours() - hoursBack);
|
||||||
|
|
||||||
|
const conversations = await this.conversationRepo.find({
|
||||||
|
where: {
|
||||||
|
status: 'ENDED',
|
||||||
|
createdAt: MoreThan(cutoffTime),
|
||||||
|
messageCount: MoreThan(minMessageCount),
|
||||||
|
},
|
||||||
|
order: { createdAt: 'DESC' },
|
||||||
|
take: limit,
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log(`[Evolution] Found ${conversations.length} conversations to analyze`);
|
||||||
|
|
||||||
|
// 2. 分析每个对话
|
||||||
|
const allKnowledgeGaps: string[] = [];
|
||||||
|
|
||||||
|
for (const conversation of conversations) {
|
||||||
|
try {
|
||||||
|
// 获取对话消息
|
||||||
|
const messages = await this.messageRepo.find({
|
||||||
|
where: { conversationId: conversation.id },
|
||||||
|
order: { createdAt: 'ASC' },
|
||||||
|
});
|
||||||
|
|
||||||
|
// 分析对话
|
||||||
|
const analysis = await this.experienceExtractor.analyzeConversation({
|
||||||
|
conversationId: conversation.id,
|
||||||
|
messages: messages.map(m => ({
|
||||||
|
role: m.role as 'user' | 'assistant',
|
||||||
|
content: m.content,
|
||||||
|
})),
|
||||||
|
category: conversation.category,
|
||||||
|
hasConverted: conversation.hasConverted,
|
||||||
|
rating: conversation.rating,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 保存提取的经验
|
||||||
|
for (const exp of analysis.experiences) {
|
||||||
|
await this.saveExperience({
|
||||||
|
experienceType: exp.type,
|
||||||
|
content: exp.content,
|
||||||
|
scenario: exp.scenario,
|
||||||
|
confidence: exp.confidence,
|
||||||
|
relatedCategory: exp.relatedCategory,
|
||||||
|
sourceConversationId: conversation.id,
|
||||||
|
});
|
||||||
|
result.experiencesExtracted++;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 收集知识缺口
|
||||||
|
allKnowledgeGaps.push(...analysis.knowledgeGaps);
|
||||||
|
|
||||||
|
result.conversationsAnalyzed++;
|
||||||
|
} catch (error) {
|
||||||
|
result.errors.push(`Conversation ${conversation.id}: ${(error as Error).message}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. 汇总知识缺口
|
||||||
|
const uniqueGaps = [...new Set(allKnowledgeGaps)];
|
||||||
|
result.knowledgeGapsFound = uniqueGaps.length;
|
||||||
|
|
||||||
|
// 可以在这里调用知识服务创建待处理的知识缺口任务
|
||||||
|
|
||||||
|
console.log(`[Evolution] Task ${taskId} completed:`, result);
|
||||||
|
|
||||||
|
if (result.errors.length > 0) {
|
||||||
|
result.status = 'partial';
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
} catch (error) {
|
||||||
|
console.error(`[Evolution] Task ${taskId} failed:`, error);
|
||||||
|
result.status = 'failed';
|
||||||
|
result.errors.push((error as Error).message);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 保存经验（带去重逻辑）
|
||||||
|
*/
|
||||||
|
private async saveExperience(params: {
|
||||||
|
experienceType: string;
|
||||||
|
content: string;
|
||||||
|
scenario: string;
|
||||||
|
confidence: number;
|
||||||
|
relatedCategory?: string;
|
||||||
|
sourceConversationId: string;
|
||||||
|
}): Promise<void> {
|
||||||
|
// 查找相似经验
|
||||||
|
const existingExperiences = await this.experienceRepo.find({
|
||||||
|
where: {
|
||||||
|
experienceType: params.experienceType,
|
||||||
|
relatedCategory: params.relatedCategory,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
// 简单的相似度检查（实际应该用向量相似度）
|
||||||
|
const similar = existingExperiences.find(
|
||||||
|
exp => this.simpleSimilarity(exp.content, params.content) > 0.8,
|
||||||
|
);
|
||||||
|
|
||||||
|
if (similar) {
|
||||||
|
// 合并到现有经验
|
||||||
|
if (!similar.sourceConversationIds.includes(params.sourceConversationId)) {
|
||||||
|
similar.sourceConversationIds.push(params.sourceConversationId);
|
||||||
|
similar.confidence = Math.min(100, similar.confidence + 5);
|
||||||
|
await this.experienceRepo.save(similar);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 创建新经验
|
||||||
|
const newExperience = this.experienceRepo.create({
|
||||||
|
id: uuidv4(),
|
||||||
|
experienceType: params.experienceType,
|
||||||
|
content: params.content,
|
||||||
|
scenario: params.scenario,
|
||||||
|
confidence: params.confidence,
|
||||||
|
relatedCategory: params.relatedCategory,
|
||||||
|
sourceConversationIds: [params.sourceConversationId],
|
||||||
|
verificationStatus: 'PENDING',
|
||||||
|
isActive: false,
|
||||||
|
});
|
||||||
|
await this.experienceRepo.save(newExperience);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 简单的文本相似度计算
|
||||||
|
*/
|
||||||
|
private simpleSimilarity(a: string, b: string): number {
|
||||||
|
const aWords = new Set(a.toLowerCase().split(/\s+/));
|
||||||
|
const bWords = new Set(b.toLowerCase().split(/\s+/));
|
||||||
|
const intersection = [...aWords].filter(x => bWords.has(x)).length;
|
||||||
|
const union = new Set([...aWords, ...bWords]).size;
|
||||||
|
return intersection / union;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取进化统计信息
|
||||||
|
*/
|
||||||
|
async getEvolutionStatistics(): Promise<{
|
||||||
|
totalExperiences: number;
|
||||||
|
pendingExperiences: number;
|
||||||
|
approvedExperiences: number;
|
||||||
|
activeExperiences: number;
|
||||||
|
recentConversationsAnalyzed: number;
|
||||||
|
topExperienceTypes: Array<{ type: string; count: number }>;
|
||||||
|
}> {
|
||||||
|
const [total, pending, approved, active] = await Promise.all([
|
||||||
|
this.experienceRepo.count(),
|
||||||
|
this.experienceRepo.count({ where: { verificationStatus: 'PENDING' } }),
|
||||||
|
this.experienceRepo.count({ where: { verificationStatus: 'APPROVED' } }),
|
||||||
|
this.experienceRepo.count({ where: { isActive: true } }),
|
||||||
|
]);
|
||||||
|
|
||||||
|
// 获取最近分析的对话数（过去7天）
|
||||||
|
const weekAgo = new Date();
|
||||||
|
weekAgo.setDate(weekAgo.getDate() - 7);
|
||||||
|
const recentConversations = await this.conversationRepo.count({
|
||||||
|
where: {
|
||||||
|
status: 'ENDED',
|
||||||
|
updatedAt: MoreThan(weekAgo),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取经验类型分布
|
||||||
|
const typeDistribution = await this.experienceRepo
|
||||||
|
.createQueryBuilder('exp')
|
||||||
|
.select('exp.experienceType', 'type')
|
||||||
|
.addSelect('COUNT(*)', 'count')
|
||||||
|
.groupBy('exp.experienceType')
|
||||||
|
.orderBy('count', 'DESC')
|
||||||
|
.limit(5)
|
||||||
|
.getRawMany();
|
||||||
|
|
||||||
|
return {
|
||||||
|
totalExperiences: total,
|
||||||
|
pendingExperiences: pending,
|
||||||
|
approvedExperiences: approved,
|
||||||
|
activeExperiences: active,
|
||||||
|
recentConversationsAnalyzed: recentConversations,
|
||||||
|
topExperienceTypes: typeDistribution.map(t => ({
|
||||||
|
type: t.type,
|
||||||
|
count: parseInt(t.count),
|
||||||
|
})),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取系统健康报告
|
||||||
|
*/
|
||||||
|
async getSystemHealthReport(): Promise<{
|
||||||
|
overall: 'healthy' | 'warning' | 'critical';
|
||||||
|
metrics: Array<{
|
||||||
|
name: string;
|
||||||
|
value: number;
|
||||||
|
threshold: number;
|
||||||
|
status: 'good' | 'warning' | 'critical';
|
||||||
|
}>;
|
||||||
|
recommendations: string[];
|
||||||
|
}> {
|
||||||
|
const stats = await this.getEvolutionStatistics();
|
||||||
|
const metrics: Array<{
|
||||||
|
name: string;
|
||||||
|
value: number;
|
||||||
|
threshold: number;
|
||||||
|
status: 'good' | 'warning' | 'critical';
|
||||||
|
}> = [];
|
||||||
|
const recommendations: string[] = [];
|
||||||
|
|
||||||
|
// 检查待验证经验堆积
|
||||||
|
const pendingRatio = stats.pendingExperiences / Math.max(1, stats.totalExperiences);
|
||||||
|
metrics.push({
|
||||||
|
name: '待验证经验比例',
|
||||||
|
value: Math.round(pendingRatio * 100),
|
||||||
|
threshold: 50,
|
||||||
|
status: pendingRatio > 0.5 ? 'warning' : 'good',
|
||||||
|
});
|
||||||
|
if (pendingRatio > 0.5) {
|
||||||
|
recommendations.push('待验证经验过多，建议及时审核');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查活跃经验数量
|
||||||
|
metrics.push({
|
||||||
|
name: '活跃经验数量',
|
||||||
|
value: stats.activeExperiences,
|
||||||
|
threshold: 10,
|
||||||
|
status: stats.activeExperiences < 10 ? 'warning' : 'good',
|
||||||
|
});
|
||||||
|
if (stats.activeExperiences < 10) {
|
||||||
|
recommendations.push('活跃经验较少，系统学习能力有限');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查最近分析的对话
|
||||||
|
metrics.push({
|
||||||
|
name: '近7天分析对话数',
|
||||||
|
value: stats.recentConversationsAnalyzed,
|
||||||
|
threshold: 50,
|
||||||
|
status: stats.recentConversationsAnalyzed < 50 ? 'warning' : 'good',
|
||||||
|
});
|
||||||
|
|
||||||
|
// 计算总体健康状态
|
||||||
|
const criticalCount = metrics.filter(m => m.status === 'critical').length;
|
||||||
|
const warningCount = metrics.filter(m => m.status === 'warning').length;
|
||||||
|
|
||||||
|
let overall: 'healthy' | 'warning' | 'critical' = 'healthy';
|
||||||
|
if (criticalCount > 0) {
|
||||||
|
overall = 'critical';
|
||||||
|
} else if (warningCount > 1) {
|
||||||
|
overall = 'warning';
|
||||||
|
}
|
||||||
|
|
||||||
|
return { overall, metrics, recommendations };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,292 @@
|
||||||
|
import { Injectable, OnModuleInit } from '@nestjs/common';
|
||||||
|
import { ConfigService } from '@nestjs/config';
|
||||||
|
import Anthropic from '@anthropic-ai/sdk';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 提取的经验结构
|
||||||
|
*/
|
||||||
|
export interface ExtractedExperience {
|
||||||
|
type: string;
|
||||||
|
content: string;
|
||||||
|
scenario: string;
|
||||||
|
confidence: number;
|
||||||
|
relatedCategory?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 对话分析结果
|
||||||
|
*/
|
||||||
|
export interface ConversationAnalysis {
|
||||||
|
experiences: ExtractedExperience[];
|
||||||
|
userInsights: UserInsight[];
|
||||||
|
knowledgeGaps: string[];
|
||||||
|
conversionSignals: ConversionSignal[];
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface UserInsight {
|
||||||
|
type: string;
|
||||||
|
content: string;
|
||||||
|
importance: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface ConversionSignal {
|
||||||
|
type: 'positive' | 'negative';
|
||||||
|
signal: string;
|
||||||
|
context: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 经验提取服务
|
||||||
|
* 使用Claude分析对话，提取有价值的系统经验
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class ExperienceExtractorService implements OnModuleInit {
|
||||||
|
private client: Anthropic;
|
||||||
|
|
||||||
|
constructor(private configService: ConfigService) {}
|
||||||
|
|
||||||
|
onModuleInit() {
|
||||||
|
const apiKey = this.configService.get<string>('ANTHROPIC_API_KEY');
|
||||||
|
|
||||||
|
if (!apiKey) {
|
||||||
|
console.warn('[ExperienceExtractor] ANTHROPIC_API_KEY not set');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.client = new Anthropic({ apiKey });
|
||||||
|
console.log('[ExperienceExtractor] Initialized');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 分析对话并提取经验
|
||||||
|
*/
|
||||||
|
async analyzeConversation(params: {
|
||||||
|
conversationId: string;
|
||||||
|
messages: Array<{ role: 'user' | 'assistant'; content: string }>;
|
||||||
|
category?: string;
|
||||||
|
hasConverted: boolean;
|
||||||
|
rating?: number;
|
||||||
|
}): Promise<ConversationAnalysis> {
|
||||||
|
if (!this.client) {
|
||||||
|
return this.getMockAnalysis();
|
||||||
|
}
|
||||||
|
|
||||||
|
const prompt = this.buildAnalysisPrompt(params);
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await this.client.messages.create({
|
||||||
|
model: 'claude-sonnet-4-20250514',
|
||||||
|
max_tokens: 2000,
|
||||||
|
messages: [{ role: 'user', content: prompt }],
|
||||||
|
});
|
||||||
|
|
||||||
|
const content = response.content[0];
|
||||||
|
if (content.type !== 'text') {
|
||||||
|
return this.getMockAnalysis();
|
||||||
|
}
|
||||||
|
|
||||||
|
return this.parseAnalysisResult(content.text);
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[ExperienceExtractor] Analysis failed:', error);
|
||||||
|
return this.getMockAnalysis();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 构建分析提示词
|
||||||
|
*/
|
||||||
|
private buildAnalysisPrompt(params: {
|
||||||
|
messages: Array<{ role: 'user' | 'assistant'; content: string }>;
|
||||||
|
category?: string;
|
||||||
|
hasConverted: boolean;
|
||||||
|
rating?: number;
|
||||||
|
}): string {
|
||||||
|
const conversationText = params.messages
|
||||||
|
.map(m => `${m.role === 'user' ? '用户' : '助手'}: ${m.content}`)
|
||||||
|
.join('\n\n');
|
||||||
|
|
||||||
|
return `你是一个专门分析香港移民咨询对话的AI专家。请分析以下对话，提取有价值的系统经验。
|
||||||
|
|
||||||
|
## 对话背景
|
||||||
|
- 移民类别: ${params.category || '未知'}
|
||||||
|
- 是否转化付费: ${params.hasConverted ? '是' : '否'}
|
||||||
|
- 用户评分: ${params.rating || '未评分'}
|
||||||
|
|
||||||
|
## 对话内容
|
||||||
|
${conversationText}
|
||||||
|
|
||||||
|
## 分析任务
|
||||||
|
请提取以下信息，以JSON格式返回：
|
||||||
|
|
||||||
|
1. **experiences** (系统经验数组)：
|
||||||
|
- type: 经验类型 (COMMON_QUESTION/ANSWER_TEMPLATE/CLARIFICATION/USER_PATTERN/CONVERSION_TRIGGER/KNOWLEDGE_GAP/CONVERSATION_SKILL/OBJECTION_HANDLING)
|
||||||
|
- content: 经验内容
|
||||||
|
- scenario: 适用场景
|
||||||
|
- confidence: 置信度 (0-100)
|
||||||
|
- relatedCategory: 相关移民类别
|
||||||
|
|
||||||
|
2. **userInsights** (用户洞察数组)：
|
||||||
|
- type: 洞察类型 (PERSONAL_INFO/WORK_EXPERIENCE/EDUCATION/LANGUAGE/IMMIGRATION_INTENT/CONCERN)
|
||||||
|
- content: 洞察内容
|
||||||
|
- importance: 重要性 (0-100)
|
||||||
|
|
||||||
|
3. **knowledgeGaps** (知识缺口数组)：
|
||||||
|
- 对话中暴露出的知识库缺失内容
|
||||||
|
|
||||||
|
4. **conversionSignals** (转化信号数组)：
|
||||||
|
- type: positive/negative
|
||||||
|
- signal: 信号描述
|
||||||
|
- context: 上下文
|
||||||
|
|
||||||
|
请只返回JSON，不要其他内容。示例格式：
|
||||||
|
{
|
||||||
|
"experiences": [...],
|
||||||
|
"userInsights": [...],
|
||||||
|
"knowledgeGaps": [...],
|
||||||
|
"conversionSignals": [...]
|
||||||
|
}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析分析结果
|
||||||
|
*/
|
||||||
|
private parseAnalysisResult(text: string): ConversationAnalysis {
|
||||||
|
try {
|
||||||
|
// 尝试从文本中提取JSON
|
||||||
|
const jsonMatch = text.match(/\{[\s\S]*\}/);
|
||||||
|
if (!jsonMatch) {
|
||||||
|
throw new Error('No JSON found in response');
|
||||||
|
}
|
||||||
|
|
||||||
|
const result = JSON.parse(jsonMatch[0]);
|
||||||
|
|
||||||
|
return {
|
||||||
|
experiences: result.experiences || [],
|
||||||
|
userInsights: result.userInsights || [],
|
||||||
|
knowledgeGaps: result.knowledgeGaps || [],
|
||||||
|
conversionSignals: result.conversionSignals || [],
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[ExperienceExtractor] Failed to parse result:', error);
|
||||||
|
return this.getMockAnalysis();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Mock分析结果（开发用）
|
||||||
|
*/
|
||||||
|
private getMockAnalysis(): ConversationAnalysis {
|
||||||
|
return {
|
||||||
|
experiences: [],
|
||||||
|
userInsights: [],
|
||||||
|
knowledgeGaps: [],
|
||||||
|
conversionSignals: [],
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 从多个对话中总结经验
|
||||||
|
*/
|
||||||
|
async summarizeExperiences(
|
||||||
|
experiences: ExtractedExperience[],
|
||||||
|
): Promise<ExtractedExperience[]> {
|
||||||
|
if (!this.client || experiences.length < 3) {
|
||||||
|
return experiences;
|
||||||
|
}
|
||||||
|
|
||||||
|
const prompt = `你是一个经验总结专家。请分析以下从多个对话中提取的经验，找出共同模式并合并相似经验。
|
||||||
|
|
||||||
|
## 原始经验
|
||||||
|
${JSON.stringify(experiences, null, 2)}
|
||||||
|
|
||||||
|
## 任务
|
||||||
|
1. 合并相似的经验
|
||||||
|
2. 提高置信度（多次出现的经验置信度更高）
|
||||||
|
3. 移除重复或低质量的经验
|
||||||
|
|
||||||
|
请返回优化后的经验数组（JSON格式）：`;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await this.client.messages.create({
|
||||||
|
model: 'claude-sonnet-4-20250514',
|
||||||
|
max_tokens: 2000,
|
||||||
|
messages: [{ role: 'user', content: prompt }],
|
||||||
|
});
|
||||||
|
|
||||||
|
const content = response.content[0];
|
||||||
|
if (content.type !== 'text') {
|
||||||
|
return experiences;
|
||||||
|
}
|
||||||
|
|
||||||
|
const jsonMatch = content.text.match(/\[[\s\S]*\]/);
|
||||||
|
if (!jsonMatch) {
|
||||||
|
return experiences;
|
||||||
|
}
|
||||||
|
|
||||||
|
return JSON.parse(jsonMatch[0]);
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[ExperienceExtractor] Summary failed:', error);
|
||||||
|
return experiences;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 分析知识库缺口
|
||||||
|
*/
|
||||||
|
async analyzeKnowledgeGaps(
|
||||||
|
gaps: string[],
|
||||||
|
existingArticles: string[],
|
||||||
|
): Promise<Array<{
|
||||||
|
topic: string;
|
||||||
|
priority: number;
|
||||||
|
suggestedContent: string;
|
||||||
|
}>> {
|
||||||
|
if (!this.client || gaps.length === 0) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
|
||||||
|
const prompt = `你是香港移民知识库的内容规划专家。
|
||||||
|
|
||||||
|
## 现有知识库文章标题
|
||||||
|
${existingArticles.join('\n')}
|
||||||
|
|
||||||
|
## 对话中发现的知识缺口
|
||||||
|
${gaps.join('\n')}
|
||||||
|
|
||||||
|
## 任务
|
||||||
|
分析哪些知识缺口是真正需要补充的，并按优先级排序。
|
||||||
|
对于每个需要补充的主题，提供建议的内容大纲。
|
||||||
|
|
||||||
|
请返回JSON数组：
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"topic": "主题",
|
||||||
|
"priority": 1-100,
|
||||||
|
"suggestedContent": "建议的内容大纲"
|
||||||
|
}
|
||||||
|
]`;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await this.client.messages.create({
|
||||||
|
model: 'claude-sonnet-4-20250514',
|
||||||
|
max_tokens: 2000,
|
||||||
|
messages: [{ role: 'user', content: prompt }],
|
||||||
|
});
|
||||||
|
|
||||||
|
const content = response.content[0];
|
||||||
|
if (content.type !== 'text') {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
|
||||||
|
const jsonMatch = content.text.match(/\[[\s\S]*\]/);
|
||||||
|
if (!jsonMatch) {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
|
||||||
|
return JSON.parse(jsonMatch[0]);
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[ExperienceExtractor] Gap analysis failed:', error);
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,52 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('admins')
|
||||||
|
export class AdminORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ length: 50, unique: true })
|
||||||
|
username: string;
|
||||||
|
|
||||||
|
@Column({ name: 'password_hash', length: 255 })
|
||||||
|
passwordHash: string;
|
||||||
|
|
||||||
|
@Column({ length: 100 })
|
||||||
|
name: string;
|
||||||
|
|
||||||
|
@Column({ length: 255, nullable: true })
|
||||||
|
email: string;
|
||||||
|
|
||||||
|
@Column({ length: 20, nullable: true })
|
||||||
|
phone: string;
|
||||||
|
|
||||||
|
@Column({ length: 20, default: 'OPERATOR' })
|
||||||
|
role: string;
|
||||||
|
|
||||||
|
@Column('jsonb', { default: '[]' })
|
||||||
|
permissions: string[];
|
||||||
|
|
||||||
|
@Column({ length: 500, nullable: true })
|
||||||
|
avatar: string;
|
||||||
|
|
||||||
|
@Column({ name: 'last_login_at', nullable: true })
|
||||||
|
lastLoginAt: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'last_login_ip', length: 50, nullable: true })
|
||||||
|
lastLoginIp?: string;
|
||||||
|
|
||||||
|
@Column({ name: 'is_active', default: true })
|
||||||
|
isActive: boolean;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,61 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('conversations')
|
||||||
|
export class ConversationORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'user_id', nullable: true })
|
||||||
|
userId: string;
|
||||||
|
|
||||||
|
@Column({ length: 20, default: 'ACTIVE' })
|
||||||
|
status: string;
|
||||||
|
|
||||||
|
@Column({ length: 255, nullable: true })
|
||||||
|
title: string;
|
||||||
|
|
||||||
|
@Column('text', { nullable: true })
|
||||||
|
summary: string;
|
||||||
|
|
||||||
|
@Column({ length: 50, nullable: true })
|
||||||
|
category: string;
|
||||||
|
|
||||||
|
@Column({ name: 'message_count', default: 0 })
|
||||||
|
messageCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'user_message_count', default: 0 })
|
||||||
|
userMessageCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'assistant_message_count', default: 0 })
|
||||||
|
assistantMessageCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'total_input_tokens', default: 0 })
|
||||||
|
totalInputTokens: number;
|
||||||
|
|
||||||
|
@Column({ name: 'total_output_tokens', default: 0 })
|
||||||
|
totalOutputTokens: number;
|
||||||
|
|
||||||
|
@Column({ type: 'smallint', nullable: true })
|
||||||
|
rating: number;
|
||||||
|
|
||||||
|
@Column('text', { nullable: true })
|
||||||
|
feedback: string;
|
||||||
|
|
||||||
|
@Column({ name: 'has_converted', default: false })
|
||||||
|
hasConverted: boolean;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'ended_at', nullable: true })
|
||||||
|
endedAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,33 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('messages')
|
||||||
|
export class MessageORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'conversation_id' })
|
||||||
|
conversationId: string;
|
||||||
|
|
||||||
|
@Column({ length: 20 })
|
||||||
|
role: string;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column({ name: 'input_tokens', default: 0 })
|
||||||
|
inputTokens: number;
|
||||||
|
|
||||||
|
@Column({ name: 'output_tokens', default: 0 })
|
||||||
|
outputTokens: number;
|
||||||
|
|
||||||
|
@Column('jsonb', { nullable: true })
|
||||||
|
metadata: Record<string, unknown>;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,61 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('system_experiences')
|
||||||
|
export class SystemExperienceORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'experience_type', length: 30 })
|
||||||
|
experienceType: string;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column({ default: 50 })
|
||||||
|
confidence: number;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
scenario: string;
|
||||||
|
|
||||||
|
@Column({ name: 'related_category', length: 50, nullable: true })
|
||||||
|
relatedCategory: string;
|
||||||
|
|
||||||
|
@Column('uuid', { name: 'source_conversation_ids', array: true, default: '{}' })
|
||||||
|
sourceConversationIds: string[];
|
||||||
|
|
||||||
|
@Column({ name: 'verification_status', length: 20, default: 'PENDING' })
|
||||||
|
verificationStatus: string;
|
||||||
|
|
||||||
|
@Column({ name: 'verified_by', nullable: true })
|
||||||
|
verifiedBy: string;
|
||||||
|
|
||||||
|
@Column({ name: 'verified_at', nullable: true })
|
||||||
|
verifiedAt: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'usage_count', default: 0 })
|
||||||
|
usageCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'positive_count', default: 0 })
|
||||||
|
positiveCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'negative_count', default: 0 })
|
||||||
|
negativeCount: number;
|
||||||
|
|
||||||
|
@Column('float', { array: true, nullable: true })
|
||||||
|
embedding: number[];
|
||||||
|
|
||||||
|
@Column({ name: 'is_active', default: false })
|
||||||
|
isActive: boolean;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,40 @@
|
||||||
|
import { NestFactory } from '@nestjs/core';
|
||||||
|
import { AppModule } from './app.module';
|
||||||
|
|
||||||
|
async function bootstrap() {
|
||||||
|
const app = await NestFactory.create(AppModule);
|
||||||
|
|
||||||
|
// 设置全局前缀
|
||||||
|
app.setGlobalPrefix('api/v1');
|
||||||
|
|
||||||
|
// 启用CORS
|
||||||
|
app.enableCors({
|
||||||
|
origin: process.env.CORS_ORIGIN || '*',
|
||||||
|
credentials: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
const port = process.env.PORT || 3005;
|
||||||
|
await app.listen(port);
|
||||||
|
|
||||||
|
console.log(`
|
||||||
|
╔══════════════════════════════════════════════════════════════╗
|
||||||
|
║ ║
|
||||||
|
║ 🧬 iConsulting Evolution Service ║
|
||||||
|
║ ║
|
||||||
|
║ Server running at: http://localhost:${port} ║
|
||||||
|
║ API prefix: /api/v1 ║
|
||||||
|
║ ║
|
||||||
|
║ Endpoints: ║
|
||||||
|
║ - POST /api/v1/admin/login Admin login ║
|
||||||
|
║ - GET /api/v1/admin/me Current admin ║
|
||||||
|
║ - POST /api/v1/admin/admins Create admin ║
|
||||||
|
║ - GET /api/v1/admin/admins List admins ║
|
||||||
|
║ - POST /api/v1/evolution/run Run evolution ║
|
||||||
|
║ - GET /api/v1/evolution/statistics Evolution stats ║
|
||||||
|
║ - GET /api/v1/evolution/health System health ║
|
||||||
|
║ ║
|
||||||
|
╚══════════════════════════════════════════════════════════════╝
|
||||||
|
`);
|
||||||
|
}
|
||||||
|
|
||||||
|
bootstrap();
|
||||||
|
|
@ -0,0 +1,17 @@
|
||||||
|
{
|
||||||
|
"extends": "../../../tsconfig.base.json",
|
||||||
|
"compilerOptions": {
|
||||||
|
"module": "CommonJS",
|
||||||
|
"moduleResolution": "Node",
|
||||||
|
"outDir": "./dist",
|
||||||
|
"rootDir": "./src",
|
||||||
|
"declaration": true,
|
||||||
|
"emitDecoratorMetadata": true,
|
||||||
|
"experimentalDecorators": true,
|
||||||
|
"strictPropertyInitialization": false,
|
||||||
|
"noUnusedLocals": false,
|
||||||
|
"noUnusedParameters": false
|
||||||
|
},
|
||||||
|
"include": ["src/**/*"],
|
||||||
|
"exclude": ["node_modules", "dist", "test"]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,45 @@
|
||||||
|
# ===========================================
|
||||||
|
# iConsulting Knowledge Service Dockerfile
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
FROM node:20-alpine AS builder
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
|
||||||
|
COPY pnpm-workspace.yaml package.json pnpm-lock.yaml ./
|
||||||
|
COPY packages/shared/package.json ./packages/shared/
|
||||||
|
COPY packages/services/knowledge-service/package.json ./packages/services/knowledge-service/
|
||||||
|
|
||||||
|
RUN pnpm install --frozen-lockfile
|
||||||
|
|
||||||
|
COPY packages/shared ./packages/shared
|
||||||
|
COPY packages/services/knowledge-service ./packages/services/knowledge-service
|
||||||
|
COPY tsconfig.base.json ./
|
||||||
|
|
||||||
|
RUN pnpm --filter @iconsulting/shared build
|
||||||
|
RUN pnpm --filter @iconsulting/knowledge-service build
|
||||||
|
|
||||||
|
FROM node:20-alpine AS runner
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
RUN addgroup --system --gid 1001 nodejs && \
|
||||||
|
adduser --system --uid 1001 nestjs
|
||||||
|
|
||||||
|
COPY --from=builder /app/packages/services/knowledge-service/dist ./dist
|
||||||
|
COPY --from=builder /app/packages/services/knowledge-service/package.json ./
|
||||||
|
COPY --from=builder /app/node_modules ./node_modules
|
||||||
|
|
||||||
|
ENV NODE_ENV=production
|
||||||
|
ENV PORT=3003
|
||||||
|
|
||||||
|
USER nestjs
|
||||||
|
EXPOSE 3003
|
||||||
|
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||||
|
CMD wget --no-verbose --tries=1 --spider http://localhost:3003/health || exit 1
|
||||||
|
|
||||||
|
CMD ["node", "dist/main.js"]
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
{
|
||||||
|
"$schema": "https://json.schemastore.org/nest-cli",
|
||||||
|
"collection": "@nestjs/schematics",
|
||||||
|
"sourceRoot": "src",
|
||||||
|
"compilerOptions": {
|
||||||
|
"deleteOutDir": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,47 @@
|
||||||
|
{
|
||||||
|
"name": "@iconsulting/knowledge-service",
|
||||||
|
"version": "0.1.0",
|
||||||
|
"description": "知识服务 - RAG检索增强生成 + Neo4j知识图谱",
|
||||||
|
"main": "dist/main.js",
|
||||||
|
"scripts": {
|
||||||
|
"build": "nest build",
|
||||||
|
"start": "nest start",
|
||||||
|
"start:dev": "nest start --watch",
|
||||||
|
"start:debug": "nest start --debug --watch",
|
||||||
|
"start:prod": "node dist/main",
|
||||||
|
"lint": "eslint \"{src,apps,libs,test}/**/*.ts\" --fix",
|
||||||
|
"test": "jest",
|
||||||
|
"test:watch": "jest --watch",
|
||||||
|
"test:cov": "jest --coverage"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@anthropic-ai/sdk": "^0.52.0",
|
||||||
|
"@nestjs/common": "^10.0.0",
|
||||||
|
"@nestjs/config": "^3.2.0",
|
||||||
|
"@nestjs/core": "^10.0.0",
|
||||||
|
"@nestjs/platform-express": "^10.0.0",
|
||||||
|
"@nestjs/typeorm": "^10.0.0",
|
||||||
|
"neo4j-driver": "^5.17.0",
|
||||||
|
"openai": "^4.28.0",
|
||||||
|
"pg": "^8.11.3",
|
||||||
|
"pgvector": "^0.1.8",
|
||||||
|
"reflect-metadata": "^0.2.1",
|
||||||
|
"rxjs": "^7.8.1",
|
||||||
|
"typeorm": "^0.3.19",
|
||||||
|
"uuid": "^9.0.1"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@nestjs/cli": "^10.0.0",
|
||||||
|
"@nestjs/schematics": "^10.0.0",
|
||||||
|
"@nestjs/testing": "^10.0.0",
|
||||||
|
"@types/express": "^4.17.21",
|
||||||
|
"@types/jest": "^29.5.11",
|
||||||
|
"@types/node": "^20.10.6",
|
||||||
|
"@types/uuid": "^9.0.7",
|
||||||
|
"jest": "^29.7.0",
|
||||||
|
"ts-jest": "^29.1.1",
|
||||||
|
"ts-loader": "^9.5.1",
|
||||||
|
"ts-node": "^10.9.2",
|
||||||
|
"typescript": "^5.3.3"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { ConfigModule, ConfigService } from '@nestjs/config';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { KnowledgeModule } from './knowledge/knowledge.module';
|
||||||
|
import { MemoryModule } from './memory/memory.module';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
// 配置模块
|
||||||
|
ConfigModule.forRoot({
|
||||||
|
isGlobal: true,
|
||||||
|
envFilePath: ['.env.local', '.env'],
|
||||||
|
}),
|
||||||
|
|
||||||
|
// 数据库连接
|
||||||
|
TypeOrmModule.forRootAsync({
|
||||||
|
imports: [ConfigModule],
|
||||||
|
inject: [ConfigService],
|
||||||
|
useFactory: (config: ConfigService) => ({
|
||||||
|
type: 'postgres',
|
||||||
|
host: config.get('DB_HOST', 'localhost'),
|
||||||
|
port: config.get('DB_PORT', 5432),
|
||||||
|
username: config.get('DB_USER', 'iconsulting'),
|
||||||
|
password: config.get('DB_PASSWORD', 'iconsulting_dev'),
|
||||||
|
database: config.get('DB_NAME', 'iconsulting'),
|
||||||
|
autoLoadEntities: true,
|
||||||
|
synchronize: config.get('NODE_ENV') !== 'production',
|
||||||
|
logging: config.get('NODE_ENV') === 'development',
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
|
||||||
|
// 功能模块
|
||||||
|
KnowledgeModule,
|
||||||
|
MemoryModule,
|
||||||
|
],
|
||||||
|
})
|
||||||
|
export class AppModule {}
|
||||||
|
|
@ -0,0 +1,311 @@
|
||||||
|
import { Injectable } from '@nestjs/common';
|
||||||
|
import {
|
||||||
|
KnowledgeChunkEntity,
|
||||||
|
ChunkType,
|
||||||
|
ChunkMetadata,
|
||||||
|
} from '../../domain/entities/knowledge-chunk.entity';
|
||||||
|
import { KnowledgeArticleEntity } from '../../domain/entities/knowledge-article.entity';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 文本分块策略配置
|
||||||
|
*/
|
||||||
|
export interface ChunkingConfig {
|
||||||
|
/** 最大块大小（字符数） */
|
||||||
|
maxChunkSize: number;
|
||||||
|
/** 块重叠大小 */
|
||||||
|
overlapSize: number;
|
||||||
|
/** 是否按语义边界分割 */
|
||||||
|
semanticSplit: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
const DEFAULT_CONFIG: ChunkingConfig = {
|
||||||
|
maxChunkSize: 500,
|
||||||
|
overlapSize: 50,
|
||||||
|
semanticSplit: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 文本分块服务
|
||||||
|
* 将长文本智能分割为适合检索的小块
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class ChunkingService {
|
||||||
|
/**
|
||||||
|
* 将文章内容分割为块
|
||||||
|
*/
|
||||||
|
chunkArticle(
|
||||||
|
article: KnowledgeArticleEntity,
|
||||||
|
config: ChunkingConfig = DEFAULT_CONFIG,
|
||||||
|
): KnowledgeChunkEntity[] {
|
||||||
|
const content = article.content;
|
||||||
|
|
||||||
|
// 检测内容格式
|
||||||
|
if (this.isMarkdown(content)) {
|
||||||
|
return this.chunkMarkdown(article.id, content, config);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 纯文本分块
|
||||||
|
return this.chunkPlainText(article.id, content, config);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检测是否为Markdown格式
|
||||||
|
*/
|
||||||
|
private isMarkdown(content: string): boolean {
|
||||||
|
const markdownPatterns = [
|
||||||
|
/^#+\s/m, // 标题
|
||||||
|
/\*\*.+\*\*/, // 加粗
|
||||||
|
/\[.+\]\(.+\)/, // 链接
|
||||||
|
/^-\s/m, // 列表
|
||||||
|
/^\d+\.\s/m, // 有序列表
|
||||||
|
/```/, // 代码块
|
||||||
|
];
|
||||||
|
|
||||||
|
return markdownPatterns.some(pattern => pattern.test(content));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Markdown内容分块
|
||||||
|
*/
|
||||||
|
private chunkMarkdown(
|
||||||
|
articleId: string,
|
||||||
|
content: string,
|
||||||
|
config: ChunkingConfig,
|
||||||
|
): KnowledgeChunkEntity[] {
|
||||||
|
const chunks: KnowledgeChunkEntity[] = [];
|
||||||
|
const sections = this.splitByHeadings(content);
|
||||||
|
|
||||||
|
let chunkIndex = 0;
|
||||||
|
let currentSectionTitle = '';
|
||||||
|
|
||||||
|
for (const section of sections) {
|
||||||
|
// 更新当前章节标题
|
||||||
|
const headingMatch = section.match(/^(#+)\s+(.+)$/m);
|
||||||
|
if (headingMatch) {
|
||||||
|
currentSectionTitle = headingMatch[2];
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果章节内容过长，进一步分割
|
||||||
|
if (section.length > config.maxChunkSize) {
|
||||||
|
const subChunks = this.splitLongSection(section, config);
|
||||||
|
|
||||||
|
for (const subContent of subChunks) {
|
||||||
|
const chunkType = this.detectChunkType(subContent);
|
||||||
|
const metadata: ChunkMetadata = {
|
||||||
|
sectionTitle: currentSectionTitle,
|
||||||
|
headingLevel: headingMatch ? headingMatch[1].length : undefined,
|
||||||
|
};
|
||||||
|
|
||||||
|
const chunk = KnowledgeChunkEntity.create({
|
||||||
|
articleId,
|
||||||
|
content: subContent.trim(),
|
||||||
|
chunkIndex,
|
||||||
|
chunkType,
|
||||||
|
metadata,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 设置前后链接
|
||||||
|
if (chunks.length > 0) {
|
||||||
|
metadata.prevChunkId = chunks[chunks.length - 1].id;
|
||||||
|
chunks[chunks.length - 1].metadata.nextChunkId = chunk.id;
|
||||||
|
}
|
||||||
|
|
||||||
|
chunks.push(chunk);
|
||||||
|
chunkIndex++;
|
||||||
|
}
|
||||||
|
} else if (section.trim()) {
|
||||||
|
const chunkType = this.detectChunkType(section);
|
||||||
|
const metadata: ChunkMetadata = {
|
||||||
|
sectionTitle: currentSectionTitle,
|
||||||
|
headingLevel: headingMatch ? headingMatch[1].length : undefined,
|
||||||
|
};
|
||||||
|
|
||||||
|
const chunk = KnowledgeChunkEntity.create({
|
||||||
|
articleId,
|
||||||
|
content: section.trim(),
|
||||||
|
chunkIndex,
|
||||||
|
chunkType,
|
||||||
|
metadata,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (chunks.length > 0) {
|
||||||
|
metadata.prevChunkId = chunks[chunks.length - 1].id;
|
||||||
|
chunks[chunks.length - 1].metadata.nextChunkId = chunk.id;
|
||||||
|
}
|
||||||
|
|
||||||
|
chunks.push(chunk);
|
||||||
|
chunkIndex++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return chunks;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 按标题分割Markdown
|
||||||
|
*/
|
||||||
|
private splitByHeadings(content: string): string[] {
|
||||||
|
// 在标题前分割，但保留标题
|
||||||
|
const parts = content.split(/(?=^#+\s)/m);
|
||||||
|
return parts.filter(part => part.trim());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 分割过长的章节
|
||||||
|
*/
|
||||||
|
private splitLongSection(content: string, config: ChunkingConfig): string[] {
|
||||||
|
const chunks: string[] = [];
|
||||||
|
|
||||||
|
if (config.semanticSplit) {
|
||||||
|
// 优先按段落分割
|
||||||
|
const paragraphs = content.split(/\n\n+/);
|
||||||
|
let currentChunk = '';
|
||||||
|
|
||||||
|
for (const para of paragraphs) {
|
||||||
|
if (currentChunk.length + para.length > config.maxChunkSize) {
|
||||||
|
if (currentChunk) {
|
||||||
|
chunks.push(currentChunk);
|
||||||
|
}
|
||||||
|
// 如果单个段落就超过限制，按句子分割
|
||||||
|
if (para.length > config.maxChunkSize) {
|
||||||
|
chunks.push(...this.splitBySentences(para, config));
|
||||||
|
} else {
|
||||||
|
currentChunk = para;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
currentChunk = currentChunk ? `${currentChunk}\n\n${para}` : para;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (currentChunk) {
|
||||||
|
chunks.push(currentChunk);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 简单按字符数分割（带重叠）
|
||||||
|
chunks.push(...this.splitBySize(content, config));
|
||||||
|
}
|
||||||
|
|
||||||
|
return chunks;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 按句子分割
|
||||||
|
*/
|
||||||
|
private splitBySentences(content: string, config: ChunkingConfig): string[] {
|
||||||
|
// 中英文句子结束符
|
||||||
|
const sentenceEndings = /([。！？.!?])/g;
|
||||||
|
const sentences = content.split(sentenceEndings);
|
||||||
|
|
||||||
|
const chunks: string[] = [];
|
||||||
|
let currentChunk = '';
|
||||||
|
|
||||||
|
for (let i = 0; i < sentences.length; i += 2) {
|
||||||
|
const sentence = sentences[i] + (sentences[i + 1] || '');
|
||||||
|
|
||||||
|
if (currentChunk.length + sentence.length > config.maxChunkSize) {
|
||||||
|
if (currentChunk) {
|
||||||
|
chunks.push(currentChunk);
|
||||||
|
}
|
||||||
|
currentChunk = sentence;
|
||||||
|
} else {
|
||||||
|
currentChunk += sentence;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (currentChunk) {
|
||||||
|
chunks.push(currentChunk);
|
||||||
|
}
|
||||||
|
|
||||||
|
return chunks;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 按大小分割（带重叠）
|
||||||
|
*/
|
||||||
|
private splitBySize(content: string, config: ChunkingConfig): string[] {
|
||||||
|
const chunks: string[] = [];
|
||||||
|
let start = 0;
|
||||||
|
|
||||||
|
while (start < content.length) {
|
||||||
|
const end = Math.min(start + config.maxChunkSize, content.length);
|
||||||
|
chunks.push(content.slice(start, end));
|
||||||
|
start = end - config.overlapSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
return chunks;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 纯文本分块
|
||||||
|
*/
|
||||||
|
private chunkPlainText(
|
||||||
|
articleId: string,
|
||||||
|
content: string,
|
||||||
|
config: ChunkingConfig,
|
||||||
|
): KnowledgeChunkEntity[] {
|
||||||
|
const textChunks = config.semanticSplit
|
||||||
|
? this.splitLongSection(content, config)
|
||||||
|
: this.splitBySize(content, config);
|
||||||
|
|
||||||
|
return textChunks.map((text, index) =>
|
||||||
|
KnowledgeChunkEntity.create({
|
||||||
|
articleId,
|
||||||
|
content: text.trim(),
|
||||||
|
chunkIndex: index,
|
||||||
|
chunkType: ChunkType.PARAGRAPH,
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检测块类型
|
||||||
|
*/
|
||||||
|
private detectChunkType(content: string): ChunkType {
|
||||||
|
// 标题
|
||||||
|
if (/^#+\s/.test(content)) {
|
||||||
|
return ChunkType.TITLE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 代码块
|
||||||
|
if (/```[\s\S]*```/.test(content)) {
|
||||||
|
return ChunkType.CODE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 列表
|
||||||
|
if (/^[-*]\s/m.test(content) || /^\d+\.\s/m.test(content)) {
|
||||||
|
return ChunkType.LIST;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 表格
|
||||||
|
if (/\|.+\|/.test(content) && /\|-+\|/.test(content)) {
|
||||||
|
return ChunkType.TABLE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// FAQ（问答格式）
|
||||||
|
if (/^[问Q][:：]/.test(content) || /^[答A][:：]/.test(content)) {
|
||||||
|
return ChunkType.FAQ;
|
||||||
|
}
|
||||||
|
|
||||||
|
return ChunkType.PARAGRAPH;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 为FAQ内容创建专门的块
|
||||||
|
*/
|
||||||
|
createFAQChunks(
|
||||||
|
articleId: string,
|
||||||
|
faqs: Array<{ question: string; answer: string }>,
|
||||||
|
): KnowledgeChunkEntity[] {
|
||||||
|
return faqs.map((faq, index) =>
|
||||||
|
KnowledgeChunkEntity.create({
|
||||||
|
articleId,
|
||||||
|
content: `问：${faq.question}\n答：${faq.answer}`,
|
||||||
|
chunkIndex: index,
|
||||||
|
chunkType: ChunkType.FAQ,
|
||||||
|
metadata: {
|
||||||
|
hasKeywords: true,
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,279 @@
|
||||||
|
import { Injectable, Inject } from '@nestjs/common';
|
||||||
|
import { EmbeddingService } from '../../infrastructure/embedding/embedding.service';
|
||||||
|
import {
|
||||||
|
IKnowledgeRepository,
|
||||||
|
KNOWLEDGE_REPOSITORY,
|
||||||
|
} from '../../domain/repositories/knowledge.repository.interface';
|
||||||
|
import {
|
||||||
|
IUserMemoryRepository,
|
||||||
|
ISystemExperienceRepository,
|
||||||
|
USER_MEMORY_REPOSITORY,
|
||||||
|
SYSTEM_EXPERIENCE_REPOSITORY,
|
||||||
|
} from '../../domain/repositories/memory.repository.interface';
|
||||||
|
import { KnowledgeArticleEntity } from '../../domain/entities/knowledge-article.entity';
|
||||||
|
import { KnowledgeChunkEntity } from '../../domain/entities/knowledge-chunk.entity';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* RAG检索结果
|
||||||
|
*/
|
||||||
|
export interface RAGResult {
|
||||||
|
/** 检索到的知识内容 */
|
||||||
|
content: string;
|
||||||
|
/** 来源（用于引用） */
|
||||||
|
sources: Array<{
|
||||||
|
articleId: string;
|
||||||
|
title: string;
|
||||||
|
similarity: number;
|
||||||
|
}>;
|
||||||
|
/** 用户相关记忆 */
|
||||||
|
userMemories?: string[];
|
||||||
|
/** 系统经验 */
|
||||||
|
systemExperiences?: string[];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* RAG服务 - 检索增强生成
|
||||||
|
* 负责从知识库中检索相关内容，增强AI回答
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class RAGService {
|
||||||
|
constructor(
|
||||||
|
private embeddingService: EmbeddingService,
|
||||||
|
@Inject(KNOWLEDGE_REPOSITORY)
|
||||||
|
private knowledgeRepo: IKnowledgeRepository,
|
||||||
|
@Inject(USER_MEMORY_REPOSITORY)
|
||||||
|
private memoryRepo: IUserMemoryRepository,
|
||||||
|
@Inject(SYSTEM_EXPERIENCE_REPOSITORY)
|
||||||
|
private experienceRepo: ISystemExperienceRepository,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检索相关知识
|
||||||
|
*/
|
||||||
|
async retrieve(params: {
|
||||||
|
query: string;
|
||||||
|
userId?: string;
|
||||||
|
category?: string;
|
||||||
|
includeMemories?: boolean;
|
||||||
|
includeExperiences?: boolean;
|
||||||
|
topK?: number;
|
||||||
|
}): Promise<RAGResult> {
|
||||||
|
const {
|
||||||
|
query,
|
||||||
|
userId,
|
||||||
|
category,
|
||||||
|
includeMemories = true,
|
||||||
|
includeExperiences = true,
|
||||||
|
topK = 5,
|
||||||
|
} = params;
|
||||||
|
|
||||||
|
// 1. 生成查询向量
|
||||||
|
const queryEmbedding = await this.embeddingService.getEmbedding(query);
|
||||||
|
|
||||||
|
// 2. 并行检索
|
||||||
|
const [chunkResults, memoryResults, experienceResults] = await Promise.all([
|
||||||
|
// 检索知识块
|
||||||
|
this.knowledgeRepo.searchChunksByVector(queryEmbedding, {
|
||||||
|
category,
|
||||||
|
limit: topK,
|
||||||
|
minSimilarity: 0.6,
|
||||||
|
}),
|
||||||
|
// 检索用户记忆
|
||||||
|
includeMemories && userId
|
||||||
|
? this.memoryRepo.searchByVector(userId, queryEmbedding, {
|
||||||
|
limit: 3,
|
||||||
|
minSimilarity: 0.7,
|
||||||
|
})
|
||||||
|
: Promise.resolve([]),
|
||||||
|
// 检索系统经验
|
||||||
|
includeExperiences
|
||||||
|
? this.experienceRepo.searchByVector(queryEmbedding, {
|
||||||
|
activeOnly: true,
|
||||||
|
limit: 3,
|
||||||
|
minSimilarity: 0.75,
|
||||||
|
})
|
||||||
|
: Promise.resolve([]),
|
||||||
|
]);
|
||||||
|
|
||||||
|
// 3. 获取完整文章信息（用于引用）
|
||||||
|
const articleIds = [...new Set(chunkResults.map(r => r.chunk.articleId))];
|
||||||
|
const articles = await Promise.all(
|
||||||
|
articleIds.map(id => this.knowledgeRepo.findArticleById(id)),
|
||||||
|
);
|
||||||
|
const articleMap = new Map(
|
||||||
|
articles.filter(Boolean).map(a => [a!.id, a!]),
|
||||||
|
);
|
||||||
|
|
||||||
|
// 4. 组装结果
|
||||||
|
const content = this.formatRetrievedContent(chunkResults.map(r => r.chunk), articleMap);
|
||||||
|
|
||||||
|
const sources = chunkResults.map(r => {
|
||||||
|
const article = articleMap.get(r.chunk.articleId);
|
||||||
|
return {
|
||||||
|
articleId: r.chunk.articleId,
|
||||||
|
title: article?.title || 'Unknown',
|
||||||
|
similarity: r.similarity,
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
|
const userMemories = memoryResults.map(r => r.memory.content);
|
||||||
|
const systemExperiences = experienceResults.map(r => r.experience.content);
|
||||||
|
|
||||||
|
// 5. 更新引用计数
|
||||||
|
this.updateCitationCounts(articleIds);
|
||||||
|
|
||||||
|
return {
|
||||||
|
content,
|
||||||
|
sources,
|
||||||
|
userMemories: userMemories.length > 0 ? userMemories : undefined,
|
||||||
|
systemExperiences: systemExperiences.length > 0 ? systemExperiences : undefined,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检索并格式化为提示词上下文
|
||||||
|
*/
|
||||||
|
async retrieveForPrompt(params: {
|
||||||
|
query: string;
|
||||||
|
userId?: string;
|
||||||
|
category?: string;
|
||||||
|
}): Promise<string> {
|
||||||
|
const result = await this.retrieve(params);
|
||||||
|
|
||||||
|
let context = '';
|
||||||
|
|
||||||
|
// 知识库内容
|
||||||
|
if (result.content) {
|
||||||
|
context += `## 相关知识\n${result.content}\n\n`;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 用户记忆
|
||||||
|
if (result.userMemories?.length) {
|
||||||
|
context += `## 用户背景信息\n`;
|
||||||
|
result.userMemories.forEach((m, i) => {
|
||||||
|
context += `${i + 1}. ${m}\n`;
|
||||||
|
});
|
||||||
|
context += '\n';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 系统经验
|
||||||
|
if (result.systemExperiences?.length) {
|
||||||
|
context += `## 参考经验\n`;
|
||||||
|
result.systemExperiences.forEach((e, i) => {
|
||||||
|
context += `${i + 1}. ${e}\n`;
|
||||||
|
});
|
||||||
|
context += '\n';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 来源引用
|
||||||
|
if (result.sources.length > 0) {
|
||||||
|
context += `## 来源\n`;
|
||||||
|
result.sources.forEach((s, i) => {
|
||||||
|
context += `[${i + 1}] ${s.title}\n`;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return context;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 格式化检索到的内容
|
||||||
|
*/
|
||||||
|
private formatRetrievedContent(
|
||||||
|
chunks: KnowledgeChunkEntity[],
|
||||||
|
articleMap: Map<string, KnowledgeArticleEntity>,
|
||||||
|
): string {
|
||||||
|
if (chunks.length === 0) {
|
||||||
|
return '';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按文章分组
|
||||||
|
const groupedByArticle = new Map<string, KnowledgeChunkEntity[]>();
|
||||||
|
chunks.forEach(chunk => {
|
||||||
|
const existing = groupedByArticle.get(chunk.articleId) || [];
|
||||||
|
existing.push(chunk);
|
||||||
|
groupedByArticle.set(chunk.articleId, existing);
|
||||||
|
});
|
||||||
|
|
||||||
|
let content = '';
|
||||||
|
let articleIndex = 1;
|
||||||
|
|
||||||
|
groupedByArticle.forEach((articleChunks, articleId) => {
|
||||||
|
const article = articleMap.get(articleId);
|
||||||
|
if (!article) return;
|
||||||
|
|
||||||
|
content += `### [${articleIndex}] ${article.title}\n`;
|
||||||
|
|
||||||
|
// 按块序号排序
|
||||||
|
articleChunks.sort((a, b) => a.chunkIndex - b.chunkIndex);
|
||||||
|
|
||||||
|
articleChunks.forEach(chunk => {
|
||||||
|
content += `${chunk.content}\n\n`;
|
||||||
|
});
|
||||||
|
|
||||||
|
articleIndex++;
|
||||||
|
});
|
||||||
|
|
||||||
|
return content;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 异步更新引用计数
|
||||||
|
*/
|
||||||
|
private async updateCitationCounts(articleIds: string[]): Promise<void> {
|
||||||
|
// 异步执行，不阻塞检索
|
||||||
|
setImmediate(async () => {
|
||||||
|
for (const id of articleIds) {
|
||||||
|
try {
|
||||||
|
const article = await this.knowledgeRepo.findArticleById(id);
|
||||||
|
if (article) {
|
||||||
|
article.incrementCitation();
|
||||||
|
await this.knowledgeRepo.updateArticle(article);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error(`Failed to update citation count for article ${id}:`, error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查是否为离题问题
|
||||||
|
*/
|
||||||
|
async checkOffTopic(query: string): Promise<{
|
||||||
|
isOffTopic: boolean;
|
||||||
|
confidence: number;
|
||||||
|
reason?: string;
|
||||||
|
}> {
|
||||||
|
// 使用向量相似度检查是否与知识库相关
|
||||||
|
const queryEmbedding = await this.embeddingService.getEmbedding(query);
|
||||||
|
|
||||||
|
const results = await this.knowledgeRepo.searchChunksByVector(queryEmbedding, {
|
||||||
|
limit: 1,
|
||||||
|
minSimilarity: 0.3, // 使用较低阈值
|
||||||
|
});
|
||||||
|
|
||||||
|
if (results.length === 0) {
|
||||||
|
return {
|
||||||
|
isOffTopic: true,
|
||||||
|
confidence: 0.8,
|
||||||
|
reason: '问题与香港移民主题无关',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const maxSimilarity = results[0].similarity;
|
||||||
|
|
||||||
|
if (maxSimilarity < 0.5) {
|
||||||
|
return {
|
||||||
|
isOffTopic: true,
|
||||||
|
confidence: 0.9 - maxSimilarity,
|
||||||
|
reason: '问题与香港移民主题相关性较低',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
isOffTopic: false,
|
||||||
|
confidence: maxSimilarity,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,193 @@
|
||||||
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 知识文章实体 - 存储移民相关的知识内容
|
||||||
|
*/
|
||||||
|
export class KnowledgeArticleEntity {
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
/** 文章标题 */
|
||||||
|
title: string;
|
||||||
|
|
||||||
|
/** 文章内容（纯文本或Markdown） */
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
/** 内容摘要（用于预览） */
|
||||||
|
summary: string;
|
||||||
|
|
||||||
|
/** 移民类别: QMAS, GEP, IANG, TTPS, CIES, TechTAS */
|
||||||
|
category: string;
|
||||||
|
|
||||||
|
/** 内容标签 */
|
||||||
|
tags: string[];
|
||||||
|
|
||||||
|
/** 来源: MANUAL(手动添加), CRAWL(爬取), EXTRACT(对话提取) */
|
||||||
|
source: KnowledgeSource;
|
||||||
|
|
||||||
|
/** 来源URL（如果是爬取的） */
|
||||||
|
sourceUrl?: string;
|
||||||
|
|
||||||
|
/** 内容向量（用于语义搜索） */
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
/** 是否已发布 */
|
||||||
|
isPublished: boolean;
|
||||||
|
|
||||||
|
/** 引用次数（被对话引用） */
|
||||||
|
citationCount: number;
|
||||||
|
|
||||||
|
/** 点赞数（用户反馈有用） */
|
||||||
|
helpfulCount: number;
|
||||||
|
|
||||||
|
/** 点踩数（用户反馈无用） */
|
||||||
|
unhelpfulCount: number;
|
||||||
|
|
||||||
|
/** 质量评分 0-100 */
|
||||||
|
qualityScore: number;
|
||||||
|
|
||||||
|
/** 创建者ID */
|
||||||
|
createdBy?: string;
|
||||||
|
|
||||||
|
/** 最后更新者ID */
|
||||||
|
updatedBy?: string;
|
||||||
|
|
||||||
|
createdAt: Date;
|
||||||
|
updatedAt: Date;
|
||||||
|
|
||||||
|
private constructor() {}
|
||||||
|
|
||||||
|
static create(params: {
|
||||||
|
title: string;
|
||||||
|
content: string;
|
||||||
|
category: string;
|
||||||
|
tags?: string[];
|
||||||
|
source: KnowledgeSource;
|
||||||
|
sourceUrl?: string;
|
||||||
|
createdBy?: string;
|
||||||
|
}): KnowledgeArticleEntity {
|
||||||
|
const article = new KnowledgeArticleEntity();
|
||||||
|
article.id = uuidv4();
|
||||||
|
article.title = params.title;
|
||||||
|
article.content = params.content;
|
||||||
|
article.summary = KnowledgeArticleEntity.generateSummary(params.content);
|
||||||
|
article.category = params.category;
|
||||||
|
article.tags = params.tags || [];
|
||||||
|
article.source = params.source;
|
||||||
|
article.sourceUrl = params.sourceUrl;
|
||||||
|
article.isPublished = false;
|
||||||
|
article.citationCount = 0;
|
||||||
|
article.helpfulCount = 0;
|
||||||
|
article.unhelpfulCount = 0;
|
||||||
|
article.qualityScore = 50; // 默认中等质量
|
||||||
|
article.createdBy = params.createdBy;
|
||||||
|
article.createdAt = new Date();
|
||||||
|
article.updatedAt = new Date();
|
||||||
|
return article;
|
||||||
|
}
|
||||||
|
|
||||||
|
static fromPersistence(data: Partial<KnowledgeArticleEntity>): KnowledgeArticleEntity {
|
||||||
|
const article = new KnowledgeArticleEntity();
|
||||||
|
Object.assign(article, data);
|
||||||
|
return article;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成内容摘要
|
||||||
|
*/
|
||||||
|
private static generateSummary(content: string, maxLength = 200): string {
|
||||||
|
const plainText = content
|
||||||
|
.replace(/#+\s/g, '') // 移除Markdown标题
|
||||||
|
.replace(/\*\*/g, '') // 移除加粗
|
||||||
|
.replace(/\n+/g, ' ') // 换行转空格
|
||||||
|
.trim();
|
||||||
|
|
||||||
|
if (plainText.length <= maxLength) {
|
||||||
|
return plainText;
|
||||||
|
}
|
||||||
|
return plainText.substring(0, maxLength) + '...';
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新内容
|
||||||
|
*/
|
||||||
|
updateContent(title: string, content: string, updatedBy?: string): void {
|
||||||
|
this.title = title;
|
||||||
|
this.content = content;
|
||||||
|
this.summary = KnowledgeArticleEntity.generateSummary(content);
|
||||||
|
this.updatedBy = updatedBy;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
// 内容更新后需要重新生成向量
|
||||||
|
this.embedding = undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 设置内容向量
|
||||||
|
*/
|
||||||
|
setEmbedding(embedding: number[]): void {
|
||||||
|
this.embedding = embedding;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 发布文章
|
||||||
|
*/
|
||||||
|
publish(): void {
|
||||||
|
this.isPublished = true;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 取消发布
|
||||||
|
*/
|
||||||
|
unpublish(): void {
|
||||||
|
this.isPublished = false;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 增加引用次数
|
||||||
|
*/
|
||||||
|
incrementCitation(): void {
|
||||||
|
this.citationCount++;
|
||||||
|
this.recalculateQualityScore();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录用户反馈
|
||||||
|
*/
|
||||||
|
recordFeedback(helpful: boolean): void {
|
||||||
|
if (helpful) {
|
||||||
|
this.helpfulCount++;
|
||||||
|
} else {
|
||||||
|
this.unhelpfulCount++;
|
||||||
|
}
|
||||||
|
this.recalculateQualityScore();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 重新计算质量评分
|
||||||
|
*/
|
||||||
|
private recalculateQualityScore(): void {
|
||||||
|
// 基础分50分
|
||||||
|
let score = 50;
|
||||||
|
|
||||||
|
// 引用次数加分（最多+20）
|
||||||
|
score += Math.min(this.citationCount * 2, 20);
|
||||||
|
|
||||||
|
// 用户反馈（最多±30）
|
||||||
|
const totalFeedback = this.helpfulCount + this.unhelpfulCount;
|
||||||
|
if (totalFeedback > 0) {
|
||||||
|
const helpfulRatio = this.helpfulCount / totalFeedback;
|
||||||
|
score += Math.round((helpfulRatio - 0.5) * 60);
|
||||||
|
}
|
||||||
|
|
||||||
|
this.qualityScore = Math.max(0, Math.min(100, score));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum KnowledgeSource {
|
||||||
|
MANUAL = 'MANUAL', // 手动添加
|
||||||
|
CRAWL = 'CRAWL', // 网页爬取
|
||||||
|
EXTRACT = 'EXTRACT', // 对话提取
|
||||||
|
IMPORT = 'IMPORT', // 批量导入
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,97 @@
|
||||||
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 知识块实体 - 将文章拆分为更小的检索单元
|
||||||
|
* 用于RAG检索时提高精确度
|
||||||
|
*/
|
||||||
|
export class KnowledgeChunkEntity {
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
/** 所属文章ID */
|
||||||
|
articleId: string;
|
||||||
|
|
||||||
|
/** 块内容 */
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
/** 块序号（在文章中的位置） */
|
||||||
|
chunkIndex: number;
|
||||||
|
|
||||||
|
/** 块类型 */
|
||||||
|
chunkType: ChunkType;
|
||||||
|
|
||||||
|
/** 内容向量 */
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
/** 元数据（如标题层级、列表项等） */
|
||||||
|
metadata: ChunkMetadata;
|
||||||
|
|
||||||
|
/** Token数量（估算） */
|
||||||
|
tokenCount: number;
|
||||||
|
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
private constructor() {}
|
||||||
|
|
||||||
|
static create(params: {
|
||||||
|
articleId: string;
|
||||||
|
content: string;
|
||||||
|
chunkIndex: number;
|
||||||
|
chunkType: ChunkType;
|
||||||
|
metadata?: ChunkMetadata;
|
||||||
|
}): KnowledgeChunkEntity {
|
||||||
|
const chunk = new KnowledgeChunkEntity();
|
||||||
|
chunk.id = uuidv4();
|
||||||
|
chunk.articleId = params.articleId;
|
||||||
|
chunk.content = params.content;
|
||||||
|
chunk.chunkIndex = params.chunkIndex;
|
||||||
|
chunk.chunkType = params.chunkType;
|
||||||
|
chunk.metadata = params.metadata || {};
|
||||||
|
chunk.tokenCount = KnowledgeChunkEntity.estimateTokens(params.content);
|
||||||
|
chunk.createdAt = new Date();
|
||||||
|
return chunk;
|
||||||
|
}
|
||||||
|
|
||||||
|
static fromPersistence(data: Partial<KnowledgeChunkEntity>): KnowledgeChunkEntity {
|
||||||
|
const chunk = new KnowledgeChunkEntity();
|
||||||
|
Object.assign(chunk, data);
|
||||||
|
return chunk;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 估算Token数量（简单估算：中文字符约1.5token，英文单词约1.3token）
|
||||||
|
*/
|
||||||
|
private static estimateTokens(content: string): number {
|
||||||
|
const chineseChars = (content.match(/[\u4e00-\u9fa5]/g) || []).length;
|
||||||
|
const englishWords = (content.match(/[a-zA-Z]+/g) || []).length;
|
||||||
|
return Math.ceil(chineseChars * 1.5 + englishWords * 1.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 设置向量
|
||||||
|
*/
|
||||||
|
setEmbedding(embedding: number[]): void {
|
||||||
|
this.embedding = embedding;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum ChunkType {
|
||||||
|
TITLE = 'TITLE', // 标题
|
||||||
|
PARAGRAPH = 'PARAGRAPH', // 段落
|
||||||
|
LIST = 'LIST', // 列表
|
||||||
|
TABLE = 'TABLE', // 表格
|
||||||
|
CODE = 'CODE', // 代码块
|
||||||
|
FAQ = 'FAQ', // 问答对
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface ChunkMetadata {
|
||||||
|
/** 所属章节标题 */
|
||||||
|
sectionTitle?: string;
|
||||||
|
/** 标题层级 */
|
||||||
|
headingLevel?: number;
|
||||||
|
/** 是否包含重要关键词 */
|
||||||
|
hasKeywords?: boolean;
|
||||||
|
/** 前一个块ID（用于上下文） */
|
||||||
|
prevChunkId?: string;
|
||||||
|
/** 后一个块ID */
|
||||||
|
nextChunkId?: string;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,209 @@
|
||||||
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 系统经验实体 - 存储系统从对话中学习到的经验
|
||||||
|
* 用于系统自我进化
|
||||||
|
*/
|
||||||
|
export class SystemExperienceEntity {
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
/** 经验类型 */
|
||||||
|
experienceType: ExperienceType;
|
||||||
|
|
||||||
|
/** 经验内容/描述 */
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
/** 经验的置信度 0-100 */
|
||||||
|
confidence: number;
|
||||||
|
|
||||||
|
/** 应用场景描述 */
|
||||||
|
scenario: string;
|
||||||
|
|
||||||
|
/** 相关移民类别 */
|
||||||
|
relatedCategory?: string;
|
||||||
|
|
||||||
|
/** 来源对话ID列表 */
|
||||||
|
sourceConversationIds: string[];
|
||||||
|
|
||||||
|
/** 验证状态 */
|
||||||
|
verificationStatus: VerificationStatus;
|
||||||
|
|
||||||
|
/** 验证者（管理员）ID */
|
||||||
|
verifiedBy?: string;
|
||||||
|
|
||||||
|
/** 验证时间 */
|
||||||
|
verifiedAt?: Date;
|
||||||
|
|
||||||
|
/** 使用次数（被应用到对话中） */
|
||||||
|
usageCount: number;
|
||||||
|
|
||||||
|
/** 正面反馈次数 */
|
||||||
|
positiveCount: number;
|
||||||
|
|
||||||
|
/** 负面反馈次数 */
|
||||||
|
negativeCount: number;
|
||||||
|
|
||||||
|
/** 内容向量 */
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
/** 是否激活使用 */
|
||||||
|
isActive: boolean;
|
||||||
|
|
||||||
|
createdAt: Date;
|
||||||
|
updatedAt: Date;
|
||||||
|
|
||||||
|
private constructor() {}
|
||||||
|
|
||||||
|
static create(params: {
|
||||||
|
experienceType: ExperienceType;
|
||||||
|
content: string;
|
||||||
|
scenario: string;
|
||||||
|
relatedCategory?: string;
|
||||||
|
sourceConversationId: string;
|
||||||
|
confidence?: number;
|
||||||
|
}): SystemExperienceEntity {
|
||||||
|
const experience = new SystemExperienceEntity();
|
||||||
|
experience.id = uuidv4();
|
||||||
|
experience.experienceType = params.experienceType;
|
||||||
|
experience.content = params.content;
|
||||||
|
experience.scenario = params.scenario;
|
||||||
|
experience.relatedCategory = params.relatedCategory;
|
||||||
|
experience.sourceConversationIds = [params.sourceConversationId];
|
||||||
|
experience.confidence = params.confidence ?? 50;
|
||||||
|
experience.verificationStatus = VerificationStatus.PENDING;
|
||||||
|
experience.usageCount = 0;
|
||||||
|
experience.positiveCount = 0;
|
||||||
|
experience.negativeCount = 0;
|
||||||
|
experience.isActive = false; // 默认不激活，需要验证后才激活
|
||||||
|
experience.createdAt = new Date();
|
||||||
|
experience.updatedAt = new Date();
|
||||||
|
return experience;
|
||||||
|
}
|
||||||
|
|
||||||
|
static fromPersistence(data: Partial<SystemExperienceEntity>): SystemExperienceEntity {
|
||||||
|
const experience = new SystemExperienceEntity();
|
||||||
|
Object.assign(experience, data);
|
||||||
|
return experience;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 添加来源对话
|
||||||
|
*/
|
||||||
|
addSourceConversation(conversationId: string): void {
|
||||||
|
if (!this.sourceConversationIds.includes(conversationId)) {
|
||||||
|
this.sourceConversationIds.push(conversationId);
|
||||||
|
// 多个对话都产生相同经验，提高置信度
|
||||||
|
this.confidence = Math.min(100, this.confidence + 5);
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 设置向量
|
||||||
|
*/
|
||||||
|
setEmbedding(embedding: number[]): void {
|
||||||
|
this.embedding = embedding;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 管理员验证通过
|
||||||
|
*/
|
||||||
|
approve(adminId: string): void {
|
||||||
|
this.verificationStatus = VerificationStatus.APPROVED;
|
||||||
|
this.verifiedBy = adminId;
|
||||||
|
this.verifiedAt = new Date();
|
||||||
|
this.isActive = true;
|
||||||
|
this.confidence = Math.min(100, this.confidence + 20);
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 管理员验证拒绝
|
||||||
|
*/
|
||||||
|
reject(adminId: string): void {
|
||||||
|
this.verificationStatus = VerificationStatus.REJECTED;
|
||||||
|
this.verifiedBy = adminId;
|
||||||
|
this.verifiedAt = new Date();
|
||||||
|
this.isActive = false;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录使用
|
||||||
|
*/
|
||||||
|
recordUsage(): void {
|
||||||
|
this.usageCount++;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录反馈
|
||||||
|
*/
|
||||||
|
recordFeedback(positive: boolean): void {
|
||||||
|
if (positive) {
|
||||||
|
this.positiveCount++;
|
||||||
|
this.confidence = Math.min(100, this.confidence + 2);
|
||||||
|
} else {
|
||||||
|
this.negativeCount++;
|
||||||
|
this.confidence = Math.max(0, this.confidence - 5);
|
||||||
|
|
||||||
|
// 负面反馈过多则自动停用
|
||||||
|
if (this.negativeCount > 5 && this.negativeCount > this.positiveCount * 2) {
|
||||||
|
this.isActive = false;
|
||||||
|
this.verificationStatus = VerificationStatus.DEPRECATED;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 合并相似经验
|
||||||
|
*/
|
||||||
|
mergeWith(other: SystemExperienceEntity): void {
|
||||||
|
// 合并来源对话
|
||||||
|
other.sourceConversationIds.forEach(id => {
|
||||||
|
if (!this.sourceConversationIds.includes(id)) {
|
||||||
|
this.sourceConversationIds.push(id);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 合并统计数据
|
||||||
|
this.usageCount += other.usageCount;
|
||||||
|
this.positiveCount += other.positiveCount;
|
||||||
|
this.negativeCount += other.negativeCount;
|
||||||
|
|
||||||
|
// 重新计算置信度
|
||||||
|
this.confidence = Math.min(100, (this.confidence + other.confidence) / 2 + 10);
|
||||||
|
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum ExperienceType {
|
||||||
|
// 问答相关
|
||||||
|
COMMON_QUESTION = 'COMMON_QUESTION', // 常见问题
|
||||||
|
ANSWER_TEMPLATE = 'ANSWER_TEMPLATE', // 回答模板
|
||||||
|
CLARIFICATION = 'CLARIFICATION', // 澄清方式
|
||||||
|
|
||||||
|
// 用户行为
|
||||||
|
USER_PATTERN = 'USER_PATTERN', // 用户行为模式
|
||||||
|
CONVERSION_TRIGGER = 'CONVERSION_TRIGGER', // 转化触发点
|
||||||
|
|
||||||
|
// 知识相关
|
||||||
|
KNOWLEDGE_GAP = 'KNOWLEDGE_GAP', // 知识缺口
|
||||||
|
KNOWLEDGE_UPDATE = 'KNOWLEDGE_UPDATE', // 知识更新
|
||||||
|
|
||||||
|
// 对话技巧
|
||||||
|
CONVERSATION_SKILL = 'CONVERSATION_SKILL', // 对话技巧
|
||||||
|
OBJECTION_HANDLING = 'OBJECTION_HANDLING', // 异议处理
|
||||||
|
|
||||||
|
// 其他
|
||||||
|
CUSTOM = 'CUSTOM', // 自定义经验
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum VerificationStatus {
|
||||||
|
PENDING = 'PENDING', // 待验证
|
||||||
|
APPROVED = 'APPROVED', // 已通过
|
||||||
|
REJECTED = 'REJECTED', // 已拒绝
|
||||||
|
DEPRECATED = 'DEPRECATED', // 已弃用
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,126 @@
|
||||||
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用户记忆实体 - 存储用户的长期记忆
|
||||||
|
* 用于个性化对话和用户画像
|
||||||
|
*/
|
||||||
|
export class UserMemoryEntity {
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
/** 用户ID（可以是匿名用户ID） */
|
||||||
|
userId: string;
|
||||||
|
|
||||||
|
/** 记忆类型 */
|
||||||
|
memoryType: MemoryType;
|
||||||
|
|
||||||
|
/** 记忆内容 */
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
/** 记忆的重要性 0-100 */
|
||||||
|
importance: number;
|
||||||
|
|
||||||
|
/** 记忆来源（哪次对话提取） */
|
||||||
|
sourceConversationId?: string;
|
||||||
|
|
||||||
|
/** 相关移民类别 */
|
||||||
|
relatedCategory?: string;
|
||||||
|
|
||||||
|
/** 内容向量 */
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
/** 访问次数（记忆被检索使用） */
|
||||||
|
accessCount: number;
|
||||||
|
|
||||||
|
/** 最后访问时间 */
|
||||||
|
lastAccessedAt?: Date;
|
||||||
|
|
||||||
|
/** 是否已过期（用户情况已变化） */
|
||||||
|
isExpired: boolean;
|
||||||
|
|
||||||
|
createdAt: Date;
|
||||||
|
updatedAt: Date;
|
||||||
|
|
||||||
|
private constructor() {}
|
||||||
|
|
||||||
|
static create(params: {
|
||||||
|
userId: string;
|
||||||
|
memoryType: MemoryType;
|
||||||
|
content: string;
|
||||||
|
importance?: number;
|
||||||
|
sourceConversationId?: string;
|
||||||
|
relatedCategory?: string;
|
||||||
|
}): UserMemoryEntity {
|
||||||
|
const memory = new UserMemoryEntity();
|
||||||
|
memory.id = uuidv4();
|
||||||
|
memory.userId = params.userId;
|
||||||
|
memory.memoryType = params.memoryType;
|
||||||
|
memory.content = params.content;
|
||||||
|
memory.importance = params.importance ?? 50;
|
||||||
|
memory.sourceConversationId = params.sourceConversationId;
|
||||||
|
memory.relatedCategory = params.relatedCategory;
|
||||||
|
memory.accessCount = 0;
|
||||||
|
memory.isExpired = false;
|
||||||
|
memory.createdAt = new Date();
|
||||||
|
memory.updatedAt = new Date();
|
||||||
|
return memory;
|
||||||
|
}
|
||||||
|
|
||||||
|
static fromPersistence(data: Partial<UserMemoryEntity>): UserMemoryEntity {
|
||||||
|
const memory = new UserMemoryEntity();
|
||||||
|
Object.assign(memory, data);
|
||||||
|
return memory;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 设置向量
|
||||||
|
*/
|
||||||
|
setEmbedding(embedding: number[]): void {
|
||||||
|
this.embedding = embedding;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录访问
|
||||||
|
*/
|
||||||
|
recordAccess(): void {
|
||||||
|
this.accessCount++;
|
||||||
|
this.lastAccessedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 标记为过期
|
||||||
|
*/
|
||||||
|
markAsExpired(): void {
|
||||||
|
this.isExpired = true;
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新重要性
|
||||||
|
*/
|
||||||
|
updateImportance(importance: number): void {
|
||||||
|
this.importance = Math.max(0, Math.min(100, importance));
|
||||||
|
this.updatedAt = new Date();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum MemoryType {
|
||||||
|
// 用户基本信息
|
||||||
|
PERSONAL_INFO = 'PERSONAL_INFO', // 个人信息（年龄、学历等）
|
||||||
|
WORK_EXPERIENCE = 'WORK_EXPERIENCE', // 工作经历
|
||||||
|
EDUCATION = 'EDUCATION', // 教育背景
|
||||||
|
LANGUAGE = 'LANGUAGE', // 语言能力
|
||||||
|
|
||||||
|
// 移民相关
|
||||||
|
IMMIGRATION_INTENT = 'IMMIGRATION_INTENT', // 移民意向
|
||||||
|
PREFERRED_CATEGORY = 'PREFERRED_CATEGORY', // 倾向的移民类别
|
||||||
|
ASSESSMENT_RESULT = 'ASSESSMENT_RESULT', // 评估结果
|
||||||
|
|
||||||
|
// 对话相关
|
||||||
|
QUESTION_ASKED = 'QUESTION_ASKED', // 问过的问题
|
||||||
|
CONCERN = 'CONCERN', // 关注点/顾虑
|
||||||
|
PREFERENCE = 'PREFERENCE', // 偏好设置
|
||||||
|
|
||||||
|
// 其他
|
||||||
|
CUSTOM = 'CUSTOM', // 自定义记忆
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,72 @@
|
||||||
|
import { KnowledgeArticleEntity } from '../entities/knowledge-article.entity';
|
||||||
|
import { KnowledgeChunkEntity } from '../entities/knowledge-chunk.entity';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 知识库仓储接口
|
||||||
|
*/
|
||||||
|
export interface IKnowledgeRepository {
|
||||||
|
// ========== 文章操作 ==========
|
||||||
|
|
||||||
|
/** 保存文章 */
|
||||||
|
saveArticle(article: KnowledgeArticleEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 根据ID获取文章 */
|
||||||
|
findArticleById(id: string): Promise<KnowledgeArticleEntity | null>;
|
||||||
|
|
||||||
|
/** 根据类别获取文章列表 */
|
||||||
|
findArticlesByCategory(category: string, options?: {
|
||||||
|
publishedOnly?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
offset?: number;
|
||||||
|
}): Promise<KnowledgeArticleEntity[]>;
|
||||||
|
|
||||||
|
/** 搜索文章（关键词） */
|
||||||
|
searchArticles(query: string, options?: {
|
||||||
|
category?: string;
|
||||||
|
publishedOnly?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
}): Promise<KnowledgeArticleEntity[]>;
|
||||||
|
|
||||||
|
/** 向量相似度搜索文章 */
|
||||||
|
searchArticlesByVector(embedding: number[], options?: {
|
||||||
|
category?: string;
|
||||||
|
publishedOnly?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
}): Promise<Array<{ article: KnowledgeArticleEntity; similarity: number }>>;
|
||||||
|
|
||||||
|
/** 更新文章 */
|
||||||
|
updateArticle(article: KnowledgeArticleEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 删除文章 */
|
||||||
|
deleteArticle(id: string): Promise<void>;
|
||||||
|
|
||||||
|
/** 获取文章总数 */
|
||||||
|
countArticles(options?: {
|
||||||
|
category?: string;
|
||||||
|
publishedOnly?: boolean;
|
||||||
|
}): Promise<number>;
|
||||||
|
|
||||||
|
// ========== 块操作 ==========
|
||||||
|
|
||||||
|
/** 保存知识块 */
|
||||||
|
saveChunk(chunk: KnowledgeChunkEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 批量保存知识块 */
|
||||||
|
saveChunks(chunks: KnowledgeChunkEntity[]): Promise<void>;
|
||||||
|
|
||||||
|
/** 根据文章ID获取所有块 */
|
||||||
|
findChunksByArticleId(articleId: string): Promise<KnowledgeChunkEntity[]>;
|
||||||
|
|
||||||
|
/** 向量相似度搜索块 */
|
||||||
|
searchChunksByVector(embedding: number[], options?: {
|
||||||
|
category?: string;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
}): Promise<Array<{ chunk: KnowledgeChunkEntity; similarity: number }>>;
|
||||||
|
|
||||||
|
/** 删除文章的所有块 */
|
||||||
|
deleteChunksByArticleId(articleId: string): Promise<void>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const KNOWLEDGE_REPOSITORY = Symbol('IKnowledgeRepository');
|
||||||
|
|
@ -0,0 +1,95 @@
|
||||||
|
import { UserMemoryEntity, MemoryType } from '../entities/user-memory.entity';
|
||||||
|
import { SystemExperienceEntity, ExperienceType, VerificationStatus } from '../entities/system-experience.entity';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用户记忆仓储接口
|
||||||
|
*/
|
||||||
|
export interface IUserMemoryRepository {
|
||||||
|
/** 保存用户记忆 */
|
||||||
|
save(memory: UserMemoryEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 根据ID获取记忆 */
|
||||||
|
findById(id: string): Promise<UserMemoryEntity | null>;
|
||||||
|
|
||||||
|
/** 获取用户所有记忆 */
|
||||||
|
findByUserId(userId: string, options?: {
|
||||||
|
memoryType?: MemoryType;
|
||||||
|
includeExpired?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
}): Promise<UserMemoryEntity[]>;
|
||||||
|
|
||||||
|
/** 向量相似度搜索用户记忆 */
|
||||||
|
searchByVector(userId: string, embedding: number[], options?: {
|
||||||
|
memoryType?: MemoryType;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
}): Promise<Array<{ memory: UserMemoryEntity; similarity: number }>>;
|
||||||
|
|
||||||
|
/** 获取用户最重要的记忆 */
|
||||||
|
findTopMemories(userId: string, limit: number): Promise<UserMemoryEntity[]>;
|
||||||
|
|
||||||
|
/** 更新记忆 */
|
||||||
|
update(memory: UserMemoryEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 删除记忆 */
|
||||||
|
delete(id: string): Promise<void>;
|
||||||
|
|
||||||
|
/** 删除用户所有记忆 */
|
||||||
|
deleteByUserId(userId: string): Promise<void>;
|
||||||
|
|
||||||
|
/** 标记用户过期的记忆 */
|
||||||
|
markExpiredMemories(userId: string, olderThanDays: number): Promise<number>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 系统经验仓储接口
|
||||||
|
*/
|
||||||
|
export interface ISystemExperienceRepository {
|
||||||
|
/** 保存系统经验 */
|
||||||
|
save(experience: SystemExperienceEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 根据ID获取经验 */
|
||||||
|
findById(id: string): Promise<SystemExperienceEntity | null>;
|
||||||
|
|
||||||
|
/** 获取待验证的经验 */
|
||||||
|
findPendingExperiences(options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
limit?: number;
|
||||||
|
offset?: number;
|
||||||
|
}): Promise<SystemExperienceEntity[]>;
|
||||||
|
|
||||||
|
/** 获取已激活的经验 */
|
||||||
|
findActiveExperiences(options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
category?: string;
|
||||||
|
minConfidence?: number;
|
||||||
|
limit?: number;
|
||||||
|
}): Promise<SystemExperienceEntity[]>;
|
||||||
|
|
||||||
|
/** 向量相似度搜索经验 */
|
||||||
|
searchByVector(embedding: number[], options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
activeOnly?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
}): Promise<Array<{ experience: SystemExperienceEntity; similarity: number }>>;
|
||||||
|
|
||||||
|
/** 查找相似经验（用于合并） */
|
||||||
|
findSimilarExperiences(embedding: number[], threshold: number): Promise<SystemExperienceEntity[]>;
|
||||||
|
|
||||||
|
/** 更新经验 */
|
||||||
|
update(experience: SystemExperienceEntity): Promise<void>;
|
||||||
|
|
||||||
|
/** 删除经验 */
|
||||||
|
delete(id: string): Promise<void>;
|
||||||
|
|
||||||
|
/** 获取统计数据 */
|
||||||
|
getStatistics(): Promise<{
|
||||||
|
total: number;
|
||||||
|
byStatus: Record<VerificationStatus, number>;
|
||||||
|
byType: Record<ExperienceType, number>;
|
||||||
|
}>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const USER_MEMORY_REPOSITORY = Symbol('IUserMemoryRepository');
|
||||||
|
export const SYSTEM_EXPERIENCE_REPOSITORY = Symbol('ISystemExperienceRepository');
|
||||||
|
|
@ -0,0 +1,422 @@
|
||||||
|
import { Injectable, OnModuleInit, OnModuleDestroy } from '@nestjs/common';
|
||||||
|
import { ConfigService } from '@nestjs/config';
|
||||||
|
import neo4j, { Driver, Session, Transaction } from 'neo4j-driver';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Neo4j知识图谱服务
|
||||||
|
* 用于存储和查询用户时间线、实体关系等
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class Neo4jService implements OnModuleInit, OnModuleDestroy {
|
||||||
|
private driver: Driver;
|
||||||
|
|
||||||
|
constructor(private configService: ConfigService) {}
|
||||||
|
|
||||||
|
async onModuleInit() {
|
||||||
|
const uri = this.configService.get<string>('NEO4J_URI') || 'bolt://localhost:7687';
|
||||||
|
const user = this.configService.get<string>('NEO4J_USER') || 'neo4j';
|
||||||
|
const password = this.configService.get<string>('NEO4J_PASSWORD') || 'password';
|
||||||
|
|
||||||
|
this.driver = neo4j.driver(uri, neo4j.auth.basic(user, password), {
|
||||||
|
maxConnectionPoolSize: 50,
|
||||||
|
connectionAcquisitionTimeout: 30000,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 验证连接
|
||||||
|
try {
|
||||||
|
await this.driver.verifyConnectivity();
|
||||||
|
console.log('[Neo4j] Connected successfully');
|
||||||
|
|
||||||
|
// 初始化schema
|
||||||
|
await this.initializeSchema();
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[Neo4j] Connection failed:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async onModuleDestroy() {
|
||||||
|
await this.driver?.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取会话
|
||||||
|
*/
|
||||||
|
getSession(): Session {
|
||||||
|
return this.driver.session();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行读查询
|
||||||
|
*/
|
||||||
|
async read<T>(query: string, params?: Record<string, unknown>): Promise<T[]> {
|
||||||
|
const session = this.getSession();
|
||||||
|
try {
|
||||||
|
const result = await session.run(query, params);
|
||||||
|
return result.records.map(record => record.toObject() as T);
|
||||||
|
} finally {
|
||||||
|
await session.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行写查询
|
||||||
|
*/
|
||||||
|
async write<T>(query: string, params?: Record<string, unknown>): Promise<T[]> {
|
||||||
|
const session = this.getSession();
|
||||||
|
try {
|
||||||
|
const result = await session.executeWrite(tx => tx.run(query, params));
|
||||||
|
return result.records.map(record => record.toObject() as T);
|
||||||
|
} finally {
|
||||||
|
await session.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 初始化图谱Schema（约束和索引）
|
||||||
|
*/
|
||||||
|
private async initializeSchema() {
|
||||||
|
const session = this.getSession();
|
||||||
|
try {
|
||||||
|
// 用户节点约束
|
||||||
|
await session.run(`
|
||||||
|
CREATE CONSTRAINT user_id IF NOT EXISTS
|
||||||
|
FOR (u:User) REQUIRE u.id IS UNIQUE
|
||||||
|
`);
|
||||||
|
|
||||||
|
// 对话节点约束
|
||||||
|
await session.run(`
|
||||||
|
CREATE CONSTRAINT conversation_id IF NOT EXISTS
|
||||||
|
FOR (c:Conversation) REQUIRE c.id IS UNIQUE
|
||||||
|
`);
|
||||||
|
|
||||||
|
// 知识实体约束
|
||||||
|
await session.run(`
|
||||||
|
CREATE CONSTRAINT entity_id IF NOT EXISTS
|
||||||
|
FOR (e:Entity) REQUIRE e.id IS UNIQUE
|
||||||
|
`);
|
||||||
|
|
||||||
|
// 事件节点约束
|
||||||
|
await session.run(`
|
||||||
|
CREATE CONSTRAINT event_id IF NOT EXISTS
|
||||||
|
FOR (e:Event) REQUIRE e.id IS UNIQUE
|
||||||
|
`);
|
||||||
|
|
||||||
|
// 时间索引
|
||||||
|
await session.run(`
|
||||||
|
CREATE INDEX event_timestamp IF NOT EXISTS
|
||||||
|
FOR (e:Event) ON (e.timestamp)
|
||||||
|
`);
|
||||||
|
|
||||||
|
// 类别索引
|
||||||
|
await session.run(`
|
||||||
|
CREATE INDEX entity_category IF NOT EXISTS
|
||||||
|
FOR (e:Entity) ON (e.category)
|
||||||
|
`);
|
||||||
|
|
||||||
|
console.log('[Neo4j] Schema initialized');
|
||||||
|
} catch (error) {
|
||||||
|
// 约束可能已存在，忽略错误
|
||||||
|
console.log('[Neo4j] Schema initialization skipped (may already exist)');
|
||||||
|
} finally {
|
||||||
|
await session.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 用户时间线操作 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建用户节点
|
||||||
|
*/
|
||||||
|
async createUserNode(userId: string, properties?: Record<string, unknown>): Promise<void> {
|
||||||
|
await this.write(
|
||||||
|
`
|
||||||
|
MERGE (u:User {id: $userId})
|
||||||
|
SET u += $properties, u.updatedAt = datetime()
|
||||||
|
`,
|
||||||
|
{ userId, properties: properties || {} },
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录用户事件（时间线）
|
||||||
|
*/
|
||||||
|
async recordUserEvent(params: {
|
||||||
|
userId: string;
|
||||||
|
eventId: string;
|
||||||
|
eventType: string;
|
||||||
|
content: string;
|
||||||
|
timestamp?: Date;
|
||||||
|
metadata?: Record<string, unknown>;
|
||||||
|
}): Promise<void> {
|
||||||
|
const { userId, eventId, eventType, content, metadata } = params;
|
||||||
|
const timestamp = params.timestamp || new Date();
|
||||||
|
|
||||||
|
await this.write(
|
||||||
|
`
|
||||||
|
MATCH (u:User {id: $userId})
|
||||||
|
CREATE (e:Event {
|
||||||
|
id: $eventId,
|
||||||
|
type: $eventType,
|
||||||
|
content: $content,
|
||||||
|
timestamp: datetime($timestamp),
|
||||||
|
metadata: $metadata
|
||||||
|
})
|
||||||
|
CREATE (u)-[:HAS_EVENT]->(e)
|
||||||
|
|
||||||
|
WITH u, e
|
||||||
|
OPTIONAL MATCH (u)-[:HAS_EVENT]->(prev:Event)
|
||||||
|
WHERE prev.timestamp < e.timestamp AND prev.id <> e.id
|
||||||
|
WITH e, prev
|
||||||
|
ORDER BY prev.timestamp DESC
|
||||||
|
LIMIT 1
|
||||||
|
FOREACH (_ IN CASE WHEN prev IS NOT NULL THEN [1] ELSE [] END |
|
||||||
|
CREATE (prev)-[:FOLLOWED_BY]->(e)
|
||||||
|
)
|
||||||
|
`,
|
||||||
|
{
|
||||||
|
userId,
|
||||||
|
eventId,
|
||||||
|
eventType,
|
||||||
|
content,
|
||||||
|
timestamp: timestamp.toISOString(),
|
||||||
|
metadata: JSON.stringify(metadata || {}),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户时间线
|
||||||
|
*/
|
||||||
|
async getUserTimeline(
|
||||||
|
userId: string,
|
||||||
|
options?: { limit?: number; beforeDate?: Date; eventTypes?: string[] },
|
||||||
|
): Promise<Array<{
|
||||||
|
id: string;
|
||||||
|
type: string;
|
||||||
|
content: string;
|
||||||
|
timestamp: Date;
|
||||||
|
metadata: Record<string, unknown>;
|
||||||
|
}>> {
|
||||||
|
let query = `
|
||||||
|
MATCH (u:User {id: $userId})-[:HAS_EVENT]->(e:Event)
|
||||||
|
`;
|
||||||
|
|
||||||
|
const params: Record<string, unknown> = { userId };
|
||||||
|
|
||||||
|
if (options?.beforeDate) {
|
||||||
|
query += ` WHERE e.timestamp < datetime($beforeDate)`;
|
||||||
|
params.beforeDate = options.beforeDate.toISOString();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.eventTypes?.length) {
|
||||||
|
const typeCondition = options.beforeDate ? ' AND' : ' WHERE';
|
||||||
|
query += `${typeCondition} e.type IN $eventTypes`;
|
||||||
|
params.eventTypes = options.eventTypes;
|
||||||
|
}
|
||||||
|
|
||||||
|
query += `
|
||||||
|
RETURN e.id as id, e.type as type, e.content as content,
|
||||||
|
e.timestamp as timestamp, e.metadata as metadata
|
||||||
|
ORDER BY e.timestamp DESC
|
||||||
|
LIMIT $limit
|
||||||
|
`;
|
||||||
|
params.limit = options?.limit || 20;
|
||||||
|
|
||||||
|
const results = await this.read<{
|
||||||
|
id: string;
|
||||||
|
type: string;
|
||||||
|
content: string;
|
||||||
|
timestamp: { toString: () => string };
|
||||||
|
metadata: string;
|
||||||
|
}>(query, params);
|
||||||
|
|
||||||
|
return results.map(r => ({
|
||||||
|
id: r.id,
|
||||||
|
type: r.type,
|
||||||
|
content: r.content,
|
||||||
|
timestamp: new Date(r.timestamp.toString()),
|
||||||
|
metadata: JSON.parse(r.metadata || '{}'),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 知识图谱操作 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建知识实体
|
||||||
|
*/
|
||||||
|
async createEntity(params: {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
type: string;
|
||||||
|
category?: string;
|
||||||
|
properties?: Record<string, unknown>;
|
||||||
|
}): Promise<void> {
|
||||||
|
await this.write(
|
||||||
|
`
|
||||||
|
MERGE (e:Entity {id: $id})
|
||||||
|
SET e.name = $name,
|
||||||
|
e.type = $type,
|
||||||
|
e.category = $category,
|
||||||
|
e += $properties,
|
||||||
|
e.updatedAt = datetime()
|
||||||
|
`,
|
||||||
|
{
|
||||||
|
id: params.id,
|
||||||
|
name: params.name,
|
||||||
|
type: params.type,
|
||||||
|
category: params.category || '',
|
||||||
|
properties: params.properties || {},
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建实体关系
|
||||||
|
*/
|
||||||
|
async createRelation(params: {
|
||||||
|
fromId: string;
|
||||||
|
toId: string;
|
||||||
|
relationType: string;
|
||||||
|
properties?: Record<string, unknown>;
|
||||||
|
}): Promise<void> {
|
||||||
|
// 动态创建关系类型
|
||||||
|
const relationQuery = `
|
||||||
|
MATCH (from:Entity {id: $fromId})
|
||||||
|
MATCH (to:Entity {id: $toId})
|
||||||
|
MERGE (from)-[r:${params.relationType}]->(to)
|
||||||
|
SET r += $properties, r.updatedAt = datetime()
|
||||||
|
`;
|
||||||
|
|
||||||
|
await this.write(relationQuery, {
|
||||||
|
fromId: params.fromId,
|
||||||
|
toId: params.toId,
|
||||||
|
properties: params.properties || {},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查询相关实体
|
||||||
|
*/
|
||||||
|
async findRelatedEntities(
|
||||||
|
entityId: string,
|
||||||
|
options?: { relationTypes?: string[]; maxDepth?: number; limit?: number },
|
||||||
|
): Promise<Array<{
|
||||||
|
entity: { id: string; name: string; type: string };
|
||||||
|
relation: string;
|
||||||
|
depth: number;
|
||||||
|
}>> {
|
||||||
|
const maxDepth = options?.maxDepth || 2;
|
||||||
|
const limit = options?.limit || 20;
|
||||||
|
|
||||||
|
let relationPattern = '*1..' + maxDepth;
|
||||||
|
if (options?.relationTypes?.length) {
|
||||||
|
relationPattern = options.relationTypes.join('|') + relationPattern;
|
||||||
|
}
|
||||||
|
|
||||||
|
const query = `
|
||||||
|
MATCH path = (start:Entity {id: $entityId})-[r:${relationPattern}]-(related:Entity)
|
||||||
|
WHERE start <> related
|
||||||
|
RETURN related.id as id, related.name as name, related.type as type,
|
||||||
|
type(r) as relation, length(path) as depth
|
||||||
|
ORDER BY depth ASC
|
||||||
|
LIMIT $limit
|
||||||
|
`;
|
||||||
|
|
||||||
|
const results = await this.read<{
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
type: string;
|
||||||
|
relation: string;
|
||||||
|
depth: number;
|
||||||
|
}>(query, { entityId, limit });
|
||||||
|
|
||||||
|
return results.map(r => ({
|
||||||
|
entity: { id: r.id, name: r.name, type: r.type },
|
||||||
|
relation: r.relation,
|
||||||
|
depth: r.depth,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 搜索实体
|
||||||
|
*/
|
||||||
|
async searchEntities(
|
||||||
|
keyword: string,
|
||||||
|
options?: { category?: string; type?: string; limit?: number },
|
||||||
|
): Promise<Array<{ id: string; name: string; type: string; score: number }>> {
|
||||||
|
let query = `
|
||||||
|
MATCH (e:Entity)
|
||||||
|
WHERE e.name CONTAINS $keyword
|
||||||
|
`;
|
||||||
|
|
||||||
|
const params: Record<string, unknown> = { keyword };
|
||||||
|
|
||||||
|
if (options?.category) {
|
||||||
|
query += ` AND e.category = $category`;
|
||||||
|
params.category = options.category;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.type) {
|
||||||
|
query += ` AND e.type = $type`;
|
||||||
|
params.type = options.type;
|
||||||
|
}
|
||||||
|
|
||||||
|
query += `
|
||||||
|
RETURN e.id as id, e.name as name, e.type as type,
|
||||||
|
CASE WHEN e.name STARTS WITH $keyword THEN 1.0 ELSE 0.5 END as score
|
||||||
|
ORDER BY score DESC
|
||||||
|
LIMIT $limit
|
||||||
|
`;
|
||||||
|
params.limit = options?.limit || 10;
|
||||||
|
|
||||||
|
return this.read(query, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 对话关联 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 关联对话与实体
|
||||||
|
*/
|
||||||
|
async linkConversationToEntity(
|
||||||
|
conversationId: string,
|
||||||
|
entityId: string,
|
||||||
|
mentionType: 'MENTIONED' | 'DISCUSSED' | 'ASKED_ABOUT',
|
||||||
|
): Promise<void> {
|
||||||
|
await this.write(
|
||||||
|
`
|
||||||
|
MERGE (c:Conversation {id: $conversationId})
|
||||||
|
WITH c
|
||||||
|
MATCH (e:Entity {id: $entityId})
|
||||||
|
MERGE (c)-[r:${mentionType}]->(e)
|
||||||
|
SET r.timestamp = datetime()
|
||||||
|
`,
|
||||||
|
{ conversationId, entityId },
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取对话相关实体
|
||||||
|
*/
|
||||||
|
async getConversationEntities(conversationId: string): Promise<Array<{
|
||||||
|
entity: { id: string; name: string; type: string };
|
||||||
|
mentionType: string;
|
||||||
|
}>> {
|
||||||
|
const results = await this.read<{
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
type: string;
|
||||||
|
mentionType: string;
|
||||||
|
}>(
|
||||||
|
`
|
||||||
|
MATCH (c:Conversation {id: $conversationId})-[r]->(e:Entity)
|
||||||
|
RETURN e.id as id, e.name as name, e.type as type, type(r) as mentionType
|
||||||
|
`,
|
||||||
|
{ conversationId },
|
||||||
|
);
|
||||||
|
|
||||||
|
return results.map(r => ({
|
||||||
|
entity: { id: r.id, name: r.name, type: r.type },
|
||||||
|
mentionType: r.mentionType,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,67 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
Index,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('knowledge_articles')
|
||||||
|
@Index('idx_knowledge_articles_category', ['category'])
|
||||||
|
@Index('idx_knowledge_articles_published', ['isPublished'])
|
||||||
|
export class KnowledgeArticleORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ length: 500 })
|
||||||
|
title: string;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column('text', { nullable: true })
|
||||||
|
summary: string;
|
||||||
|
|
||||||
|
@Column({ length: 50 })
|
||||||
|
category: string;
|
||||||
|
|
||||||
|
@Column('text', { array: true, default: '{}' })
|
||||||
|
tags: string[];
|
||||||
|
|
||||||
|
@Column({ length: 20 })
|
||||||
|
source: string;
|
||||||
|
|
||||||
|
@Column({ name: 'source_url', length: 1000, nullable: true })
|
||||||
|
sourceUrl?: string;
|
||||||
|
|
||||||
|
@Column('float', { array: true, nullable: true })
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
@Column({ name: 'is_published', default: false })
|
||||||
|
isPublished: boolean;
|
||||||
|
|
||||||
|
@Column({ name: 'citation_count', default: 0 })
|
||||||
|
citationCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'helpful_count', default: 0 })
|
||||||
|
helpfulCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'unhelpful_count', default: 0 })
|
||||||
|
unhelpfulCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'quality_score', default: 50 })
|
||||||
|
qualityScore: number;
|
||||||
|
|
||||||
|
@Column({ name: 'created_by', nullable: true })
|
||||||
|
createdBy?: string;
|
||||||
|
|
||||||
|
@Column({ name: 'updated_by', nullable: true })
|
||||||
|
updatedBy?: string;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,38 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
Index,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('knowledge_chunks')
|
||||||
|
@Index('idx_knowledge_chunks_article', ['articleId'])
|
||||||
|
export class KnowledgeChunkORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'article_id' })
|
||||||
|
articleId: string;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column({ name: 'chunk_index' })
|
||||||
|
chunkIndex: number;
|
||||||
|
|
||||||
|
@Column({ name: 'chunk_type', length: 20 })
|
||||||
|
chunkType: string;
|
||||||
|
|
||||||
|
@Column('float', { array: true, nullable: true })
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
@Column('jsonb', { default: '{}' })
|
||||||
|
metadata: Record<string, unknown>;
|
||||||
|
|
||||||
|
@Column({ name: 'token_count', default: 0 })
|
||||||
|
tokenCount: number;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,65 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
Index,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('system_experiences')
|
||||||
|
@Index('idx_system_experiences_type', ['experienceType'])
|
||||||
|
@Index('idx_system_experiences_status', ['verificationStatus'])
|
||||||
|
@Index('idx_system_experiences_active', ['isActive'])
|
||||||
|
export class SystemExperienceORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'experience_type', length: 30 })
|
||||||
|
experienceType: string;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column({ default: 50 })
|
||||||
|
confidence: number;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
scenario: string;
|
||||||
|
|
||||||
|
@Column({ name: 'related_category', length: 50, nullable: true })
|
||||||
|
relatedCategory?: string;
|
||||||
|
|
||||||
|
@Column('text', { name: 'source_conversation_ids', array: true, default: '{}' })
|
||||||
|
sourceConversationIds: string[];
|
||||||
|
|
||||||
|
@Column({ name: 'verification_status', length: 20, default: 'PENDING' })
|
||||||
|
verificationStatus: string;
|
||||||
|
|
||||||
|
@Column({ name: 'verified_by', nullable: true })
|
||||||
|
verifiedBy?: string;
|
||||||
|
|
||||||
|
@Column({ name: 'verified_at', nullable: true })
|
||||||
|
verifiedAt?: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'usage_count', default: 0 })
|
||||||
|
usageCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'positive_count', default: 0 })
|
||||||
|
positiveCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'negative_count', default: 0 })
|
||||||
|
negativeCount: number;
|
||||||
|
|
||||||
|
@Column('float', { array: true, nullable: true })
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
@Column({ name: 'is_active', default: false })
|
||||||
|
isActive: boolean;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,52 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
Column,
|
||||||
|
PrimaryColumn,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
Index,
|
||||||
|
} from 'typeorm';
|
||||||
|
|
||||||
|
@Entity('user_memories')
|
||||||
|
@Index('idx_user_memories_user', ['userId'])
|
||||||
|
@Index('idx_user_memories_type', ['memoryType'])
|
||||||
|
export class UserMemoryORM {
|
||||||
|
@PrimaryColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'user_id' })
|
||||||
|
userId: string;
|
||||||
|
|
||||||
|
@Column({ name: 'memory_type', length: 30 })
|
||||||
|
memoryType: string;
|
||||||
|
|
||||||
|
@Column('text')
|
||||||
|
content: string;
|
||||||
|
|
||||||
|
@Column({ default: 50 })
|
||||||
|
importance: number;
|
||||||
|
|
||||||
|
@Column({ name: 'source_conversation_id', nullable: true })
|
||||||
|
sourceConversationId?: string;
|
||||||
|
|
||||||
|
@Column({ name: 'related_category', length: 50, nullable: true })
|
||||||
|
relatedCategory?: string;
|
||||||
|
|
||||||
|
@Column('float', { array: true, nullable: true })
|
||||||
|
embedding?: number[];
|
||||||
|
|
||||||
|
@Column({ name: 'access_count', default: 0 })
|
||||||
|
accessCount: number;
|
||||||
|
|
||||||
|
@Column({ name: 'last_accessed_at', nullable: true })
|
||||||
|
lastAccessedAt?: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'is_expired', default: false })
|
||||||
|
isExpired: boolean;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,321 @@
|
||||||
|
import { Injectable } from '@nestjs/common';
|
||||||
|
import { InjectRepository } from '@nestjs/typeorm';
|
||||||
|
import { Repository, ILike } from 'typeorm';
|
||||||
|
import { IKnowledgeRepository } from '../../../domain/repositories/knowledge.repository.interface';
|
||||||
|
import { KnowledgeArticleEntity, KnowledgeSource } from '../../../domain/entities/knowledge-article.entity';
|
||||||
|
import { KnowledgeChunkEntity, ChunkType } from '../../../domain/entities/knowledge-chunk.entity';
|
||||||
|
import { KnowledgeArticleORM } from './entities/knowledge-article.orm';
|
||||||
|
import { KnowledgeChunkORM } from './entities/knowledge-chunk.orm';
|
||||||
|
|
||||||
|
@Injectable()
|
||||||
|
export class KnowledgePostgresRepository implements IKnowledgeRepository {
|
||||||
|
constructor(
|
||||||
|
@InjectRepository(KnowledgeArticleORM)
|
||||||
|
private articleRepo: Repository<KnowledgeArticleORM>,
|
||||||
|
@InjectRepository(KnowledgeChunkORM)
|
||||||
|
private chunkRepo: Repository<KnowledgeChunkORM>,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
// ========== 文章操作 ==========
|
||||||
|
|
||||||
|
async saveArticle(article: KnowledgeArticleEntity): Promise<void> {
|
||||||
|
const orm = this.toArticleORM(article);
|
||||||
|
await this.articleRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async findArticleById(id: string): Promise<KnowledgeArticleEntity | null> {
|
||||||
|
const orm = await this.articleRepo.findOne({ where: { id } });
|
||||||
|
return orm ? this.toArticleEntity(orm) : null;
|
||||||
|
}
|
||||||
|
|
||||||
|
async findArticlesByCategory(
|
||||||
|
category: string,
|
||||||
|
options?: { publishedOnly?: boolean; limit?: number; offset?: number },
|
||||||
|
): Promise<KnowledgeArticleEntity[]> {
|
||||||
|
const query = this.articleRepo.createQueryBuilder('article')
|
||||||
|
.where('article.category = :category', { category });
|
||||||
|
|
||||||
|
if (options?.publishedOnly) {
|
||||||
|
query.andWhere('article.isPublished = true');
|
||||||
|
}
|
||||||
|
|
||||||
|
query.orderBy('article.qualityScore', 'DESC')
|
||||||
|
.addOrderBy('article.createdAt', 'DESC');
|
||||||
|
|
||||||
|
if (options?.limit) {
|
||||||
|
query.take(options.limit);
|
||||||
|
}
|
||||||
|
if (options?.offset) {
|
||||||
|
query.skip(options.offset);
|
||||||
|
}
|
||||||
|
|
||||||
|
const orms = await query.getMany();
|
||||||
|
return orms.map(orm => this.toArticleEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async searchArticles(
|
||||||
|
queryStr: string,
|
||||||
|
options?: { category?: string; publishedOnly?: boolean; limit?: number },
|
||||||
|
): Promise<KnowledgeArticleEntity[]> {
|
||||||
|
const query = this.articleRepo.createQueryBuilder('article')
|
||||||
|
.where('(article.title ILIKE :search OR article.content ILIKE :search)', {
|
||||||
|
search: `%${queryStr}%`,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (options?.category) {
|
||||||
|
query.andWhere('article.category = :category', { category: options.category });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.publishedOnly) {
|
||||||
|
query.andWhere('article.isPublished = true');
|
||||||
|
}
|
||||||
|
|
||||||
|
query.orderBy('article.qualityScore', 'DESC')
|
||||||
|
.take(options?.limit || 10);
|
||||||
|
|
||||||
|
const orms = await query.getMany();
|
||||||
|
return orms.map(orm => this.toArticleEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async searchArticlesByVector(
|
||||||
|
embedding: number[],
|
||||||
|
options?: {
|
||||||
|
category?: string;
|
||||||
|
publishedOnly?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
},
|
||||||
|
): Promise<Array<{ article: KnowledgeArticleEntity; similarity: number }>> {
|
||||||
|
const embeddingStr = `[${embedding.join(',')}]`;
|
||||||
|
const limit = options?.limit || 5;
|
||||||
|
const minSimilarity = options?.minSimilarity || 0.7;
|
||||||
|
|
||||||
|
let sql = `
|
||||||
|
SELECT *,
|
||||||
|
1 - (embedding <=> '${embeddingStr}'::vector) as similarity
|
||||||
|
FROM knowledge_articles
|
||||||
|
WHERE embedding IS NOT NULL
|
||||||
|
`;
|
||||||
|
|
||||||
|
if (options?.category) {
|
||||||
|
sql += ` AND category = '${options.category}'`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.publishedOnly) {
|
||||||
|
sql += ` AND is_published = true`;
|
||||||
|
}
|
||||||
|
|
||||||
|
sql += `
|
||||||
|
HAVING 1 - (embedding <=> '${embeddingStr}'::vector) >= ${minSimilarity}
|
||||||
|
ORDER BY similarity DESC
|
||||||
|
LIMIT ${limit}
|
||||||
|
`;
|
||||||
|
|
||||||
|
// 使用原生查询以利用pgvector
|
||||||
|
const results = await this.articleRepo.query(sql);
|
||||||
|
|
||||||
|
return results.map((row: any) => ({
|
||||||
|
article: this.toArticleEntityFromRaw(row),
|
||||||
|
similarity: parseFloat(row.similarity),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
async updateArticle(article: KnowledgeArticleEntity): Promise<void> {
|
||||||
|
const orm = this.toArticleORM(article);
|
||||||
|
await this.articleRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteArticle(id: string): Promise<void> {
|
||||||
|
await this.articleRepo.delete(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
async countArticles(options?: { category?: string; publishedOnly?: boolean }): Promise<number> {
|
||||||
|
const query = this.articleRepo.createQueryBuilder('article');
|
||||||
|
|
||||||
|
if (options?.category) {
|
||||||
|
query.andWhere('article.category = :category', { category: options.category });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.publishedOnly) {
|
||||||
|
query.andWhere('article.isPublished = true');
|
||||||
|
}
|
||||||
|
|
||||||
|
return query.getCount();
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 块操作 ==========
|
||||||
|
|
||||||
|
async saveChunk(chunk: KnowledgeChunkEntity): Promise<void> {
|
||||||
|
const orm = this.toChunkORM(chunk);
|
||||||
|
await this.chunkRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async saveChunks(chunks: KnowledgeChunkEntity[]): Promise<void> {
|
||||||
|
const orms = chunks.map(chunk => this.toChunkORM(chunk));
|
||||||
|
await this.chunkRepo.save(orms);
|
||||||
|
}
|
||||||
|
|
||||||
|
async findChunksByArticleId(articleId: string): Promise<KnowledgeChunkEntity[]> {
|
||||||
|
const orms = await this.chunkRepo.find({
|
||||||
|
where: { articleId },
|
||||||
|
order: { chunkIndex: 'ASC' },
|
||||||
|
});
|
||||||
|
return orms.map(orm => this.toChunkEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async searchChunksByVector(
|
||||||
|
embedding: number[],
|
||||||
|
options?: {
|
||||||
|
category?: string;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
},
|
||||||
|
): Promise<Array<{ chunk: KnowledgeChunkEntity; similarity: number }>> {
|
||||||
|
const embeddingStr = `[${embedding.join(',')}]`;
|
||||||
|
const limit = options?.limit || 5;
|
||||||
|
const minSimilarity = options?.minSimilarity || 0.7;
|
||||||
|
|
||||||
|
let sql = `
|
||||||
|
SELECT c.*,
|
||||||
|
1 - (c.embedding <=> '${embeddingStr}'::vector) as similarity
|
||||||
|
FROM knowledge_chunks c
|
||||||
|
JOIN knowledge_articles a ON c.article_id = a.id
|
||||||
|
WHERE c.embedding IS NOT NULL
|
||||||
|
AND a.is_published = true
|
||||||
|
`;
|
||||||
|
|
||||||
|
if (options?.category) {
|
||||||
|
sql += ` AND a.category = '${options.category}'`;
|
||||||
|
}
|
||||||
|
|
||||||
|
sql += `
|
||||||
|
AND 1 - (c.embedding <=> '${embeddingStr}'::vector) >= ${minSimilarity}
|
||||||
|
ORDER BY similarity DESC
|
||||||
|
LIMIT ${limit}
|
||||||
|
`;
|
||||||
|
|
||||||
|
const results = await this.chunkRepo.query(sql);
|
||||||
|
|
||||||
|
return results.map((row: any) => ({
|
||||||
|
chunk: this.toChunkEntityFromRaw(row),
|
||||||
|
similarity: parseFloat(row.similarity),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteChunksByArticleId(articleId: string): Promise<void> {
|
||||||
|
await this.chunkRepo.delete({ articleId });
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 转换方法 ==========
|
||||||
|
|
||||||
|
private toArticleORM(entity: KnowledgeArticleEntity): KnowledgeArticleORM {
|
||||||
|
const orm = new KnowledgeArticleORM();
|
||||||
|
orm.id = entity.id;
|
||||||
|
orm.title = entity.title;
|
||||||
|
orm.content = entity.content;
|
||||||
|
orm.summary = entity.summary;
|
||||||
|
orm.category = entity.category;
|
||||||
|
orm.tags = entity.tags;
|
||||||
|
orm.source = entity.source;
|
||||||
|
orm.sourceUrl = entity.sourceUrl;
|
||||||
|
orm.embedding = entity.embedding;
|
||||||
|
orm.isPublished = entity.isPublished;
|
||||||
|
orm.citationCount = entity.citationCount;
|
||||||
|
orm.helpfulCount = entity.helpfulCount;
|
||||||
|
orm.unhelpfulCount = entity.unhelpfulCount;
|
||||||
|
orm.qualityScore = entity.qualityScore;
|
||||||
|
orm.createdBy = entity.createdBy;
|
||||||
|
orm.updatedBy = entity.updatedBy;
|
||||||
|
orm.createdAt = entity.createdAt;
|
||||||
|
orm.updatedAt = entity.updatedAt;
|
||||||
|
return orm;
|
||||||
|
}
|
||||||
|
|
||||||
|
private toArticleEntity(orm: KnowledgeArticleORM): KnowledgeArticleEntity {
|
||||||
|
return KnowledgeArticleEntity.fromPersistence({
|
||||||
|
id: orm.id,
|
||||||
|
title: orm.title,
|
||||||
|
content: orm.content,
|
||||||
|
summary: orm.summary,
|
||||||
|
category: orm.category,
|
||||||
|
tags: orm.tags,
|
||||||
|
source: orm.source as KnowledgeSource,
|
||||||
|
sourceUrl: orm.sourceUrl,
|
||||||
|
embedding: orm.embedding,
|
||||||
|
isPublished: orm.isPublished,
|
||||||
|
citationCount: orm.citationCount,
|
||||||
|
helpfulCount: orm.helpfulCount,
|
||||||
|
unhelpfulCount: orm.unhelpfulCount,
|
||||||
|
qualityScore: orm.qualityScore,
|
||||||
|
createdBy: orm.createdBy,
|
||||||
|
updatedBy: orm.updatedBy,
|
||||||
|
createdAt: orm.createdAt,
|
||||||
|
updatedAt: orm.updatedAt,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private toArticleEntityFromRaw(row: any): KnowledgeArticleEntity {
|
||||||
|
return KnowledgeArticleEntity.fromPersistence({
|
||||||
|
id: row.id,
|
||||||
|
title: row.title,
|
||||||
|
content: row.content,
|
||||||
|
summary: row.summary,
|
||||||
|
category: row.category,
|
||||||
|
tags: row.tags,
|
||||||
|
source: row.source as KnowledgeSource,
|
||||||
|
sourceUrl: row.source_url,
|
||||||
|
embedding: row.embedding,
|
||||||
|
isPublished: row.is_published,
|
||||||
|
citationCount: row.citation_count,
|
||||||
|
helpfulCount: row.helpful_count,
|
||||||
|
unhelpfulCount: row.unhelpful_count,
|
||||||
|
qualityScore: row.quality_score,
|
||||||
|
createdBy: row.created_by,
|
||||||
|
updatedBy: row.updated_by,
|
||||||
|
createdAt: new Date(row.created_at),
|
||||||
|
updatedAt: new Date(row.updated_at),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private toChunkORM(entity: KnowledgeChunkEntity): KnowledgeChunkORM {
|
||||||
|
const orm = new KnowledgeChunkORM();
|
||||||
|
orm.id = entity.id;
|
||||||
|
orm.articleId = entity.articleId;
|
||||||
|
orm.content = entity.content;
|
||||||
|
orm.chunkIndex = entity.chunkIndex;
|
||||||
|
orm.chunkType = entity.chunkType;
|
||||||
|
orm.embedding = entity.embedding;
|
||||||
|
orm.metadata = entity.metadata as Record<string, unknown>;
|
||||||
|
orm.tokenCount = entity.tokenCount;
|
||||||
|
orm.createdAt = entity.createdAt;
|
||||||
|
return orm;
|
||||||
|
}
|
||||||
|
|
||||||
|
private toChunkEntity(orm: KnowledgeChunkORM): KnowledgeChunkEntity {
|
||||||
|
return KnowledgeChunkEntity.fromPersistence({
|
||||||
|
id: orm.id,
|
||||||
|
articleId: orm.articleId,
|
||||||
|
content: orm.content,
|
||||||
|
chunkIndex: orm.chunkIndex,
|
||||||
|
chunkType: orm.chunkType as ChunkType,
|
||||||
|
embedding: orm.embedding,
|
||||||
|
metadata: orm.metadata,
|
||||||
|
tokenCount: orm.tokenCount,
|
||||||
|
createdAt: orm.createdAt,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private toChunkEntityFromRaw(row: any): KnowledgeChunkEntity {
|
||||||
|
return KnowledgeChunkEntity.fromPersistence({
|
||||||
|
id: row.id,
|
||||||
|
articleId: row.article_id,
|
||||||
|
content: row.content,
|
||||||
|
chunkIndex: row.chunk_index,
|
||||||
|
chunkType: row.chunk_type as ChunkType,
|
||||||
|
embedding: row.embedding,
|
||||||
|
metadata: row.metadata,
|
||||||
|
tokenCount: row.token_count,
|
||||||
|
createdAt: new Date(row.created_at),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,426 @@
|
||||||
|
import { Injectable } from '@nestjs/common';
|
||||||
|
import { InjectRepository } from '@nestjs/typeorm';
|
||||||
|
import { Repository, LessThan, MoreThan } from 'typeorm';
|
||||||
|
import {
|
||||||
|
IUserMemoryRepository,
|
||||||
|
ISystemExperienceRepository,
|
||||||
|
} from '../../../domain/repositories/memory.repository.interface';
|
||||||
|
import { UserMemoryEntity, MemoryType } from '../../../domain/entities/user-memory.entity';
|
||||||
|
import {
|
||||||
|
SystemExperienceEntity,
|
||||||
|
ExperienceType,
|
||||||
|
VerificationStatus,
|
||||||
|
} from '../../../domain/entities/system-experience.entity';
|
||||||
|
import { UserMemoryORM } from './entities/user-memory.orm';
|
||||||
|
import { SystemExperienceORM } from './entities/system-experience.orm';
|
||||||
|
|
||||||
|
@Injectable()
|
||||||
|
export class UserMemoryPostgresRepository implements IUserMemoryRepository {
|
||||||
|
constructor(
|
||||||
|
@InjectRepository(UserMemoryORM)
|
||||||
|
private memoryRepo: Repository<UserMemoryORM>,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
async save(memory: UserMemoryEntity): Promise<void> {
|
||||||
|
const orm = this.toORM(memory);
|
||||||
|
await this.memoryRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async findById(id: string): Promise<UserMemoryEntity | null> {
|
||||||
|
const orm = await this.memoryRepo.findOne({ where: { id } });
|
||||||
|
return orm ? this.toEntity(orm) : null;
|
||||||
|
}
|
||||||
|
|
||||||
|
async findByUserId(
|
||||||
|
userId: string,
|
||||||
|
options?: { memoryType?: MemoryType; includeExpired?: boolean; limit?: number },
|
||||||
|
): Promise<UserMemoryEntity[]> {
|
||||||
|
const query = this.memoryRepo.createQueryBuilder('memory')
|
||||||
|
.where('memory.userId = :userId', { userId });
|
||||||
|
|
||||||
|
if (options?.memoryType) {
|
||||||
|
query.andWhere('memory.memoryType = :type', { type: options.memoryType });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!options?.includeExpired) {
|
||||||
|
query.andWhere('memory.isExpired = false');
|
||||||
|
}
|
||||||
|
|
||||||
|
query.orderBy('memory.importance', 'DESC')
|
||||||
|
.addOrderBy('memory.createdAt', 'DESC');
|
||||||
|
|
||||||
|
if (options?.limit) {
|
||||||
|
query.take(options.limit);
|
||||||
|
}
|
||||||
|
|
||||||
|
const orms = await query.getMany();
|
||||||
|
return orms.map(orm => this.toEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async searchByVector(
|
||||||
|
userId: string,
|
||||||
|
embedding: number[],
|
||||||
|
options?: { memoryType?: MemoryType; limit?: number; minSimilarity?: number },
|
||||||
|
): Promise<Array<{ memory: UserMemoryEntity; similarity: number }>> {
|
||||||
|
const embeddingStr = `[${embedding.join(',')}]`;
|
||||||
|
const limit = options?.limit || 5;
|
||||||
|
const minSimilarity = options?.minSimilarity || 0.7;
|
||||||
|
|
||||||
|
let sql = `
|
||||||
|
SELECT *,
|
||||||
|
1 - (embedding <=> '${embeddingStr}'::vector) as similarity
|
||||||
|
FROM user_memories
|
||||||
|
WHERE user_id = '${userId}'
|
||||||
|
AND embedding IS NOT NULL
|
||||||
|
AND is_expired = false
|
||||||
|
`;
|
||||||
|
|
||||||
|
if (options?.memoryType) {
|
||||||
|
sql += ` AND memory_type = '${options.memoryType}'`;
|
||||||
|
}
|
||||||
|
|
||||||
|
sql += `
|
||||||
|
AND 1 - (embedding <=> '${embeddingStr}'::vector) >= ${minSimilarity}
|
||||||
|
ORDER BY similarity DESC
|
||||||
|
LIMIT ${limit}
|
||||||
|
`;
|
||||||
|
|
||||||
|
const results = await this.memoryRepo.query(sql);
|
||||||
|
|
||||||
|
return results.map((row: any) => ({
|
||||||
|
memory: this.toEntityFromRaw(row),
|
||||||
|
similarity: parseFloat(row.similarity),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
async findTopMemories(userId: string, limit: number): Promise<UserMemoryEntity[]> {
|
||||||
|
const orms = await this.memoryRepo.find({
|
||||||
|
where: { userId, isExpired: false },
|
||||||
|
order: { importance: 'DESC', accessCount: 'DESC' },
|
||||||
|
take: limit,
|
||||||
|
});
|
||||||
|
return orms.map(orm => this.toEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async update(memory: UserMemoryEntity): Promise<void> {
|
||||||
|
const orm = this.toORM(memory);
|
||||||
|
await this.memoryRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async delete(id: string): Promise<void> {
|
||||||
|
await this.memoryRepo.delete(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteByUserId(userId: string): Promise<void> {
|
||||||
|
await this.memoryRepo.delete({ userId });
|
||||||
|
}
|
||||||
|
|
||||||
|
async markExpiredMemories(userId: string, olderThanDays: number): Promise<number> {
|
||||||
|
const cutoffDate = new Date();
|
||||||
|
cutoffDate.setDate(cutoffDate.getDate() - olderThanDays);
|
||||||
|
|
||||||
|
const result = await this.memoryRepo.update(
|
||||||
|
{
|
||||||
|
userId,
|
||||||
|
isExpired: false,
|
||||||
|
updatedAt: LessThan(cutoffDate),
|
||||||
|
},
|
||||||
|
{ isExpired: true },
|
||||||
|
);
|
||||||
|
|
||||||
|
return result.affected || 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
private toORM(entity: UserMemoryEntity): UserMemoryORM {
|
||||||
|
const orm = new UserMemoryORM();
|
||||||
|
orm.id = entity.id;
|
||||||
|
orm.userId = entity.userId;
|
||||||
|
orm.memoryType = entity.memoryType;
|
||||||
|
orm.content = entity.content;
|
||||||
|
orm.importance = entity.importance;
|
||||||
|
orm.sourceConversationId = entity.sourceConversationId;
|
||||||
|
orm.relatedCategory = entity.relatedCategory;
|
||||||
|
orm.embedding = entity.embedding;
|
||||||
|
orm.accessCount = entity.accessCount;
|
||||||
|
orm.lastAccessedAt = entity.lastAccessedAt;
|
||||||
|
orm.isExpired = entity.isExpired;
|
||||||
|
orm.createdAt = entity.createdAt;
|
||||||
|
orm.updatedAt = entity.updatedAt;
|
||||||
|
return orm;
|
||||||
|
}
|
||||||
|
|
||||||
|
private toEntity(orm: UserMemoryORM): UserMemoryEntity {
|
||||||
|
return UserMemoryEntity.fromPersistence({
|
||||||
|
id: orm.id,
|
||||||
|
userId: orm.userId,
|
||||||
|
memoryType: orm.memoryType as MemoryType,
|
||||||
|
content: orm.content,
|
||||||
|
importance: orm.importance,
|
||||||
|
sourceConversationId: orm.sourceConversationId,
|
||||||
|
relatedCategory: orm.relatedCategory,
|
||||||
|
embedding: orm.embedding,
|
||||||
|
accessCount: orm.accessCount,
|
||||||
|
lastAccessedAt: orm.lastAccessedAt,
|
||||||
|
isExpired: orm.isExpired,
|
||||||
|
createdAt: orm.createdAt,
|
||||||
|
updatedAt: orm.updatedAt,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private toEntityFromRaw(row: any): UserMemoryEntity {
|
||||||
|
return UserMemoryEntity.fromPersistence({
|
||||||
|
id: row.id,
|
||||||
|
userId: row.user_id,
|
||||||
|
memoryType: row.memory_type as MemoryType,
|
||||||
|
content: row.content,
|
||||||
|
importance: row.importance,
|
||||||
|
sourceConversationId: row.source_conversation_id,
|
||||||
|
relatedCategory: row.related_category,
|
||||||
|
embedding: row.embedding,
|
||||||
|
accessCount: row.access_count,
|
||||||
|
lastAccessedAt: row.last_accessed_at ? new Date(row.last_accessed_at) : undefined,
|
||||||
|
isExpired: row.is_expired,
|
||||||
|
createdAt: new Date(row.created_at),
|
||||||
|
updatedAt: new Date(row.updated_at),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Injectable()
|
||||||
|
export class SystemExperiencePostgresRepository implements ISystemExperienceRepository {
|
||||||
|
constructor(
|
||||||
|
@InjectRepository(SystemExperienceORM)
|
||||||
|
private experienceRepo: Repository<SystemExperienceORM>,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
async save(experience: SystemExperienceEntity): Promise<void> {
|
||||||
|
const orm = this.toORM(experience);
|
||||||
|
await this.experienceRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async findById(id: string): Promise<SystemExperienceEntity | null> {
|
||||||
|
const orm = await this.experienceRepo.findOne({ where: { id } });
|
||||||
|
return orm ? this.toEntity(orm) : null;
|
||||||
|
}
|
||||||
|
|
||||||
|
async findPendingExperiences(options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
limit?: number;
|
||||||
|
offset?: number;
|
||||||
|
}): Promise<SystemExperienceEntity[]> {
|
||||||
|
const query = this.experienceRepo.createQueryBuilder('exp')
|
||||||
|
.where('exp.verificationStatus = :status', { status: VerificationStatus.PENDING });
|
||||||
|
|
||||||
|
if (options?.experienceType) {
|
||||||
|
query.andWhere('exp.experienceType = :type', { type: options.experienceType });
|
||||||
|
}
|
||||||
|
|
||||||
|
query.orderBy('exp.confidence', 'DESC')
|
||||||
|
.addOrderBy('exp.createdAt', 'DESC');
|
||||||
|
|
||||||
|
if (options?.limit) query.take(options.limit);
|
||||||
|
if (options?.offset) query.skip(options.offset);
|
||||||
|
|
||||||
|
const orms = await query.getMany();
|
||||||
|
return orms.map(orm => this.toEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async findActiveExperiences(options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
category?: string;
|
||||||
|
minConfidence?: number;
|
||||||
|
limit?: number;
|
||||||
|
}): Promise<SystemExperienceEntity[]> {
|
||||||
|
const query = this.experienceRepo.createQueryBuilder('exp')
|
||||||
|
.where('exp.isActive = true');
|
||||||
|
|
||||||
|
if (options?.experienceType) {
|
||||||
|
query.andWhere('exp.experienceType = :type', { type: options.experienceType });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.category) {
|
||||||
|
query.andWhere('exp.relatedCategory = :category', { category: options.category });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.minConfidence) {
|
||||||
|
query.andWhere('exp.confidence >= :minConf', { minConf: options.minConfidence });
|
||||||
|
}
|
||||||
|
|
||||||
|
query.orderBy('exp.confidence', 'DESC');
|
||||||
|
|
||||||
|
if (options?.limit) query.take(options.limit);
|
||||||
|
|
||||||
|
const orms = await query.getMany();
|
||||||
|
return orms.map(orm => this.toEntity(orm));
|
||||||
|
}
|
||||||
|
|
||||||
|
async searchByVector(
|
||||||
|
embedding: number[],
|
||||||
|
options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
activeOnly?: boolean;
|
||||||
|
limit?: number;
|
||||||
|
minSimilarity?: number;
|
||||||
|
},
|
||||||
|
): Promise<Array<{ experience: SystemExperienceEntity; similarity: number }>> {
|
||||||
|
const embeddingStr = `[${embedding.join(',')}]`;
|
||||||
|
const limit = options?.limit || 5;
|
||||||
|
const minSimilarity = options?.minSimilarity || 0.7;
|
||||||
|
|
||||||
|
let sql = `
|
||||||
|
SELECT *,
|
||||||
|
1 - (embedding <=> '${embeddingStr}'::vector) as similarity
|
||||||
|
FROM system_experiences
|
||||||
|
WHERE embedding IS NOT NULL
|
||||||
|
`;
|
||||||
|
|
||||||
|
if (options?.activeOnly !== false) {
|
||||||
|
sql += ` AND is_active = true`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (options?.experienceType) {
|
||||||
|
sql += ` AND experience_type = '${options.experienceType}'`;
|
||||||
|
}
|
||||||
|
|
||||||
|
sql += `
|
||||||
|
AND 1 - (embedding <=> '${embeddingStr}'::vector) >= ${minSimilarity}
|
||||||
|
ORDER BY similarity DESC
|
||||||
|
LIMIT ${limit}
|
||||||
|
`;
|
||||||
|
|
||||||
|
const results = await this.experienceRepo.query(sql);
|
||||||
|
|
||||||
|
return results.map((row: any) => ({
|
||||||
|
experience: this.toEntityFromRaw(row),
|
||||||
|
similarity: parseFloat(row.similarity),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
async findSimilarExperiences(
|
||||||
|
embedding: number[],
|
||||||
|
threshold: number,
|
||||||
|
): Promise<SystemExperienceEntity[]> {
|
||||||
|
const embeddingStr = `[${embedding.join(',')}]`;
|
||||||
|
|
||||||
|
const sql = `
|
||||||
|
SELECT *
|
||||||
|
FROM system_experiences
|
||||||
|
WHERE embedding IS NOT NULL
|
||||||
|
AND 1 - (embedding <=> '${embeddingStr}'::vector) >= ${threshold}
|
||||||
|
ORDER BY 1 - (embedding <=> '${embeddingStr}'::vector) DESC
|
||||||
|
LIMIT 10
|
||||||
|
`;
|
||||||
|
|
||||||
|
const results = await this.experienceRepo.query(sql);
|
||||||
|
return results.map((row: any) => this.toEntityFromRaw(row));
|
||||||
|
}
|
||||||
|
|
||||||
|
async update(experience: SystemExperienceEntity): Promise<void> {
|
||||||
|
const orm = this.toORM(experience);
|
||||||
|
await this.experienceRepo.save(orm);
|
||||||
|
}
|
||||||
|
|
||||||
|
async delete(id: string): Promise<void> {
|
||||||
|
await this.experienceRepo.delete(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getStatistics(): Promise<{
|
||||||
|
total: number;
|
||||||
|
byStatus: Record<VerificationStatus, number>;
|
||||||
|
byType: Record<ExperienceType, number>;
|
||||||
|
}> {
|
||||||
|
const total = await this.experienceRepo.count();
|
||||||
|
|
||||||
|
const statusCounts = await this.experienceRepo
|
||||||
|
.createQueryBuilder('exp')
|
||||||
|
.select('exp.verificationStatus', 'status')
|
||||||
|
.addSelect('COUNT(*)', 'count')
|
||||||
|
.groupBy('exp.verificationStatus')
|
||||||
|
.getRawMany();
|
||||||
|
|
||||||
|
const typeCounts = await this.experienceRepo
|
||||||
|
.createQueryBuilder('exp')
|
||||||
|
.select('exp.experienceType', 'type')
|
||||||
|
.addSelect('COUNT(*)', 'count')
|
||||||
|
.groupBy('exp.experienceType')
|
||||||
|
.getRawMany();
|
||||||
|
|
||||||
|
const byStatus = {} as Record<VerificationStatus, number>;
|
||||||
|
statusCounts.forEach((item: any) => {
|
||||||
|
byStatus[item.status as VerificationStatus] = parseInt(item.count);
|
||||||
|
});
|
||||||
|
|
||||||
|
const byType = {} as Record<ExperienceType, number>;
|
||||||
|
typeCounts.forEach((item: any) => {
|
||||||
|
byType[item.type as ExperienceType] = parseInt(item.count);
|
||||||
|
});
|
||||||
|
|
||||||
|
return { total, byStatus, byType };
|
||||||
|
}
|
||||||
|
|
||||||
|
private toORM(entity: SystemExperienceEntity): SystemExperienceORM {
|
||||||
|
const orm = new SystemExperienceORM();
|
||||||
|
orm.id = entity.id;
|
||||||
|
orm.experienceType = entity.experienceType;
|
||||||
|
orm.content = entity.content;
|
||||||
|
orm.confidence = entity.confidence;
|
||||||
|
orm.scenario = entity.scenario;
|
||||||
|
orm.relatedCategory = entity.relatedCategory;
|
||||||
|
orm.sourceConversationIds = entity.sourceConversationIds;
|
||||||
|
orm.verificationStatus = entity.verificationStatus;
|
||||||
|
orm.verifiedBy = entity.verifiedBy;
|
||||||
|
orm.verifiedAt = entity.verifiedAt;
|
||||||
|
orm.usageCount = entity.usageCount;
|
||||||
|
orm.positiveCount = entity.positiveCount;
|
||||||
|
orm.negativeCount = entity.negativeCount;
|
||||||
|
orm.embedding = entity.embedding;
|
||||||
|
orm.isActive = entity.isActive;
|
||||||
|
orm.createdAt = entity.createdAt;
|
||||||
|
orm.updatedAt = entity.updatedAt;
|
||||||
|
return orm;
|
||||||
|
}
|
||||||
|
|
||||||
|
private toEntity(orm: SystemExperienceORM): SystemExperienceEntity {
|
||||||
|
return SystemExperienceEntity.fromPersistence({
|
||||||
|
id: orm.id,
|
||||||
|
experienceType: orm.experienceType as ExperienceType,
|
||||||
|
content: orm.content,
|
||||||
|
confidence: orm.confidence,
|
||||||
|
scenario: orm.scenario,
|
||||||
|
relatedCategory: orm.relatedCategory,
|
||||||
|
sourceConversationIds: orm.sourceConversationIds,
|
||||||
|
verificationStatus: orm.verificationStatus as VerificationStatus,
|
||||||
|
verifiedBy: orm.verifiedBy,
|
||||||
|
verifiedAt: orm.verifiedAt,
|
||||||
|
usageCount: orm.usageCount,
|
||||||
|
positiveCount: orm.positiveCount,
|
||||||
|
negativeCount: orm.negativeCount,
|
||||||
|
embedding: orm.embedding,
|
||||||
|
isActive: orm.isActive,
|
||||||
|
createdAt: orm.createdAt,
|
||||||
|
updatedAt: orm.updatedAt,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private toEntityFromRaw(row: any): SystemExperienceEntity {
|
||||||
|
return SystemExperienceEntity.fromPersistence({
|
||||||
|
id: row.id,
|
||||||
|
experienceType: row.experience_type as ExperienceType,
|
||||||
|
content: row.content,
|
||||||
|
confidence: row.confidence,
|
||||||
|
scenario: row.scenario,
|
||||||
|
relatedCategory: row.related_category,
|
||||||
|
sourceConversationIds: row.source_conversation_ids,
|
||||||
|
verificationStatus: row.verification_status as VerificationStatus,
|
||||||
|
verifiedBy: row.verified_by,
|
||||||
|
verifiedAt: row.verified_at ? new Date(row.verified_at) : undefined,
|
||||||
|
usageCount: row.usage_count,
|
||||||
|
positiveCount: row.positive_count,
|
||||||
|
negativeCount: row.negative_count,
|
||||||
|
embedding: row.embedding,
|
||||||
|
isActive: row.is_active,
|
||||||
|
createdAt: new Date(row.created_at),
|
||||||
|
updatedAt: new Date(row.updated_at),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,153 @@
|
||||||
|
import { Injectable, OnModuleInit } from '@nestjs/common';
|
||||||
|
import { ConfigService } from '@nestjs/config';
|
||||||
|
import OpenAI from 'openai';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 向量嵌入服务 - 使用OpenAI的text-embedding-3-small模型
|
||||||
|
* 用于将文本转换为向量，支持语义搜索
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class EmbeddingService implements OnModuleInit {
|
||||||
|
private openai: OpenAI;
|
||||||
|
private readonly modelName = 'text-embedding-3-small';
|
||||||
|
private readonly dimensions = 1536; // text-embedding-3-small 默认维度
|
||||||
|
|
||||||
|
constructor(private configService: ConfigService) {}
|
||||||
|
|
||||||
|
onModuleInit() {
|
||||||
|
const apiKey = this.configService.get<string>('OPENAI_API_KEY');
|
||||||
|
|
||||||
|
if (!apiKey) {
|
||||||
|
console.warn('[EmbeddingService] OPENAI_API_KEY not set, using mock embeddings');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.openai = new OpenAI({
|
||||||
|
apiKey,
|
||||||
|
baseURL: this.configService.get<string>('OPENAI_BASE_URL'), // 支持自定义endpoint
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log('[EmbeddingService] Initialized with OpenAI embedding model');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取单个文本的向量
|
||||||
|
*/
|
||||||
|
async getEmbedding(text: string): Promise<number[]> {
|
||||||
|
if (!this.openai) {
|
||||||
|
return this.getMockEmbedding(text);
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await this.openai.embeddings.create({
|
||||||
|
model: this.modelName,
|
||||||
|
input: this.preprocessText(text),
|
||||||
|
});
|
||||||
|
|
||||||
|
return response.data[0].embedding;
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[EmbeddingService] Failed to get embedding:', error);
|
||||||
|
// 降级到mock embedding
|
||||||
|
return this.getMockEmbedding(text);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 批量获取向量
|
||||||
|
*/
|
||||||
|
async getEmbeddings(texts: string[]): Promise<number[][]> {
|
||||||
|
if (!this.openai) {
|
||||||
|
return texts.map(text => this.getMockEmbedding(text));
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const processedTexts = texts.map(text => this.preprocessText(text));
|
||||||
|
|
||||||
|
const response = await this.openai.embeddings.create({
|
||||||
|
model: this.modelName,
|
||||||
|
input: processedTexts,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 按原始顺序返回
|
||||||
|
return response.data
|
||||||
|
.sort((a, b) => a.index - b.index)
|
||||||
|
.map(item => item.embedding);
|
||||||
|
} catch (error) {
|
||||||
|
console.error('[EmbeddingService] Failed to get batch embeddings:', error);
|
||||||
|
return texts.map(text => this.getMockEmbedding(text));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 计算两个向量的余弦相似度
|
||||||
|
*/
|
||||||
|
cosineSimilarity(a: number[], b: number[]): number {
|
||||||
|
if (a.length !== b.length) {
|
||||||
|
throw new Error('Vectors must have the same length');
|
||||||
|
}
|
||||||
|
|
||||||
|
let dotProduct = 0;
|
||||||
|
let normA = 0;
|
||||||
|
let normB = 0;
|
||||||
|
|
||||||
|
for (let i = 0; i < a.length; i++) {
|
||||||
|
dotProduct += a[i] * b[i];
|
||||||
|
normA += a[i] * a[i];
|
||||||
|
normB += b[i] * b[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
if (normA === 0 || normB === 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return dotProduct / (Math.sqrt(normA) * Math.sqrt(normB));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取向量维度
|
||||||
|
*/
|
||||||
|
getDimensions(): number {
|
||||||
|
return this.dimensions;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 预处理文本
|
||||||
|
*/
|
||||||
|
private preprocessText(text: string): string {
|
||||||
|
return text
|
||||||
|
.replace(/\s+/g, ' ') // 合并多个空白
|
||||||
|
.trim()
|
||||||
|
.substring(0, 8000); // OpenAI限制
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成Mock向量（用于开发测试）
|
||||||
|
* 基于文本hash生成确定性的伪随机向量
|
||||||
|
*/
|
||||||
|
private getMockEmbedding(text: string): number[] {
|
||||||
|
const embedding: number[] = [];
|
||||||
|
let hash = 0;
|
||||||
|
|
||||||
|
// 简单hash
|
||||||
|
for (let i = 0; i < text.length; i++) {
|
||||||
|
const char = text.charCodeAt(i);
|
||||||
|
hash = ((hash << 5) - hash) + char;
|
||||||
|
hash = hash & hash;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 基于hash生成伪随机向量
|
||||||
|
const random = (seed: number) => {
|
||||||
|
const x = Math.sin(seed) * 10000;
|
||||||
|
return x - Math.floor(x);
|
||||||
|
};
|
||||||
|
|
||||||
|
for (let i = 0; i < this.dimensions; i++) {
|
||||||
|
const value = random(hash + i) * 2 - 1; // -1 to 1
|
||||||
|
embedding.push(value);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 归一化
|
||||||
|
const norm = Math.sqrt(embedding.reduce((sum, v) => sum + v * v, 0));
|
||||||
|
return embedding.map(v => v / norm);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,267 @@
|
||||||
|
import {
|
||||||
|
Controller,
|
||||||
|
Get,
|
||||||
|
Post,
|
||||||
|
Put,
|
||||||
|
Delete,
|
||||||
|
Body,
|
||||||
|
Param,
|
||||||
|
Query,
|
||||||
|
HttpCode,
|
||||||
|
HttpStatus,
|
||||||
|
} from '@nestjs/common';
|
||||||
|
import { KnowledgeService } from './knowledge.service';
|
||||||
|
import { RAGService } from '../application/services/rag.service';
|
||||||
|
import { KnowledgeSource } from '../domain/entities/knowledge-article.entity';
|
||||||
|
|
||||||
|
// ========== DTOs ==========
|
||||||
|
|
||||||
|
class CreateArticleDto {
|
||||||
|
title: string;
|
||||||
|
content: string;
|
||||||
|
category: string;
|
||||||
|
tags?: string[];
|
||||||
|
sourceUrl?: string;
|
||||||
|
autoPublish?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
class UpdateArticleDto {
|
||||||
|
title?: string;
|
||||||
|
content?: string;
|
||||||
|
category?: string;
|
||||||
|
tags?: string[];
|
||||||
|
}
|
||||||
|
|
||||||
|
class SearchArticlesDto {
|
||||||
|
query: string;
|
||||||
|
category?: string;
|
||||||
|
useVector?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
class RetrieveKnowledgeDto {
|
||||||
|
query: string;
|
||||||
|
userId?: string;
|
||||||
|
category?: string;
|
||||||
|
includeMemories?: boolean;
|
||||||
|
includeExperiences?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
class ImportArticlesDto {
|
||||||
|
articles: Array<{
|
||||||
|
title: string;
|
||||||
|
content: string;
|
||||||
|
category: string;
|
||||||
|
tags?: string[];
|
||||||
|
}>;
|
||||||
|
}
|
||||||
|
|
||||||
|
class FeedbackDto {
|
||||||
|
helpful: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== Controller ==========
|
||||||
|
|
||||||
|
@Controller('knowledge')
|
||||||
|
export class KnowledgeController {
|
||||||
|
constructor(
|
||||||
|
private knowledgeService: KnowledgeService,
|
||||||
|
private ragService: RAGService,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建文章
|
||||||
|
*/
|
||||||
|
@Post('articles')
|
||||||
|
@HttpCode(HttpStatus.CREATED)
|
||||||
|
async createArticle(@Body() dto: CreateArticleDto) {
|
||||||
|
const article = await this.knowledgeService.createArticle({
|
||||||
|
...dto,
|
||||||
|
source: KnowledgeSource.MANUAL,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: article,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取文章详情
|
||||||
|
*/
|
||||||
|
@Get('articles/:id')
|
||||||
|
async getArticle(@Param('id') id: string) {
|
||||||
|
const article = await this.knowledgeService.getArticle(id);
|
||||||
|
if (!article) {
|
||||||
|
return {
|
||||||
|
success: false,
|
||||||
|
error: 'Article not found',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: article,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取文章列表
|
||||||
|
*/
|
||||||
|
@Get('articles')
|
||||||
|
async listArticles(
|
||||||
|
@Query('category') category?: string,
|
||||||
|
@Query('publishedOnly') publishedOnly?: string,
|
||||||
|
@Query('page') page?: string,
|
||||||
|
@Query('pageSize') pageSize?: string,
|
||||||
|
) {
|
||||||
|
const result = await this.knowledgeService.listArticles({
|
||||||
|
category,
|
||||||
|
publishedOnly: publishedOnly === 'true',
|
||||||
|
page: page ? parseInt(page) : 1,
|
||||||
|
pageSize: pageSize ? parseInt(pageSize) : 20,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新文章
|
||||||
|
*/
|
||||||
|
@Put('articles/:id')
|
||||||
|
async updateArticle(
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() dto: UpdateArticleDto,
|
||||||
|
) {
|
||||||
|
const article = await this.knowledgeService.updateArticle(id, dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: article,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除文章
|
||||||
|
*/
|
||||||
|
@Delete('articles/:id')
|
||||||
|
@HttpCode(HttpStatus.NO_CONTENT)
|
||||||
|
async deleteArticle(@Param('id') id: string) {
|
||||||
|
await this.knowledgeService.deleteArticle(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 发布文章
|
||||||
|
*/
|
||||||
|
@Post('articles/:id/publish')
|
||||||
|
async publishArticle(@Param('id') id: string) {
|
||||||
|
await this.knowledgeService.publishArticle(id);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Article published',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 取消发布
|
||||||
|
*/
|
||||||
|
@Post('articles/:id/unpublish')
|
||||||
|
async unpublishArticle(@Param('id') id: string) {
|
||||||
|
await this.knowledgeService.unpublishArticle(id);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Article unpublished',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 搜索文章
|
||||||
|
*/
|
||||||
|
@Post('articles/search')
|
||||||
|
async searchArticles(@Body() dto: SearchArticlesDto) {
|
||||||
|
const results = await this.knowledgeService.searchArticles(dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: results,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录反馈
|
||||||
|
*/
|
||||||
|
@Post('articles/:id/feedback')
|
||||||
|
async recordFeedback(
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() dto: FeedbackDto,
|
||||||
|
) {
|
||||||
|
await this.knowledgeService.recordFeedback(id, dto.helpful);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Feedback recorded',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 批量导入
|
||||||
|
*/
|
||||||
|
@Post('articles/import')
|
||||||
|
async importArticles(@Body() dto: ImportArticlesDto) {
|
||||||
|
const result = await this.knowledgeService.importArticles(dto.articles);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取统计信息
|
||||||
|
*/
|
||||||
|
@Get('statistics')
|
||||||
|
async getStatistics() {
|
||||||
|
const stats = await this.knowledgeService.getStatistics();
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: stats,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== RAG检索接口 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* RAG知识检索（供对话服务调用）
|
||||||
|
*/
|
||||||
|
@Post('retrieve')
|
||||||
|
async retrieveKnowledge(@Body() dto: RetrieveKnowledgeDto) {
|
||||||
|
const result = await this.ragService.retrieve(dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检索并格式化为提示词
|
||||||
|
*/
|
||||||
|
@Post('retrieve/prompt')
|
||||||
|
async retrieveForPrompt(@Body() dto: RetrieveKnowledgeDto) {
|
||||||
|
const context = await this.ragService.retrieveForPrompt({
|
||||||
|
query: dto.query,
|
||||||
|
userId: dto.userId,
|
||||||
|
category: dto.category,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: { context },
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查是否离题
|
||||||
|
*/
|
||||||
|
@Post('check-off-topic')
|
||||||
|
async checkOffTopic(@Body() body: { query: string }) {
|
||||||
|
const result = await this.ragService.checkOffTopic(body.query);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,53 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { KnowledgeController } from './knowledge.controller';
|
||||||
|
import { KnowledgeService } from './knowledge.service';
|
||||||
|
import { RAGService } from '../application/services/rag.service';
|
||||||
|
import { ChunkingService } from '../application/services/chunking.service';
|
||||||
|
import { EmbeddingService } from '../infrastructure/embedding/embedding.service';
|
||||||
|
import { KnowledgePostgresRepository } from '../infrastructure/database/postgres/knowledge-postgres.repository';
|
||||||
|
import {
|
||||||
|
UserMemoryPostgresRepository,
|
||||||
|
SystemExperiencePostgresRepository,
|
||||||
|
} from '../infrastructure/database/postgres/memory-postgres.repository';
|
||||||
|
import { KnowledgeArticleORM } from '../infrastructure/database/postgres/entities/knowledge-article.orm';
|
||||||
|
import { KnowledgeChunkORM } from '../infrastructure/database/postgres/entities/knowledge-chunk.orm';
|
||||||
|
import { UserMemoryORM } from '../infrastructure/database/postgres/entities/user-memory.orm';
|
||||||
|
import { SystemExperienceORM } from '../infrastructure/database/postgres/entities/system-experience.orm';
|
||||||
|
import { KNOWLEDGE_REPOSITORY } from '../domain/repositories/knowledge.repository.interface';
|
||||||
|
import {
|
||||||
|
USER_MEMORY_REPOSITORY,
|
||||||
|
SYSTEM_EXPERIENCE_REPOSITORY,
|
||||||
|
} from '../domain/repositories/memory.repository.interface';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
TypeOrmModule.forFeature([
|
||||||
|
KnowledgeArticleORM,
|
||||||
|
KnowledgeChunkORM,
|
||||||
|
UserMemoryORM,
|
||||||
|
SystemExperienceORM,
|
||||||
|
]),
|
||||||
|
],
|
||||||
|
controllers: [KnowledgeController],
|
||||||
|
providers: [
|
||||||
|
KnowledgeService,
|
||||||
|
RAGService,
|
||||||
|
ChunkingService,
|
||||||
|
EmbeddingService,
|
||||||
|
{
|
||||||
|
provide: KNOWLEDGE_REPOSITORY,
|
||||||
|
useClass: KnowledgePostgresRepository,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provide: USER_MEMORY_REPOSITORY,
|
||||||
|
useClass: UserMemoryPostgresRepository,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provide: SYSTEM_EXPERIENCE_REPOSITORY,
|
||||||
|
useClass: SystemExperiencePostgresRepository,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
exports: [KnowledgeService, RAGService, EmbeddingService],
|
||||||
|
})
|
||||||
|
export class KnowledgeModule {}
|
||||||
|
|
@ -0,0 +1,336 @@
|
||||||
|
import { Injectable, Inject } from '@nestjs/common';
|
||||||
|
import { EmbeddingService } from '../infrastructure/embedding/embedding.service';
|
||||||
|
import { ChunkingService } from '../application/services/chunking.service';
|
||||||
|
import {
|
||||||
|
IKnowledgeRepository,
|
||||||
|
KNOWLEDGE_REPOSITORY,
|
||||||
|
} from '../domain/repositories/knowledge.repository.interface';
|
||||||
|
import {
|
||||||
|
KnowledgeArticleEntity,
|
||||||
|
KnowledgeSource,
|
||||||
|
} from '../domain/entities/knowledge-article.entity';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 知识管理服务
|
||||||
|
* 提供知识库的CRUD操作和处理
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class KnowledgeService {
|
||||||
|
constructor(
|
||||||
|
private embeddingService: EmbeddingService,
|
||||||
|
private chunkingService: ChunkingService,
|
||||||
|
@Inject(KNOWLEDGE_REPOSITORY)
|
||||||
|
private knowledgeRepo: IKnowledgeRepository,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建知识文章
|
||||||
|
*/
|
||||||
|
async createArticle(params: {
|
||||||
|
title: string;
|
||||||
|
content: string;
|
||||||
|
category: string;
|
||||||
|
tags?: string[];
|
||||||
|
source?: KnowledgeSource;
|
||||||
|
sourceUrl?: string;
|
||||||
|
createdBy?: string;
|
||||||
|
autoPublish?: boolean;
|
||||||
|
}): Promise<KnowledgeArticleEntity> {
|
||||||
|
// 1. 创建文章实体
|
||||||
|
const article = KnowledgeArticleEntity.create({
|
||||||
|
title: params.title,
|
||||||
|
content: params.content,
|
||||||
|
category: params.category,
|
||||||
|
tags: params.tags,
|
||||||
|
source: params.source || KnowledgeSource.MANUAL,
|
||||||
|
sourceUrl: params.sourceUrl,
|
||||||
|
createdBy: params.createdBy,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 2. 生成文章向量
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(
|
||||||
|
`${article.title}\n${article.summary}`,
|
||||||
|
);
|
||||||
|
article.setEmbedding(embedding);
|
||||||
|
|
||||||
|
// 3. 自动发布（可选）
|
||||||
|
if (params.autoPublish) {
|
||||||
|
article.publish();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. 保存文章
|
||||||
|
await this.knowledgeRepo.saveArticle(article);
|
||||||
|
|
||||||
|
// 5. 分块并保存
|
||||||
|
await this.processArticleChunks(article);
|
||||||
|
|
||||||
|
console.log(`[KnowledgeService] Created article: ${article.id} - ${article.title}`);
|
||||||
|
|
||||||
|
return article;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理文章分块
|
||||||
|
*/
|
||||||
|
private async processArticleChunks(article: KnowledgeArticleEntity): Promise<void> {
|
||||||
|
// 1. 删除旧的块
|
||||||
|
await this.knowledgeRepo.deleteChunksByArticleId(article.id);
|
||||||
|
|
||||||
|
// 2. 分块
|
||||||
|
const chunks = this.chunkingService.chunkArticle(article);
|
||||||
|
|
||||||
|
// 3. 批量生成向量
|
||||||
|
const embeddings = await this.embeddingService.getEmbeddings(
|
||||||
|
chunks.map(c => c.content),
|
||||||
|
);
|
||||||
|
|
||||||
|
// 4. 设置向量
|
||||||
|
chunks.forEach((chunk, index) => {
|
||||||
|
chunk.setEmbedding(embeddings[index]);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 5. 保存块
|
||||||
|
await this.knowledgeRepo.saveChunks(chunks);
|
||||||
|
|
||||||
|
console.log(`[KnowledgeService] Processed ${chunks.length} chunks for article ${article.id}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新文章
|
||||||
|
*/
|
||||||
|
async updateArticle(
|
||||||
|
articleId: string,
|
||||||
|
params: {
|
||||||
|
title?: string;
|
||||||
|
content?: string;
|
||||||
|
category?: string;
|
||||||
|
tags?: string[];
|
||||||
|
updatedBy?: string;
|
||||||
|
},
|
||||||
|
): Promise<KnowledgeArticleEntity> {
|
||||||
|
const article = await this.knowledgeRepo.findArticleById(articleId);
|
||||||
|
if (!article) {
|
||||||
|
throw new Error(`Article not found: ${articleId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新字段
|
||||||
|
if (params.title || params.content) {
|
||||||
|
article.updateContent(
|
||||||
|
params.title || article.title,
|
||||||
|
params.content || article.content,
|
||||||
|
params.updatedBy,
|
||||||
|
);
|
||||||
|
|
||||||
|
// 重新生成向量
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(
|
||||||
|
`${article.title}\n${article.summary}`,
|
||||||
|
);
|
||||||
|
article.setEmbedding(embedding);
|
||||||
|
|
||||||
|
// 重新分块
|
||||||
|
await this.processArticleChunks(article);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (params.category) {
|
||||||
|
article.category = params.category;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (params.tags) {
|
||||||
|
article.tags = params.tags;
|
||||||
|
}
|
||||||
|
|
||||||
|
await this.knowledgeRepo.updateArticle(article);
|
||||||
|
|
||||||
|
return article;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取文章详情
|
||||||
|
*/
|
||||||
|
async getArticle(articleId: string): Promise<KnowledgeArticleEntity | null> {
|
||||||
|
return this.knowledgeRepo.findArticleById(articleId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取文章列表
|
||||||
|
*/
|
||||||
|
async listArticles(params: {
|
||||||
|
category?: string;
|
||||||
|
publishedOnly?: boolean;
|
||||||
|
page?: number;
|
||||||
|
pageSize?: number;
|
||||||
|
}): Promise<{
|
||||||
|
items: KnowledgeArticleEntity[];
|
||||||
|
total: number;
|
||||||
|
page: number;
|
||||||
|
pageSize: number;
|
||||||
|
}> {
|
||||||
|
const page = params.page || 1;
|
||||||
|
const pageSize = params.pageSize || 20;
|
||||||
|
const offset = (page - 1) * pageSize;
|
||||||
|
|
||||||
|
const [items, total] = await Promise.all([
|
||||||
|
this.knowledgeRepo.findArticlesByCategory(params.category || '', {
|
||||||
|
publishedOnly: params.publishedOnly,
|
||||||
|
limit: pageSize,
|
||||||
|
offset,
|
||||||
|
}),
|
||||||
|
this.knowledgeRepo.countArticles({
|
||||||
|
category: params.category,
|
||||||
|
publishedOnly: params.publishedOnly,
|
||||||
|
}),
|
||||||
|
]);
|
||||||
|
|
||||||
|
return { items, total, page, pageSize };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 搜索文章
|
||||||
|
*/
|
||||||
|
async searchArticles(params: {
|
||||||
|
query: string;
|
||||||
|
category?: string;
|
||||||
|
useVector?: boolean;
|
||||||
|
}): Promise<Array<{
|
||||||
|
article: KnowledgeArticleEntity;
|
||||||
|
similarity?: number;
|
||||||
|
}>> {
|
||||||
|
if (params.useVector) {
|
||||||
|
// 向量搜索
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(params.query);
|
||||||
|
const results = await this.knowledgeRepo.searchArticlesByVector(embedding, {
|
||||||
|
category: params.category,
|
||||||
|
publishedOnly: true,
|
||||||
|
limit: 10,
|
||||||
|
});
|
||||||
|
return results;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 关键词搜索
|
||||||
|
const articles = await this.knowledgeRepo.searchArticles(params.query, {
|
||||||
|
category: params.category,
|
||||||
|
publishedOnly: true,
|
||||||
|
limit: 10,
|
||||||
|
});
|
||||||
|
return articles.map(article => ({ article }));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 发布文章
|
||||||
|
*/
|
||||||
|
async publishArticle(articleId: string): Promise<void> {
|
||||||
|
const article = await this.knowledgeRepo.findArticleById(articleId);
|
||||||
|
if (!article) {
|
||||||
|
throw new Error(`Article not found: ${articleId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
article.publish();
|
||||||
|
await this.knowledgeRepo.updateArticle(article);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 取消发布
|
||||||
|
*/
|
||||||
|
async unpublishArticle(articleId: string): Promise<void> {
|
||||||
|
const article = await this.knowledgeRepo.findArticleById(articleId);
|
||||||
|
if (!article) {
|
||||||
|
throw new Error(`Article not found: ${articleId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
article.unpublish();
|
||||||
|
await this.knowledgeRepo.updateArticle(article);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除文章
|
||||||
|
*/
|
||||||
|
async deleteArticle(articleId: string): Promise<void> {
|
||||||
|
// 先删除分块
|
||||||
|
await this.knowledgeRepo.deleteChunksByArticleId(articleId);
|
||||||
|
// 再删除文章
|
||||||
|
await this.knowledgeRepo.deleteArticle(articleId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录用户反馈
|
||||||
|
*/
|
||||||
|
async recordFeedback(articleId: string, helpful: boolean): Promise<void> {
|
||||||
|
const article = await this.knowledgeRepo.findArticleById(articleId);
|
||||||
|
if (!article) {
|
||||||
|
throw new Error(`Article not found: ${articleId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
article.recordFeedback(helpful);
|
||||||
|
await this.knowledgeRepo.updateArticle(article);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 批量导入文章
|
||||||
|
*/
|
||||||
|
async importArticles(
|
||||||
|
articles: Array<{
|
||||||
|
title: string;
|
||||||
|
content: string;
|
||||||
|
category: string;
|
||||||
|
tags?: string[];
|
||||||
|
}>,
|
||||||
|
createdBy?: string,
|
||||||
|
): Promise<{ success: number; failed: number; errors: string[] }> {
|
||||||
|
let success = 0;
|
||||||
|
let failed = 0;
|
||||||
|
const errors: string[] = [];
|
||||||
|
|
||||||
|
for (const articleData of articles) {
|
||||||
|
try {
|
||||||
|
await this.createArticle({
|
||||||
|
...articleData,
|
||||||
|
source: KnowledgeSource.IMPORT,
|
||||||
|
createdBy,
|
||||||
|
autoPublish: false, // 导入后需要审核
|
||||||
|
});
|
||||||
|
success++;
|
||||||
|
} catch (error) {
|
||||||
|
failed++;
|
||||||
|
errors.push(`Failed to import "${articleData.title}": ${(error as Error).message}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return { success, failed, errors };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取知识库统计
|
||||||
|
*/
|
||||||
|
async getStatistics(): Promise<{
|
||||||
|
totalArticles: number;
|
||||||
|
publishedArticles: number;
|
||||||
|
byCategory: Record<string, number>;
|
||||||
|
recentArticles: KnowledgeArticleEntity[];
|
||||||
|
}> {
|
||||||
|
const categories = ['QMAS', 'GEP', 'IANG', 'TTPS', 'CIES', 'TechTAS', 'GENERAL'];
|
||||||
|
|
||||||
|
const [total, published, byCategoryResults, recent] = await Promise.all([
|
||||||
|
this.knowledgeRepo.countArticles(),
|
||||||
|
this.knowledgeRepo.countArticles({ publishedOnly: true }),
|
||||||
|
Promise.all(
|
||||||
|
categories.map(async cat => ({
|
||||||
|
category: cat,
|
||||||
|
count: await this.knowledgeRepo.countArticles({ category: cat }),
|
||||||
|
})),
|
||||||
|
),
|
||||||
|
this.knowledgeRepo.findArticlesByCategory('', { limit: 5 }),
|
||||||
|
]);
|
||||||
|
|
||||||
|
const byCategory: Record<string, number> = {};
|
||||||
|
byCategoryResults.forEach(r => {
|
||||||
|
byCategory[r.category] = r.count;
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
totalArticles: total,
|
||||||
|
publishedArticles: published,
|
||||||
|
byCategory,
|
||||||
|
recentArticles: recent,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,39 @@
|
||||||
|
import { NestFactory } from '@nestjs/core';
|
||||||
|
import { AppModule } from './app.module';
|
||||||
|
|
||||||
|
async function bootstrap() {
|
||||||
|
const app = await NestFactory.create(AppModule);
|
||||||
|
|
||||||
|
// 设置全局前缀
|
||||||
|
app.setGlobalPrefix('api/v1');
|
||||||
|
|
||||||
|
// 启用CORS
|
||||||
|
app.enableCors({
|
||||||
|
origin: process.env.CORS_ORIGIN || '*',
|
||||||
|
credentials: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
const port = process.env.PORT || 3004;
|
||||||
|
await app.listen(port);
|
||||||
|
|
||||||
|
console.log(`
|
||||||
|
╔══════════════════════════════════════════════════════════════╗
|
||||||
|
║ ║
|
||||||
|
║ 🧠 iConsulting Knowledge Service ║
|
||||||
|
║ ║
|
||||||
|
║ Server running at: http://localhost:${port} ║
|
||||||
|
║ API prefix: /api/v1 ║
|
||||||
|
║ ║
|
||||||
|
║ Endpoints: ║
|
||||||
|
║ - POST /api/v1/knowledge/articles Create article ║
|
||||||
|
║ - GET /api/v1/knowledge/articles List articles ║
|
||||||
|
║ - POST /api/v1/knowledge/retrieve RAG retrieval ║
|
||||||
|
║ - POST /api/v1/memory/user Save memory ║
|
||||||
|
║ - POST /api/v1/memory/experience Save experience ║
|
||||||
|
║ - GET /api/v1/memory/experience/pending Pending exp. ║
|
||||||
|
║ ║
|
||||||
|
╚══════════════════════════════════════════════════════════════╝
|
||||||
|
`);
|
||||||
|
}
|
||||||
|
|
||||||
|
bootstrap();
|
||||||
|
|
@ -0,0 +1,286 @@
|
||||||
|
import {
|
||||||
|
Controller,
|
||||||
|
Get,
|
||||||
|
Post,
|
||||||
|
Put,
|
||||||
|
Delete,
|
||||||
|
Body,
|
||||||
|
Param,
|
||||||
|
Query,
|
||||||
|
HttpCode,
|
||||||
|
HttpStatus,
|
||||||
|
} from '@nestjs/common';
|
||||||
|
import { MemoryService } from './memory.service';
|
||||||
|
import { MemoryType } from '../domain/entities/user-memory.entity';
|
||||||
|
import { ExperienceType } from '../domain/entities/system-experience.entity';
|
||||||
|
|
||||||
|
// ========== DTOs ==========
|
||||||
|
|
||||||
|
class SaveMemoryDto {
|
||||||
|
userId: string;
|
||||||
|
memoryType: MemoryType;
|
||||||
|
content: string;
|
||||||
|
importance?: number;
|
||||||
|
sourceConversationId?: string;
|
||||||
|
relatedCategory?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
class SearchMemoryDto {
|
||||||
|
userId: string;
|
||||||
|
query: string;
|
||||||
|
limit?: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
class ExtractExperienceDto {
|
||||||
|
experienceType: ExperienceType;
|
||||||
|
content: string;
|
||||||
|
scenario: string;
|
||||||
|
relatedCategory?: string;
|
||||||
|
sourceConversationId: string;
|
||||||
|
confidence?: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
class FeedbackDto {
|
||||||
|
positive: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== Controller ==========
|
||||||
|
|
||||||
|
@Controller('memory')
|
||||||
|
export class MemoryController {
|
||||||
|
constructor(private memoryService: MemoryService) {}
|
||||||
|
|
||||||
|
// ========== 用户记忆 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 保存用户记忆
|
||||||
|
*/
|
||||||
|
@Post('user')
|
||||||
|
@HttpCode(HttpStatus.CREATED)
|
||||||
|
async saveUserMemory(@Body() dto: SaveMemoryDto) {
|
||||||
|
const memory = await this.memoryService.saveUserMemory(dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: memory,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 搜索用户相关记忆
|
||||||
|
*/
|
||||||
|
@Post('user/search')
|
||||||
|
async searchUserMemories(@Body() dto: SearchMemoryDto) {
|
||||||
|
const memories = await this.memoryService.getUserRelevantMemories(
|
||||||
|
dto.userId,
|
||||||
|
dto.query,
|
||||||
|
{ limit: dto.limit },
|
||||||
|
);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: memories,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户最重要的记忆
|
||||||
|
*/
|
||||||
|
@Get('user/:userId/top')
|
||||||
|
async getUserTopMemories(
|
||||||
|
@Param('userId') userId: string,
|
||||||
|
@Query('limit') limit?: string,
|
||||||
|
) {
|
||||||
|
const memories = await this.memoryService.getUserTopMemories(
|
||||||
|
userId,
|
||||||
|
limit ? parseInt(limit) : 5,
|
||||||
|
);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: memories,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户所有记忆
|
||||||
|
*/
|
||||||
|
@Get('user/:userId')
|
||||||
|
async getUserMemories(
|
||||||
|
@Param('userId') userId: string,
|
||||||
|
@Query('type') memoryType?: MemoryType,
|
||||||
|
@Query('includeExpired') includeExpired?: string,
|
||||||
|
) {
|
||||||
|
const memories = await this.memoryService.getUserMemories(userId, {
|
||||||
|
memoryType,
|
||||||
|
includeExpired: includeExpired === 'true',
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: memories,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新记忆重要性
|
||||||
|
*/
|
||||||
|
@Put('user/memory/:id/importance')
|
||||||
|
async updateMemoryImportance(
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() body: { importance: number },
|
||||||
|
) {
|
||||||
|
await this.memoryService.updateMemoryImportance(id, body.importance);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Importance updated',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 标记记忆为过期
|
||||||
|
*/
|
||||||
|
@Post('user/memory/:id/expire')
|
||||||
|
async expireMemory(@Param('id') id: string) {
|
||||||
|
await this.memoryService.expireMemory(id);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Memory expired',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除用户所有记忆
|
||||||
|
*/
|
||||||
|
@Delete('user/:userId')
|
||||||
|
@HttpCode(HttpStatus.NO_CONTENT)
|
||||||
|
async deleteUserMemories(@Param('userId') userId: string) {
|
||||||
|
await this.memoryService.deleteUserMemories(userId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户时间线
|
||||||
|
*/
|
||||||
|
@Get('user/:userId/timeline')
|
||||||
|
async getUserTimeline(
|
||||||
|
@Param('userId') userId: string,
|
||||||
|
@Query('limit') limit?: string,
|
||||||
|
) {
|
||||||
|
const timeline = await this.memoryService.getUserTimeline(userId, {
|
||||||
|
limit: limit ? parseInt(limit) : 20,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: timeline,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 系统经验 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 提取并保存经验
|
||||||
|
*/
|
||||||
|
@Post('experience')
|
||||||
|
@HttpCode(HttpStatus.CREATED)
|
||||||
|
async extractExperience(@Body() dto: ExtractExperienceDto) {
|
||||||
|
const experience = await this.memoryService.extractAndSaveExperience(dto);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: experience,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 搜索相关经验
|
||||||
|
*/
|
||||||
|
@Post('experience/search')
|
||||||
|
async searchExperiences(@Body() body: {
|
||||||
|
query: string;
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
category?: string;
|
||||||
|
limit?: number;
|
||||||
|
}) {
|
||||||
|
const experiences = await this.memoryService.getRelevantExperiences(body.query, {
|
||||||
|
experienceType: body.experienceType,
|
||||||
|
category: body.category,
|
||||||
|
limit: body.limit,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: experiences,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取待验证的经验
|
||||||
|
*/
|
||||||
|
@Get('experience/pending')
|
||||||
|
async getPendingExperiences(
|
||||||
|
@Query('type') experienceType?: ExperienceType,
|
||||||
|
@Query('page') page?: string,
|
||||||
|
@Query('pageSize') pageSize?: string,
|
||||||
|
) {
|
||||||
|
const result = await this.memoryService.getPendingExperiences({
|
||||||
|
experienceType,
|
||||||
|
page: page ? parseInt(page) : 1,
|
||||||
|
pageSize: pageSize ? parseInt(pageSize) : 20,
|
||||||
|
});
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: result,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 审批经验
|
||||||
|
*/
|
||||||
|
@Post('experience/:id/approve')
|
||||||
|
async approveExperience(
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() body: { adminId: string },
|
||||||
|
) {
|
||||||
|
await this.memoryService.approveExperience(id, body.adminId);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Experience approved',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 拒绝经验
|
||||||
|
*/
|
||||||
|
@Post('experience/:id/reject')
|
||||||
|
async rejectExperience(
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() body: { adminId: string },
|
||||||
|
) {
|
||||||
|
await this.memoryService.rejectExperience(id, body.adminId);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Experience rejected',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录经验反馈
|
||||||
|
*/
|
||||||
|
@Post('experience/:id/feedback')
|
||||||
|
async recordExperienceFeedback(
|
||||||
|
@Param('id') id: string,
|
||||||
|
@Body() dto: FeedbackDto,
|
||||||
|
) {
|
||||||
|
await this.memoryService.recordExperienceFeedback(id, dto.positive);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
message: 'Feedback recorded',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取经验统计
|
||||||
|
*/
|
||||||
|
@Get('experience/statistics')
|
||||||
|
async getExperienceStatistics() {
|
||||||
|
const stats = await this.memoryService.getExperienceStatistics();
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
data: stats,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,41 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { MemoryController } from './memory.controller';
|
||||||
|
import { MemoryService } from './memory.service';
|
||||||
|
import { EmbeddingService } from '../infrastructure/embedding/embedding.service';
|
||||||
|
import { Neo4jService } from '../infrastructure/database/neo4j/neo4j.service';
|
||||||
|
import {
|
||||||
|
UserMemoryPostgresRepository,
|
||||||
|
SystemExperiencePostgresRepository,
|
||||||
|
} from '../infrastructure/database/postgres/memory-postgres.repository';
|
||||||
|
import { UserMemoryORM } from '../infrastructure/database/postgres/entities/user-memory.orm';
|
||||||
|
import { SystemExperienceORM } from '../infrastructure/database/postgres/entities/system-experience.orm';
|
||||||
|
import {
|
||||||
|
USER_MEMORY_REPOSITORY,
|
||||||
|
SYSTEM_EXPERIENCE_REPOSITORY,
|
||||||
|
} from '../domain/repositories/memory.repository.interface';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
TypeOrmModule.forFeature([
|
||||||
|
UserMemoryORM,
|
||||||
|
SystemExperienceORM,
|
||||||
|
]),
|
||||||
|
],
|
||||||
|
controllers: [MemoryController],
|
||||||
|
providers: [
|
||||||
|
MemoryService,
|
||||||
|
EmbeddingService,
|
||||||
|
Neo4jService,
|
||||||
|
{
|
||||||
|
provide: USER_MEMORY_REPOSITORY,
|
||||||
|
useClass: UserMemoryPostgresRepository,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
provide: SYSTEM_EXPERIENCE_REPOSITORY,
|
||||||
|
useClass: SystemExperiencePostgresRepository,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
exports: [MemoryService, Neo4jService],
|
||||||
|
})
|
||||||
|
export class MemoryModule {}
|
||||||
|
|
@ -0,0 +1,323 @@
|
||||||
|
import { Injectable, Inject } from '@nestjs/common';
|
||||||
|
import { EmbeddingService } from '../infrastructure/embedding/embedding.service';
|
||||||
|
import { Neo4jService } from '../infrastructure/database/neo4j/neo4j.service';
|
||||||
|
import {
|
||||||
|
IUserMemoryRepository,
|
||||||
|
ISystemExperienceRepository,
|
||||||
|
USER_MEMORY_REPOSITORY,
|
||||||
|
SYSTEM_EXPERIENCE_REPOSITORY,
|
||||||
|
} from '../domain/repositories/memory.repository.interface';
|
||||||
|
import { UserMemoryEntity, MemoryType } from '../domain/entities/user-memory.entity';
|
||||||
|
import {
|
||||||
|
SystemExperienceEntity,
|
||||||
|
ExperienceType,
|
||||||
|
VerificationStatus,
|
||||||
|
} from '../domain/entities/system-experience.entity';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记忆管理服务
|
||||||
|
* 管理用户长期记忆和系统经验
|
||||||
|
*/
|
||||||
|
@Injectable()
|
||||||
|
export class MemoryService {
|
||||||
|
constructor(
|
||||||
|
private embeddingService: EmbeddingService,
|
||||||
|
private neo4jService: Neo4jService,
|
||||||
|
@Inject(USER_MEMORY_REPOSITORY)
|
||||||
|
private memoryRepo: IUserMemoryRepository,
|
||||||
|
@Inject(SYSTEM_EXPERIENCE_REPOSITORY)
|
||||||
|
private experienceRepo: ISystemExperienceRepository,
|
||||||
|
) {}
|
||||||
|
|
||||||
|
// ========== 用户记忆 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 保存用户记忆
|
||||||
|
*/
|
||||||
|
async saveUserMemory(params: {
|
||||||
|
userId: string;
|
||||||
|
memoryType: MemoryType;
|
||||||
|
content: string;
|
||||||
|
importance?: number;
|
||||||
|
sourceConversationId?: string;
|
||||||
|
relatedCategory?: string;
|
||||||
|
}): Promise<UserMemoryEntity> {
|
||||||
|
const memory = UserMemoryEntity.create(params);
|
||||||
|
|
||||||
|
// 生成向量
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(params.content);
|
||||||
|
memory.setEmbedding(embedding);
|
||||||
|
|
||||||
|
// 保存到PostgreSQL
|
||||||
|
await this.memoryRepo.save(memory);
|
||||||
|
|
||||||
|
// 同时记录到Neo4j时间线
|
||||||
|
await this.neo4jService.recordUserEvent({
|
||||||
|
userId: params.userId,
|
||||||
|
eventId: memory.id,
|
||||||
|
eventType: `MEMORY_${params.memoryType}`,
|
||||||
|
content: params.content,
|
||||||
|
metadata: {
|
||||||
|
importance: params.importance,
|
||||||
|
relatedCategory: params.relatedCategory,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log(`[MemoryService] Saved memory for user ${params.userId}: ${params.memoryType}`);
|
||||||
|
|
||||||
|
return memory;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户的相关记忆
|
||||||
|
*/
|
||||||
|
async getUserRelevantMemories(
|
||||||
|
userId: string,
|
||||||
|
query: string,
|
||||||
|
options?: { limit?: number; memoryTypes?: MemoryType[] },
|
||||||
|
): Promise<UserMemoryEntity[]> {
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(query);
|
||||||
|
|
||||||
|
const results = await this.memoryRepo.searchByVector(userId, embedding, {
|
||||||
|
limit: options?.limit || 5,
|
||||||
|
minSimilarity: 0.6,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 记录访问
|
||||||
|
for (const { memory } of results) {
|
||||||
|
memory.recordAccess();
|
||||||
|
await this.memoryRepo.update(memory);
|
||||||
|
}
|
||||||
|
|
||||||
|
return results.map(r => r.memory);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户最重要的记忆（用于对话上下文）
|
||||||
|
*/
|
||||||
|
async getUserTopMemories(userId: string, limit = 5): Promise<UserMemoryEntity[]> {
|
||||||
|
return this.memoryRepo.findTopMemories(userId, limit);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户所有记忆
|
||||||
|
*/
|
||||||
|
async getUserMemories(
|
||||||
|
userId: string,
|
||||||
|
options?: { memoryType?: MemoryType; includeExpired?: boolean },
|
||||||
|
): Promise<UserMemoryEntity[]> {
|
||||||
|
return this.memoryRepo.findByUserId(userId, options);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新记忆重要性
|
||||||
|
*/
|
||||||
|
async updateMemoryImportance(memoryId: string, importance: number): Promise<void> {
|
||||||
|
const memory = await this.memoryRepo.findById(memoryId);
|
||||||
|
if (memory) {
|
||||||
|
memory.updateImportance(importance);
|
||||||
|
await this.memoryRepo.update(memory);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 标记记忆为过期
|
||||||
|
*/
|
||||||
|
async expireMemory(memoryId: string): Promise<void> {
|
||||||
|
const memory = await this.memoryRepo.findById(memoryId);
|
||||||
|
if (memory) {
|
||||||
|
memory.markAsExpired();
|
||||||
|
await this.memoryRepo.update(memory);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 清理过期记忆
|
||||||
|
*/
|
||||||
|
async cleanupExpiredMemories(userId: string, olderThanDays = 180): Promise<number> {
|
||||||
|
return this.memoryRepo.markExpiredMemories(userId, olderThanDays);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除用户所有记忆（GDPR合规）
|
||||||
|
*/
|
||||||
|
async deleteUserMemories(userId: string): Promise<void> {
|
||||||
|
await this.memoryRepo.deleteByUserId(userId);
|
||||||
|
console.log(`[MemoryService] Deleted all memories for user ${userId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 系统经验 ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 提取并保存系统经验
|
||||||
|
*/
|
||||||
|
async extractAndSaveExperience(params: {
|
||||||
|
experienceType: ExperienceType;
|
||||||
|
content: string;
|
||||||
|
scenario: string;
|
||||||
|
relatedCategory?: string;
|
||||||
|
sourceConversationId: string;
|
||||||
|
confidence?: number;
|
||||||
|
}): Promise<SystemExperienceEntity> {
|
||||||
|
// 生成向量
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(
|
||||||
|
`${params.scenario}\n${params.content}`,
|
||||||
|
);
|
||||||
|
|
||||||
|
// 查找相似经验（用于合并）
|
||||||
|
const similarExperiences = await this.experienceRepo.findSimilarExperiences(
|
||||||
|
embedding,
|
||||||
|
0.9, // 高阈值，只合并非常相似的
|
||||||
|
);
|
||||||
|
|
||||||
|
if (similarExperiences.length > 0) {
|
||||||
|
// 合并到现有经验
|
||||||
|
const existingExperience = similarExperiences[0];
|
||||||
|
existingExperience.addSourceConversation(params.sourceConversationId);
|
||||||
|
existingExperience.setEmbedding(embedding); // 可以选择更新向量
|
||||||
|
await this.experienceRepo.update(existingExperience);
|
||||||
|
|
||||||
|
console.log(`[MemoryService] Merged experience into ${existingExperience.id}`);
|
||||||
|
return existingExperience;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建新经验
|
||||||
|
const experience = SystemExperienceEntity.create(params);
|
||||||
|
experience.setEmbedding(embedding);
|
||||||
|
await this.experienceRepo.save(experience);
|
||||||
|
|
||||||
|
console.log(`[MemoryService] Created new experience: ${experience.id}`);
|
||||||
|
return experience;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取相关系统经验
|
||||||
|
*/
|
||||||
|
async getRelevantExperiences(
|
||||||
|
query: string,
|
||||||
|
options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
category?: string;
|
||||||
|
limit?: number;
|
||||||
|
},
|
||||||
|
): Promise<SystemExperienceEntity[]> {
|
||||||
|
const embedding = await this.embeddingService.getEmbedding(query);
|
||||||
|
|
||||||
|
const results = await this.experienceRepo.searchByVector(embedding, {
|
||||||
|
experienceType: options?.experienceType,
|
||||||
|
activeOnly: true,
|
||||||
|
limit: options?.limit || 5,
|
||||||
|
minSimilarity: 0.7,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 记录使用
|
||||||
|
for (const { experience } of results) {
|
||||||
|
experience.recordUsage();
|
||||||
|
await this.experienceRepo.update(experience);
|
||||||
|
}
|
||||||
|
|
||||||
|
return results.map(r => r.experience);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取待验证的经验
|
||||||
|
*/
|
||||||
|
async getPendingExperiences(options?: {
|
||||||
|
experienceType?: ExperienceType;
|
||||||
|
page?: number;
|
||||||
|
pageSize?: number;
|
||||||
|
}): Promise<{
|
||||||
|
items: SystemExperienceEntity[];
|
||||||
|
total: number;
|
||||||
|
}> {
|
||||||
|
const page = options?.page || 1;
|
||||||
|
const pageSize = options?.pageSize || 20;
|
||||||
|
|
||||||
|
const items = await this.experienceRepo.findPendingExperiences({
|
||||||
|
experienceType: options?.experienceType,
|
||||||
|
limit: pageSize,
|
||||||
|
offset: (page - 1) * pageSize,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 简单计数（实际应该有专门的count方法）
|
||||||
|
const stats = await this.experienceRepo.getStatistics();
|
||||||
|
const total = stats.byStatus[VerificationStatus.PENDING] || 0;
|
||||||
|
|
||||||
|
return { items, total };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 审批经验
|
||||||
|
*/
|
||||||
|
async approveExperience(experienceId: string, adminId: string): Promise<void> {
|
||||||
|
const experience = await this.experienceRepo.findById(experienceId);
|
||||||
|
if (!experience) {
|
||||||
|
throw new Error(`Experience not found: ${experienceId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
experience.approve(adminId);
|
||||||
|
await this.experienceRepo.update(experience);
|
||||||
|
|
||||||
|
console.log(`[MemoryService] Experience ${experienceId} approved by ${adminId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 拒绝经验
|
||||||
|
*/
|
||||||
|
async rejectExperience(experienceId: string, adminId: string): Promise<void> {
|
||||||
|
const experience = await this.experienceRepo.findById(experienceId);
|
||||||
|
if (!experience) {
|
||||||
|
throw new Error(`Experience not found: ${experienceId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
experience.reject(adminId);
|
||||||
|
await this.experienceRepo.update(experience);
|
||||||
|
|
||||||
|
console.log(`[MemoryService] Experience ${experienceId} rejected by ${adminId}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 记录经验反馈
|
||||||
|
*/
|
||||||
|
async recordExperienceFeedback(experienceId: string, positive: boolean): Promise<void> {
|
||||||
|
const experience = await this.experienceRepo.findById(experienceId);
|
||||||
|
if (experience) {
|
||||||
|
experience.recordFeedback(positive);
|
||||||
|
await this.experienceRepo.update(experience);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取经验统计
|
||||||
|
*/
|
||||||
|
async getExperienceStatistics(): Promise<{
|
||||||
|
total: number;
|
||||||
|
byStatus: Record<VerificationStatus, number>;
|
||||||
|
byType: Record<ExperienceType, number>;
|
||||||
|
}> {
|
||||||
|
return this.experienceRepo.getStatistics();
|
||||||
|
}
|
||||||
|
|
||||||
|
// ========== 用户时间线（Neo4j） ==========
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取用户时间线
|
||||||
|
*/
|
||||||
|
async getUserTimeline(
|
||||||
|
userId: string,
|
||||||
|
options?: {
|
||||||
|
limit?: number;
|
||||||
|
beforeDate?: Date;
|
||||||
|
eventTypes?: string[];
|
||||||
|
},
|
||||||
|
) {
|
||||||
|
return this.neo4jService.getUserTimeline(userId, options);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 初始化用户节点
|
||||||
|
*/
|
||||||
|
async initializeUserNode(userId: string, properties?: Record<string, unknown>): Promise<void> {
|
||||||
|
await this.neo4jService.createUserNode(userId, properties);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,17 @@
|
||||||
|
{
|
||||||
|
"extends": "../../../tsconfig.base.json",
|
||||||
|
"compilerOptions": {
|
||||||
|
"module": "CommonJS",
|
||||||
|
"moduleResolution": "Node",
|
||||||
|
"outDir": "./dist",
|
||||||
|
"rootDir": "./src",
|
||||||
|
"declaration": true,
|
||||||
|
"emitDecoratorMetadata": true,
|
||||||
|
"experimentalDecorators": true,
|
||||||
|
"strictPropertyInitialization": false,
|
||||||
|
"noUnusedLocals": false,
|
||||||
|
"noUnusedParameters": false
|
||||||
|
},
|
||||||
|
"include": ["src/**/*"],
|
||||||
|
"exclude": ["node_modules", "dist", "test"]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,45 @@
|
||||||
|
# ===========================================
|
||||||
|
# iConsulting Payment Service Dockerfile
|
||||||
|
# ===========================================
|
||||||
|
|
||||||
|
FROM node:20-alpine AS builder
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
|
||||||
|
COPY pnpm-workspace.yaml package.json pnpm-lock.yaml ./
|
||||||
|
COPY packages/shared/package.json ./packages/shared/
|
||||||
|
COPY packages/services/payment-service/package.json ./packages/services/payment-service/
|
||||||
|
|
||||||
|
RUN pnpm install --frozen-lockfile
|
||||||
|
|
||||||
|
COPY packages/shared ./packages/shared
|
||||||
|
COPY packages/services/payment-service ./packages/services/payment-service
|
||||||
|
COPY tsconfig.base.json ./
|
||||||
|
|
||||||
|
RUN pnpm --filter @iconsulting/shared build
|
||||||
|
RUN pnpm --filter @iconsulting/payment-service build
|
||||||
|
|
||||||
|
FROM node:20-alpine AS runner
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
RUN corepack enable && corepack prepare pnpm@latest --activate
|
||||||
|
RUN addgroup --system --gid 1001 nodejs && \
|
||||||
|
adduser --system --uid 1001 nestjs
|
||||||
|
|
||||||
|
COPY --from=builder /app/packages/services/payment-service/dist ./dist
|
||||||
|
COPY --from=builder /app/packages/services/payment-service/package.json ./
|
||||||
|
COPY --from=builder /app/node_modules ./node_modules
|
||||||
|
|
||||||
|
ENV NODE_ENV=production
|
||||||
|
ENV PORT=3002
|
||||||
|
|
||||||
|
USER nestjs
|
||||||
|
EXPOSE 3002
|
||||||
|
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||||
|
CMD wget --no-verbose --tries=1 --spider http://localhost:3002/health || exit 1
|
||||||
|
|
||||||
|
CMD ["node", "dist/main.js"]
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
{
|
||||||
|
"$schema": "https://json.schemastore.org/nest-cli",
|
||||||
|
"collection": "@nestjs/schematics",
|
||||||
|
"sourceRoot": "src",
|
||||||
|
"compilerOptions": {
|
||||||
|
"deleteOutDir": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,39 @@
|
||||||
|
{
|
||||||
|
"name": "@iconsulting/payment-service",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "Payment service for handling Alipay, WeChat Pay, and Stripe",
|
||||||
|
"main": "dist/main.js",
|
||||||
|
"scripts": {
|
||||||
|
"build": "nest build",
|
||||||
|
"dev": "nest start --watch",
|
||||||
|
"start": "node dist/main.js",
|
||||||
|
"lint": "eslint \"{src,apps,libs,test}/**/*.ts\"",
|
||||||
|
"test": "jest"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"@nestjs/common": "^10.0.0",
|
||||||
|
"@nestjs/config": "^3.2.0",
|
||||||
|
"@nestjs/core": "^10.0.0",
|
||||||
|
"@nestjs/platform-express": "^10.0.0",
|
||||||
|
"@nestjs/typeorm": "^10.0.0",
|
||||||
|
"typeorm": "^0.3.19",
|
||||||
|
"pg": "^8.11.0",
|
||||||
|
"alipay-sdk": "^3.6.0",
|
||||||
|
"stripe": "^14.0.0",
|
||||||
|
"qrcode": "^1.5.0",
|
||||||
|
"rxjs": "^7.8.0",
|
||||||
|
"class-validator": "^0.14.0",
|
||||||
|
"class-transformer": "^0.5.1",
|
||||||
|
"uuid": "^9.0.0",
|
||||||
|
"@iconsulting/shared": "workspace:*"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@nestjs/cli": "^10.0.0",
|
||||||
|
"@nestjs/testing": "^10.0.0",
|
||||||
|
"@types/express": "^4.17.21",
|
||||||
|
"@types/qrcode": "^1.5.0",
|
||||||
|
"@types/node": "^20.10.0",
|
||||||
|
"@types/uuid": "^9.0.0",
|
||||||
|
"typescript": "^5.3.0"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,33 @@
|
||||||
|
import { Module } from '@nestjs/common';
|
||||||
|
import { ConfigModule, ConfigService } from '@nestjs/config';
|
||||||
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
||||||
|
import { PaymentModule } from './payment/payment.module';
|
||||||
|
import { OrderModule } from './order/order.module';
|
||||||
|
|
||||||
|
@Module({
|
||||||
|
imports: [
|
||||||
|
ConfigModule.forRoot({
|
||||||
|
isGlobal: true,
|
||||||
|
envFilePath: ['.env.local', '.env'],
|
||||||
|
}),
|
||||||
|
|
||||||
|
TypeOrmModule.forRootAsync({
|
||||||
|
imports: [ConfigModule],
|
||||||
|
inject: [ConfigService],
|
||||||
|
useFactory: (configService: ConfigService) => ({
|
||||||
|
type: 'postgres',
|
||||||
|
host: configService.get('POSTGRES_HOST', 'localhost'),
|
||||||
|
port: configService.get<number>('POSTGRES_PORT', 5432),
|
||||||
|
username: configService.get('POSTGRES_USER', 'iconsulting'),
|
||||||
|
password: configService.get('POSTGRES_PASSWORD'),
|
||||||
|
database: configService.get('POSTGRES_DB', 'iconsulting'),
|
||||||
|
entities: [__dirname + '/**/*.entity{.ts,.js}'],
|
||||||
|
synchronize: configService.get('NODE_ENV') === 'development',
|
||||||
|
}),
|
||||||
|
}),
|
||||||
|
|
||||||
|
OrderModule,
|
||||||
|
PaymentModule,
|
||||||
|
],
|
||||||
|
})
|
||||||
|
export class AppModule {}
|
||||||
|
|
@ -0,0 +1,84 @@
|
||||||
|
import {
|
||||||
|
Entity,
|
||||||
|
PrimaryGeneratedColumn,
|
||||||
|
Column,
|
||||||
|
CreateDateColumn,
|
||||||
|
UpdateDateColumn,
|
||||||
|
OneToMany,
|
||||||
|
} from 'typeorm';
|
||||||
|
import { PaymentEntity } from './payment.entity';
|
||||||
|
|
||||||
|
export enum OrderStatus {
|
||||||
|
CREATED = 'CREATED',
|
||||||
|
PENDING_PAYMENT = 'PENDING_PAYMENT',
|
||||||
|
PAID = 'PAID',
|
||||||
|
PROCESSING = 'PROCESSING',
|
||||||
|
COMPLETED = 'COMPLETED',
|
||||||
|
CANCELLED = 'CANCELLED',
|
||||||
|
REFUNDED = 'REFUNDED',
|
||||||
|
}
|
||||||
|
|
||||||
|
export enum ServiceType {
|
||||||
|
ASSESSMENT = 'ASSESSMENT',
|
||||||
|
CONSULTATION = 'CONSULTATION',
|
||||||
|
DOCUMENT_REVIEW = 'DOCUMENT_REVIEW',
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity('orders')
|
||||||
|
export class OrderEntity {
|
||||||
|
@PrimaryGeneratedColumn('uuid')
|
||||||
|
id: string;
|
||||||
|
|
||||||
|
@Column({ name: 'user_id', type: 'uuid' })
|
||||||
|
userId: string;
|
||||||
|
|
||||||
|
@Column({ name: 'conversation_id', type: 'uuid', nullable: true })
|
||||||
|
conversationId: string;
|
||||||
|
|
||||||
|
@Column({
|
||||||
|
name: 'service_type',
|
||||||
|
type: 'enum',
|
||||||
|
enum: ServiceType,
|
||||||
|
})
|
||||||
|
serviceType: ServiceType;
|
||||||
|
|
||||||
|
@Column({ name: 'service_category', nullable: true })
|
||||||
|
serviceCategory: string;
|
||||||
|
|
||||||
|
@Column({ type: 'decimal', precision: 10, scale: 2 })
|
||||||
|
amount: number;
|
||||||
|
|
||||||
|
@Column({ default: 'CNY' })
|
||||||
|
currency: string;
|
||||||
|
|
||||||
|
@Column({
|
||||||
|
type: 'enum',
|
||||||
|
enum: OrderStatus,
|
||||||
|
default: OrderStatus.CREATED,
|
||||||
|
})
|
||||||
|
status: OrderStatus;
|
||||||
|
|
||||||
|
@Column({ name: 'payment_method', nullable: true })
|
||||||
|
paymentMethod: string;
|
||||||
|
|
||||||
|
@Column({ name: 'payment_id', type: 'uuid', nullable: true })
|
||||||
|
paymentId: string;
|
||||||
|
|
||||||
|
@Column({ name: 'paid_at', nullable: true })
|
||||||
|
paidAt: Date;
|
||||||
|
|
||||||
|
@Column({ name: 'completed_at', nullable: true })
|
||||||
|
completedAt: Date;
|
||||||
|
|
||||||
|
@Column({ type: 'jsonb', nullable: true })
|
||||||
|
metadata: Record<string, unknown>;
|
||||||
|
|
||||||
|
@CreateDateColumn({ name: 'created_at' })
|
||||||
|
createdAt: Date;
|
||||||
|
|
||||||
|
@UpdateDateColumn({ name: 'updated_at' })
|
||||||
|
updatedAt: Date;
|
||||||
|
|
||||||
|
@OneToMany(() => PaymentEntity, (payment) => payment.order)
|
||||||
|
payments: PaymentEntity[];
|
||||||
|
}
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue