@bolloon/bolloon-agent 0.1.0

package/src/test/p2p-agent-dialogue.ts

@@ -0,0 +1,423 @@
+/**
+ * P2P Agent Multi-Turn Dialogue (单向发送模式)
+ * 模拟真实场景：Alice 发送消息，Bob 处理并回复
+ *
+ * 运行: npx tsx src/test/p2p-agent-dialogue.ts
+ */
+
+import { config } from 'dotenv';
+import express from 'express';
+import { createServer } from 'http';
+import crypto from 'crypto';
+
+config();
+
+const PORT_ALICE = 7001;
+const PORT_BOB = 7002;
+
+// ============================================================================
+// Harness 判断引擎
+// ============================================================================
+
+interface HarnessResult {
+  called: boolean;
+  gate: number;
+  skills: string[];
+  result: string;
+}
+
+class JudgmentEngine {
+  private context: string[] = [];
+  private gateHistory: number[] = [];
+
+  decide(message: string): HarnessResult {
+    this.context.push(message);
+
+    // 根据消息内容和历史上下文判断
+    const history = this.context.join(' ').toLowerCase();
+    const current = message.toLowerCase();
+
+    // 上下文敏感的判断
+    if (current.includes('架构') || current.includes('设计') || current.includes('方案')) {
+      this.gateHistory.push(1);
+      return {
+        called: true,
+        gate: 1,
+        skills: ['arch', 'lead'],
+        result: '[Gate 1 - 架构设计] 建议采用三层架构：表现层、业务层、数据层分离'
+      };
+    }
+
+    if (current.includes('review') || current.includes('检查') || current.includes('审核')) {
+      this.gateHistory.push(2);
+      return {
+        called: true,
+        gate: 2,
+        skills: ['arch', 'guardian-fixer'],
+        result: '[Gate 2 - 代码审查] 建议添加输入验证和错误处理'
+      };
+    }
+
+    if (current.includes('安全') || current.includes('权限') || current.includes('认证')) {
+      this.gateHistory.push(4);
+      return {
+        called: true,
+        gate: 4,
+        skills: ['arch', 'guardian-fixer'],
+        result: '[Gate 4 - 安全检查] 建议实现 JWT token 过期和刷新机制'
+      };
+    }
+
+    if (current.includes('任务') || current.includes('分解') || current.includes('计划')) {
+      this.gateHistory.push(5);
+      return {
+        called: true,
+        gate: 5,
+        skills: ['task-arch', 'crystal-learn'],
+        result: '[Gate 5 - 任务分解] 已分解为 4 个子任务：1)登录 2)注册 3)权限 4)测试'
+      };
+    }
+
+    if (current.includes('实现') || current.includes('写代码') || current.includes('优化')) {
+      this.gateHistory.push(7);
+      return {
+        called: true,
+        gate: 7,
+        skills: ['harness-eng', 'crystal-learn'],
+        result: '[Gate 7 - 代码实现] 建议使用策略模式重构业务逻辑'
+      };
+    }
+
+    if (current.includes('测试') || current.includes('验证')) {
+      this.gateHistory.push(8);
+      return {
+        called: true,
+        gate: 8,
+        skills: ['harness-eng', 'harness-eng-test'],
+        result: '[Gate 8 - 测试验证] 测试覆盖率目标 80%，优先测试核心业务'
+      };
+    }
+
+    return { called: false, gate: 0, skills: [], result: '' };
+  }
+
+  getLastGate(): number {
+    return this.gateHistory.length > 0 ? this.gateHistory[this.gateHistory.length - 1] : 0;
+  }
+
+  clear(): void {
+    this.context = [];
+    this.gateHistory = [];
+  }
+}
+
+// ============================================================================
+// 智能体 (只响应，不主动发送)
+// ============================================================================
+
+interface DialogEntry {
+  id: string;
+  speaker: string;
+  message: string;
+  timestamp: number;
+  harnessCalled: boolean;
+  gate?: number;
+}
+
+class Agent {
+  id = crypto.randomUUID();
+  name: string;
+  port: number;
+  judgment = new JudgmentEngine();
+  history: DialogEntry[] = [];
+
+  constructor(name: string, port: number) {
+    this.name = name;
+    this.port = port;
+  }
+
+  async start(): Promise<void> {
+    const app = express();
+    app.use(express.json());
+
+    app.get('/health', (req, res) => {
+      res.json({ id: this.id, name: this.name, port: this.port, historyCount: this.history.length });
+    });
+
+    app.get('/discovery', (req, res) => {
+      res.json({ id: this.id, name: this.name, port: this.port });
+    });
+
+    app.get('/history', (req, res) => {
+      res.json(this.history);
+    });
+
+    // 接收消息并生成回复
+    app.post('/receive', async (req, res) => {
+      const { fromName, content } = req.body;
+
+      console.log(`\n📨 [${this.name}] 收到 [${fromName}]: "${content.substring(0, 50)}..."`);
+
+      // 判断是否需要 Harness
+      const decision = this.judgment.decide(content);
+
+      let response: string;
+      if (decision.called) {
+        console.log(`🧠 [${this.name}] 调用 Harness: Gate ${decision.gate}`);
+        console.log(`   Skills: ${decision.skills.join(', ')}`);
+        console.log(`   ${decision.result}`);
+
+        // 根据 Gate 生成回复
+        response = this.generateResponse(decision, fromName);
+      } else {
+        console.log(`🧠 [${this.name}] 普通回复 (无需 Harness)`);
+        response = this.generateNaturalResponse(content, fromName);
+      }
+
+      // 记录对话
+      this.history.push({
+        id: crypto.randomUUID(),
+        speaker: fromName,
+        message: content,
+        timestamp: Date.now(),
+        harnessCalled: decision.called,
+        gate: decision.gate || undefined
+      });
+
+      this.history.push({
+        id: crypto.randomUUID(),
+        speaker: this.name,
+        message: response,
+        timestamp: Date.now(),
+        harnessCalled: decision.called,
+        gate: decision.gate || undefined
+      });
+
+      console.log(`📤 [${this.name}] 回复: "${response.substring(0, 50)}..."`);
+
+      res.json({
+        ok: true,
+        response,
+        harnessCalled: decision.called,
+        gate: decision.gate
+      });
+    });
+
+    const server = createServer(app);
+    return new Promise(resolve => {
+      server.listen(this.port, () => {
+        console.log(`✓ [${this.name}] 监听 port ${this.port}`);
+        resolve();
+      });
+    });
+  }
+}
+
+// ============================================================================
+// 响应生成
+// ============================================================================
+
+const responses: Record<number, string[]> = {
+  1: [
+    '关于架构设计，Harness 分析建议采用分层架构，将表现层、业务逻辑和数据访问层分离。这样可以提高代码的可维护性和可测试性。',
+    'Harness Gate 1 分析完成。建议使用依赖注入来降低模块间的耦合，这样更利于单元测试。',
+  ],
+  2: [
+    '代码审查完成，Harness 建议添加输入验证和错误边界处理。特别是 API 接口，需要统一错误响应格式。',
+    'Gate 2 审查结果：建议为每个函数添加 JSDoc 注释，并使用 ESLint 统一代码风格。',
+  ],
+  4: [
+    '安全检查完成。建议实现 JWT token 的过期机制和刷新策略，同时添加 IP 白名单功能。',
+    'Harness Gate 4 建议：使用 HTTPS 加密传输，对敏感操作添加二次验证。',
+  ],
+  5: [
+    '任务已分解为 4 个子任务：\n1. 用户登录模块 (优先级：高)\n2. 用户注册模块 (优先级：高)\n3. 权限管理 (优先级：中)\n4. 集成测试 (优先级：中)\n建议按顺序执行。',
+    'Harness Gate 5 分解完成。每个子任务都有明确的完成标准和验收条件。',
+  ],
+  7: [
+    '代码实现中，Harness 建议使用策略模式重构条件判断逻辑，这样更容易扩展新的验证规则。',
+    'Gate 7 分析：建议在服务层添加缓存机制，提高响应速度。',
+  ],
+  8: [
+    '测试策略制定完成。目标覆盖率 80%，优先测试核心业务逻辑和边界情况。',
+    'Harness Gate 8：建议先写单元测试，再写集成测试，最后进行 E2E 测试。',
+  ],
+};
+
+function generateNaturalResponses(): string[] {
+  return [
+    '好的，明白。',
+    '我理解了，继续。',
+    '没问题，我们继续。',
+    '明白，让我看看...',
+    '收到，我会处理的。',
+    '好的，这个信息很有用。',
+    '明白了，还有其他需要讨论的吗？',
+  ];
+}
+
+class ResponseGenerator {
+  private responseIndex = 0;
+
+  generate(decision: HarnessResult, fromName: string): string {
+    if (!decision.called) {
+      const responses = generateNaturalResponses();
+      return responses[this.responseIndex++ % responses.length];
+    }
+
+    const gateResponses = responses[decision.gate] || ['已记录你的请求。'];
+    return gateResponses[this.responseIndex++ % gateResponses.length];
+  }
+}
+
+// ============================================================================
+// 测试场景
+// ============================================================================
+
+async function runDialogue() {
+  console.log('\n========================================');
+  console.log('  P2P Agent 多轮对话测试');
+  console.log('  场景: 完成一个用户认证系统的需求讨论');
+  console.log('========================================\n');
+
+  const alice = new Agent('Alice', PORT_ALICE);
+  const bob = new Agent('Bob', PORT_BOB);
+
+  await alice.start();
+  await bob.start();
+
+  // 对话脚本 (Alice 发送，Bob 接收并回复)
+  const script = [
+    'Bob，我们需要设计一个用户认证系统。',
+    '好的，具体需要哪些功能？',
+    '需要登录、注册、找回密码，还要支持社交登录。',
+    '明白了。这个系统需要什么级别的安全性？',
+    '需要处理金融数据，所以安全性很重要。帮我分析一下架构。',
+    '我来分析架构设计。',
+    '架构看起来不错。帮我 review 一下登录模块的实现。',
+    '好的，检查登录代码...',
+    'Review 完成，发现了一些命名规范问题。',
+    '好的，我们需要优化错误处理。',
+    '同意，现在分解任务吧。',
+    '我来把任务分解成可执行的小块。',
+    '任务分解得很好。我们开始实现第一个任务。',
+    '好的，开始实现登录模块。',
+    '实现完成，帮我验证一下代码质量。',
+    '运行测试中...',
+    '测试通过了！继续下一个任务。',
+    '好的，继续实现注册模块。',
+    '注册模块也完成了。',
+    '测试通过，项目进展顺利！',
+  ];
+
+  const responseGen = new ResponseGenerator();
+
+  console.log('━━━ 开始对话 ━━━\n');
+
+  for (let i = 0; i < script.length; i++) {
+    const isAliceSending = i % 2 === 0;
+    const speaker = isAliceSending ? 'Alice' : 'Bob';
+    const content = script[i];
+
+    // Alice 发送消息给 Bob
+    console.log(`[${speaker}] >>> ${content}`);
+
+    if (isAliceSending) {
+      // Alice -> Bob
+      try {
+        const resp = await fetch(`http://localhost:${PORT_BOB}/receive`, {
+          method: 'POST',
+          headers: { 'Content-Type': 'application/json' },
+          body: JSON.stringify({ fromName: 'Alice', content })
+        });
+
+        const result = await resp.json();
+
+        if (result.harnessCalled) {
+          console.log(`🧠 [Bob] Harness: Gate ${result.gate}`);
+        } else {
+          console.log(`🧠 [Bob] 普通回复`);
+        }
+        console.log(`📤 [Bob] >>> ${result.response?.substring(0, 60)}...`);
+      } catch (err) {
+        console.log(`❌ 发送失败:`, err);
+      }
+    } else {
+      // Bob -> Alice
+      try {
+        const resp = await fetch(`http://localhost:${PORT_ALICE}/receive`, {
+          method: 'POST',
+          headers: { 'Content-Type': 'application/json' },
+          body: JSON.stringify({ fromName: 'Bob', content })
+        });
+
+        const result = await resp.json();
+
+        if (result.harnessCalled) {
+          console.log(`🧠 [Alice] Harness: Gate ${result.gate}`);
+        } else {
+          console.log(`🧠 [Alice] 普通回复`);
+        }
+        console.log(`📤 [Alice] >>> ${result.response?.substring(0, 60)}...`);
+      } catch (err) {
+        console.log(`❌ 发送失败:`, err);
+      }
+    }
+
+    await new Promise(r => setTimeout(r, 400));
+  }
+
+  // 结果汇总
+  console.log('\n━━━ 结果汇总 ━━━\n');
+
+  const aliceHistory = alice.history;
+  const bobHistory = bob.history;
+
+  console.log(`[Alice] 对话记录: ${aliceHistory.length} 条`);
+  console.log(`[Bob] 对话记录: ${bobHistory.length} 条`);
+
+  const harnessCalls = [...aliceHistory, ...bobHistory].filter(e => e.harnessCalled);
+  console.log(`\nHarness 调用次数: ${harnessCalls.length}`);
+
+  // Gate 统计
+  const gateStats = new Map<number, number>();
+  for (const call of harnessCalls) {
+    if (call.gate) {
+      gateStats.set(call.gate, (gateStats.get(call.gate) || 0) + 1);
+    }
+  }
+
+  console.log('\nGate 统计:');
+  for (const [gate, count] of gateStats) {
+    console.log(`  Gate ${gate}: ${count} 次`);
+  }
+
+  // 对话流程
+  console.log('\n━━━ 对话流程 ━━━\n');
+  for (let i = 0; i < aliceHistory.length; i++) {
+    const entry = aliceHistory[i];
+    const icon = entry.harnessCalled ? '🧠' : '  ';
+    const gateInfo = entry.harnessCalled ? ` [Gate ${entry.gate}]` : '';
+    console.log(`${icon} [${entry.speaker}]${gateInfo}: ${entry.message.substring(0, 50)}...`);
+  }
+
+  // 健康检查
+  console.log('\n━━━ 最终状态 ━━━\n');
+  const aliceHealth = await fetch(`http://localhost:${PORT_ALICE}/health`).then(r => r.json());
+  const bobHealth = await fetch(`http://localhost:${PORT_BOB}/health`).then(r => r.json());
+
+  console.log(`[Alice] ID: ${aliceHealth.id.substring(0, 16)}...`);
+  console.log(`[Bob]   ID: ${bobHealth.id.substring(0, 16)}...`);
+  console.log(`\n对话总数: ${aliceHistory.length + bobHistory.length}`);
+  console.log(`Harness 调用: ${harnessCalls.length} 次`);
+
+  const success = harnessCalls.length >= 5;
+  console.log(`\n${success ? '✅' : '⚠️'} ${success ? '测试成功' : '测试部分成功'}`);
+
+  console.log('\n========================================\n');
+}
+
+runDialogue().catch(err => {
+  console.error('测试失败:', err);
+  process.exit(1);
+});