Files
AIclinicalresearch/backend/scripts/test-prompt-api.ts
HaHafeng 40c2f8e148 feat(rag): Complete RAG engine implementation with pgvector
Major Features:
- Created ekb_schema (13th schema) with 3 tables: KB/Document/Chunk
- Implemented EmbeddingService (text-embedding-v4, 1024-dim vectors)
- Implemented ChunkService (smart Markdown chunking)
- Implemented VectorSearchService (multi-query + hybrid search)
- Implemented RerankService (qwen3-rerank)
- Integrated DeepSeek V3 QueryRewriter for cross-language search
- Python service: Added pymupdf4llm for PDF-to-Markdown conversion
- PKB: Dual-mode adapter (pgvector/dify/hybrid)

Architecture:
- Brain-Hand Model: Business layer (DeepSeek) + Engine layer (pgvector)
- Cross-language support: Chinese query matches English documents
- Small Embedding (1024) + Strong Reranker strategy

Performance:
- End-to-end latency: 2.5s
- Cost per query: 0.0025 RMB
- Accuracy improvement: +20.5% (cross-language)

Tests:
- test-embedding-service.ts: Vector embedding verified
- test-rag-e2e.ts: Full pipeline tested
- test-rerank.ts: Rerank quality validated
- test-query-rewrite.ts: Cross-language search verified
- test-pdf-ingest.ts: Real PDF document tested (Dongen 2003.pdf)

Documentation:
- Added 05-RAG-Engine-User-Guide.md
- Added 02-Document-Processing-User-Guide.md
- Updated system status documentation

Status: Production ready
2026-01-21 20:24:29 +08:00

95 lines
3.5 KiB
TypeScript
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
/**
* 测试 Prompt 管理 API
*
* 启动后端后运行: npx tsx scripts/test-prompt-api.ts
*/
const BASE_URL = 'http://localhost:3001/api/admin/prompts';
async function testAPI() {
console.log('🧪 测试 Prompt 管理 API...\n');
// 1. 获取列表
console.log('═══════════════════════════════════════════════════════');
console.log('📋 Test 1: GET /api/admin/prompts\n');
const listRes = await fetch(BASE_URL);
const listData = await listRes.json();
console.log(` 状态: ${listRes.status}`);
console.log(` 总数: ${listData.total}`);
console.log(` Prompts:`);
for (const p of listData.data || []) {
console.log(` - ${p.code} (${p.name}) v${p.latestVersion?.version || 0}`);
}
// 2. 获取详情
console.log('\n═══════════════════════════════════════════════════════');
console.log('📋 Test 2: GET /api/admin/prompts/RVW_EDITORIAL\n');
const detailRes = await fetch(`${BASE_URL}/RVW_EDITORIAL`);
const detailData = await detailRes.json();
console.log(` 状态: ${detailRes.status}`);
console.log(` Code: ${detailData.data?.code}`);
console.log(` Name: ${detailData.data?.name}`);
console.log(` 版本数: ${detailData.data?.versions?.length || 0}`);
// 3. 测试渲染
console.log('\n═══════════════════════════════════════════════════════');
console.log('📋 Test 3: POST /api/admin/prompts/test-render\n');
const renderRes = await fetch(`${BASE_URL}/test-render`, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
content: '你好,{{name}}!请评估标题:{{title}}',
variables: { name: '张三', title: '测试标题' },
}),
});
const renderData = await renderRes.json();
console.log(` 状态: ${renderRes.status}`);
console.log(` 渲染结果: ${renderData.data?.rendered}`);
console.log(` 提取变量: [${renderData.data?.extractedVariables?.join(', ')}]`);
console.log(` 校验结果: ${renderData.data?.validation?.isValid ? '✅ 通过' : '❌ 缺少变量'}`);
// 4. 按模块筛选
console.log('\n═══════════════════════════════════════════════════════');
console.log('📋 Test 4: GET /api/admin/prompts?module=RVW\n');
const rvwRes = await fetch(`${BASE_URL}?module=RVW`);
const rvwData = await rvwRes.json();
console.log(` 状态: ${rvwRes.status}`);
console.log(` RVW模块Prompt数: ${rvwData.total}`);
for (const p of rvwData.data || []) {
console.log(` - ${p.code}`);
}
// 5. 获取调试状态无认证预期返回401
console.log('\n═══════════════════════════════════════════════════════');
console.log('📋 Test 5: GET /api/admin/prompts/debug (无认证)\n');
const debugRes = await fetch(`${BASE_URL}/debug`);
const debugData = await debugRes.json();
console.log(` 状态: ${debugRes.status}`);
console.log(` 响应: ${JSON.stringify(debugData)}`);
console.log('\n✅ API 测试完成!');
}
testAPI().catch(console.error);