fix: missing onMessageProcessed or occur multi times

Author: hexqi

packages/plugins/robot/src/composables/core/useConversation.ts

@@ -11,6 +11,7 @@ export interface ConversationAdapterOptions {
   statusManager: {
     isProcessing: () => boolean
     setProcessing: () => void
+    resetProcessing: () => void
   }
 }
 
@@ -43,7 +44,8 @@ export function useConversationAdapter(options: ConversationAdapterOptions) {
       const contextMessages = toRaw(messages.value.slice(0, -1))
       await onFinishRequest(finishReason ?? 'unknown', messages.value, contextMessages, messageState)
       const lastMessage = messages.value.at(-1)
-      if (lastMessage) {
+      if (lastMessage && finishReason === 'stop' && !lastMessage.tool_calls && statusManager.isProcessing()) {
+        statusManager.resetProcessing()
         await onMessageProcessed(finishReason ?? 'unknown', lastMessage.content ?? '', messages.value, {})
       }
     }

packages/plugins/robot/src/composables/features/useToolCalls.ts

@@ -77,6 +77,7 @@ export interface ToolCallHandlerConfig {
   statusManager?: {
     isProcessing: () => boolean
     setProcessing: () => void
+    resetProcessing: () => void
   }
 }
 

packages/plugins/robot/src/composables/useChat.ts

@@ -110,6 +110,10 @@ const handleFinishRequest = async (
 
   if (finishReason === 'aborted' || messageState?.status === STATUS.ABORTED) {
     messageState.status = STATUS.ABORTED
+  } else if (finishReason === 'stop' && !lastMessage.tool_calls) {
+    messageState.status = STATUS.FINISHED
+    chatStatus.value = CHAT_STATUS.FINISHED
+    await await onMessageProcessed(finishReason, lastMessage.content ?? '', messages.value, {})
   }
 }
 
@@ -146,6 +150,9 @@ const {
     isProcessing: () => chatStatus.value === CHAT_STATUS.PROCESSING,
     setProcessing: () => {
       chatStatus.value = CHAT_STATUS.PROCESSING
+    },
+    resetProcessing: () => {
+      chatStatus.value = CHAT_STATUS.FINISHED
     }
   }
 })
@@ -173,6 +180,9 @@ const handleToolCall = createToolCallHandler({
     isProcessing: () => chatStatus.value === CHAT_STATUS.PROCESSING,
     setProcessing: () => {
       chatStatus.value = CHAT_STATUS.PROCESSING
+    },
+    resetProcessing: () => {
+      chatStatus.value = CHAT_STATUS.FINISHED
     }
   }
 })