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

Merge pull request #451 from buster-so/dallin/bus-1339-make-sure-only-major-assumption-make-it-into-summary 

Refactor follow-up and initial message formatting to handle major assumptions more effectively. Added checks for major assumptions and updated message structure for clarity in output.
This commit is contained in:
dal 2025-07-09 08:18:17 -07:00 committed by GitHub
parent 250e9c914a cba5e99223
commit 38243a133d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 31 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
packages/ai/src/steps/post-processing/format-follow-up-message-step.ts
View File

@ -90,24 +90,38 @@ export const formatFollowUpMessageStepExecution = async ({
inputData: z.infer<typeof inputSchema>;
}): Promise<z.infer<typeof formatFollowUpMessageOutputSchema>> => {
try {
// Check if there are any major assumptions
const majorAssumptions = inputData.assumptions?.filter((a) => a.label === 'major') || [];
// If no major assumptions, return null for formatted_message
if (majorAssumptions.length === 0) {
return {
...inputData,
};
}
// Prepare context about issues and assumptions for the agent
const issuesAndAssumptions = {
flagged_issues: inputData.flagChatMessage || 'No issues flagged',
assumptions: inputData.assumptions || [],
major_assumptions: majorAssumptions,
};
const contextMessage = `New issues and assumptions identified from the latest chat messages:
User: ${inputData.userName}
Issues Flagged: ${issuesAndAssumptions.flagged_issues}
Assumptions Identified: ${
issuesAndAssumptions.assumptions.length > 0
? issuesAndAssumptions.assumptions
Issues Flagged:
${issuesAndAssumptions.flagged_issues}
Major Assumptions Identified:
${
issuesAndAssumptions.major_assumptions.length > 0
? issuesAndAssumptions.major_assumptions
.map((a) => `- ${a.descriptiveTitle}: ${a.explanation}`)
.join('\n')
: 'No new assumptions identified'
.join('\n\n')
: 'No major assumptions identified'
}
Generate a concise update message for the data team.`;

18
packages/ai/src/steps/post-processing/format-initial-message-step.ts
View File

@ -152,15 +152,17 @@ export const formatInitialMessageStepExecution = async ({
User: ${inputData.userName}
Issues Flagged: ${issuesAndAssumptions.flagged_issues}
Issues Flagged:
${issuesAndAssumptions.flagged_issues}
Major Assumptions Identified: ${
issuesAndAssumptions.major_assumptions.length > 0
? issuesAndAssumptions.major_assumptions
.map((a) => `- ${a.descriptiveTitle}: ${a.explanation}`)
.join('\n')
: 'No major assumptions identified'
}
Major Assumptions Identified:
${
issuesAndAssumptions.major_assumptions.length > 0
? issuesAndAssumptions.major_assumptions
.map((a) => `- ${a.descriptiveTitle}: ${a.explanation}`)
.join('\n\n')
: 'No major assumptions identified'
}
Generate a cohesive summary with title for the data team.`;