diff --git a/src/models/chat_model.py b/src/models/chat_model.py index e3a83d1e..43b00aa5 100644 --- a/src/models/chat_model.py +++ b/src/models/chat_model.py @@ -31,7 +31,8 @@ class OpenAIBase(): stream=True, ) for chunk in response: - yield chunk.choices[0].delta + if len(chunk.choices) > 0: + yield chunk.choices[0].delta def _get_response(self, messages): response = self.client.chat.completions.create( diff --git a/web/src/components/ChatComponent.vue b/web/src/components/ChatComponent.vue index efd98363..60e61f5d 100644 --- a/web/src/components/ChatComponent.vue +++ b/web/src/components/ChatComponent.vue @@ -77,6 +77,7 @@
@@ -100,7 +102,7 @@ :is-loading="isStreaming" :send-button-disabled="!conv.inputText && !isStreaming" :auto-size="{ minRows: 2, maxRows: 10 }" - @send="sendMessage" + @send="handleSendOrStop" @keydown="handleKeyDown" >