feat(iit-manager): Integrate Dify knowledge base for hybrid retrieval
Completed features: - Created Dify dataset (Dify_test0102) with 2 processed documents - Linked test0102 project with Dify dataset ID - Extended intent detection to recognize query_protocol intent - Implemented queryDifyKnowledge method (semantic search Top 5) - Integrated hybrid retrieval (REDCap data + Dify documents) - Fixed AI hallucination bugs (intent detection + API field path) - Developed debugging scripts - Completed end-to-end testing (5 scenarios passed) - Generated comprehensive documentation (600+ lines) - Updated development plans and module status Technical highlights: - Single project single knowledge base architecture - Smart routing based on user intent - Prevent AI hallucination by injecting real data/documents - Session memory for multi-turn conversations - Reused LLMFactory for DeepSeek-V3 integration Bug fixes: - Fixed intent detection missing keywords - Fixed Dify API response field path error Testing: All scenarios verified in WeChat production environment Status: Fully tested and deployed
This commit is contained in:
128
backend/src/modules/iit-manager/test-chatservice-dify.ts
Normal file
128
backend/src/modules/iit-manager/test-chatservice-dify.ts
Normal file
@@ -0,0 +1,128 @@
|
||||
/**
|
||||
* 测试ChatService的Dify知识库集成
|
||||
*
|
||||
* 测试场景:
|
||||
* 1. 询问研究方案相关问题(触发Dify检索)
|
||||
* 2. 询问患者数据(触发REDCap查询)
|
||||
* 3. 混合查询(同时涉及文档和数据)
|
||||
*/
|
||||
|
||||
import { ChatService } from './services/ChatService.js';
|
||||
|
||||
const chatService = new ChatService();
|
||||
|
||||
async function testDifyIntegration() {
|
||||
console.log('='.repeat(80));
|
||||
console.log('🧪 测试ChatService的Dify知识库集成');
|
||||
console.log('='.repeat(80));
|
||||
console.log('');
|
||||
|
||||
const testUserId = 'FengZhiBo';
|
||||
|
||||
// 测试1:研究方案相关问题(应该触发Dify检索)
|
||||
console.log('📝 测试1:询问研究的纳入排除标准(应触发Dify检索)');
|
||||
console.log('-'.repeat(80));
|
||||
try {
|
||||
const answer1 = await chatService.handleMessage(
|
||||
testUserId,
|
||||
'这个研究的排除标准是什么?'
|
||||
);
|
||||
console.log('✅ AI回答:');
|
||||
console.log(answer1);
|
||||
console.log('');
|
||||
} catch (error: any) {
|
||||
console.error('❌ 测试1失败:', error.message);
|
||||
console.log('');
|
||||
}
|
||||
|
||||
// 等待2秒
|
||||
await new Promise(resolve => setTimeout(resolve, 2000));
|
||||
|
||||
// 测试2:CRF相关问题(应该触发Dify检索)
|
||||
console.log('📝 测试2:询问CRF表格内容(应触发Dify检索)');
|
||||
console.log('-'.repeat(80));
|
||||
try {
|
||||
const answer2 = await chatService.handleMessage(
|
||||
testUserId,
|
||||
'CRF表格中有哪些观察指标?'
|
||||
);
|
||||
console.log('✅ AI回答:');
|
||||
console.log(answer2);
|
||||
console.log('');
|
||||
} catch (error: any) {
|
||||
console.error('❌ 测试2失败:', error.message);
|
||||
console.log('');
|
||||
}
|
||||
|
||||
// 等待2秒
|
||||
await new Promise(resolve => setTimeout(resolve, 2000));
|
||||
|
||||
// 测试3:患者数据查询(应该触发REDCap查询)
|
||||
console.log('📊 测试3:询问患者记录(应触发REDCap查询)');
|
||||
console.log('-'.repeat(80));
|
||||
try {
|
||||
const answer3 = await chatService.handleMessage(
|
||||
testUserId,
|
||||
'查询一下ID 7的患者情况'
|
||||
);
|
||||
console.log('✅ AI回答:');
|
||||
console.log(answer3);
|
||||
console.log('');
|
||||
} catch (error: any) {
|
||||
console.error('❌ 测试3失败:', error.message);
|
||||
console.log('');
|
||||
}
|
||||
|
||||
// 等待2秒
|
||||
await new Promise(resolve => setTimeout(resolve, 2000));
|
||||
|
||||
// 测试4:混合查询(可能同时触发Dify和REDCap)
|
||||
console.log('🔀 测试4:混合查询(询问研究目的)');
|
||||
console.log('-'.repeat(80));
|
||||
try {
|
||||
const answer4 = await chatService.handleMessage(
|
||||
testUserId,
|
||||
'这个研究的主要研究目的是什么?'
|
||||
);
|
||||
console.log('✅ AI回答:');
|
||||
console.log(answer4);
|
||||
console.log('');
|
||||
} catch (error: any) {
|
||||
console.error('❌ 测试4失败:', error.message);
|
||||
console.log('');
|
||||
}
|
||||
|
||||
// 测试5:统计查询(REDCap)
|
||||
console.log('📈 测试5:统计查询(应触发REDCap查询)');
|
||||
console.log('-'.repeat(80));
|
||||
try {
|
||||
const answer5 = await chatService.handleMessage(
|
||||
testUserId,
|
||||
'目前有多少位患者入组?'
|
||||
);
|
||||
console.log('✅ AI回答:');
|
||||
console.log(answer5);
|
||||
console.log('');
|
||||
} catch (error: any) {
|
||||
console.error('❌ 测试5失败:', error.message);
|
||||
console.log('');
|
||||
}
|
||||
|
||||
console.log('='.repeat(80));
|
||||
console.log('✅ 测试完成!');
|
||||
console.log('='.repeat(80));
|
||||
console.log('');
|
||||
console.log('📝 测试总结:');
|
||||
console.log(' - Dify知识库检索(研究方案、CRF)');
|
||||
console.log(' - REDCap数据查询(患者记录、统计)');
|
||||
console.log(' - 上下文记忆(SessionMemory)');
|
||||
console.log('');
|
||||
console.log('🚀 下一步:企业微信端到端测试');
|
||||
}
|
||||
|
||||
// 执行测试
|
||||
testDifyIntegration().catch(error => {
|
||||
console.error('❌ 测试脚本执行失败:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user