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

flush on braintrust while in async mode still

This commit is contained in:
dal 2025-07-08 12:24:39 -06:00
parent f4ca39a3f7
commit e8848f1d90
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
2 changed files with 12 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
apps/trigger/src/tasks/analyst-agent-task/analyst-agent-task.ts
View File

@ -290,26 +290,10 @@ export const analystAgentTask: ReturnType<
throw new Error('BRAINTRUST_KEY is not set');
}
// Start Braintrust initialization immediately but don't block the critical path
const braintrustInitStart = Date.now();
const braintrustInitPromise = Promise.resolve().then(async () => {
try {
initLogger({
apiKey: process.env.BRAINTRUST_KEY,
projectName: process.env.ENVIRONMENT || 'development',
});
logger.log('Braintrust initialization completed', {
messageId: payload.message_id,
braintrustInitTimeMs: Date.now() - braintrustInitStart,
});
} catch (error) {
logger.error('Braintrust initialization failed', {
messageId: payload.message_id,
error: error instanceof Error ? error.message : 'Unknown error',
braintrustInitTimeMs: Date.now() - braintrustInitStart,
});
// Don't throw - allow workflow to continue without Braintrust
}
// Initialize Braintrust logger
const braintrustLogger = initLogger({
apiKey: process.env.BRAINTRUST_KEY,
projectName: process.env.ENVIRONMENT || 'development',
});
try {
@ -424,19 +408,6 @@ export const analystAgentTask: ReturnType<
// Log performance after workflow run creation
logPerformanceMetrics('post-createrun', payload.message_id, taskStartTime, resourceTracker);
// Wait for Braintrust initialization if it's not ready yet
const braintrustWaitStart = Date.now();
await braintrustInitPromise;
const braintrustWaitTime = Date.now() - braintrustWaitStart;
if (braintrustWaitTime > 10) {
// Only log if we actually had to wait
logger.log('Waited for Braintrust initialization', {
messageId: payload.message_id,
braintrustWaitTimeMs: braintrustWaitTime,
});
}
// Execute workflow with tracing
const workflowStartMethodStart = Date.now();
const tracedWorkflow = wrapTraced(
@ -472,7 +443,6 @@ export const analystAgentTask: ReturnType<
workflowStartMethodTimeMs: workflowStartMethodTime,
totalWorkflowTimeMs: totalWorkflowTime,
createRunTimeMs: createRunTime,
braintrustWaitTimeMs: braintrustWaitTime,
});
// Log final performance metrics
@ -491,7 +461,6 @@ export const analystAgentTask: ReturnType<
dataLoadTimeMs: dataLoadTime,
contextSetupTimeMs: contextSetupTime,
createRunTimeMs: createRunTime,
braintrustWaitTimeMs: braintrustWaitTime,
workflowStartMethodTimeMs: workflowStartMethodTime,
totalWorkflowTimeMs: totalWorkflowTime,
},
@ -549,6 +518,9 @@ export const analystAgentTask: ReturnType<
executionTimeMs: totalExecutionTime,
});
// Need to flush the Braintrust logger to ensure all traces are sent
await braintrustLogger.flush();
return {
success: false,
messageId: payload.message_id,

7
apps/trigger/src/tasks/message-post-processing/message-post-processing.ts
View File

@ -135,8 +135,8 @@ export const messagePostProcessingTask: ReturnType<
throw new Error('BRAINTRUST_KEY is not set');
}
// Initialize Braintrust logging for observability
initLogger({
// Initialize Braintrust logger
const braintrustLogger = initLogger({
apiKey: process.env.BRAINTRUST_KEY,
projectName: process.env.ENVIRONMENT || 'development',
});
@ -400,6 +400,9 @@ export const messagePostProcessingTask: ReturnType<
executionTimeMs: Date.now() - startTime,
});
// Need to flush the Braintrust logger to ensure all traces are sent
await braintrustLogger.flush();
return {
success: false,
messageId: payload.messageId,