Merge pull request #135 from moxin-org/support-chat-without-streaming

Support chat responses when stream option is off
moxin-org · Jul 4, 2024 · 39ffc3f · 39ffc3f
2 parents 2d4d5ed + ccd7987
commit 39ffc3f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/data/chats/chat.rs b/src/data/chats/chat.rs
@@ -251,6 +251,14 @@ impl Chat {
                             break;
                         }
                     }
+                    Ok(ChatResponse::ChatFinalResponseData(data)) => {
+                        let _ = store_chat_tx.send(ChatTokenArrivalAction::AppendDelta(
+                            data.choices[0].message.content.clone(),
+                        ));
+                        let _ = store_chat_tx.send(ChatTokenArrivalAction::StreamingDone);
+                        SignalToUI::set_ui_signal();
+                        break;
+                    }
                     Err(err) => eprintln!("Error receiving response chunk: {:?}", err),
                     _ => (),
                 }