diff --git a/components/QuillEditor.tsx b/components/QuillEditor.tsx
index ba54fe5..d98e907 100644
--- a/components/QuillEditor.tsx
+++ b/components/QuillEditor.tsx
@@ -286,66 +286,40 @@ const QEditor = ({ lng }) => {
const newValue = parseInt(event.target.value, 10);
setGeneratedPaperNumber(newValue);
};
- // 处理AI写作
- const handleAIWrite = async () => {
+
+ // 处理handleAIAction
+ async function handleAIAction(topic: string, actionType: string) {
+ // 创建一个新的 AbortController 实例
+ const newController = new AbortController();
+ setController(newController);
+ quill!.setSelection(cursorPosition!, 0); // 将光标移动到原来的位置
+ setOpenProgressBar(true); //开启进度条
try {
- setOpenProgressBar(true); //开启进度条
- // 创建一个新的 AbortController 实例
- const newController = new AbortController();
- setController(newController);
- quill!.setSelection(cursorPosition!, 0); // 将光标移动到原来的位置
+ if (actionType === "write") {
+ // 写作逻辑
+ const prompt = "请帮助用户完成论文写作，使用用户所说的语言完成";
+ await sendMessageToOpenAI(
+ userInput,
+ quill!,
+ selectedModel!,
+ apiKey,
+ upsreamUrl,
+ prompt,
+ cursorPosition!,
+ true,
+ newController.signal // 传递 AbortSignal
+ );
+ } else if (actionType === "paper2AI") {
+ // paper2AI 逻辑，根据 actionParam 处理特定任务
+ let offset = -1;
+ if (generatedPaperNumber != 1) offset = 0; //如果生成的数量不为1，则从0开始
+ //如果说要评估主题是否匹配的话,就要多获取一些文献
+ let limit = 2;
+ if (isEvaluateTopicMatch) {
+ limit = 4;
+ }
- const prompt = "请帮助用户完成论文写作，使用用户所说的语言完成";
- await sendMessageToOpenAI(
- userInput,
- quill!,
- selectedModel!,
- apiKey,
- upsreamUrl,
- prompt,
- cursorPosition!,
- true,
- newController.signal // 传递 AbortSignal
- );
- // 清空input内容
- setUserInput("");
- // 重新获取更新后的内容并更新 Redux store
- const updatedContent = quill!.root.innerHTML;
- dispatch(setEditorContent(updatedContent));
- toast.success(`AI写作完成`, {
- position: "top-center",
- autoClose: 2000,
- pauseOnHover: true,
- });
- } catch (error) {
- toast.error(`AI写作出现错误: ${error}`, {
- position: "top-center",
- autoClose: 3000,
- pauseOnHover: true,
- });
- } finally {
- setOpenProgressBar(false); //关闭进度条
- }
- };
-
- // 处理paper2AI
- async function paper2AI(topic: string) {
- try {
- // 创建一个新的 AbortController 实例
- const newController = new AbortController();
- setController(newController);
- quill!.setSelection(cursorPosition!, 0); // 将光标移动到原来的位置
- let offset = -1;
- if (generatedPaperNumber != 1) offset = 0; //如果生成的数量不为1，则从0开始
- setOpenProgressBar(true); //开启进度条
- //如果说要评估主题是否匹配的话,就要多获取一些文献
- let limit = 2;
- if (isEvaluateTopicMatch) {
- limit = 4;
- }
-
- for (let i = 0; i < generatedPaperNumber!; i++) {
- try {
+ for (let i = 0; i < generatedPaperNumber!; i++) {
if (!topic) {
//使用ai提取当前要请求的论文主题
const prompt =
@@ -370,9 +344,14 @@ const QEditor = ({ lng }) => {
topic = topic.slice(0, 10);
}
}
- console.log("topic in AI", topic);
- console.log("offset in paper2AI", offset);
- console.log("limit in paper2AI", limit);
+ console.log(
+ "topic in AI:",
+ topic,
+ "offset in paper2AI:",
+ offset,
+ "limit in paper2AI:",
+ limit
+ );
let rawData, dataString, newReferences;
if (selectedSource === "arxiv") {
rawData = await getArxivPapers(topic, limit, offset);
@@ -484,10 +463,7 @@ const QEditor = ({ lng }) => {
// 确保搜索到的论文不超过 3000 个字符
const trimmedMessage =
dataString.length > 3000 ? dataString.slice(0, 3000) : dataString;
- //slate的方法
- // const content = `需要完成的论文主题：${topic}, 搜索到的论文内容:${trimmedMessage},之前已经完成的内容上下文：${extractText(
- // editorValue
- // )}`;
+ // 生成AI PROMPT
const content = `之前用户已经完成的内容上下文：${getTextBeforeCursor(
quill!,
800
@@ -504,10 +480,6 @@ const QEditor = ({ lng }) => {
true,
newController.signal // 传递 AbortSignal
);
- setUserInput("");
- // 重新获取更新后的内容并更新 Redux store
- const updatedContent = quill!.root.innerHTML;
- dispatch(setEditorContent(updatedContent));
//在对应的位置添加文献
const nearestNumber = getNumberBeforeCursor(quill!);
dispatch(
@@ -516,43 +488,41 @@ const QEditor = ({ lng }) => {
position: nearestNumber,
})
);
- if (isVip) {
- //在云端同步supabase
- const data = await submitPaper(
- supabase,
- updatedContent,
- references,
- paperNumberRedux
- );
- }
//修改offset使得按照接下来的顺序进行获取文献
offset += 2;
setGenerateNumber(i + 1);
- toast.success(`AI写作完成`, {
- position: "top-center",
- autoClose: 2000,
- pauseOnHover: true,
- });
- } catch (error) {
- console.error("Paper2AI出现错误", error);
- // 在处理错误后，再次抛出这个错误
- // throw new Error(`Paper2AI出现错误: ${error}`);
- toast.error(`Paper2AI出现错误: ${error}`, {
- position: "top-center",
- autoClose: 3000,
- pauseOnHover: true,
- });
}
+ setUserInput(""); // 只有在全部成功之后才清空input内容
}
+ toast.success(
+ `AI ${actionType == "write" ? "写作" : "文献获取总结"}完成`,
+ {
+ position: "top-center",
+ autoClose: 2000,
+ pauseOnHover: true,
+ }
+ );
} catch (error) {
- toast.error(`Paper2AI出现错误: ${error}`, {
+ toast.error(`AI写作出现错误: ${error}`, {
position: "top-center",
autoClose: 3000,
pauseOnHover: true,
});
} finally {
+ // 通用的后处理逻辑
+ const updatedContent = quill!.root.innerHTML;
+ dispatch(setEditorContent(updatedContent));
+ if (isVip) {
+ //在云端同步supabase
+ const data = await submitPaper(
+ supabase,
+ updatedContent,
+ references,
+ paperNumberRedux
+ );
+ }
setOpenProgressBar(false);
- setGenerateNumber(0); //总的已经生成的数量
+ setGenerateNumber(0); //总的已经生成的数量设置为0 以便下次使用
}
}
@@ -574,13 +544,13 @@ const QEditor = ({ lng }) => {
)}
/>