From 35e0fc27d7c4b3a9a5f2edb1a26de978def6672f Mon Sep 17 00:00:00 2001
From: dal <dallin@buster.so>
Date: Mon, 18 Aug 2025 10:43:15 -0600
Subject: [PATCH] done tool fix

---
 .../communication-tools/done-tool/done-tool-constants.ts    | 1 +
 .../tools/communication-tools/done-tool/done-tool-delta.ts  | 5 ++++-
 .../done-tool/helpers/done-tool-transform-helper.ts         | 6 +-----
 3 files changed, 6 insertions(+), 6 deletions(-)
 create mode 100644 packages/ai/src/tools/communication-tools/done-tool/done-tool-constants.ts

diff --git a/packages/ai/src/tools/communication-tools/done-tool/done-tool-constants.ts b/packages/ai/src/tools/communication-tools/done-tool/done-tool-constants.ts
new file mode 100644
index 000000000..0519ecba6
--- /dev/null
+++ b/packages/ai/src/tools/communication-tools/done-tool/done-tool-constants.ts
@@ -0,0 +1 @@
+ 
\ No newline at end of file
diff --git a/packages/ai/src/tools/communication-tools/done-tool/done-tool-delta.ts b/packages/ai/src/tools/communication-tools/done-tool/done-tool-delta.ts
index 69dd2a111..0b5710d3b 100644
--- a/packages/ai/src/tools/communication-tools/done-tool/done-tool-delta.ts
+++ b/packages/ai/src/tools/communication-tools/done-tool/done-tool-delta.ts
@@ -39,7 +39,10 @@ export function createDoneToolDelta(context: DoneToolContext, doneToolState: Don
         doneToolState,
         options.toolCallId
       );
-      const doneToolMessage = createDoneToolRawLlmMessageEntry(doneToolState, options.toolCallId);
+      const doneToolMessage = createDoneToolRawLlmMessageEntry(
+        doneToolState,
+        options.toolCallId || ''
+      );
 
       const entries: UpdateMessageEntriesParams = {
         messageId: context.messageId,
diff --git a/packages/ai/src/tools/communication-tools/done-tool/helpers/done-tool-transform-helper.ts b/packages/ai/src/tools/communication-tools/done-tool/helpers/done-tool-transform-helper.ts
index 9f82b394c..6e28193a1 100644
--- a/packages/ai/src/tools/communication-tools/done-tool/helpers/done-tool-transform-helper.ts
+++ b/packages/ai/src/tools/communication-tools/done-tool/helpers/done-tool-transform-helper.ts
@@ -38,12 +38,8 @@ export function createDoneToolRawLlmMessageEntry(
         type: 'tool-call',
         toolCallId: id,
         toolName: 'doneTool',
-        input: {},
+        input: { finalResponse: doneToolState.finalResponse || '' },
       },
-      // Optionally include any accumulated text content
-      ...(doneToolState.finalResponse
-        ? [{ type: 'text' as const, text: doneToolState.finalResponse }]
-        : []),
     ],
   };
 }