thatgfsj-code 0.4.1 → 0.6.0

package/src/app/agent.ts

@@ -1,140 +0,0 @@
-/**
- * Agent Loop - Core agent logic with streaming and tool execution
- * Extracted from old src/core/ai-engine.ts
- *
- * The agent loop: while(true) { stream response → check tool calls → execute → repeat }
- */
-
-import chalk from 'chalk';
-import type { ChatMessage, ChatResponse } from '../types.js';
-import type { Tool } from '../tools/types.js';
-import type { HookManager } from '../hooks/index.js';
-import { LLMService } from '../llm/index.js';
-import { ToolRegistry } from '../tools/index.js';
-
-export interface AgentConfig {
-  maxIterations?: number;
-  hooks?: HookManager;
-}
-
-export class Agent {
-  private llm: LLMService;
-  private registry: ToolRegistry;
-  private hooks?: HookManager;
-  private maxIterations: number;
-
-  constructor(llm: LLMService, registry: ToolRegistry, config: AgentConfig = {}) {
-    this.llm = llm;
-    this.registry = registry;
-    this.hooks = config.hooks;
-    this.maxIterations = config.maxIterations ?? 10;
-
-    // Register tools with LLM service
-    this.llm.registerTools(registry.list());
-  }
-
-  /**
-   * Run agent loop with streaming output
-   * Returns the final response
-   */
-  async *run(messages: ChatMessage[]): AsyncGenerator<string, ChatResponse> {
-    let currentMessages = [...messages];
-    let iterations = 0;
-
-    while (iterations < this.maxIterations) {
-      iterations++;
-
-      // Hook: beforeAgentLoop
-      if (this.hooks) {
-        await this.hooks.emit('beforeAgentLoop', { messages: currentMessages, iteration: iterations });
-      }
-
-      // Stream the response
-      let fullContent = '';
-
-      for await (const chunk of this.llm.chatStream(currentMessages, { maxIterations: 1 })) {
-        fullContent += chunk;
-        yield chunk;
-      }
-
-      // Get the full response to check for tool calls
-      const response: ChatResponse = {
-        content: fullContent,
-        role: 'assistant',
-      };
-
-      // Check if the response contains tool calls
-      // For now, we rely on the LLM service to handle tool calls internally
-      // If the LLM service returned tool_calls, we need to process them
-      if (response.tool_calls && response.tool_calls.length > 0) {
-        currentMessages.push({
-          role: 'assistant',
-          content: response.content,
-          tool_calls: response.tool_calls,
-        });
-
-        for (const toolCall of response.tool_calls) {
-          // Hook: beforeToolCall
-          if (this.hooks) {
-            await this.hooks.emit('beforeToolCall', {
-              toolName: toolCall.function.name,
-              toolParams: JSON.parse(toolCall.function.arguments || '{}'),
-              toolCallId: toolCall.id,
-            });
-          }
-
-          const result = await this.registry.execute(
-            toolCall.function.name,
-            JSON.parse(toolCall.function.arguments || '{}')
-          );
-
-          // Hook: afterToolCall
-          if (this.hooks) {
-            await this.hooks.emit('afterToolCall', {
-              toolName: toolCall.function.name,
-              toolParams: JSON.parse(toolCall.function.arguments || '{}'),
-              toolResult: result,
-              toolCallId: toolCall.id,
-            });
-          }
-
-          currentMessages.push({
-            role: 'tool',
-            content: result.success ? (result.output || JSON.stringify(result.data)) : (result.error || 'Tool failed'),
-            tool_call_id: toolCall.id,
-            name: toolCall.function.name,
-          });
-
-          yield `\n${chalk.gray(`[tool: ${toolCall.function.name}]`)}`;
-        }
-
-        continue; // Next iteration
-      }
-
-      // No tool calls - done
-      if (response.content) {
-        currentMessages.push({ role: 'assistant', content: response.content });
-      }
-
-      // Hook: afterAgentLoop
-      if (this.hooks) {
-        await this.hooks.emit('afterAgentLoop', { messages: currentMessages, iteration: iterations });
-      }
-
-      return response;
-    }
-
-    return { content: '[Agent loop exceeded maximum iterations]', role: 'assistant' };
-  }
-
-  /**
-   * Non-streaming chat (convenience method)
-   */
-  async chat(messages: ChatMessage[]): Promise<ChatResponse> {
-    let fullContent = '';
-    for await (const chunk of this.run(messages)) {
-      fullContent += chunk;
-    }
-    return { content: fullContent, role: 'assistant' };
-  }
-}