🐛 (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,
           })
         }

packages/forge/blocks/difyAi/package.json

@@ -12,5 +12,8 @@
     "@types/react": "18.2.15",
     "ky": "1.2.3",
     "typescript": "5.4.5"
+  },
+  "dependencies": {
+    "ai": "3.1.12"
   }
 }

packages/forge/blocks/difyAi/types.ts

@@ -11,7 +11,7 @@ type AgentMessageChunk = {
 type MessageEndChunk = {
   event: 'message_end'
   metadata: {
-    usage: {
+    usage?: {
       total_tokens: number
     }
   }