Refactor: Move config and types to separate files, enhance tool call handling.

src/index.ts

@@ -1,4 +1,5 @@
 import {
+  AIMessage,
   HumanMessage,
   SystemMessage,
   ToolMessage,
@@ -27,7 +28,6 @@ class MockMcpToolsService {
       description: 'Возвращает введенную строку.',
       schema: z.object({ input: z.string().describe('Строка для эхо.') }),
       func: async (input: { input: string }) => {
-        console.log(`Вызов инструмента 'echo' с входом: ${input.input}`);
         return input.input;
       },
     });
@@ -37,7 +37,6 @@ class MockMcpToolsService {
       description: 'Возвращает текущую дату в формате ISO.',
       schema: z.object({}),
       func: async () => {
-        console.log('Вызов инструмента "get_current_date"');
         return new Date().toISOString();
       },
     });
@@ -48,13 +47,21 @@ class MockMcpToolsService {
 
 class PipelineExecutor {
   private readonly mcpToolsService: MockMcpToolsService;
+  private cachedTools: DynamicStructuredTool[] | null = null;
 
   constructor() {
     this.mcpToolsService = new MockMcpToolsService();
   }
 
+  private async _getOrLoadTools(): Promise<DynamicStructuredTool[]> {
+    if (!this.cachedTools) {
+      this.cachedTools = await this.mcpToolsService.getTools();
+    }
+    return this.cachedTools;
+  }
+
   private async initChain(openAIApiKey: string): Promise<Runnable> {
-    const tools = await this.mcpToolsService.getTools();
+    const tools = await this._getOrLoadTools();
     const llm = new ChatOpenAI({
       model: 'gpt-4o',
       temperature: 0.2,
@@ -68,10 +75,8 @@ class PipelineExecutor {
     variables: Record<string, string> = {},
   ): (HumanMessage | SystemMessage)[] {
     return [
-      ...systemPrompts.map((prompt) =>
+      ...systemPrompts.map(
-        Object.keys(variables).length
+        (prompt) => new SystemMessage(Handlebars.compile(prompt)(variables)),
-          ? new SystemMessage(Handlebars.compile(prompt)(variables))
-          : new SystemMessage(prompt),
       ),
       new SystemMessage(`Текущая дата и время: ${new Date().toISOString()}`),
     ];
@@ -81,51 +86,71 @@ class PipelineExecutor {
     prompt: string,
     variables: Record<string, string> = {},
   ): string {
-    return Object.keys(variables).length
+    return Handlebars.compile(prompt)(variables);
-      ? Handlebars.compile(prompt)(variables)
-      : prompt;
   }
 
-  async executeChain(params: {
+  async executeChain(options: {
     pipeline: Pipeline;
     variables?: Record<string, string>;
     systemPrompts?: string[];
     openAIApiKey: string;
   }): Promise<string[]> {
-    const variables = params.variables ?? {};
+    const variables = options.variables ?? {};
-    const systemPrompts = params.systemPrompts ?? [];
+    const systemPrompts = options.systemPrompts ?? [];
-    const chain = await this.initChain(params.openAIApiKey);
+    const chain = await this.initChain(options.openAIApiKey);
     const initialMessages = this.buildMessages(
-      [params.pipeline.systemPrompt, ...systemPrompts],
+      [options.pipeline.systemPrompt, ...systemPrompts],
       variables,
     );
     const results: string[] = [];
 
-    for (const { prompt } of params.pipeline.steps) {
+    for (const { prompt } of options.pipeline.steps) {
       const renderedPrompt = this.renderPrompt(prompt, variables);
-      const currentMessages = [
+      const currentMessages: (
-        ...initialMessages,
+        | HumanMessage
-        new HumanMessage(renderedPrompt),
+        | SystemMessage
-      ];
+        | AIMessage
-      let response = await chain.invoke(currentMessages);
+        | ToolMessage
+      )[] = [...initialMessages, new HumanMessage(renderedPrompt)];
+
+      const finalResponse = await this._resolveToolCalls(
+        chain,
+        currentMessages,
+      );
+      results.push(finalResponse?.content.toString() ?? '<empty>');
+    }
+    return results;
+  }
+
+  private async _resolveToolCalls(
+    chain: Runnable,
+    messages: (HumanMessage | SystemMessage | AIMessage | ToolMessage)[],
+  ): Promise<AIMessage> {
+    let response: AIMessage = await chain.invoke(messages);
 
     while (response.tool_calls && response.tool_calls.length > 0) {
-        const toolCalls = response.tool_calls;
+      messages.push(response);
-        currentMessages.push(response);
 
-        for (const toolCall of toolCalls) {
+      for (const toolCall of response.tool_calls) {
         try {
-            const tools = await this.mcpToolsService.getTools();
+          const tools = await this._getOrLoadTools();
           const tool = tools.find((t) => t.name === toolCall.name);
 
           if (!tool) {
-              throw new Error(`Tool ${toolCall.name} not found`);
+            console.error(`Инструмент ${toolCall.name} не найден.`);
+            messages.push(
+              new ToolMessage({
+                tool_call_id: toolCall.id!,
+                content: `Ошибка: Инструмент ${toolCall.name} не найден.`,
+              }),
+            );
+            continue;
           }
 
           const toolResult = await tool.func(toolCall.args);
-            currentMessages.push(
+          messages.push(
             new ToolMessage({
-                tool_call_id: toolCall.id,
+              tool_call_id: toolCall.id!,
               content: JSON.stringify(toolResult),
             }),
           );
@@ -134,19 +159,17 @@ class PipelineExecutor {
             `Ошибка выполнения инструмента ${toolCall.name}:`,
             error,
           );
-            currentMessages.push(
+          messages.push(
             new ToolMessage({
-                tool_call_id: toolCall.id,
+              tool_call_id: toolCall.id!,
-                content: `Error: ${error.message}`,
+              content: `Ошибка: ${error.message}`,
             }),
           );
         }
       }
-        response = await chain.invoke(currentMessages);
+      response = await chain.invoke(messages);
     }
-      results.push(response?.content ?? '<empty>');
+    return response;
-    }
-    return results;
   }
 }
 
@@ -181,7 +204,7 @@ async function main() {
         name: 'Получение даты',
         prompt: 'Какая сейчас дата? Используй инструмент `get_current_date`.',
       },
-      { name: 'Завершение', prompt: 'Напиши резюме, что ты сделал' },
+      { name: 'Завершение', prompt: 'Напиши резюме нашего с тобой диалога' },
     ],
   };