Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

sequential thinking

This commit is contained in:
dal 2025-08-15 15:03:58 -06:00
parent 37aa14c43b
commit 2059988494
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/ai/src/tools/database-tools/execute-sql/execute-sql-execute.ts
View File

@ -6,6 +6,7 @@ import {
createPermissionErrorMessage, createPermissionErrorMessage,
validateSqlPermissions, validateSqlPermissions,
} from '../../../utils/sql-permissions'; } from '../../../utils/sql-permissions';
import { createRawToolResultEntry } from '../../shared/create-raw-llm-tool-result-entry';
import { import {
EXECUTE_SQL_TOOL_NAME, EXECUTE_SQL_TOOL_NAME,
type ExecuteSqlContext, type ExecuteSqlContext,
@ -327,6 +328,13 @@ export function createExecuteSqlExecute(state: ExecuteSqlState, context: Execute
// Create final reasoning entry with complete status // Create final reasoning entry with complete status
const reasoningEntry = createExecuteSqlReasoningEntry(state, state.toolCallId || ''); const reasoningEntry = createExecuteSqlReasoningEntry(state, state.toolCallId || '');
const rawLlmMessage = createExecuteSqlRawLlmMessageEntry(state, state.toolCallId || ''); const rawLlmMessage = createExecuteSqlRawLlmMessageEntry(state, state.toolCallId || '');
const rawLlmResultEntry = createRawToolResultEntry(
state.toolCallId || '',
EXECUTE_SQL_TOOL_NAME,
{
results,
}
);
// Update database with final status // Update database with final status
const messagesToSave: Parameters<typeof updateMessageEntries>[0] = { const messagesToSave: Parameters<typeof updateMessageEntries>[0] = {
@ -338,7 +346,7 @@ export function createExecuteSqlExecute(state: ExecuteSqlState, context: Execute
} }
if (rawLlmMessage) { if (rawLlmMessage) {
messagesToSave.rawLlmMessages = [rawLlmMessage]; messagesToSave.rawLlmMessages = [rawLlmMessage, rawLlmResultEntry];
} }
if (messagesToSave.reasoningMessages || messagesToSave.rawLlmMessages) { if (messagesToSave.reasoningMessages || messagesToSave.rawLlmMessages) {