my-ai-chat-framework 2.0.0 → 3.0.0

package/test.js

@@ -1,107 +0,0 @@
-// test.js - 测试新框架核心功能
-import 'dotenv/config'; // 如果已安装 dotenv，否则手动读取 .env
-import { ChatService, openaiAdapter, toolCallingPlugin } from './src/index.js';
-
-// 从环境变量读取配置（确保 .env 文件存在）
-const apiKey = process.env.DEEPSEEK_API_KEY;
-const apiUrl = process.env.DEEPSEEK_API_URL || 'https://api.deepseek.com/v1/chat/completions';
-const model = process.env.TEST_MODEL || 'deepseek-chat';
-
-if (!apiKey) {
-  console.error('❌ 未找到 API Key，请设置 DEEPSEEK_API_KEY 环境变量或创建 .env 文件');
-  process.exit(1);
-}
-
-console.log('🚀 开始测试新框架...\n');
-
-// 创建聊天服务实例
-const chat = new ChatService({
-  apiKey,
-  model,
-  apiUrl,
-  temperature: 0.7,
-  maxTokens: 2000
-});
-
-// 加载适配器和插件
-chat.use(openaiAdapter);
-chat.use(toolCallingPlugin);
-
-// 监听事件（便于观察）
-chat.on('sending', msg => {
-  console.log('📤 发送消息:', msg.content?.substring(0, 50) || msg);
-});
-chat.on('message', msg => {
-  console.log('📥 收到回复:', msg.content?.substring(0, 100) || msg);
-});
-chat.on('stream-progress', chunk => {
-  console.log('🌊 流式数据块:', chunk.content?.substring(0, 30) || '');
-});
-chat.on('error', err => {
-  console.error('❌ 错误:', err.message);
-});
-
-// 注册一个简单测试工具（不实际调用，只验证注册和工具调用流程）
-chat.registerTool('greet', '打招呼工具', async (args) => {
-  console.log('🔧 工具被调用:', args);
-  return `你好，${args.name || '朋友'}！`;
-});
-
-// 辅助函数：等待指定毫秒
-const sleep = ms => new Promise(resolve => setTimeout(resolve, ms));
-
-// 测试 1：基础发送消息（非流式）
-console.log('===== 测试1：基础发送消息 =====');
-const response = await chat.send('请简单说一句：你好，世界！');
-console.log('最终回复:', response.content);
-console.log();
-
-// 等待一下，避免请求过快
-await sleep(1000);
-
-// 测试 2：流式发送消息
-console.log('===== 测试2：流式发送消息 =====');
-let streamContent = '';
-await chat.stream('请用流式方式回答：告诉我一个好消息。',
-  (chunk) => {
-    streamContent += chunk.content;
-    // 实时打印已接收内容（可选）
-    // process.stdout.write(chunk.content);
-  },
-  (final) => {
-    console.log('流式完成，完整内容:', final.content);
-  }
-);
-console.log();
-
-await sleep(1000);
-
-// 测试 3：工具调用
-console.log('===== 测试3：工具调用 =====');
-// 注意：需要让 AI 主动调用 greet 工具。由于 deepseek-chat 支持函数调用，我们可以在系统提示或用户消息中引导。
-// 我们在用户消息中明确要求调用工具，但 AI 是否调用取决于模型。这里做一个简单引导。
-const toolResponse = await chat.send('请使用 greet 工具，向 "小明" 打招呼。');
-console.log('工具调用后的回复:', toolResponse.content);
-console.log();
-
-await sleep(1000);
-
-// 测试 4：验证消息存储
-console.log('===== 测试4：消息存储验证 =====');
-const allMessages = chat.messages.getAll();
-console.log(`当前消息总数: ${allMessages.length}`);
-allMessages.forEach((msg, i) => {
-  console.log(`[${i}] ${msg.role}: ${msg.content?.substring(0, 40)}${msg.content?.length > 40 ? '...' : ''}`);
-  if (msg.toolCalls) console.log('  工具调用:', msg.toolCalls);
-  if (msg.toolCallId) console.log('  工具调用ID:', msg.toolCallId);
-});
-console.log();
-
-// 测试 5：撤回功能（可选）
-console.log('===== 测试5：撤回最后一条助手消息 =====');
-const removed = chat.messages.undoToLastAssistant();
-console.log(`撤回了 ${removed.length} 条消息`);
-const afterUndo = chat.messages.getAll();
-console.log(`剩余消息数: ${afterUndo.length}`);
-
-console.log('\n✅ 所有测试完成！');