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

/**
 * PKB璅∪<E79285>API<50>芸𢆡<E88AB8>𡝗<EFBFBD>霂閗<E99C82><E99697>? * 
 * <20>蠘<EFBFBD>嚗? * 1. 瘚贝<E7989A><E8B49D><EFBFBD><EFBFBD>侨KB API蝡舐<E89DA1>嚗ǒ1<C792>綋2嚗? * 2. 撖寞<E69296>v1<76>綋2<E7B68B><32><EFBFBD><EFBFBD>䂿<EFBFBD><E482BF>? * 3. 撉諹<E69289><E8ABB9>唳旿銝<E697BF><E98A9D>湔<EFBFBD>? * 4. <20>扯<EFBFBD>撖寞<E69296>
 * 5. 颲寧<E9A2B2><E5AFA7>∩辣瘚贝<E7989A>
 * 
 * 餈鞱<E9A488><E99EB1>孵<EFBFBD>嚗? * npx tsx scripts/test-pkb-apis.ts
 */

import axios, { AxiosError } from 'axios';

const BASE_URL = 'http://localhost:3000';
const TEST_KB_NAME = `瘚贝<EFBFBD><EFBFBD>亥<EFBFBD>摨?${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;

// 撌亙<E6928C><E4BA99>賣㺭嚗𡁏<E59A97>颲<EFBFBD>舅銝芸<E98A9D>摨娍糓<E5A88D>虫<EFBFBD><E899AB>?function compareResponses(v1: any, v2: any): boolean {
  return JSON.stringify(v1) === JSON.stringify(v2);
}

// 撌亙<E6928C><E4BA99>賣㺭嚗𡁏<E59A97><F0A1818F>唳<EFBFBD>霂閧<E99C82><E996A7>?function printResult(result: TestResult) {
  const icon = result.status === 'pass' ? '<27>? : result.status === 'fail' ? '<EFBFBD>? : '<27>哨<EFBFBD>';
  console.log(`${icon} ${result.name} ${result.duration ? `(${result.duration}ms)` : ''}`);
  if (result.message) {
    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' && response.data.module === 'pkb' && response.data.version === 'v2') {
      return {
        name: '<27>亙熒璉<E78692><E79289>?,
        status: 'pass',
        message: `<60>嗆<EFBFBD>? ${response.data.status}, <20>亥<EFBFBD>摨𤘪㺭: ${response.data.database.knowledgeBases}`,
        duration,
      };
    } else {
      return {
        name: '<EFBFBD>亙熒璉<EFBFBD><EFBFBD>?,
        status: 'fail',
        message: '餈𥪜<E9A488><F0A5AA9C>唳旿<E594B3>澆<EFBFBD>銝齿迤蝖?,
        duration,
      };
    }
  } catch (error: any) {
    return {
      name: '<EFBFBD>亙熒璉<EFBFBD><EFBFBD>?,
      status: 'fail',
      message: error.message,
      duration: Date.now() - startTime,
    };
  }
}

// 瘚贝<E7989A>2嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82><EFBFBD>𡑒”嚗<E2809D>笆瘥畕1<E79595>綋2嚗?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>唳旿銝<E697BF><E98A9D>氯<EFBFBD>`,
        duration: v1Duration + v2Duration,
        v1Response: v1Response.data,
        v2Response: v2Response.data,
      };
    } 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嚗𡁜<E59A97>撱箇䰻霂<E4B0BB><E99C82>嚗ǒ2嚗?async function testCreateKnowledgeBase(): Promise<TestResult> {
  const startTime = Date.now();
  try {
    const response = await axios.post(`${BASE_URL}/api/v1/pkb/knowledge/knowledge-bases`, {
      name: TEST_KB_NAME,
      description: '餈蹱糓銝<E7B393>銝芾䌊<E88ABE>典<EFBFBD>瘚贝<E7989A><E8B49D>𥕦遣<F0A595A6><E981A3>䰻霂<E4B0BB><E99C82>',
    }, {
      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: '<27>𥕦遣<F0A595A6>亥<EFBFBD>摨橒<E691A8>v2嚗?,
        status: 'pass',
        message: `<EFBFBD>𣂼<EFBFBD><EFBFBD>𥕦遣嚗䬠D: ${testKbId}`,
        duration,
      };
    } else {
      return {
        name: '<27>𥕦遣<F0A595A6>亥<EFBFBD>摨橒<E691A8>v2嚗?,
        status: 'fail',
        message: '<27>𥕦遣憭梯揖<E6A2AF>𤥁<EFBFBD><F0A4A581>墧聢撘譍<E69298>甇<EFBFBD>＆',
        duration,
      };
    }
  } catch (error: any) {
    const errorDetail = error.response?.data ? 
      JSON.stringify(error.response.data) : 
      (error.response?.data?.message || error.message);
    return {
      name: '<27>𥕦遣<F0A595A6>亥<EFBFBD>摨橒<E691A8>v2嚗?,
      status: 'fail',
      message: errorDetail,
      duration: Date.now() - startTime,
    };
  }
}

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

// 瘚贝<E7989A>5嚗𡁏凒<F0A1818F>啁䰻霂<E4B0BB><E99C82>嚗ǒ2嚗?async function testUpdateKnowledgeBase(kbId: string): Promise<TestResult> {
  const startTime = Date.now();
  try {
    const response = await axios.put(`${BASE_URL}/api/v1/pkb/knowledge/knowledge-bases/${kbId}`, {
      name: `${TEST_KB_NAME}-撌脫凒<E884AB>躬,
      description: '<27>讛膩撌脫凒<E884AB>?,
    });
    const duration = Date.now() - startTime;
    
    if (response.data.success) {
      return {
        name: '<27>湔鰵<E6B994>亥<EFBFBD>摨橒<E691A8>v2嚗?,
        status: 'pass',
        message: '<27>湔鰵<E6B994>𣂼<EFBFBD>',
        duration,
      };
    } else {
      return {
        name: '<27>湔鰵<E6B994>亥<EFBFBD>摨橒<E691A8>v2嚗?,
        status: 'fail',
        message: '<27>湔鰵憭梯揖',
        duration,
      };
    }
  } catch (error: any) {
    return {
      name: '<27>湔鰵<E6B994>亥<EFBFBD>摨橒<E691A8>v2嚗?,
      status: 'fail',
      message: error.response?.data?.message || error.message,
      duration: Date.now() - startTime,
    };
  }
}

// 瘚贝<E7989A>6嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82>蝏蠘恣嚗<E681A3>笆瘥畕1<E79595>綋2嚗?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 v1Stats = v1Response.data.data;
    const v2Stats = v2Response.data.data;
    
    if (v1Stats.totalDocuments === v2Stats.totalDocuments) {
      return {
        name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨梶<E691A8>霈∴<E99C88>v1 vs v2嚗?,
        status: 'pass',
        message: `<EFBFBD><EFBFBD>﹝<EFBFBD>唬<EFBFBD><EFBFBD>? ${v1Stats.totalDocuments}銝芬<EFBFBD>`,
      };
    } else {
      return {
        name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨梶<E691A8>霈∴<E99C88>v1 vs v2嚗?,
        status: 'fail',
        message: `<EFBFBD><EFBFBD>﹝<EFBFBD>唬<EFBFBD>銝<EFBFBD><EFBFBD>湛<EFBFBD>v1: ${v1Stats.totalDocuments}, v2: ${v2Stats.totalDocuments}`,
      };
    }
  } catch (error: any) {
    return {
      name: '<27>瑕<EFBFBD><E79195>亥<EFBFBD>摨梶<E691A8>霈∴<E99C88>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>7嚗鑹AG璉<47>蝝ｇ<E89D9D>撖寞<E69296>v1<76>綋2嚗?async function testSearchKnowledgeBase(kbId: string): Promise<TestResult> {
  try {
    const query = '瘚贝<E7989A><E8B49D>亥砭';
    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/v1/pkb/knowledge/knowledge-bases/${kbId}/search`, {
      params: { query, top_k: 5 },
    });
    
    return {
      name: 'RAG璉<47>蝝ｇ<E89D9D>v1 vs v2嚗?,
      status: 'pass',
      message: '璉<>蝝Ｘ<E89D9D><EFBCB8><EFBFBD><EFBFBD>銝支葵<E694AF><E891B5>𧋦<EFBFBD>質<EFBFBD><E8B3AA>硺<EFBFBD>蝏𤘪<E89D8F>',
    };
  } catch (error: any) {
    return {
      name: 'RAG璉<47>蝝ｇ<E89D9D>v1 vs v2嚗?,
      status: 'fail',
      message: error.message,
    };
  }
}

// 瘚贝<E7989A>8嚗朞器<E69C9E>峕辺隞?- 銝滚<E98A9D><E6BB9A>函<EFBFBD><E587BD>亥<EFBFBD>摨?async function testNotFoundKnowledgeBase(): Promise<TestResult> {
  try {
    await axios.get(`${BASE_URL}/api/v1/pkb/knowledge/knowledge-bases/00000000-0000-0000-0000-000000000000`);
    return {
      name: '颲寧<E9A2B2>瘚贝<E7989A>嚗帋<E59A97>摮睃銁<E79D83><E98A81>䰻霂<E4B0BB><E99C82>',
      status: 'fail',
      message: '摨磰砲餈𥪜<E9A488>404<30>躰秤嚗䔶<E59A97>瘝⊥<E7989D>',
    };
  } catch (error: any) {
    if (error.response?.status === 404 || error.response?.status === 500) {
      return {
        name: '颲寧<E9A2B2>瘚贝<E7989A>嚗帋<E59A97>摮睃銁<E79D83><E98A81>䰻霂<E4B0BB><E99C82>',
        status: 'pass',
        message: `甇<EFBFBD>＆餈𥪜<EFBFBD><EFBFBD>躰秤<EFBFBD>嗆<EFBFBD>? ${error.response.status}<EFBFBD><EFBFBD>,
      };
    } else {
      return {
        name: '颲寧<E9A2B2>瘚贝<E7989A>嚗帋<E59A97>摮睃銁<E79D83><E98A81>䰻霂<E4B0BB><E99C82>',
        status: 'fail',
        message: `<EFBFBD>誩<EFBFBD><EFBFBD><EFBFBD>𠶖<EFBFBD><EFBFBD><EFBFBD>: ${error.response?.status}`,
      };
    }
  }
}

// 瘚贝<E7989A>9嚗𡁏<E59A97><F0A1818F>?- <20>𣳇膄瘚贝<E7989A><E8B49D>亥<EFBFBD>摨?async function testDeleteKnowledgeBase(kbId: string): Promise<TestResult> {
  const startTime = Date.now();
  try {
    const response = await axios.delete(`${BASE_URL}/api/v1/pkb/knowledge/knowledge-bases/${kbId}`);
    const duration = Date.now() - startTime;
    
    if (response.data.success) {
      return {
        name: '<27>𣳇膄<F0A3B387>亥<EFBFBD>摨橒<E691A8>v2嚗?,
        status: 'pass',
        message: '<EFBFBD>𣳇膄<EFBFBD>𣂼<EFBFBD>',
        duration,
      };
    } else {
      return {
        name: '<EFBFBD>𣳇膄<EFBFBD>亥<EFBFBD>摨橒<EFBFBD>v2嚗?,
        status: 'fail',
        message: '<27>𣳇膄憭梯揖',
        duration,
      };
    }
  } catch (error: any) {
    return {
      name: '<27>𣳇膄<F0A3B387>亥<EFBFBD>摨橒<E691A8>v2嚗?,
      status: 'fail',
      message: error.response?.data?.message || error.message,
      duration: Date.now() - startTime,
    };
  }
}

// 銝餅<E98A9D>霂訫遆<E8A8AB>?async function runTests() {
  console.log('<EFBFBD><EFBFBD> 撘<EFBFBD>憪閪KB API<EFBFBD>芸𢆡<EFBFBD>𡝗<EFBFBD>霂?..\n');
  console.log('='.repeat(80));
  
  // 瘚贝<E7989A>1嚗𡁜<E59A97>摨瑟<E691A8><E7919F>?  console.log('\n<EFBFBD><EFBFBD> <EFBFBD>嗆挾1嚗𡁜<EFBFBD>摨瑟<EFBFBD><EFBFBD>?);
  console.log('-'.repeat(80));
  results.push(await testHealthCheck());
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>2嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82><EFBFBD>𡑒”
  console.log('\n<><6E> <20>嗆挾2嚗𡁶䰻霂<E4B0BB><E99C82><EFBFBD>𡑒”');
  console.log('-'.repeat(80));
  results.push(await testGetKnowledgeBases());
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>3嚗𡁜<E59A97>撱箇䰻霂<E4B0BB><E99C82>
  console.log('\n<><6E> <20>嗆挾3嚗𡁜<E59A97>撱箇䰻霂<E4B0BB><E99C82>');
  console.log('-'.repeat(80));
  results.push(await testCreateKnowledgeBase());
  printResult(results[results.length - 1]);
  
  if (!testKbId) {
    console.log('\n<>?<3F>䭾<EFBFBD><E4ADBE>瑕<EFBFBD>瘚贝<E7989A><E8B49D>亥<EFBFBD>摨𨧻D嚗<44><E59A97>蝏剜<E89D8F>霂閗歲餈?);
    return;
  }
  
  // 瘚贝<E7989A>4嚗朞繮<E69C9E>𣇉䰻霂<E4B0BB><E99C82>霂行<E99C82>
  console.log('\n<EFBFBD><EFBFBD> <EFBFBD>嗆挾4嚗𡁶䰻霂<EFBFBD><EFBFBD>霂行<EFBFBD>');
  console.log('-'.repeat(80));
  results.push(await testGetKnowledgeBaseById(testKbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>5嚗𡁏凒<F0A1818F>啁䰻霂<E4B0BB><E99C82>
  console.log('\n<EFBFBD><EFBFBD> <EFBFBD>嗆挾5嚗𡁏凒<EFBFBD>啁䰻霂<EFBFBD><EFBFBD>');
  console.log('-'.repeat(80));
  results.push(await testUpdateKnowledgeBase(testKbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>6嚗朞繮<E69C9E>𣇉<EFBFBD>霈∩縑<E288A9>?  console.log('\n<EFBFBD><EFBFBD> <EFBFBD>嗆挾6嚗𡁶䰻霂<EFBFBD><EFBFBD>蝏蠘恣');
  console.log('-'.repeat(80));
  results.push(await testGetKnowledgeBaseStats(testKbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>7嚗鑹AG璉<47>蝝?  console.log('\n<EFBFBD><EFBFBD> <EFBFBD>嗆挾7嚗鑹AG璉<EFBFBD>蝝?);
  console.log('-'.repeat(80));
  results.push(await testSearchKnowledgeBase(testKbId));
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>8嚗朞器<E69C9E>峕辺隞?  console.log('\n<><6E> <20>嗆挾8嚗朞器<E69C9E>峕辺隞嗆<E99A9E>霂?);
  console.log('-'.repeat(80));
  results.push(await testNotFoundKnowledgeBase());
  printResult(results[results.length - 1]);
  
  // 瘚贝<E7989A>9嚗𡁏<E59A97><F0A1818F>?  console.log('\n<><6E> <20>嗆挾9嚗𡁏<E59A97><F0A1818F><EFBFBD><EFBFBD>霂閙㺭<E99699>?);
  console.log('-'.repeat(80));
  results.push(await testDeleteKnowledgeBase(testKbId));
  printResult(results[results.length - 1]);
  
  // <20>餌<EFBFBD>
  console.log('\n' + '='.repeat(80));
  console.log('<27><> 瘚贝<E7989A><E8B49D>餌<EFBFBD>');
  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<EFBFBD>餉恣: ${results.length}銝芣<EFBFBD>霂𧄧);
  console.log(`<EFBFBD>?<EFBFBD>朞<EFBFBD>: ${passCount}銝注);
  console.log(`<EFBFBD>?憭梯揖: ${failCount}銝注);
  console.log(`<EFBFBD>哨<EFBFBD>  頝唾<EFBFBD>: ${skipCount}銝注);
  console.log(`<EFBFBD>梧<EFBFBD>  <20>餉<EFBFBD>埈𧒄: ${totalDuration}ms`);
  
  if (failCount === 0) {
    console.log('\n<><6E> <20><><EFBFBD>㗇<EFBFBD>霂閖<E99C82>朞<EFBFBD>嚗?);
  } else {
    console.log('\n<EFBFBD>𩤃<EFBFBD>  <EFBFBD>典<EFBFBD>瘚贝<EFBFBD>憭梯揖嚗諹窈<EFBFBD>亦<EFBFBD>霂行<EFBFBD>');
  }
}

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