000

frontend-web/src/pages/StudioNotebookDetailPage.tsx

@@ -1,6 +1,6 @@
 "use client";
 
-import { useEffect, useMemo, useRef, useState } from "react";
+import { useEffect, useMemo, useState } from "react";
 import { useParams } from "react-router-dom";
 import {
   AlignLeft,
@@ -191,7 +191,7 @@ export default function StudioNotebookDetailPage() {
   const queryClient = useQueryClient();
   const [sourcesOpen, setSourcesOpen] = useState(false);
   const [selectedSourceIds, setSelectedSourceIds] = useState<number[]>([]);
-  const sourcesInitializedRef = useRef(false);
+  const [initializedForNotebook, setInitializedForNotebook] = useState<number | null>(null);
   const [selectedModel, setSelectedModel] = useState<string>("");
   const [input, setInput] = useState("");
   // null = 显示历史线程列表；string = 显示该线程的聊天
@@ -211,11 +211,11 @@ export default function StudioNotebookDetailPage() {
   const { sources, isLoading: isLoadingSources } = useNotebookSources(notebook?.id ?? null);
 
   useEffect(() => {
-    if (sources.length > 0 && !sourcesInitializedRef.current) {
-      sourcesInitializedRef.current = true;
+    if (sources.length > 0 && initializedForNotebook !== notebook?.id) {
+      setInitializedForNotebook(notebook?.id ?? null);
       setSelectedSourceIds(sources.map((s) => s.id));
     }
-  }, [sources]);
+  }, [sources, notebook?.id, initializedForNotebook]);
   const citationRehypePlugins = useMemo(() => {
     const mapped = sources.map((s) => ({
       id: s.id,