fix: refresh during generation

fixes #116
2024-11-25 19:33:59 +01:00 · 2023-03-25 13:58:17 +08:00 · 2023-03-25 13:58:17 +08:00 · b7848e8b8b
parent b04dbfa76b
commit b7848e8b8b
1 changed files with 5 additions and 3 deletions
--- a/src/components/Chat/ChatContent/Message/MessageContent.tsx
+++ b/src/components/Chat/ChatContent/Message/MessageContent.tsx
@ -157,7 +157,9 @@ const ContentView = React.memo(
        <div className='flex justify-end gap-2 w-full mt-2'>
          {isDelete || (
            <>
-              {role === 'assistant' && messageIndex === lastMessageIndex && (
+              {!useStore.getState().generating &&
+                role === 'assistant' &&
+                messageIndex === lastMessageIndex && (
                  <RefreshButton onClick={handleRefresh} />
                )}
              {messageIndex !== 0 && <UpButton onClick={handleMoveUp} />}