mirror of
https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git
synced 2025-11-18 06:53:41 +08:00
init
This commit is contained in:
sijinhui
@@ -11,7 +11,12 @@ import {
|
||||
StoreKey,
|
||||
SUMMARIZE_MODEL,
|
||||
} from "../constant";
|
||||
import { api, RequestMessage } from "../client/api";
|
||||
import {
|
||||
api,
|
||||
getHeaders,
|
||||
useGetMidjourneySelfProxyUrl,
|
||||
RequestMessage,
|
||||
} from "../client/api";
|
||||
import { ChatControllerPool } from "../client/controller";
|
||||
import { prettyObject } from "../utils/format";
|
||||
import { estimateTokenLength } from "../utils/token";
|
||||
@@ -24,6 +29,7 @@ export type ChatMessage = RequestMessage & {
|
||||
isError?: boolean;
|
||||
id: string;
|
||||
model?: ModelType;
|
||||
attr?: any;
|
||||
};
|
||||
|
||||
export function createMessage(override: Partial<ChatMessage>): ChatMessage {
|
||||
@@ -80,6 +86,8 @@ function createEmptySession(): ChatSession {
|
||||
};
|
||||
}
|
||||
|
||||
const ChatFetchTaskPool: Record<string, any> = {};
|
||||
|
||||
function getSummarizeModel(currentModel: string) {
|
||||
// if it is using gpt-* models, force to use 3.5 to summarize
|
||||
return currentModel.startsWith("gpt") ? SUMMARIZE_MODEL : currentModel;
|
||||
@@ -266,12 +274,143 @@ export const useChatStore = createPersistStore(
|
||||
get().summarizeSession();
|
||||
},
|
||||
|
||||
async onUserInput(content: string) {
|
||||
fetchMidjourneyStatus(botMessage: ChatMessage, extAttr?: any) {
|
||||
const taskId = botMessage?.attr?.taskId;
|
||||
if (
|
||||
!taskId ||
|
||||
["SUCCESS", "FAILURE"].includes(botMessage?.attr?.status) ||
|
||||
ChatFetchTaskPool[taskId]
|
||||
)
|
||||
return;
|
||||
ChatFetchTaskPool[taskId] = setTimeout(async () => {
|
||||
ChatFetchTaskPool[taskId] = null;
|
||||
const statusRes = await fetch(
|
||||
`/api/midjourney/mj/task/${taskId}/fetch`,
|
||||
{
|
||||
method: "GET",
|
||||
headers: getHeaders(),
|
||||
},
|
||||
);
|
||||
const statusResJson = await statusRes.json();
|
||||
if (statusRes.status < 200 || statusRes.status >= 300) {
|
||||
botMessage.content =
|
||||
Locale.Midjourney.TaskStatusFetchFail +
|
||||
": " +
|
||||
(statusResJson?.error || statusResJson?.description) ||
|
||||
Locale.Midjourney.UnknownReason;
|
||||
console.log("【mid】状态码不对");
|
||||
} else {
|
||||
let isFinished = false;
|
||||
let content;
|
||||
const prefixContent = Locale.Midjourney.TaskPrefix(
|
||||
statusResJson.prompt,
|
||||
taskId,
|
||||
);
|
||||
console.log("【mid】请求成功了", statusResJson);
|
||||
switch (statusResJson?.status) {
|
||||
case "SUCCESS":
|
||||
console.log("[mid] SUCCESS", statusResJson);
|
||||
|
||||
content = statusResJson.imageUrl;
|
||||
isFinished = true;
|
||||
if (statusResJson.imageUrl) {
|
||||
let imgUrl = useGetMidjourneySelfProxyUrl(
|
||||
statusResJson.imageUrl,
|
||||
);
|
||||
botMessage.attr.imgUrl = imgUrl;
|
||||
botMessage.content =
|
||||
prefixContent + `[](${imgUrl})`;
|
||||
}
|
||||
if (
|
||||
statusResJson.action === "DESCRIBE" &&
|
||||
statusResJson.prompt
|
||||
) {
|
||||
botMessage.content += `\n${statusResJson.prompt}`;
|
||||
}
|
||||
break;
|
||||
case "FAILURE":
|
||||
console.log("[mid] FAILURE", statusResJson);
|
||||
content =
|
||||
statusResJson.failReason || Locale.Midjourney.UnknownReason;
|
||||
isFinished = true;
|
||||
botMessage.content =
|
||||
prefixContent +
|
||||
`**${
|
||||
Locale.Midjourney.TaskStatus
|
||||
}:** [${new Date().toLocaleString()}] - ${content}`;
|
||||
break;
|
||||
case "NOT_START":
|
||||
content = Locale.Midjourney.TaskNotStart;
|
||||
break;
|
||||
case "IN_PROGRESS":
|
||||
console.log("[mid] ", statusResJson);
|
||||
content = Locale.Midjourney.TaskProgressTip(
|
||||
statusResJson.progress,
|
||||
);
|
||||
break;
|
||||
case "SUBMITTED":
|
||||
content = Locale.Midjourney.TaskRemoteSubmit;
|
||||
break;
|
||||
default:
|
||||
console.log("[mid] ", statusResJson);
|
||||
content = statusResJson.status;
|
||||
}
|
||||
botMessage.attr.status = statusResJson.status;
|
||||
if (isFinished) {
|
||||
botMessage.attr.finished = true;
|
||||
} else {
|
||||
botMessage.content =
|
||||
prefixContent +
|
||||
`**${
|
||||
Locale.Midjourney.TaskStatus
|
||||
}:** [${new Date().toLocaleString()}] - ${content}`;
|
||||
if (
|
||||
statusResJson.status === "IN_PROGRESS" &&
|
||||
statusResJson.imageUrl
|
||||
) {
|
||||
let imgUrl = useGetMidjourneySelfProxyUrl(
|
||||
statusResJson.imageUrl,
|
||||
);
|
||||
botMessage.attr.imgUrl = imgUrl;
|
||||
botMessage.content += `\n[](${imgUrl})`;
|
||||
}
|
||||
this.fetchMidjourneyStatus(taskId, botMessage);
|
||||
}
|
||||
set(() => ({}));
|
||||
if (isFinished) {
|
||||
extAttr?.setAutoScroll(true);
|
||||
}
|
||||
}
|
||||
}, 3000);
|
||||
},
|
||||
|
||||
async onUserInput(content: string, extAttr?: any) {
|
||||
const session = get().currentSession();
|
||||
const modelConfig = session.mask.modelConfig;
|
||||
let userContent: string = "";
|
||||
if (
|
||||
extAttr?.mjImageMode &&
|
||||
(extAttr?.useImages?.length ?? 0) > 0 &&
|
||||
extAttr.mjImageMode !== "IMAGINE"
|
||||
) {
|
||||
if (
|
||||
extAttr.mjImageMode === "BLEND" &&
|
||||
(extAttr.useImages.length < 2 || extAttr.useImages.length > 5)
|
||||
) {
|
||||
alert(Locale.Midjourney.BlendMinImg(2, 5));
|
||||
return new Promise((resolve: any, reject) => {
|
||||
resolve(false);
|
||||
});
|
||||
}
|
||||
userContent = `/mj ${extAttr?.mjImageMode}`;
|
||||
extAttr.useImages.forEach((img: any, index: number) => {
|
||||
userContent += `::[${index + 1}]${img.filename}`;
|
||||
});
|
||||
} else {
|
||||
userContent = fillTemplateWith(content, modelConfig);
|
||||
}
|
||||
|
||||
const userContent = fillTemplateWith(content, modelConfig);
|
||||
console.log("[User Input] after template: ", userContent);
|
||||
// console.log("[User Input] after template: ", userContent);
|
||||
|
||||
const userMessage: ChatMessage = createMessage({
|
||||
role: "user",
|
||||
@@ -282,6 +421,7 @@ export const useChatStore = createPersistStore(
|
||||
role: "assistant",
|
||||
streaming: true,
|
||||
model: modelConfig.model,
|
||||
attr: {},
|
||||
});
|
||||
|
||||
// get recent messages
|
||||
@@ -301,57 +441,222 @@ export const useChatStore = createPersistStore(
|
||||
]);
|
||||
});
|
||||
|
||||
// make request
|
||||
api.llm.chat({
|
||||
messages: sendMessages,
|
||||
config: { ...modelConfig, stream: true },
|
||||
onUpdate(message) {
|
||||
botMessage.streaming = true;
|
||||
if (message) {
|
||||
botMessage.content = message;
|
||||
if (
|
||||
content.toLowerCase().startsWith("/mj") ||
|
||||
content.toLowerCase().startsWith("/MJ")
|
||||
) {
|
||||
botMessage.model = "midjourney";
|
||||
const startFn = async () => {
|
||||
const prompt = content.substring(3).trim();
|
||||
let action: string = "IMAGINE";
|
||||
const firstSplitIndex = prompt.indexOf("::");
|
||||
if (firstSplitIndex > 0) {
|
||||
action = prompt.substring(0, firstSplitIndex);
|
||||
}
|
||||
get().updateCurrentSession((session) => {
|
||||
session.messages = session.messages.concat();
|
||||
});
|
||||
},
|
||||
onFinish(message) {
|
||||
botMessage.streaming = false;
|
||||
if (message) {
|
||||
botMessage.content = message;
|
||||
get().onNewMessage(botMessage);
|
||||
if (
|
||||
![
|
||||
"UPSCALE",
|
||||
"VARIATION",
|
||||
"IMAGINE",
|
||||
"DESCRIBE",
|
||||
"BLEND",
|
||||
"REROLL",
|
||||
].includes(action)
|
||||
) {
|
||||
botMessage.content = Locale.Midjourney.TaskErrUnknownType;
|
||||
botMessage.streaming = false;
|
||||
return;
|
||||
}
|
||||
ChatControllerPool.remove(session.id, botMessage.id);
|
||||
},
|
||||
onError(error) {
|
||||
const isAborted = error.message.includes("aborted");
|
||||
botMessage.content +=
|
||||
"\n\n" +
|
||||
prettyObject({
|
||||
error: true,
|
||||
message: error.message,
|
||||
console.log("[action] ", action);
|
||||
botMessage.attr.action = action;
|
||||
let actionIndex: any = null;
|
||||
let actionUseTaskId: any = null;
|
||||
if (
|
||||
action === "VARIATION" ||
|
||||
action == "UPSCALE" ||
|
||||
action == "REROLL"
|
||||
) {
|
||||
actionIndex = parseInt(
|
||||
prompt.substring(firstSplitIndex + 2, firstSplitIndex + 3),
|
||||
);
|
||||
actionUseTaskId = prompt.substring(firstSplitIndex + 5);
|
||||
}
|
||||
try {
|
||||
let res = null;
|
||||
const reqFn = (path: string, method: string, body?: any) => {
|
||||
return fetch("/api/midjourney/mj/" + path, {
|
||||
method: method,
|
||||
headers: getHeaders(),
|
||||
body: body,
|
||||
});
|
||||
};
|
||||
switch (action) {
|
||||
case "IMAGINE": {
|
||||
res = await reqFn(
|
||||
"submit/imagine",
|
||||
"POST",
|
||||
JSON.stringify({
|
||||
prompt: prompt,
|
||||
// base64Array: extAttr?.useImages?.[0]?.base64 ?? null,
|
||||
base64Array: extAttr?.useImages?.[0]?.base64
|
||||
? [extAttr?.useImages?.[0]?.base64]
|
||||
: null,
|
||||
}),
|
||||
);
|
||||
break;
|
||||
}
|
||||
case "DESCRIBE": {
|
||||
res = await reqFn(
|
||||
"submit/describe",
|
||||
"POST",
|
||||
JSON.stringify({
|
||||
base64: extAttr.useImages[0].base64,
|
||||
}),
|
||||
);
|
||||
break;
|
||||
}
|
||||
case "BLEND": {
|
||||
const base64Array = extAttr.useImages.map(
|
||||
(ui: any) => ui.base64,
|
||||
);
|
||||
res = await reqFn(
|
||||
"submit/blend",
|
||||
"POST",
|
||||
JSON.stringify({ base64Array }),
|
||||
);
|
||||
break;
|
||||
}
|
||||
case "UPSCALE":
|
||||
case "VARIATION":
|
||||
case "REROLL": {
|
||||
res = await reqFn(
|
||||
"submit/change",
|
||||
"POST",
|
||||
JSON.stringify({
|
||||
action: action,
|
||||
index: actionIndex,
|
||||
taskId: actionUseTaskId,
|
||||
}),
|
||||
);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
}
|
||||
if (res == null) {
|
||||
botMessage.content =
|
||||
Locale.Midjourney.TaskErrNotSupportType(action);
|
||||
botMessage.streaming = false;
|
||||
return;
|
||||
}
|
||||
if (!res.ok) {
|
||||
const text = await res.text();
|
||||
throw new Error(
|
||||
`\n${Locale.Midjourney.StatusCode(
|
||||
res.status,
|
||||
)}\n${Locale.Midjourney.RespBody(
|
||||
text || Locale.Midjourney.None,
|
||||
)}`,
|
||||
);
|
||||
}
|
||||
const resJson = await res.json();
|
||||
if (
|
||||
res.status < 200 ||
|
||||
res.status >= 300 ||
|
||||
(resJson.code != 1 && resJson.code != 22)
|
||||
) {
|
||||
botMessage.content = Locale.Midjourney.TaskSubmitErr(
|
||||
resJson?.msg ||
|
||||
resJson?.error ||
|
||||
resJson?.description ||
|
||||
Locale.Midjourney.UnknownError,
|
||||
);
|
||||
} else {
|
||||
const taskId: string = resJson.result;
|
||||
const prefixContent = Locale.Midjourney.TaskPrefix(
|
||||
prompt,
|
||||
taskId,
|
||||
);
|
||||
botMessage.content =
|
||||
prefixContent +
|
||||
`[${new Date().toLocaleString()}] - ${
|
||||
Locale.Midjourney.TaskSubmitOk
|
||||
}: ` +
|
||||
resJson?.description || Locale.Midjourney.PleaseWait;
|
||||
botMessage.attr.taskId = taskId;
|
||||
botMessage.attr.status = resJson.status;
|
||||
this.fetchMidjourneyStatus(botMessage, extAttr);
|
||||
}
|
||||
} catch (e: any) {
|
||||
console.error(e);
|
||||
botMessage.content = Locale.Midjourney.TaskSubmitErr(
|
||||
e?.error || e?.message || Locale.Midjourney.UnknownError,
|
||||
);
|
||||
} finally {
|
||||
ChatControllerPool.remove(
|
||||
session.id,
|
||||
botMessage.id ?? messageIndex,
|
||||
);
|
||||
botMessage.streaming = false;
|
||||
}
|
||||
};
|
||||
await startFn();
|
||||
get().onNewMessage(botMessage);
|
||||
set(() => ({}));
|
||||
extAttr?.setAutoScroll(true);
|
||||
} else {
|
||||
// make request
|
||||
api.llm.chat({
|
||||
messages: sendMessages,
|
||||
config: { ...modelConfig, stream: true },
|
||||
onUpdate(message) {
|
||||
botMessage.streaming = true;
|
||||
if (message) {
|
||||
botMessage.content = message;
|
||||
}
|
||||
get().updateCurrentSession((session) => {
|
||||
session.messages = session.messages.concat();
|
||||
});
|
||||
botMessage.streaming = false;
|
||||
userMessage.isError = !isAborted;
|
||||
botMessage.isError = !isAborted;
|
||||
get().updateCurrentSession((session) => {
|
||||
session.messages = session.messages.concat();
|
||||
});
|
||||
ChatControllerPool.remove(
|
||||
session.id,
|
||||
botMessage.id ?? messageIndex,
|
||||
);
|
||||
},
|
||||
onFinish(message) {
|
||||
botMessage.streaming = false;
|
||||
if (message) {
|
||||
botMessage.content = message;
|
||||
get().onNewMessage(botMessage);
|
||||
}
|
||||
ChatControllerPool.remove(session.id, botMessage.id);
|
||||
},
|
||||
onError(error) {
|
||||
const isAborted = error.message.includes("aborted");
|
||||
botMessage.content =
|
||||
"\n\n" +
|
||||
prettyObject({
|
||||
error: true,
|
||||
message: error.message,
|
||||
message2: "用上面刷新按钮试试。",
|
||||
});
|
||||
botMessage.streaming = false;
|
||||
userMessage.isError = !isAborted;
|
||||
botMessage.isError = !isAborted;
|
||||
get().updateCurrentSession((session) => {
|
||||
session.messages = session.messages.concat();
|
||||
});
|
||||
ChatControllerPool.remove(
|
||||
session.id,
|
||||
botMessage.id ?? messageIndex,
|
||||
);
|
||||
|
||||
console.error("[Chat] failed ", error);
|
||||
},
|
||||
onController(controller) {
|
||||
// collect controller for stop/retry
|
||||
ChatControllerPool.addController(
|
||||
session.id,
|
||||
botMessage.id ?? messageIndex,
|
||||
controller,
|
||||
);
|
||||
},
|
||||
});
|
||||
console.error("[Chat] failed ", error);
|
||||
},
|
||||
onController(controller) {
|
||||
// collect controller for stop/retry
|
||||
ChatControllerPool.addController(
|
||||
session.id,
|
||||
botMessage.id ?? messageIndex,
|
||||
controller,
|
||||
);
|
||||
},
|
||||
});
|
||||
}
|
||||
},
|
||||
|
||||
getMemoryPrompt() {
|
||||
@@ -528,12 +833,12 @@ export const useChatStore = createPersistStore(
|
||||
|
||||
const lastSummarizeIndex = session.messages.length;
|
||||
|
||||
console.log(
|
||||
"[Chat History] ",
|
||||
toBeSummarizedMsgs,
|
||||
historyMsgLength,
|
||||
modelConfig.compressMessageLengthThreshold,
|
||||
);
|
||||
// console.log(
|
||||
// "[Chat History] ",
|
||||
// toBeSummarizedMsgs,
|
||||
// historyMsgLength,
|
||||
// modelConfig.compressMessageLengthThreshold,
|
||||
// );
|
||||
|
||||
if (
|
||||
historyMsgLength > modelConfig.compressMessageLengthThreshold &&
|
||||
|
||||
Reference in New Issue
Block a user