From bac87c46dc3238bfc8c91be098d07bfb288030df Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Fri, 1 Dec 2023 21:32:38 +0800
Subject: [PATCH] docs: update basdic todos

---
 editor/src/components/editor/action/command-functions.ts | 4 ++--
 editor/src/components/editor/advice/advice-view.tsx      | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/editor/src/components/editor/action/command-functions.ts b/editor/src/components/editor/action/command-functions.ts
index d4696bd..4d37ddb 100644
--- a/editor/src/components/editor/action/command-functions.ts
+++ b/editor/src/components/editor/action/command-functions.ts
@@ -71,7 +71,7 @@ export const CommandFunctions = Extension.create({
 
 						switch (action.outputForm) {
 							case OutputForm.STREAMING:
-								const content = await fetch("/api/completion/qwen", {
+								const content = await fetch("/api/completion/yiyan", {
 									method: "POST",
 									body: JSON.stringify({ prompt: prompt }),
 								}).then(it => it.text());
@@ -80,7 +80,7 @@ export const CommandFunctions = Extension.create({
 								editor.chain().focus().insertContentAt(pos, content).run();
 								break;
 							case OutputForm.NORMAL:
-								const msg = await fetch("/api/completion/qwen", {
+								const msg = await fetch("/api/completion/yiyan", {
 									method: "POST",
 									body: JSON.stringify({ prompt: prompt }),
 								}).then(it => it.text());
diff --git a/editor/src/components/editor/advice/advice-view.tsx b/editor/src/components/editor/advice/advice-view.tsx
index f3679c6..8137f67 100644
--- a/editor/src/components/editor/advice/advice-view.tsx
+++ b/editor/src/components/editor/advice/advice-view.tsx
@@ -87,7 +87,8 @@ export const AdviceView = ({ editor }: AdviceViewProps) => {
 							<button
 								className='rounded-md bg-red-500 px-2.5 py-1.5 text-sm font-semibold text-white shadow-sm hover:bg-white/20'
 								onClick={() => {
-									setActiveId(null)
+									editor.commands.unsetAdvice(activeCommentId)
+									editor.commands.undo()
 									editor.commands.focus()
 								}}
 							>
@@ -97,6 +98,7 @@ export const AdviceView = ({ editor }: AdviceViewProps) => {
 								className='rounded-md bg-white/10 px-2.5 py-1.5 text-sm font-semibold text-white shadow-sm hover:bg-white/20'
 								onClick={() => {
 									setActiveId(null)
+									editor.commands.unsetAdvice(activeCommentId)
 									editor.commands.focus()
 								}}
 							>