feat(pkb): Complete PKB module frontend migration with V3 design
Summary: - Implement PKB Dashboard and Workspace pages based on V3 prototype - Add single-layer header with integrated Tab navigation - Implement 3 work modes: Full Text, Deep Read, Batch Processing - Integrate Ant Design X Chat component for AI conversations - Create BatchModeComplete with template selection and document processing - Add compact work mode selector with dropdown design Backend: - Migrate PKB controllers and services to /modules/pkb structure - Register v2 API routes at /api/v2/pkb/knowledge - Maintain dual API routes for backward compatibility Technical details: - Use Zustand for state management - Handle SSE streaming responses for AI chat - Support document selection for Deep Read mode - Implement batch processing with progress tracking Known issues: - Batch processing API integration pending - Knowledge assets page navigation needs optimization Status: Frontend functional, pending refinement
This commit is contained in:
@@ -233,6 +233,10 @@ checkDCTables();
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -185,6 +185,10 @@ createAiHistoryTable()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -172,6 +172,10 @@ createToolCTable()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -169,6 +169,10 @@ createToolCTable()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
327
backend/scripts/test-pkb-apis-simple.ts
Normal file
327
backend/scripts/test-pkb-apis-simple.ts
Normal file
@@ -0,0 +1,327 @@
|
||||
/**
|
||||
* PKB模块API简化测试脚本
|
||||
* 测试现有知识库的各项功能
|
||||
*/
|
||||
|
||||
import axios from 'axios';
|
||||
|
||||
const BASE_URL = 'http://localhost:3000';
|
||||
|
||||
interface TestResult {
|
||||
name: string;
|
||||
status: 'pass' | 'fail';
|
||||
message: string;
|
||||
duration?: number;
|
||||
}
|
||||
|
||||
const results: TestResult[] = [];
|
||||
|
||||
function printResult(result: TestResult) {
|
||||
const icon = result.status === 'pass' ? '✅' : '❌';
|
||||
console.log(`${icon} ${result.name} ${result.duration ? `(${result.duration}ms)` : ''}`);
|
||||
console.log(` ${result.message}`);
|
||||
}
|
||||
|
||||
// 测试1：健康检查
|
||||
async function testHealthCheck(): Promise<TestResult> {
|
||||
const startTime = Date.now();
|
||||
try {
|
||||
const response = await axios.get(`${BASE_URL}/api/v2/pkb/health`);
|
||||
const duration = Date.now() - startTime;
|
||||
|
||||
if (response.data.status === 'ok') {
|
||||
return {
|
||||
name: '健康检查（v2）',
|
||||
status: 'pass',
|
||||
message: `知识库数: ${response.data.database.knowledgeBases}, schema: ${response.data.database.schema}`,
|
||||
duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '健康检查（v2）',
|
||||
status: 'fail',
|
||||
message: '返回状态异常',
|
||||
duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '健康检查（v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
duration: Date.now() - startTime,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试2：获取知识库列表（v1 vs v2）
|
||||
async function testGetKnowledgeBases(): Promise<TestResult> {
|
||||
try {
|
||||
const startV1 = Date.now();
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases`);
|
||||
const v1Duration = Date.now() - startV1;
|
||||
|
||||
const startV2 = Date.now();
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases`);
|
||||
const v2Duration = Date.now() - startV2;
|
||||
|
||||
const v1Count = v1Response.data.data?.length || 0;
|
||||
const v2Count = v2Response.data.data?.length || 0;
|
||||
|
||||
if (v1Count === v2Count) {
|
||||
return {
|
||||
name: '获取知识库列表（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `v1: ${v1Count}个 (${v1Duration}ms), v2: ${v2Count}个 (${v2Duration}ms) ✅`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '获取知识库列表（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: `数量不一致！v1: ${v1Count}, v2: ${v2Count}`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '获取知识库列表（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试3：获取知识库详情（v1 vs v2）
|
||||
async function testGetKnowledgeBaseById(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const startV1 = Date.now();
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}`);
|
||||
const v1Duration = Date.now() - startV1;
|
||||
|
||||
const startV2 = Date.now();
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}`);
|
||||
const v2Duration = Date.now() - startV2;
|
||||
|
||||
const v1Name = v1Response.data.data?.name;
|
||||
const v2Name = v2Response.data.data?.name;
|
||||
|
||||
if (v1Name === v2Name) {
|
||||
return {
|
||||
name: '获取知识库详情（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `名称一致: "${v1Name}", v1: ${v1Duration}ms, v2: ${v2Duration}ms ✅`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '获取知识库详情（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: `名称不一致！v1: "${v1Name}", v2: "${v2Name}"`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '获取知识库详情（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试4：获取知识库统计（v1 vs v2）
|
||||
async function testGetKnowledgeBaseStats(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}/stats`);
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}/stats`);
|
||||
|
||||
const v1Docs = v1Response.data.data.totalDocuments;
|
||||
const v2Docs = v2Response.data.data.totalDocuments;
|
||||
|
||||
if (v1Docs === v2Docs) {
|
||||
return {
|
||||
name: '获取知识库统计（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `文档数一致: ${v1Docs}个 ✅`,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '获取知识库统计（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: `文档数不一致！v1: ${v1Docs}, v2: ${v2Docs}`,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '获取知识库统计（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试5：RAG检索（v1 vs v2）
|
||||
async function testSearchKnowledgeBase(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const query = '治疗';
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}/search`, {
|
||||
params: { query, top_k: 3 },
|
||||
});
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}/search`, {
|
||||
params: { query, top_k: 3 },
|
||||
});
|
||||
|
||||
const v1Count = v1Response.data.data?.records?.length || 0;
|
||||
const v2Count = v2Response.data.data?.records?.length || 0;
|
||||
|
||||
return {
|
||||
name: 'RAG检索（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `v1返回${v1Count}条, v2返回${v2Count}条 ✅`,
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: 'RAG检索（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试6：文档选择（全文阅读模式）
|
||||
async function testDocumentSelection(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}/document-selection`, {
|
||||
params: { max_files: 5, max_tokens: 100000 },
|
||||
});
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}/document-selection`, {
|
||||
params: { max_files: 5, max_tokens: 100000 },
|
||||
});
|
||||
|
||||
const v1Docs = v1Response.data.data?.selectedDocuments?.length || 0;
|
||||
const v2Docs = v2Response.data.data?.selectedDocuments?.length || 0;
|
||||
|
||||
return {
|
||||
name: '文档选择-全文阅读模式（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `v1选择${v1Docs}个文档, v2选择${v2Docs}个文档 ✅`,
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '文档选择-全文阅读模式（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试7：批处理模板
|
||||
async function testBatchTemplates(): Promise<TestResult> {
|
||||
try {
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/batch/templates`);
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/batch-tasks/batch/templates`);
|
||||
|
||||
const v1Count = v1Response.data.data?.length || 0;
|
||||
const v2Count = v2Response.data.data?.length || 0;
|
||||
|
||||
return {
|
||||
name: '批处理模板（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `v1: ${v1Count}个模板, v2: ${v2Count}个模板 ✅`,
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '批处理模板（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 主测试函数
|
||||
async function runTests() {
|
||||
console.log('🚀 PKB API测试开始...\n');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
// 测试1：健康检查
|
||||
console.log('\n📋 测试1：健康检查');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testHealthCheck());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试2：获取知识库列表
|
||||
console.log('\n📋 测试2：知识库列表');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testGetKnowledgeBases());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 获取第一个知识库ID用于后续测试
|
||||
const kbListResponse = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases`);
|
||||
const firstKb = kbListResponse.data.data?.[0];
|
||||
|
||||
if (!firstKb) {
|
||||
console.log('\n❌ 没有可用的知识库，后续测试跳过');
|
||||
return;
|
||||
}
|
||||
|
||||
const kbId = firstKb.id;
|
||||
console.log(`\n使用知识库: ${firstKb.name} (ID: ${kbId})`);
|
||||
|
||||
// 测试3：获取知识库详情
|
||||
console.log('\n📋 测试3：知识库详情');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testGetKnowledgeBaseById(kbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试4：知识库统计
|
||||
console.log('\n📋 测试4：知识库统计');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testGetKnowledgeBaseStats(kbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试5：RAG检索
|
||||
console.log('\n📋 测试5：RAG检索');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testSearchKnowledgeBase(kbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试6：文档选择
|
||||
console.log('\n📋 测试6：文档选择（全文阅读）');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testDocumentSelection(kbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试7：批处理模板
|
||||
console.log('\n📋 测试7：批处理模板');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testBatchTemplates());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 总结
|
||||
console.log('\n' + '='.repeat(80));
|
||||
console.log('📊 测试总结');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
const passCount = results.filter(r => r.status === 'pass').length;
|
||||
const failCount = results.filter(r => r.status === 'fail').length;
|
||||
const totalDuration = results.reduce((sum, r) => sum + (r.duration || 0), 0);
|
||||
|
||||
console.log(`\n总计: ${results.length}个测试`);
|
||||
console.log(`✅ 通过: ${passCount}个`);
|
||||
console.log(`❌ 失败: ${failCount}个`);
|
||||
console.log(`⏱️ 总耗时: ${totalDuration}ms`);
|
||||
|
||||
if (failCount === 0) {
|
||||
console.log('\n🎉 所有测试通过！v1和v2功能完全一致！');
|
||||
} else {
|
||||
console.log('\n⚠️ 部分测试失败，请查看详情');
|
||||
}
|
||||
}
|
||||
|
||||
runTests().catch(error => {
|
||||
console.error('❌ 测试执行失败:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
|
||||
440
backend/scripts/test-pkb-apis.ts
Normal file
440
backend/scripts/test-pkb-apis.ts
Normal file
@@ -0,0 +1,440 @@
|
||||
/**
|
||||
* PKB模块API自动化测试脚本
|
||||
*
|
||||
* 功能：
|
||||
* 1. 测试所有PKB API端点（v1和v2）
|
||||
* 2. 对比v1和v2的返回结果
|
||||
* 3. 验证数据一致性
|
||||
* 4. 性能对比
|
||||
* 5. 边界条件测试
|
||||
*
|
||||
* 运行方式：
|
||||
* npx tsx scripts/test-pkb-apis.ts
|
||||
*/
|
||||
|
||||
import axios, { AxiosError } from 'axios';
|
||||
|
||||
const BASE_URL = 'http://localhost:3000';
|
||||
const TEST_KB_NAME = `测试知识库-${Date.now()}`;
|
||||
|
||||
interface TestResult {
|
||||
name: string;
|
||||
status: 'pass' | 'fail' | 'skip';
|
||||
message: string;
|
||||
duration?: number;
|
||||
v1Response?: any;
|
||||
v2Response?: any;
|
||||
}
|
||||
|
||||
const results: TestResult[] = [];
|
||||
let testKbId: string | null = null;
|
||||
|
||||
// 工具函数：比较两个响应是否一致
|
||||
function compareResponses(v1: any, v2: any): boolean {
|
||||
return JSON.stringify(v1) === JSON.stringify(v2);
|
||||
}
|
||||
|
||||
// 工具函数：打印测试结果
|
||||
function printResult(result: TestResult) {
|
||||
const icon = result.status === 'pass' ? '✅' : result.status === 'fail' ? '❌' : '⏭️';
|
||||
console.log(`${icon} ${result.name} ${result.duration ? `(${result.duration}ms)` : ''}`);
|
||||
if (result.message) {
|
||||
console.log(` ${result.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
// 测试1：健康检查
|
||||
async function testHealthCheck(): Promise<TestResult> {
|
||||
const startTime = Date.now();
|
||||
try {
|
||||
const response = await axios.get(`${BASE_URL}/api/v2/pkb/health`);
|
||||
const duration = Date.now() - startTime;
|
||||
|
||||
if (response.data.status === 'ok' && response.data.module === 'pkb' && response.data.version === 'v2') {
|
||||
return {
|
||||
name: '健康检查',
|
||||
status: 'pass',
|
||||
message: `状态: ${response.data.status}, 知识库数: ${response.data.database.knowledgeBases}`,
|
||||
duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '健康检查',
|
||||
status: 'fail',
|
||||
message: '返回数据格式不正确',
|
||||
duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '健康检查',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
duration: Date.now() - startTime,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试2：获取知识库列表（对比v1和v2）
|
||||
async function testGetKnowledgeBases(): Promise<TestResult> {
|
||||
try {
|
||||
const startV1 = Date.now();
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases`);
|
||||
const v1Duration = Date.now() - startV1;
|
||||
|
||||
const startV2 = Date.now();
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases`);
|
||||
const v2Duration = Date.now() - startV2;
|
||||
|
||||
const v1Count = v1Response.data.data?.length || 0;
|
||||
const v2Count = v2Response.data.data?.length || 0;
|
||||
|
||||
if (v1Count === v2Count) {
|
||||
return {
|
||||
name: '获取知识库列表（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `v1: ${v1Count}个 (${v1Duration}ms), v2: ${v2Count}个 (${v2Duration}ms), 数据一致✅`,
|
||||
duration: v1Duration + v2Duration,
|
||||
v1Response: v1Response.data,
|
||||
v2Response: v2Response.data,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '获取知识库列表（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: `数量不一致！v1: ${v1Count}个, v2: ${v2Count}个`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '获取知识库列表（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试3：创建知识库（v2）
|
||||
async function testCreateKnowledgeBase(): Promise<TestResult> {
|
||||
const startTime = Date.now();
|
||||
try {
|
||||
const response = await axios.post(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases`, {
|
||||
name: TEST_KB_NAME,
|
||||
description: '这是一个自动化测试创建的知识库',
|
||||
}, {
|
||||
headers: {
|
||||
'Content-Type': 'application/json'
|
||||
}
|
||||
});
|
||||
const duration = Date.now() - startTime;
|
||||
|
||||
if (response.data.success && response.data.data.id) {
|
||||
testKbId = response.data.data.id;
|
||||
return {
|
||||
name: '创建知识库（v2）',
|
||||
status: 'pass',
|
||||
message: `成功创建，ID: ${testKbId}`,
|
||||
duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '创建知识库（v2）',
|
||||
status: 'fail',
|
||||
message: '创建失败或返回格式不正确',
|
||||
duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
const errorDetail = error.response?.data ?
|
||||
JSON.stringify(error.response.data) :
|
||||
(error.response?.data?.message || error.message);
|
||||
return {
|
||||
name: '创建知识库（v2）',
|
||||
status: 'fail',
|
||||
message: errorDetail,
|
||||
duration: Date.now() - startTime,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试4：获取知识库详情（对比v1和v2）
|
||||
async function testGetKnowledgeBaseById(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const startV1 = Date.now();
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}`);
|
||||
const v1Duration = Date.now() - startV1;
|
||||
|
||||
const startV2 = Date.now();
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}`);
|
||||
const v2Duration = Date.now() - startV2;
|
||||
|
||||
const v1Name = v1Response.data.data?.name;
|
||||
const v2Name = v2Response.data.data?.name;
|
||||
|
||||
if (v1Name === v2Name) {
|
||||
return {
|
||||
name: '获取知识库详情（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `v1: ${v1Duration}ms, v2: ${v2Duration}ms, 名称一致: "${v1Name}"✅`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '获取知识库详情（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: `名称不一致！v1: "${v1Name}", v2: "${v2Name}"`,
|
||||
duration: v1Duration + v2Duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '获取知识库详情（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试5：更新知识库（v2）
|
||||
async function testUpdateKnowledgeBase(kbId: string): Promise<TestResult> {
|
||||
const startTime = Date.now();
|
||||
try {
|
||||
const response = await axios.put(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}`, {
|
||||
name: `${TEST_KB_NAME}-已更新`,
|
||||
description: '描述已更新',
|
||||
});
|
||||
const duration = Date.now() - startTime;
|
||||
|
||||
if (response.data.success) {
|
||||
return {
|
||||
name: '更新知识库（v2）',
|
||||
status: 'pass',
|
||||
message: '更新成功',
|
||||
duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '更新知识库（v2）',
|
||||
status: 'fail',
|
||||
message: '更新失败',
|
||||
duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '更新知识库（v2）',
|
||||
status: 'fail',
|
||||
message: error.response?.data?.message || error.message,
|
||||
duration: Date.now() - startTime,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试6：获取知识库统计（对比v1和v2）
|
||||
async function testGetKnowledgeBaseStats(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}/stats`);
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}/stats`);
|
||||
|
||||
const v1Stats = v1Response.data.data;
|
||||
const v2Stats = v2Response.data.data;
|
||||
|
||||
if (v1Stats.totalDocuments === v2Stats.totalDocuments) {
|
||||
return {
|
||||
name: '获取知识库统计（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: `文档数一致: ${v1Stats.totalDocuments}个✅`,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '获取知识库统计（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: `文档数不一致！v1: ${v1Stats.totalDocuments}, v2: ${v2Stats.totalDocuments}`,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '获取知识库统计（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试7：RAG检索（对比v1和v2）
|
||||
async function testSearchKnowledgeBase(kbId: string): Promise<TestResult> {
|
||||
try {
|
||||
const query = '测试查询';
|
||||
const v1Response = await axios.get(`${BASE_URL}/api/v1/knowledge-bases/${kbId}/search`, {
|
||||
params: { query, top_k: 5 },
|
||||
});
|
||||
const v2Response = await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}/search`, {
|
||||
params: { query, top_k: 5 },
|
||||
});
|
||||
|
||||
return {
|
||||
name: 'RAG检索（v1 vs v2）',
|
||||
status: 'pass',
|
||||
message: '检索成功，两个版本都返回了结果',
|
||||
};
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: 'RAG检索（v1 vs v2）',
|
||||
status: 'fail',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 测试8：边界条件 - 不存在的知识库
|
||||
async function testNotFoundKnowledgeBase(): Promise<TestResult> {
|
||||
try {
|
||||
await axios.get(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/00000000-0000-0000-0000-000000000000`);
|
||||
return {
|
||||
name: '边界测试：不存在的知识库',
|
||||
status: 'fail',
|
||||
message: '应该返回404错误，但没有',
|
||||
};
|
||||
} catch (error: any) {
|
||||
if (error.response?.status === 404 || error.response?.status === 500) {
|
||||
return {
|
||||
name: '边界测试：不存在的知识库',
|
||||
status: 'pass',
|
||||
message: `正确返回错误状态: ${error.response.status}✅`,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '边界测试：不存在的知识库',
|
||||
status: 'fail',
|
||||
message: `意外的状态码: ${error.response?.status}`,
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 测试9：清理 - 删除测试知识库
|
||||
async function testDeleteKnowledgeBase(kbId: string): Promise<TestResult> {
|
||||
const startTime = Date.now();
|
||||
try {
|
||||
const response = await axios.delete(`${BASE_URL}/api/v2/pkb/knowledge/knowledge-bases/${kbId}`);
|
||||
const duration = Date.now() - startTime;
|
||||
|
||||
if (response.data.success) {
|
||||
return {
|
||||
name: '删除知识库（v2）',
|
||||
status: 'pass',
|
||||
message: '删除成功',
|
||||
duration,
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
name: '删除知识库（v2）',
|
||||
status: 'fail',
|
||||
message: '删除失败',
|
||||
duration,
|
||||
};
|
||||
}
|
||||
} catch (error: any) {
|
||||
return {
|
||||
name: '删除知识库（v2）',
|
||||
status: 'fail',
|
||||
message: error.response?.data?.message || error.message,
|
||||
duration: Date.now() - startTime,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
// 主测试函数
|
||||
async function runTests() {
|
||||
console.log('🚀 开始PKB API自动化测试...\n');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
// 测试1：健康检查
|
||||
console.log('\n📋 阶段1：健康检查');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testHealthCheck());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试2：获取知识库列表
|
||||
console.log('\n📋 阶段2：知识库列表');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testGetKnowledgeBases());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试3：创建知识库
|
||||
console.log('\n📋 阶段3：创建知识库');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testCreateKnowledgeBase());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
if (!testKbId) {
|
||||
console.log('\n❌ 无法获取测试知识库ID，后续测试跳过');
|
||||
return;
|
||||
}
|
||||
|
||||
// 测试4：获取知识库详情
|
||||
console.log('\n📋 阶段4：知识库详情');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testGetKnowledgeBaseById(testKbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试5：更新知识库
|
||||
console.log('\n📋 阶段5：更新知识库');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testUpdateKnowledgeBase(testKbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试6：获取统计信息
|
||||
console.log('\n📋 阶段6：知识库统计');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testGetKnowledgeBaseStats(testKbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试7：RAG检索
|
||||
console.log('\n📋 阶段7：RAG检索');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testSearchKnowledgeBase(testKbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试8：边界条件
|
||||
console.log('\n📋 阶段8：边界条件测试');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testNotFoundKnowledgeBase());
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 测试9：清理
|
||||
console.log('\n📋 阶段9：清理测试数据');
|
||||
console.log('-'.repeat(80));
|
||||
results.push(await testDeleteKnowledgeBase(testKbId));
|
||||
printResult(results[results.length - 1]);
|
||||
|
||||
// 总结
|
||||
console.log('\n' + '='.repeat(80));
|
||||
console.log('📊 测试总结');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
const passCount = results.filter(r => r.status === 'pass').length;
|
||||
const failCount = results.filter(r => r.status === 'fail').length;
|
||||
const skipCount = results.filter(r => r.status === 'skip').length;
|
||||
const totalDuration = results.reduce((sum, r) => sum + (r.duration || 0), 0);
|
||||
|
||||
console.log(`\n总计: ${results.length}个测试`);
|
||||
console.log(`✅ 通过: ${passCount}个`);
|
||||
console.log(`❌ 失败: ${failCount}个`);
|
||||
console.log(`⏭️ 跳过: ${skipCount}个`);
|
||||
console.log(`⏱️ 总耗时: ${totalDuration}ms`);
|
||||
|
||||
if (failCount === 0) {
|
||||
console.log('\n🎉 所有测试通过！');
|
||||
} else {
|
||||
console.log('\n⚠️ 部分测试失败，请查看详情');
|
||||
}
|
||||
}
|
||||
|
||||
// 执行测试
|
||||
runTests().catch(error => {
|
||||
console.error('❌ 测试执行失败:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
292
backend/scripts/verify-pkb-rvw-schema.ts
Normal file
292
backend/scripts/verify-pkb-rvw-schema.ts
Normal file
@@ -0,0 +1,292 @@
|
||||
/**
|
||||
* 验证PKB和RVW的数据库Schema状态
|
||||
*
|
||||
* 目的：
|
||||
* 1. 检查pkb_schema是否存在及其表结构
|
||||
* 2. 检查rvw_schema是否存在
|
||||
* 3. 检查ReviewTask在哪个Schema中
|
||||
* 4. 检查是否有旧数据需要迁移
|
||||
*
|
||||
* 运行方式：
|
||||
* npx tsx scripts/verify-pkb-rvw-schema.ts
|
||||
*/
|
||||
|
||||
import { PrismaClient } from '@prisma/client';
|
||||
|
||||
const prisma = new PrismaClient();
|
||||
|
||||
interface SchemaInfo {
|
||||
schema_name: string;
|
||||
}
|
||||
|
||||
interface TableInfo {
|
||||
table_schema: string;
|
||||
table_name: string;
|
||||
}
|
||||
|
||||
interface ColumnInfo {
|
||||
column_name: string;
|
||||
data_type: string;
|
||||
is_nullable: string;
|
||||
column_default: string | null;
|
||||
}
|
||||
|
||||
interface RowCount {
|
||||
count: bigint;
|
||||
}
|
||||
|
||||
async function verifySchemas() {
|
||||
console.log('🔍 开始验证PKB和RVW的数据库Schema状态...\n');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
try {
|
||||
// ========================================
|
||||
// 1. 检查所有Schema
|
||||
// ========================================
|
||||
console.log('\n📦 1. 检查数据库中所有的Schema');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
const schemas = await prisma.$queryRaw<SchemaInfo[]>`
|
||||
SELECT schema_name
|
||||
FROM information_schema.schemata
|
||||
WHERE schema_name IN ('pkb_schema', 'rvw_schema', 'public', 'platform_schema')
|
||||
ORDER BY schema_name
|
||||
`;
|
||||
|
||||
console.log('✅ 找到以下Schema：');
|
||||
schemas.forEach(s => console.log(` - ${s.schema_name}`));
|
||||
|
||||
const hasRvwSchema = schemas.some(s => s.schema_name === 'rvw_schema');
|
||||
const hasPkbSchema = schemas.some(s => s.schema_name === 'pkb_schema');
|
||||
|
||||
console.log(`\n📊 Schema存在性检查：`);
|
||||
console.log(` pkb_schema: ${hasPkbSchema ? '✅ 存在' : '❌ 不存在'}`);
|
||||
console.log(` rvw_schema: ${hasRvwSchema ? '✅ 存在' : '❌ 不存在'}`);
|
||||
|
||||
// ========================================
|
||||
// 2. 检查PKB相关表
|
||||
// ========================================
|
||||
console.log('\n');
|
||||
console.log('='.repeat(80));
|
||||
console.log('📚 2. 检查PKB相关表（在pkb_schema中）');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
if (hasPkbSchema) {
|
||||
const pkbTables = await prisma.$queryRaw<TableInfo[]>`
|
||||
SELECT table_schema, table_name
|
||||
FROM information_schema.tables
|
||||
WHERE table_schema = 'pkb_schema'
|
||||
ORDER BY table_name
|
||||
`;
|
||||
|
||||
console.log(`✅ pkb_schema中共有 ${pkbTables.length} 个表：`);
|
||||
pkbTables.forEach(t => console.log(` - ${t.table_name}`));
|
||||
|
||||
// 检查每个表的行数
|
||||
console.log('\n📊 PKB表数据统计：');
|
||||
for (const table of pkbTables) {
|
||||
try {
|
||||
const result = await prisma.$queryRaw<RowCount[]>`
|
||||
SELECT COUNT(*) as count FROM pkb_schema.${prisma.$queryRawUnsafe(table.table_name)}
|
||||
`;
|
||||
const count = Number(result[0]?.count || 0);
|
||||
console.log(` - ${table.table_name}: ${count} 行`);
|
||||
} catch (error) {
|
||||
console.log(` - ${table.table_name}: 查询失败`);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
console.log('❌ pkb_schema不存在！需要创建！');
|
||||
}
|
||||
|
||||
// ========================================
|
||||
// 3. 检查RVW相关表
|
||||
// ========================================
|
||||
console.log('\n');
|
||||
console.log('='.repeat(80));
|
||||
console.log('📝 3. 检查RVW相关表（review_tasks）');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
// 查找review_tasks表在哪个schema
|
||||
const reviewTaskLocation = await prisma.$queryRaw<TableInfo[]>`
|
||||
SELECT table_schema, table_name
|
||||
FROM information_schema.tables
|
||||
WHERE table_name = 'review_tasks'
|
||||
`;
|
||||
|
||||
if (reviewTaskLocation.length > 0) {
|
||||
console.log('✅ 找到review_tasks表：');
|
||||
reviewTaskLocation.forEach(t => {
|
||||
console.log(` - 位置: ${t.table_schema}.${t.table_name}`);
|
||||
});
|
||||
|
||||
// 查看review_tasks表结构
|
||||
const reviewSchema = reviewTaskLocation[0].table_schema;
|
||||
console.log(`\n📋 review_tasks表结构（在${reviewSchema}中）：`);
|
||||
|
||||
const reviewColumns = await prisma.$queryRaw<ColumnInfo[]>`
|
||||
SELECT column_name, data_type, is_nullable, column_default
|
||||
FROM information_schema.columns
|
||||
WHERE table_schema = ${reviewSchema}
|
||||
AND table_name = 'review_tasks'
|
||||
ORDER BY ordinal_position
|
||||
`;
|
||||
|
||||
console.log('='.repeat(80));
|
||||
console.log(
|
||||
'Column Name'.padEnd(25) +
|
||||
'Data Type'.padEnd(20) +
|
||||
'Nullable'.padEnd(12) +
|
||||
'Default'
|
||||
);
|
||||
console.log('='.repeat(80));
|
||||
|
||||
reviewColumns.forEach(col => {
|
||||
const colName = col.column_name.padEnd(25);
|
||||
const dataType = col.data_type.padEnd(20);
|
||||
const nullable = (col.is_nullable === 'YES' ? 'YES' : 'NO').padEnd(12);
|
||||
const defaultVal = col.column_default || '';
|
||||
console.log(`${colName}${dataType}${nullable}${defaultVal}`);
|
||||
});
|
||||
|
||||
console.log('='.repeat(80));
|
||||
console.log(`总计: ${reviewColumns.length} 个字段`);
|
||||
|
||||
// 查询数据量
|
||||
const reviewCount = await prisma.$queryRaw<RowCount[]>`
|
||||
SELECT COUNT(*) as count FROM ${prisma.$queryRawUnsafe(reviewSchema)}.review_tasks
|
||||
`;
|
||||
const count = Number(reviewCount[0]?.count || 0);
|
||||
console.log(`\n📊 review_tasks表数据量: ${count} 行`);
|
||||
|
||||
if (reviewSchema === 'public') {
|
||||
console.log('\n⚠️ WARNING: review_tasks当前在public schema中！');
|
||||
console.log(' 建议迁移到rvw_schema以保持架构一致性！');
|
||||
}
|
||||
} else {
|
||||
console.log('❌ 未找到review_tasks表！');
|
||||
}
|
||||
|
||||
// ========================================
|
||||
// 4. 检查rvw_schema状态
|
||||
// ========================================
|
||||
console.log('\n');
|
||||
console.log('='.repeat(80));
|
||||
console.log('🔍 4. 检查rvw_schema状态');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
if (hasRvwSchema) {
|
||||
const rvwTables = await prisma.$queryRaw<TableInfo[]>`
|
||||
SELECT table_schema, table_name
|
||||
FROM information_schema.tables
|
||||
WHERE table_schema = 'rvw_schema'
|
||||
ORDER BY table_name
|
||||
`;
|
||||
|
||||
if (rvwTables.length > 0) {
|
||||
console.log(`✅ rvw_schema中共有 ${rvwTables.length} 个表：`);
|
||||
rvwTables.forEach(t => console.log(` - ${t.table_name}`));
|
||||
} else {
|
||||
console.log('⚠️ rvw_schema存在但为空（未创建任何表）');
|
||||
}
|
||||
} else {
|
||||
console.log('❌ rvw_schema不存在！需要创建！');
|
||||
}
|
||||
|
||||
// ========================================
|
||||
// 5. 检查User表状态
|
||||
// ========================================
|
||||
console.log('\n');
|
||||
console.log('='.repeat(80));
|
||||
console.log('👤 5. 检查User表状态');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
const userTables = await prisma.$queryRaw<TableInfo[]>`
|
||||
SELECT table_schema, table_name
|
||||
FROM information_schema.tables
|
||||
WHERE table_name = 'users'
|
||||
AND table_schema IN ('public', 'platform_schema')
|
||||
`;
|
||||
|
||||
console.log('✅ 找到以下users表：');
|
||||
for (const userTable of userTables) {
|
||||
console.log(` - ${userTable.table_schema}.${userTable.table_name}`);
|
||||
|
||||
const userCount = await prisma.$queryRaw<RowCount[]>`
|
||||
SELECT COUNT(*) as count FROM ${prisma.$queryRawUnsafe(userTable.table_schema)}.users
|
||||
`;
|
||||
const count = Number(userCount[0]?.count || 0);
|
||||
console.log(` 数据量: ${count} 行`);
|
||||
}
|
||||
|
||||
// ========================================
|
||||
// 6. 总结和建议
|
||||
// ========================================
|
||||
console.log('\n');
|
||||
console.log('='.repeat(80));
|
||||
console.log('📋 6. 迁移建议总结');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
console.log('\n✅ PKB模块迁移建议：');
|
||||
if (hasPkbSchema && pkbTables.length > 0) {
|
||||
console.log(' 1. ✅ pkb_schema已存在且有表');
|
||||
console.log(' 2. ✅ 可以直接迁移代码到新架构');
|
||||
console.log(' 3. ⚠️ 注意检查旧代码是否连接到正确的schema');
|
||||
} else {
|
||||
console.log(' 1. ❌ 需要先创建pkb_schema和相关表');
|
||||
console.log(' 2. ❌ 需要运行Prisma迁移');
|
||||
}
|
||||
|
||||
console.log('\n⚠️ RVW模块迁移建议：');
|
||||
if (reviewTaskLocation.length > 0) {
|
||||
const currentSchema = reviewTaskLocation[0].table_schema;
|
||||
if (currentSchema === 'public') {
|
||||
console.log(' 1. ⚠️ review_tasks当前在public schema');
|
||||
console.log(' 2. 🔄 建议：先完整迁移功能，后续再迁移Schema');
|
||||
console.log(' 3. 📋 迁移步骤：');
|
||||
console.log(' a. 创建rvw_schema（如果不存在）');
|
||||
console.log(' b. 在rvw_schema中创建新表');
|
||||
console.log(' c. 迁移数据');
|
||||
console.log(' d. 更新代码');
|
||||
console.log(' e. 删除旧表');
|
||||
} else if (currentSchema === 'rvw_schema') {
|
||||
console.log(' 1. ✅ review_tasks已在rvw_schema中');
|
||||
console.log(' 2. ✅ 可以直接迁移代码');
|
||||
}
|
||||
} else {
|
||||
console.log(' 1. ❌ review_tasks表不存在');
|
||||
console.log(' 2. ❌ 需要从头创建');
|
||||
}
|
||||
|
||||
console.log('\n💡 推荐迁移策略：');
|
||||
console.log(' 【方案A：渐进式迁移（推荐）】');
|
||||
console.log(' 1. PKB: 直接迁移代码到新架构（Schema已就绪）');
|
||||
console.log(' 2. RVW: 先迁移代码保持在public schema');
|
||||
console.log(' 3. 验证功能完整性');
|
||||
console.log(' 4. 后续独立任务：将RVW迁移到rvw_schema');
|
||||
console.log('');
|
||||
console.log(' 【方案B：一步到位（风险较高）】');
|
||||
console.log(' 1. 同时迁移代码+Schema');
|
||||
console.log(' 2. 需要数据迁移脚本');
|
||||
console.log(' 3. 需要更长的测试时间');
|
||||
|
||||
console.log('\n='.repeat(80));
|
||||
console.log('✅ 验证完成！');
|
||||
console.log('='.repeat(80));
|
||||
|
||||
} catch (error) {
|
||||
console.error('❌ 验证过程中出错：', error);
|
||||
throw error;
|
||||
} finally {
|
||||
await prisma.$disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
// 运行验证
|
||||
verifySchemas()
|
||||
.catch((error) => {
|
||||
console.error('Fatal error:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
|
||||
Reference in New Issue
Block a user