🐛 (dify) Fix invalid parsing when text delta is multiline

packages/forge/blocks/difyAi/actions/createChatMessage.ts

@@ -5,6 +5,7 @@ import { defaultBaseUrl } from '../constants'
 import { Chunk } from '../types'
 import ky, { HTTPError } from 'ky'
 import { deprecatedCreateChatMessageOptions } from '../deprecated'
+import { formatStreamPart } from 'ai'
 
 export const createChatMessage = createAction({
   auth,
@@ -120,7 +121,7 @@ export const createChatMessage = createAction({
                     onMessage: (message) => {
                       controller.enqueue(
                         new TextEncoder().encode(
-                          '0:"' + message.replace(/"/g, '\\"') + '"\n'
+                          formatStreamPart('text', message)
                         )
                       )
                     },
@@ -299,7 +300,7 @@ const processDifyStream = async (
       totalTokens,
       conversationId,
     }: {
-      totalTokens: number
+      totalTokens?: number
       conversationId: string
     }) => void
   }
@@ -333,7 +334,7 @@ const processDifyStream = async (
         }
         if (data.event === 'message_end') {
           callbacks.onMessageEnd?.({
-            totalTokens: data.metadata.usage.total_tokens,
+            totalTokens: data.metadata.usage?.total_tokens,
             conversationId: data.conversation_id,
           })
         }