From 4e31d7b64ff1676e496828acc4b5a8a87efb6527 Mon Sep 17 00:00:00 2001
From: StyleZhang <jasonapring2015@outlook.com>
Date: Mon, 1 Apr 2024 21:17:17 +0800
Subject: [PATCH] chat

---
 web/app/components/base/chat/chat/answer/index.tsx          | 4 +++-
 web/app/components/base/chat/chat/index.tsx                 | 3 +++
 .../workflow/panel/debug-and-preview/chat-wrapper.tsx       | 1 +
 .../components/workflow/panel/debug-and-preview/hooks.ts    | 6 +-----
 4 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/web/app/components/base/chat/chat/answer/index.tsx b/web/app/components/base/chat/chat/answer/index.tsx
index 4b1d757a0e..64a0c3c8f3 100644
--- a/web/app/components/base/chat/chat/answer/index.tsx
+++ b/web/app/components/base/chat/chat/answer/index.tsx
@@ -29,6 +29,7 @@ type AnswerProps = {
   responding?: boolean
   allToolIcons?: Record<string, string | Emoji>
   showPromptLog?: boolean
+  chatAnswerContainerInner?: string
 }
 const Answer: FC<AnswerProps> = ({
   item,
@@ -39,6 +40,7 @@ const Answer: FC<AnswerProps> = ({
   responding,
   allToolIcons,
   showPromptLog,
+  chatAnswerContainerInner,
 }) => {
   const { t } = useTranslation()
   const {
@@ -70,7 +72,7 @@ const Answer: FC<AnswerProps> = ({
         }
       </div>
       <div className='chat-answer-container grow w-0 ml-4'>
-        <div className='relative pr-10'>
+        <div className={`relative pr-10 ${chatAnswerContainerInner}`}>
           <AnswerTriangle className='absolute -left-2 top-0 w-2 h-3 text-gray-100' />
           <div
             className={`
diff --git a/web/app/components/base/chat/chat/index.tsx b/web/app/components/base/chat/chat/index.tsx
index 8e723b06f7..f0a2b345bf 100644
--- a/web/app/components/base/chat/chat/index.tsx
+++ b/web/app/components/base/chat/chat/index.tsx
@@ -50,6 +50,7 @@ export type ChatProps = {
   onAnnotationRemoved?: (index: number) => void
   chatNode?: ReactNode
   onFeedback?: (messageId: string, feedback: Feedback) => void
+  chatAnswerContainerInner?: string
 }
 const Chat: FC<ChatProps> = ({
   config,
@@ -73,6 +74,7 @@ const Chat: FC<ChatProps> = ({
   onAnnotationRemoved,
   chatNode,
   onFeedback,
+  chatAnswerContainerInner,
 }) => {
   const { t } = useTranslation()
   const { currentLogItem, setCurrentLogItem, showPromptLogModal, setShowPromptLogModal } = useAppStore()
@@ -191,6 +193,7 @@ const Chat: FC<ChatProps> = ({
                       responding={isLast && isResponding}
                       allToolIcons={allToolIcons}
                       showPromptLog={showPromptLog}
+                      chatAnswerContainerInner={chatAnswerContainerInner}
                     />
                   )
                 }
diff --git a/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx b/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx
index fe5c3aa898..d3baa75cc5 100644
--- a/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx
+++ b/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx
@@ -112,6 +112,7 @@ const ChatWrapper = forwardRef<ChatWrapperRefType>((_, ref) => {
       )}
       suggestedQuestions={suggestedQuestions}
       showPromptLog
+      chatAnswerContainerInner='!pr-2'
     />
   )
 })
diff --git a/web/app/components/workflow/panel/debug-and-preview/hooks.ts b/web/app/components/workflow/panel/debug-and-preview/hooks.ts
index 3eb1d0ddea..52082600a3 100644
--- a/web/app/components/workflow/panel/debug-and-preview/hooks.ts
+++ b/web/app/components/workflow/panel/debug-and-preview/hooks.ts
@@ -203,7 +203,7 @@ export const useChat = (
     handleRun(
       params,
       {
-        onData: (message: string, isFirstMessage: boolean, { conversationId: newConversationId, messageId, taskId }: any) => {
+        onData: (message: string, isFirstMessage: boolean, { conversationId: newConversationId, messageId, taskId, errorMessage }: any) => {
           responseItem.content = responseItem.content + message
 
           if (messageId && !hasSetResponseId) {
@@ -257,10 +257,6 @@ export const useChat = (
         },
         onError() {
           handleResponding(false)
-          const newChatList = produce(chatListRef.current, (draft) => {
-            draft.splice(draft.findIndex(item => item.id === placeholderAnswerId), 1)
-          })
-          handleUpdateChatList(newChatList)
         },
         onWorkflowStarted: ({ workflow_run_id, task_id }) => {
           taskIdRef.current = task_id