From 0b8f8665d6810f3b35b057dc4a6e5a888ce9980f Mon Sep 17 00:00:00 2001 From: dal Date: Tue, 12 Aug 2025 16:13:41 -0600 Subject: [PATCH] Refactor agent steps and message roles - Removed unused `steps` array from `createThinkAndPrepAgent` function to simplify the code. - Updated message role from 'assistant' to 'user' in `create-todos-step` for better context alignment. - Adjusted `valuesMessage` role in `extract-values-search-step` to 'user' when extracted values are present, enhancing message clarity. These changes improve code clarity and ensure consistent message roles across agent steps. --- .../think-and-prep-agent/think-and-prep-agent.ts | 6 ------ .../create-todos-step/create-todos-step.ts | 2 +- .../extract-values-step/extract-values-search-step.ts | 11 +++++++---- 3 files changed, 8 insertions(+), 11 deletions(-) diff --git a/packages/ai/src/agents/think-and-prep-agent/think-and-prep-agent.ts b/packages/ai/src/agents/think-and-prep-agent/think-and-prep-agent.ts index 43fdd6ac8..0009dbff3 100644 --- a/packages/ai/src/agents/think-and-prep-agent/think-and-prep-agent.ts +++ b/packages/ai/src/agents/think-and-prep-agent/think-and-prep-agent.ts @@ -53,7 +53,6 @@ export type ThinkAndPrepAgentOptions = z.infer; export function createThinkAndPrepAgent(thinkAndPrepAgentSchema: ThinkAndPrepAgentOptions) { - const steps: never[] = []; const { messageId } = thinkAndPrepAgentSchema; const systemMessage = { @@ -147,12 +146,7 @@ export function createThinkAndPrepAgent(thinkAndPrepAgentSchema: ThinkAndPrepAge throw new Error('Max retry attempts exceeded'); } - async function getSteps() { - return steps; - } - return { stream, - getSteps, }; } diff --git a/packages/ai/src/steps/analyst-agent-steps/create-todos-step/create-todos-step.ts b/packages/ai/src/steps/analyst-agent-steps/create-todos-step/create-todos-step.ts index 3e8741187..65fc8d11a 100644 --- a/packages/ai/src/steps/analyst-agent-steps/create-todos-step/create-todos-step.ts +++ b/packages/ai/src/steps/analyst-agent-steps/create-todos-step/create-todos-step.ts @@ -74,7 +74,7 @@ export async function runCreateTodosStep(params: CreateTodosParams): Promise 0 + ? { + role: 'user', + content: storedValuesResult.searchResults, + } + : undefined, }; } catch (error) { // Handle AbortError gracefully