Improve message handling in AIChat hook for non-standard message types

YaoApp · Feb 7, 2025 · b773842 · b773842
1 parent 7935659
commit b773842
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/xgen/layouts/components/Neo/hooks/useAIChat.ts b/packages/xgen/layouts/components/Neo/hooks/useAIChat.ts
@@ -218,6 +218,12 @@ export default ({ assistant_id, chat_id, upload_options = {} }: Args) => {
 				}
 			})
 
+			// if message type is not text, tool, think, then append directly
+			if (lastItem.type != 'text' && lastItem.type != 'tool' && lastItem.type != 'think') {
+				res.push({ ...baseMessage, ...lastItem })
+				return
+			}
+
 			// Create final message based on last item's type
 			const finalMessage = {
 				...baseMessage,