build fixes

This commit is contained in:
dal 2025-07-07 12:19:50 -06:00
parent 1a617a1c47
commit fa73b7b35b
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 12 additions and 11 deletions

View File

@ -1,5 +1,5 @@
import { logger, schemaTask } from '@trigger.dev/sdk';
import { initLogger, wrapTraced, currentSpan } from 'braintrust';
import { logger, schemaTask, tasks } from '@trigger.dev/sdk';
import { currentSpan, initLogger, wrapTraced } from 'braintrust';
import { AnalystAgentTaskInputSchema, type AnalystAgentTaskOutput } from './types';
// Task 2 & 4: Database helpers (IMPLEMENTED)
@ -340,13 +340,14 @@ export const analystAgentTask: ReturnType<
const braintrustMetadataPromise = getBraintrustMetadata({ messageId: payload.message_id });
// Wait for all operations to complete
const [messageContext, conversationHistory, dataSource, dashboardFiles, braintrustMetadata] = await Promise.all([
messageContextPromise,
conversationHistoryPromise,
dataSourcePromise,
dashboardFilesPromise,
braintrustMetadataPromise,
]);
const [messageContext, conversationHistory, dataSource, dashboardFiles, braintrustMetadata] =
await Promise.all([
messageContextPromise,
conversationHistoryPromise,
dataSourcePromise,
dashboardFilesPromise,
braintrustMetadataPromise,
]);
const dataLoadEnd = Date.now();
const dataLoadTime = dataLoadEnd - dataLoadStart;
@ -444,7 +445,7 @@ export const analystAgentTask: ReturnType<
inputData: workflowInput,
runtimeContext,
});
// Log the metadata as part of the span
currentSpan().log({
metadata: {
@ -456,7 +457,7 @@ export const analystAgentTask: ReturnType<
chatId: braintrustMetadata.chatId,
},
});
return result;
},
{