createDataStreamResponse doesn't allow streamText to be aborted via signals/useChat.stop method

[raffaele-filiberti]

I was referring to the docs about how to stop streams, and this works perfectly when using directly streamText but it's not working the same way when wrapping it with createDataStreamResponse.

return createDataStreamResponse({
  onError(error) {
    console.error("Error in data stream:", error)
    return "Error in data stream"
  },
  execute: async (dataStream) => {
    const result = streamText({
      model: openrouter(model as string),
      messages: messages,
      maxSteps: 5,
      abortSignal: req.signal,
      system: systemPrompt,
      temperature: 0,
      experimental_activeTools: activeTools,
      tools: activeTools && {
        web_search: generateWebSearchTool(dataStream),
        retrieve: generateRetrieveTool(),
      },
      onChunk(event) {
        console.log("Chunk: ", event, req.signal.aborted)
      },
      async onFinish({ response }) {
        // onFinish will be called even if I'm aborting
        await saveChat({
          id,
          userId,
          messages: appendResponseMessages({
            messages,
            responseMessages: response.messages,
          }),
          model,
        })
      },
    })

    result.consumeStream()

    return result.mergeIntoDataStream(dataStream)
  },
})

Thanks in advance!