From dcf1687b507e726ae16eb3eed7773692375bb976 Mon Sep 17 00:00:00 2001
From: dal <dallin@buster.so>
Date: Fri, 22 Aug 2025 16:50:55 -0600
Subject: [PATCH] Refactor sequential thinking tool to remove raw LLM message
 entries

- Removed the creation and handling of raw LLM message entries in both start and finish functions.
- Updated message entry updates to only include reasoning messages, simplifying the logic and improving clarity.
---
 .../sequential-thinking-tool-finish.ts            | 15 ++-------------
 .../sequential-thinking-tool-start.ts             | 15 ++-------------
 2 files changed, 4 insertions(+), 26 deletions(-)

diff --git a/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-finish.ts b/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-finish.ts
index 3052ff9c0..eeb43935b 100644
--- a/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-finish.ts
+++ b/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-finish.ts
@@ -1,10 +1,7 @@
 import { updateMessageEntries } from '@buster/database';
 import type { ToolCallOptions } from 'ai';
 import { normalizeEscapedText } from '../../../utils/streaming/escape-normalizer';
-import {
-  createSequentialThinkingRawLlmMessageEntry,
-  createSequentialThinkingReasoningMessage,
-} from './helpers/sequential-thinking-tool-transform-helper';
+import { createSequentialThinkingReasoningMessage } from './helpers/sequential-thinking-tool-transform-helper';
 import type {
   SequentialThinkingContext,
   SequentialThinkingInput,
@@ -31,22 +28,14 @@ export function createSequentialThinkingFinish(
       'completed' // Mark as completed when finish is called
     );
 
-    // Create final raw LLM message entry
-    const rawLlmMessage = createSequentialThinkingRawLlmMessageEntry(
-      sequentialThinkingState,
-      options.toolCallId
-    );
-
     try {
       if (context.messageId) {
         const reasoningMessages = reasoningEntry ? [reasoningEntry] : [];
-        const rawLlmMessages = rawLlmMessage ? [rawLlmMessage] : [];
 
-        if (reasoningMessages.length > 0 || rawLlmMessages.length > 0) {
+        if (reasoningMessages.length > 0) {
           await updateMessageEntries({
             messageId: context.messageId,
             reasoningMessages,
-            rawLlmMessages,
           });
 
           console.info('[sequential-thinking] Completed sequential thinking:', {
diff --git a/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-start.ts b/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-start.ts
index ff28644ba..90570aca6 100644
--- a/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-start.ts
+++ b/packages/ai/src/tools/planning-thinking-tools/sequential-thinking-tool/sequential-thinking-tool-start.ts
@@ -1,9 +1,6 @@
 import { updateMessageEntries } from '@buster/database';
 import type { ToolCallOptions } from 'ai';
-import {
-  createSequentialThinkingRawLlmMessageEntry,
-  createSequentialThinkingReasoningMessage,
-} from './helpers/sequential-thinking-tool-transform-helper';
+import { createSequentialThinkingReasoningMessage } from './helpers/sequential-thinking-tool-transform-helper';
 import type {
   SequentialThinkingContext,
   SequentialThinkingState,
@@ -29,22 +26,14 @@ export function createSequentialThinkingStart(
       options.toolCallId
     );
 
-    // Create initial raw LLM message entry
-    const rawLlmMessage = createSequentialThinkingRawLlmMessageEntry(
-      sequentialThinkingState,
-      options.toolCallId
-    );
-
     try {
       if (context.messageId) {
         const reasoningMessages = reasoningEntry ? [reasoningEntry] : [];
-        const rawLlmMessages = rawLlmMessage ? [rawLlmMessage] : [];
 
-        if (reasoningMessages.length > 0 || rawLlmMessages.length > 0) {
+        if (reasoningMessages.length > 0) {
           await updateMessageEntries({
             messageId: context.messageId,
             reasoningMessages,
-            rawLlmMessages,
           });
 
           console.info('[sequential-thinking] Started sequential thinking:', {