chore: retain the source code from the underlying tools (#756)

src/loop/loop.ts

@@ -47,7 +47,7 @@ export interface LLMDelegate {
   checkDoneToolCall(toolCall: LLMToolCall): string | null;
 }
 
-export async function runTask(delegate: LLMDelegate, client: Client, task: string, oneShot: boolean = false): Promise<string> {
+export async function runTask(delegate: LLMDelegate, client: Client, task: string, oneShot: boolean = false): Promise<LLMMessage[]> {
   const { tools } = await client.listTools();
   const taskContent = oneShot ? `Perform following task: ${task}.` : `Perform following task: ${task}. Once the task is complete, call the "done" tool.`;
   const conversation = delegate.createConversation(taskContent, tools, oneShot);
@@ -60,10 +60,9 @@ export async function runTask(delegate: LLMDelegate, client: Client, task: strin
 
     const toolResults: Array<{ toolCallId: string; content: string; isError?: boolean }> = [];
     for (const toolCall of toolCalls) {
-      // Check if this is the "done" tool
       const doneResult = delegate.checkDoneToolCall(toolCall);
       if (doneResult !== null)
-        return doneResult;
+        return conversation.messages;
 
       const { name, arguments: args, id } = toolCall;
       try {
@@ -100,10 +99,9 @@ export async function runTask(delegate: LLMDelegate, client: Client, task: strin
       }
     }
 
+    delegate.addToolResults(conversation, toolResults);
     if (oneShot)
-      return toolResults.map(result => result.content).join('\n');
-    else
-      delegate.addToolResults(conversation, toolResults);
+      return conversation.messages;
   }
 
   throw new Error('Failed to perform step, max attempts reached');

src/loop/loopClaude.ts

@@ -44,10 +44,7 @@ export class ClaudeDelegate implements LLMDelegate {
         description: 'Call this tool when the task is complete.',
         inputSchema: {
           type: 'object',
-          properties: {
-            result: { type: 'string', description: 'The result of the task.' },
-          },
-          required: ['result'],
+          properties: {},
         },
       });
     }

src/loop/loopOpenAI.ts

@@ -44,10 +44,7 @@ export class OpenAIDelegate implements LLMDelegate {
         description: 'Call this tool when the task is complete.',
         inputSchema: {
           type: 'object',
-          properties: {
-            result: { type: 'string', description: 'The result of the task.' },
-          },
-          required: ['result'],
+          properties: {},
         },
       });
     }

src/loop/main.ts

@@ -49,10 +49,11 @@ async function run(delegate: LLMDelegate) {
   await client.connect(transport);
   await client.ping();
 
-  let lastResult: string | undefined;
-  for (const task of tasks)
-    lastResult = await runTask(delegate, client, task);
-  console.log(lastResult);
+  for (const task of tasks) {
+    const messages = await runTask(delegate, client, task);
+    for (const message of messages)
+      console.log(`${message.role}: ${message.content}`);
+  }
   await client.close();
 }