Fix bug: resolveMessageId never called if aisdk errors

Author: jahooma

backend/src/llm-apis/vercel-ai-sdk/ai-sdk.ts

@@ -82,7 +82,7 @@ export const promptAiSdkStream = async function* (
     maxRetries?: number
     onCostCalculated?: (credits: number) => Promise<void>
     includeCacheControl?: boolean
-    resolveMessageId?: (messageId: string) => unknown
+    resolveMessageId: (messageId: string | undefined) => unknown
   } & Omit<Parameters<typeof streamText>[0], 'model' | 'messages'>,
 ): AsyncGenerator<StreamChunk> {
   if (
@@ -150,6 +150,10 @@ export const promptAiSdkStream = async function* (
         type: 'error',
         message: errorMessage,
       }
+
+      // Important: we need to resolve the message id before returning.
+      options.resolveMessageId(undefined)
+
       return
     }
     if (chunk.type === 'reasoning-delta') {

backend/src/prompt-agent-stream.ts

@@ -36,7 +36,7 @@ export const getAgentStreamFromTemplate = (params: {
   const { model } = template
 
   const { promise: messageIdPromise, resolve: resolveMessageId } =
-    Promise.withResolvers<string>()
+    Promise.withResolvers<string | undefined>()
 
   const getStream = (messages: Message[]) => {
     const options: Parameters<typeof promptAiSdkStream>[0] = {