fix: correct incorrect assistant display when quick ai access (#779)

ayangweb · web-flow · commit 657df482bfd1 · 2025-07-19T13:54:39.000+08:00
* fix: correct incorrect assistant display when quick ai access

* docs: update changelog
diff --git a/docs/content.en/docs/release-notes/_index.md b/docs/content.en/docs/release-notes/_index.md
@@ -26,7 +26,7 @@ Information about release notes of Coco Server is provided here.
 - fix: unregister ext hotkey when it gets deleted #770
 - fix: indexing apps does not respect search scope config #773
 - fix: restore missing category titles on subpages #772
-
+- fix: correct incorrect assistant display when quick ai access #779
 
 ### ✈️ Improvements
 
diff --git a/src/components/Search/AssistantManager.tsx b/src/components/Search/AssistantManager.tsx
@@ -1,4 +1,4 @@
-import { useCallback, useRef, useMemo, useState } from "react";
+import { useCallback, useRef, useMemo, useState, useEffect } from "react";
 import { cloneDeep, isEmpty } from "lodash-es";
 
 import { useSearchStore } from "@/stores/searchStore";
@@ -43,12 +43,17 @@ export function useAssistantManager({
   const askAIRef = useRef<Assistant | null>(null);
 
   const askAI = useMemo(() => {
-    const newAssistant = selectedAssistant ?? quickAiAccessAssistant;
-    return newAssistant;
+    return selectedAssistant ?? quickAiAccessAssistant;
   }, [quickAiAccessAssistant, selectedAssistant]);
 
   const [assistantDetail, setAssistantDetail] = useState<any>({});
 
+  useEffect(() => {
+    if (goAskAi) return;
+
+    askAIRef.current = null;
+  }, [goAskAi]);
+
   const assistant_get = useCallback(async () => {
     if (!askAI?.id) return;
     if (disabledExtensions.includes("QuickAIAccess")) return;
@@ -75,7 +80,8 @@ export function useAssistantManager({
 
     if (disabledExtensions.includes("QuickAIAccess")) return;
 
-    askAIRef.current = cloneDeep(askAI);
+    askAIRef.current ??= cloneDeep(askAI);
+
     if (!askAIRef.current) return;
 
     let value = inputValue.trim();
