Trigger message event on generating

This commit is contained in:
Luo Tim 2024-06-07 01:55:09 +08:00
parent e23f343d0f
commit 05bf0a7f3d

View File

@ -5,6 +5,7 @@ import yaml from "js-yaml";
import { RootInstance } from "./RootStore"; import { RootInstance } from "./RootStore";
import { useTranslation } from "react-i18next"; import { useTranslation } from "react-i18next";
import i18next from "i18next"; import i18next from "i18next";
import APIUtil from "@/util/APIUtil";
interface Context { interface Context {
content: string; content: string;
@ -171,7 +172,8 @@ export const ChatStore = types
self.hasDone = false; self.hasDone = false;
self.errorMessage = ""; self.errorMessage = "";
self.currentMessage = ""; self.currentMessage = "";
const chatModel = getParent<RootInstance>(self).config.getDefaultModel(); const config = getParent<RootInstance>(self).config
const chatModel = config.getDefaultModel();
messageUtil.sendMessage({ messageUtil.sendMessage({
command: "sendMessage", command: "sendMessage",
text: text, text: text,
@ -179,6 +181,7 @@ export const ChatStore = types
parent_hash: lastNonEmptyHash(), parent_hash: lastNonEmptyHash(),
model: chatModel, model: chatModel,
}); });
APIUtil.createMessage(config.getAppURL(), config.getUserKey(), {content: text});
}; };
const helpMessage = (originalMessage = false) => { const helpMessage = (originalMessage = false) => {
@ -390,7 +393,7 @@ Thinking...
self.messages.push(...messages); self.messages.push(...messages);
}, },
updateLastMessage: (message: string) => { updateLastMessage: (message: string) => {
console.log("message: ", message); // console.log("message: ", message);
if (self.messages.length > 0) { if (self.messages.length > 0) {
self.messages[self.messages.length - 1].message = message; self.messages[self.messages.length - 1].message = message;
// if (message === "") { // if (message === "") {