refractor: 主要的和ai交互的函数重构
This commit is contained in:
liuweiqing
parent
4c78494e4d
commit
efabb39dfc
|
|
@ -286,15 +286,17 @@ const QEditor = ({ lng }) => {
|
|||
const newValue = parseInt(event.target.value, 10);
|
||||
setGeneratedPaperNumber(newValue);
|
||||
};
|
||||
// 处理AI写作
|
||||
const handleAIWrite = async () => {
|
||||
try {
|
||||
setOpenProgressBar(true); //开启进度条
|
||||
|
||||
// 处理handleAIAction
|
||||
async function handleAIAction(topic: string, actionType: string) {
|
||||
// 创建一个新的 AbortController 实例
|
||||
const newController = new AbortController();
|
||||
setController(newController);
|
||||
quill!.setSelection(cursorPosition!, 0); // 将光标移动到原来的位置
|
||||
|
||||
setOpenProgressBar(true); //开启进度条
|
||||
try {
|
||||
if (actionType === "write") {
|
||||
// 写作逻辑
|
||||
const prompt = "请帮助用户完成论文写作，使用用户所说的语言完成";
|
||||
await sendMessageToOpenAI(
|
||||
userInput,
|
||||
|
|
@ -307,37 +309,10 @@ const QEditor = ({ lng }) => {
|
|||
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); // 将光标移动到原来的位置
|
||||
} else if (actionType === "paper2AI") {
|
||||
// paper2AI 逻辑，根据 actionParam 处理特定任务
|
||||
let offset = -1;
|
||||
if (generatedPaperNumber != 1) offset = 0; //如果生成的数量不为1，则从0开始
|
||||
setOpenProgressBar(true); //开启进度条
|
||||
//如果说要评估主题是否匹配的话,就要多获取一些文献
|
||||
let limit = 2;
|
||||
if (isEvaluateTopicMatch) {
|
||||
|
|
@ -345,7 +320,6 @@ const QEditor = ({ lng }) => {
|
|||
}
|
||||
|
||||
for (let i = 0; i < generatedPaperNumber!; i++) {
|
||||
try {
|
||||
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,6 +488,30 @@ const QEditor = ({ lng }) => {
|
|||
position: nearestNumber,
|
||||
})
|
||||
);
|
||||
//修改offset使得按照接下来的顺序进行获取文献
|
||||
offset += 2;
|
||||
setGenerateNumber(i + 1);
|
||||
}
|
||||
setUserInput(""); // 只有在全部成功之后才清空input内容
|
||||
}
|
||||
toast.success(
|
||||
`AI ${actionType == "write" ? "写作" : "文献获取总结"}完成`,
|
||||
{
|
||||
position: "top-center",
|
||||
autoClose: 2000,
|
||||
pauseOnHover: true,
|
||||
}
|
||||
);
|
||||
} catch (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(
|
||||
|
|
@ -525,34 +521,8 @@ const QEditor = ({ lng }) => {
|
|||
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,
|
||||
});
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
toast.error(`Paper2AI出现错误: ${error}`, {
|
||||
position: "top-center",
|
||||
autoClose: 3000,
|
||||
pauseOnHover: true,
|
||||
});
|
||||
} finally {
|
||||
setOpenProgressBar(false);
|
||||
setGenerateNumber(0); //总的已经生成的数量
|
||||
setGenerateNumber(0); //总的已经生成的数量设置为0 以便下次使用
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -574,13 +544,13 @@ const QEditor = ({ lng }) => {
|
|||
)}
|
||||
/>
|
||||
<button
|
||||
onClick={handleAIWrite}
|
||||
onClick={() => handleAIAction(userInput, "write")}
|
||||
className="bg-gray-300 hover:bg-gray-400 text-black font-bold py-2 px-4 mr-2 rounded"
|
||||
>
|
||||
{t("AI写作")}
|
||||
</button>
|
||||
<button
|
||||
onClick={() => paper2AI(userInput)}
|
||||
onClick={() => handleAIAction(userInput, "paper2AI")}
|
||||
className="bg-yellow-500 hover:bg-yellow-600 text-white font-bold py-2 px-4 mr-2 rounded"
|
||||
>
|
||||
{t("Paper2AI")}
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user