AIclinicalresearch/backend/scripts/test-pkb-apis-simple.ts

/**
 * PKB璅∪<E79285>API蝞<49><E89D9E>𡝗<EFBFBD>霂閗<E99C82><E99697>?
 * 瘚贝<E7989A><E8B49D>唳<EFBFBD><E594B3>亥<EFBFBD>摨梶<E691A8><E6A2B6><EFBFBD>★<EFBFBD>蠘<EFBFBD>
 */

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' ? '<27>? : '<EFBFBD>?;
  console.log(`${icon} ${result.name} ${result.duration ? `(${result.duration}ms)` : ''}`);
  console.log(`   ${result.message}`);
}

// 瘚贝<E7989A>1嚗𡁜<E59A97>摨瑟<E691A8><E7919F>?
async function testHealthCheck(): Promise<TestResult> {
  const startTime = Date.now();
  try {
    const response = await axios.get(`${BASE_URL}/api/v1/pkb/health`);
    const duration = Date.now() - startTime;
    
    if (response.data.status === 'ok') {
      return {
        name: '<27>亙熒璉<E78692><E79289>伐<EFBFBD>v2嚗?,
        status: 'pass',
        message: `<60>亥<EFBFBD>摨𤘪㺭: ${response.data.database.knowledgeBases}, schema: ${response.data.database.schema}`,
        duration,
      };
    } else {
      return {
        name: '<EFBFBD>亙熒璉<EFBFBD><EFBFBD>伐<EFBFBD>v2嚗?,
        status: 'fail',
        message: '餈𥪜<E9A488><F0A5AA9C>嗆<EFBFBD><E59786><EFBFBD>撣?,
        duration,
      };
    }
  } catch (error: any) {
    return {
      name: '<EFBFBD>亙熒璉<EFBFBD><EFBFBD>伐<EFBFBD>v2嚗?,
      status: 'fail',
      message: error.message,
      duration: Date.now() - startTime,
    };
  }
}

// 瘚贝<E7989A>2嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82><EFBFBD>𡑒”嚗ǒ1 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/v1/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: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨枏<E691A8>銵剁<E98AB5>v1 vs v2嚗?,
        status: 'pass',
        message: `v1: ${v1Count}銝?(${v1Duration}ms), v2: ${v2Count}銝?(${v2Duration}ms) <20><>,
        duration: v1Duration + v2Duration,
      };
    } else {
      return {
        name: '<EFBFBD>瑕<EFBFBD><EFBFBD>亥<EFBFBD>摨枏<EFBFBD>銵剁<EFBFBD>v1 vs v2嚗?,
        status: 'fail',
        message: `<EFBFBD>圈<EFBFBD>銝滢<EFBFBD><EFBFBD>湛<EFBFBD>v1: ${v1Count}, v2: ${v2Count}`,
        duration: v1Duration + v2Duration,
      };
    }
  } catch (error: any) {
    return {
      name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨枏<E691A8>銵剁<E98AB5>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>3嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82>霂行<E99C82>嚗ǒ1 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/v1/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: '<EFBFBD>瑕<EFBFBD><EFBFBD>亥<EFBFBD>摨栞祕<EFBFBD><EFBFBD><EFBFBD>v1 vs v2嚗?,
        status: 'pass',
        message: `<EFBFBD>滨妍銝<EFBFBD><EFBFBD>? "${v1Name}", v1: ${v1Duration}ms, v2: ${v2Duration}ms <20><>,
        duration: v1Duration + v2Duration,
      };
    } else {
      return {
        name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨栞祕<E6A09E><E7A595><EFBFBD>v1 vs v2嚗?,
        status: 'fail',
        message: `<EFBFBD>滨妍銝滢<EFBFBD><EFBFBD>湛<EFBFBD>v1: "${v1Name}", v2: "${v2Name}"`,
        duration: v1Duration + v2Duration,
      };
    }
  } catch (error: any) {
    return {
      name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨栞祕<E6A09E><E7A595><EFBFBD>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>4嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82>蝏蠘恣嚗ǒ1 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/v1/pkb/knowledge/knowledge-bases/${kbId}/stats`);
    
    const v1Docs = v1Response.data.data.totalDocuments;
    const v2Docs = v2Response.data.data.totalDocuments;
    
    if (v1Docs === v2Docs) {
      return {
        name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨梶<E691A8>霈∴<E99C88>v1 vs v2嚗?,
        status: 'pass',
        message: `<EFBFBD><EFBFBD>﹝<EFBFBD>唬<EFBFBD><EFBFBD>? ${v1Docs}銝?<EFBFBD><EFBFBD>,
      };
    } else {
      return {
        name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨梶<E691A8>霈∴<E99C88>v1 vs v2嚗?,
        status: 'fail',
        message: `<60><>﹝<EFBFBD>唬<EFBFBD>銝<EFBFBD><E98A9D>湛<EFBFBD>v1: ${v1Docs}, v2: ${v2Docs}`,
      };
    }
  } catch (error: any) {
    return {
      name: '<EFBFBD>瑕<EFBFBD><EFBFBD>亥<EFBFBD>摨梶<EFBFBD>霈∴<EFBFBD>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>5嚗鑹AG璉<47>蝝ｇ<E89D9D>v1 vs v2嚗?
async function testSearchKnowledgeBase(kbId: string): Promise<TestResult> {
  try {
    const query = '瘝餌<E7989D>';
    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/v1/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璉<47>蝝ｇ<E89D9D>v1 vs v2嚗?,
      status: 'pass',
      message: `v1餈𥪜<E9A488>${v1Count}<7D>? v2餈𥪜<E9A488>${v2Count}<7D>?<3F><>,
    };
  } catch (error: any) {
    return {
      name: 'RAG璉<EFBFBD>蝝ｇ<EFBFBD>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>6嚗𡁏<E59A97>獢<EFBFBD><E78DA2>㗇𥋘嚗<F0A58B98><E59A97><EFBFBD><EFBFBD><EFBFBD>霂餅芋撘𧶏<E69298>
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/v1/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: '<27><>﹝<EFBFBD>㗇𥋘-<2D>冽<EFBFBD><E586BD><EFBFBD>粉璅∪<E79285>嚗ǒ1 vs v2嚗?,
      status: 'pass',
      message: `v1<76>㗇𥋘${v1Docs}銝芣<E98A9D>獢? v2<76>㗇𥋘${v2Docs}銝芣<E98A9D>獢?<3F><>,
    };
  } catch (error: any) {
    return {
      name: '<EFBFBD><EFBFBD>﹝<EFBFBD>㗇𥋘-<EFBFBD>冽<EFBFBD><EFBFBD><EFBFBD>粉璅∪<EFBFBD>嚗ǒ1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>7嚗𡁏鸌憭<E9B88C><E686AD>璅⊥踎
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/v1/pkb/batch-tasks/batch/templates`);
    
    const v1Count = v1Response.data.data?.length || 0;
    const v2Count = v2Response.data.data?.length || 0;
    
    return {
      name: '<27>孵<EFBFBD><E5ADB5><EFBFBD>芋<EFBFBD>選<EFBFBD>v1 vs v2嚗?,
      status: 'pass',
      message: `v1: ${v1Count}銝芣芋<E88AA3>? v2: ${v2Count}銝芣芋<E88AA3>?<3F><>,
    };
  } catch (error: any) {
    return {
      name: '<EFBFBD>孵<EFBFBD><EFBFBD><EFBFBD>芋<EFBFBD>選<EFBFBD>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 銝餅<E98A9D>霂訫遆<E8A8AB>?
async function runTests() {
  console.log('<27><> PKB API瘚贝<E7989A>撘<EFBFBD>憪?..\n');
  console.log('='.repeat(80));
  
  // 瘚贝<E7989A>1嚗𡁜<E59A97>摨瑟<E691A8><E7919F>?
  console.log('\n<><6E> 瘚贝<E7989A>1嚗𡁜<E59A97>摨瑟<E691A8><E7919F>?);
  console.log('-'.repeat(80));
  results.push(await testHealthCheck());
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>2嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82><EFBFBD>𡑒”
  console.log('\n<EFBFBD><EFBFBD> 瘚贝<EFBFBD>2嚗𡁶䰻霂<EFBFBD><EFBFBD><EFBFBD>𡑒”');
  console.log('-'.repeat(80));
  results.push(await testGetKnowledgeBases());
  printResult(results[results.length - 1]);
  
  // <20>瑕<EFBFBD>蝚砌<E89D9A>銝芰䰻霂<E4B0BB><E99C82>ID<49>其<EFBFBD><E585B6>𡒊賒瘚贝<E7989A>
  const kbListResponse = await axios.get(`${BASE_URL}/api/v1/pkb/knowledge/knowledge-bases`);
  const firstKb = kbListResponse.data.data?.[0];
  
  if (!firstKb) {
    console.log('\n<EFBFBD>?瘝⊥<EFBFBD><EFBFBD>舐鍂<EFBFBD><EFBFBD>䰻霂<EFBFBD><EFBFBD>嚗<EFBFBD><EFBFBD>蝏剜<EFBFBD>霂閗歲餈?);
    return;
  }
  
  const kbId = firstKb.id;
  console.log(`\n雿輻鍂<EFBFBD>亥<EFBFBD>摨? ${firstKb.name} (ID: ${kbId})`);
  
  // 瘚贝<E7989A>3嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82>霂行<E99C82>
  console.log('\n<><6E> 瘚贝<E7989A>3嚗𡁶䰻霂<E4B0BB><E99C82>霂行<E99C82>');
  console.log('-'.repeat(80));
  results.push(await testGetKnowledgeBaseById(kbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>4嚗𡁶䰻霂<E4B0BB><E99C82>蝏蠘恣
  console.log('\n<><6E> 瘚贝<E7989A>4嚗𡁶䰻霂<E4B0BB><E99C82>蝏蠘恣');
  console.log('-'.repeat(80));
  results.push(await testGetKnowledgeBaseStats(kbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>5嚗鑹AG璉<47>蝝?
  console.log('\n<><6E> 瘚贝<E7989A>5嚗鑹AG璉<47>蝝?);
  console.log('-'.repeat(80));
  results.push(await testSearchKnowledgeBase(kbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>6嚗𡁏<E59A97>獢<EFBFBD><E78DA2>㗇𥋘
  console.log('\n<EFBFBD><EFBFBD> 瘚贝<EFBFBD>6嚗𡁏<EFBFBD>獢<EFBFBD><EFBFBD>㗇𥋘嚗<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>霂鳴<EFBFBD>');
  console.log('-'.repeat(80));
  results.push(await testDocumentSelection(kbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>7嚗𡁏鸌憭<E9B88C><E686AD>璅⊥踎
  console.log('\n<EFBFBD><EFBFBD> 瘚贝<EFBFBD>7嚗𡁏鸌憭<EFBFBD><EFBFBD>璅⊥踎');
  console.log('-'.repeat(80));
  results.push(await testBatchTemplates());
  printResult(results[results.length - 1]);
  
  // <20>餌<EFBFBD>
  console.log('\n' + '='.repeat(80));
  console.log('<EFBFBD><EFBFBD> 瘚贝<EFBFBD><EFBFBD>餌<EFBFBD>');
  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}銝芣<E98A9D>霂𧄧);
  console.log(`<60>?<3F>朞<EFBFBD>: ${passCount}銝注);
  console.log(`<60>?憭梯揖: ${failCount}銝注);
  console.log(`<60>梧<EFBFBD>  <20>餉<EFBFBD>埈𧒄: ${totalDuration}ms`);
  
  if (failCount === 0) {
    console.log('\n<EFBFBD><EFBFBD> <EFBFBD><EFBFBD><EFBFBD>㗇<EFBFBD>霂閖<EFBFBD>朞<EFBFBD>嚗<EFBFBD>1<EFBFBD>綋2<EFBFBD>蠘<EFBFBD>摰<EFBFBD><EFBFBD>銝<EFBFBD><EFBFBD>湛<EFBFBD>');
  } else {
    console.log('\n<EFBFBD>𩤃<EFBFBD>  <EFBFBD>典<EFBFBD>瘚贝<EFBFBD>憭梯揖嚗諹窈<EFBFBD>亦<EFBFBD>霂行<EFBFBD>');
  }
}

runTests().catch(error => {
  console.error('<EFBFBD>?瘚贝<EFBFBD><EFBFBD>扯<EFBFBD>憭梯揖:', error);
  process.exit(1);
});