222

frontend-web/src/components/workspace/workspace-nav-chat-list.tsx

@@ -165,7 +165,7 @@ export function WorkspaceNavChatList({
         )}
       </SidebarMenu>
     </SidebarGroup>
-    {/* {!showCollapseTrigger && <WorkspaceNavStudio />} */}
+    {!showCollapseTrigger && <WorkspaceNavStudio />}
     </>
   );
 }

frontend-web/src/pages/StudioNotebookDetailPage.tsx

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