diff --git a/packages/ai/src/agents/analyst-agent/analyst-agent.ts b/packages/ai/src/agents/analyst-agent/analyst-agent.ts index f900fc496..dc4780797 100644 --- a/packages/ai/src/agents/analyst-agent/analyst-agent.ts +++ b/packages/ai/src/agents/analyst-agent/analyst-agent.ts @@ -127,10 +127,10 @@ export function createAnalystAgent(analystAgentOptions: AnalystAgentOptions) { const docsSystemMessage = docsContent ? ({ - role: 'system', - content: `\n${docsContent}\n`, - providerOptions: DEFAULT_ANTHROPIC_OPTIONS, - } as ModelMessage) + role: 'system', + content: `\n${docsContent}\n`, + providerOptions: DEFAULT_ANTHROPIC_OPTIONS, + } as ModelMessage) : null; async function stream({ messages }: AnalystStreamOptions) { @@ -187,19 +187,19 @@ export function createAnalystAgent(analystAgentOptions: AnalystAgentOptions) { // Create analyst instructions system message with proper escaping const analystInstructionsMessage = analystInstructions ? ({ - role: 'system', - content: `\n${analystInstructions}\n`, - providerOptions: DEFAULT_ANTHROPIC_OPTIONS, - } as ModelMessage) + role: 'system', + content: `\n${analystInstructions}\n`, + providerOptions: DEFAULT_ANTHROPIC_OPTIONS, + } as ModelMessage) : null; // Create user personalization system message const userPersonalizationSystemMessage = userPersonalizationMessageContent ? ({ - role: 'system', - content: userPersonalizationMessageContent, - providerOptions: DEFAULT_ANTHROPIC_OPTIONS, - } as ModelMessage) + role: 'system', + content: userPersonalizationMessageContent, + providerOptions: DEFAULT_ANTHROPIC_OPTIONS, + } as ModelMessage) : null; return wrapTraced( diff --git a/packages/ai/src/agents/analyst-agent/get-analyst-agent-system-prompt.test.ts b/packages/ai/src/agents/analyst-agent/get-analyst-agent-system-prompt.test.ts index 71b8947b5..66647e799 100644 --- a/packages/ai/src/agents/analyst-agent/get-analyst-agent-system-prompt.test.ts +++ b/packages/ai/src/agents/analyst-agent/get-analyst-agent-system-prompt.test.ts @@ -63,7 +63,7 @@ describe('Analyst Agent Instructions', () => { expect(result).toContain(''); expect(result).toContain(''); expect(result).toContain(''); - expect(result).toContain('You are a Buster'); + expect(result).toContain('You are an agent'); }); it('should throw an error for empty SQL dialect guidance', () => { diff --git a/packages/ai/src/workflows/analyst-agent-workflow/analyst-workflow.ts b/packages/ai/src/workflows/analyst-agent-workflow/analyst-workflow.ts index 60b1bcdc9..54f7e3885 100644 --- a/packages/ai/src/workflows/analyst-agent-workflow/analyst-workflow.ts +++ b/packages/ai/src/workflows/analyst-agent-workflow/analyst-workflow.ts @@ -123,27 +123,27 @@ export async function runAnalystWorkflow( // earlyTermination: thinkAndPrepAgentStepResults.earlyTermination, // }); - analystAgentStepResults = await runAnalystAgentStep({ - options: { - messageId: input.messageId, - chatId: input.chatId, - organizationId: input.organizationId, - dataSourceId: input.dataSourceId, - dataSourceSyntax: input.dataSourceSyntax, - userId: input.userId, - datasets: input.datasets, - workflowStartTime, - analysisMode, - analystInstructions, - organizationDocs, - userPersonalizationMessageContent, - }, - streamOptions: { - messages, - }, - }); + analystAgentStepResults = await runAnalystAgentStep({ + options: { + messageId: input.messageId, + chatId: input.chatId, + organizationId: input.organizationId, + dataSourceId: input.dataSourceId, + dataSourceSyntax: input.dataSourceSyntax, + userId: input.userId, + datasets: input.datasets, + workflowStartTime, + analysisMode, + analystInstructions, + organizationDocs, + userPersonalizationMessageContent, + }, + streamOptions: { + messages, + }, + }); - messages.push(...analystAgentStepResults.messages); + messages.push(...analystAgentStepResults.messages); // } else { // console.info('[runAnalystWorkflow] DEBUG: SKIPPING analyst agent due to early termination', { // workflowId,