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

test fix

This commit is contained in:
dal 2025-10-01 20:45:03 -06:00
parent 5000e8aca3
commit fe7af072cf
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
3 changed files with 33 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
packages/ai/src/agents/analyst-agent/analyst-agent.ts
View File

@ -127,10 +127,10 @@ export function createAnalystAgent(analystAgentOptions: AnalystAgentOptions) {
const docsSystemMessage = docsContent
? ({
role: 'system',
content: `<data_catalog_docs>\n${docsContent}\n</data_catalog_docs>`,
providerOptions: DEFAULT_ANTHROPIC_OPTIONS,
} as ModelMessage)
role: 'system',
content: `<data_catalog_docs>\n${docsContent}\n</data_catalog_docs>`,
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: `<organization_instructions>\n${analystInstructions}\n</organization_instructions>`,
providerOptions: DEFAULT_ANTHROPIC_OPTIONS,
} as ModelMessage)
role: 'system',
content: `<organization_instructions>\n${analystInstructions}\n</organization_instructions>`,
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(

2
packages/ai/src/agents/analyst-agent/get-analyst-agent-system-prompt.test.ts
View File

@ -63,7 +63,7 @@ describe('Analyst Agent Instructions', () => {
expect(result).toContain('<intro>');
expect(result).toContain('<sql_best_practices>');
expect(result).toContain('<visualization_and_charting_guidelines>');
expect(result).toContain('You are a Buster');
expect(result).toContain('You are an agent');
});
it('should throw an error for empty SQL dialect guidance', () => {

40
packages/ai/src/workflows/analyst-agent-workflow/analyst-workflow.ts
View File

@ -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,