src/pages/views/workspace/WorkspaceInput.tsx

@@ -9,7 +9,6 @@ import React, {
   useRef,
   useEffect,
   useImperativeHandle,
-  useMemo,
   forwardRef,
 } from 'react';
 import { ProviderMultiSelect } from '../../../components/ProviderSelector';
@@ -70,8 +69,6 @@ interface WorkspaceInputProps {
   workflowId?: string | null;
   loadedAttachedDataSourceIds?: string[];
   loadedAttachedFeatureDataSourceIds?: string[];
-  loadedDataSourceLabels?: Record<string, string>;
-  loadedFeatureDataSourceLabels?: Record<string, string>;
   loadedNonce?: number;
 }
 
@@ -138,8 +135,6 @@ export const WorkspaceInput = forwardRef<WorkspaceInputHandle, WorkspaceInputPro
   workflowId,
   loadedAttachedDataSourceIds,
   loadedAttachedFeatureDataSourceIds,
-  loadedDataSourceLabels,
-  loadedFeatureDataSourceLabels,
   loadedNonce,
 }, ref) {
   const { t } = useLanguage();
@@ -159,34 +154,6 @@ export const WorkspaceInput = forwardRef<WorkspaceInputHandle, WorkspaceInputPro
   const [neutralizeActive, setNeutralizeActive] = useState(false);
   const textareaRef = useRef<HTMLTextAreaElement>(null);
   const dsLabelCache = useRef<Map<string, string>>(new Map());
-  const dsLabelMap = useMemo(() => {
-    const m = new Map<string, string>();
-    if (loadedDataSourceLabels) {
-      for (const [k, v] of Object.entries(loadedDataSourceLabels)) {
-        if (v) m.set(k, v);
-      }
-    }
-    dsLabelCache.current.forEach((v, k) => { if (!m.has(k)) m.set(k, v); });
-    for (const ds of dataSources) {
-      const lbl = ds.label || ds.path;
-      if (lbl) m.set(ds.id, lbl);
-    }
-    return m;
-  }, [dataSources, loadedDataSourceLabels]);
-
-  const fdsLabelMap = useMemo(() => {
-    const m = new Map<string, string>();
-    if (loadedFeatureDataSourceLabels) {
-      for (const [k, v] of Object.entries(loadedFeatureDataSourceLabels)) {
-        if (v) m.set(k, v);
-      }
-    }
-    for (const fds of featureDataSources) {
-      const lbl = fds.label || fds.tableName;
-      if (lbl) m.set(fds.id, lbl);
-    }
-    return m;
-  }, [featureDataSources, loadedFeatureDataSourceLabels]);
 
   const _appendAttachment = useCallback((item: AttachmentItem) => {
     setAttachments(prev => prev.some(a => a.id === item.id) ? prev : [...prev, item]);
@@ -267,32 +234,26 @@ export const WorkspaceInput = forwardRef<WorkspaceInputHandle, WorkspaceInputPro
   useEffect(() => {
     if (loadedNonce === undefined) return;
     if (_reconciledDsForNonce.current === loadedNonce) return;
-    if (dataSources.length === 0 && !loadedDataSourceLabels) return;
+    if (dataSources.length === 0) return;
     _reconciledDsForNonce.current = loadedNonce;
-    for (const ds of dataSources) {
-      const lbl = ds.label || ds.path;
-      if (lbl) dsLabelCache.current.set(ds.id, lbl);
-    }
     const validIds = new Set(dataSources.map(d => d.id));
-    const labeledIds = new Set(Object.keys(loadedDataSourceLabels || {}));
     setAttachedDataSourceIds(prev => {
-      const filtered = prev.filter(id => validIds.has(id) || labeledIds.has(id));
+      const filtered = prev.filter(id => validIds.has(id));
       return filtered.length === prev.length ? prev : filtered;
     });
-  }, [loadedNonce, dataSources, loadedDataSourceLabels]);
+  }, [loadedNonce, dataSources]);
 
   useEffect(() => {
     if (loadedNonce === undefined) return;
     if (_reconciledFdsForNonce.current === loadedNonce) return;
-    if (featureDataSources.length === 0 && !loadedFeatureDataSourceLabels) return;
+    if (featureDataSources.length === 0) return;
     _reconciledFdsForNonce.current = loadedNonce;
     const validIds = new Set(featureDataSources.map(d => d.id));
-    const labeledIds = new Set(Object.keys(loadedFeatureDataSourceLabels || {}));
     setAttachedFeatureDataSourceIds(prev => {
-      const filtered = prev.filter(id => validIds.has(id) || labeledIds.has(id));
+      const filtered = prev.filter(id => validIds.has(id));
       return filtered.length === prev.length ? prev : filtered;
     });
-  }, [loadedNonce, featureDataSources, loadedFeatureDataSourceLabels]);
+  }, [loadedNonce, featureDataSources]);
 
   const promptBeforeVoiceRef = useRef('');
   const finalizedTextRef = useRef('');
@@ -702,7 +663,7 @@ export const WorkspaceInput = forwardRef<WorkspaceInputHandle, WorkspaceInputPro
                   background: '#e8f5e9', color: '#2e7d32', fontWeight: 500,
                 }}
               >
-                🔗 {ds?.label || ds?.path || dsLabelMap.get(dsId) || dsId}
+                🔗 {ds?.label || ds?.path || dsLabelCache.current.get(dsId) || dsId}
                 <button
                   type="button"
                   onClick={() => _removeAttachedDataSource(dsId)}
@@ -729,7 +690,7 @@ export const WorkspaceInput = forwardRef<WorkspaceInputHandle, WorkspaceInputPro
                 }}
               >
                 <span style={{ display: 'flex', alignItems: 'center', fontSize: 12 }}>{fdsIcon || '\uD83D\uDDC3\uFE0F'}</span>
-                {fds?.label || fdsLabelMap.get(fdsId) || fdsId} – {fds?.tableName || ''}
+                {fds?.label || fdsId} – {fds?.tableName || ''}
                 <button
                   type="button"
                   onClick={() => _toggleFeatureDataSource(fdsId)}

src/pages/views/workspace/WorkspacePage.tsx

@@ -571,8 +571,6 @@ export const WorkspacePage: React.FC<WorkspacePageProps> = ({ persistentInstance
           workflowId={workspace.workflowId}
           loadedAttachedDataSourceIds={workspace.loadedAttachedDataSourceIds}
           loadedAttachedFeatureDataSourceIds={workspace.loadedAttachedFeatureDataSourceIds}
-          loadedDataSourceLabels={workspace.loadedDataSourceLabels}
-          loadedFeatureDataSourceLabels={workspace.loadedFeatureDataSourceLabels}
           loadedNonce={workspace.loadedNonce}
         />
       </main>

src/pages/views/workspace/useWorkspace.ts

@@ -120,8 +120,6 @@ interface UseWorkspaceReturn {
    */
   loadedAttachedDataSourceIds: string[];
   loadedAttachedFeatureDataSourceIds: string[];
-  loadedDataSourceLabels: Record<string, string>;
-  loadedFeatureDataSourceLabels: Record<string, string>;
   loadedNonce: number;
 }
 
@@ -140,8 +138,6 @@ export function useWorkspace(instanceId: string): UseWorkspaceReturn {
   const [dataSourceAccesses, setDataSourceAccesses] = useState<DataSourceAccessEvent[]>([]);
   const [loadedAttachedDataSourceIds, setLoadedAttachedDataSourceIds] = useState<string[]>([]);
   const [loadedAttachedFeatureDataSourceIds, setLoadedAttachedFeatureDataSourceIds] = useState<string[]>([]);
-  const [loadedDataSourceLabels, setLoadedDataSourceLabels] = useState<Record<string, string>>({});
-  const [loadedFeatureDataSourceLabels, setLoadedFeatureDataSourceLabels] = useState<Record<string, string>>({});
   const [loadedNonce, setLoadedNonce] = useState(0);
   const cleanupRef = useRef<(() => void) | null>(null);
 
@@ -196,12 +192,8 @@ export function useWorkspace(instanceId: string): UseWorkspaceReturn {
         const fdsIds: string[] = Array.isArray(res.data.attachedFeatureDataSourceIds)
           ? res.data.attachedFeatureDataSourceIds.map((x: unknown) => String(x))
           : [];
-        const dsLabels: Record<string, string> = res.data.attachedDataSourceLabels || {};
-        const fdsLabels: Record<string, string> = res.data.attachedFeatureDataSourceLabels || {};
         setLoadedAttachedDataSourceIds(dsIds);
         setLoadedAttachedFeatureDataSourceIds(fdsIds);
-        setLoadedDataSourceLabels(dsLabels);
-        setLoadedFeatureDataSourceLabels(fdsLabels);
         setLoadedNonce(n => n + 1);
       })
       .catch(() => {});
@@ -216,8 +208,6 @@ export function useWorkspace(instanceId: string): UseWorkspaceReturn {
     setDataSourceAccesses([]);
     setLoadedAttachedDataSourceIds([]);
     setLoadedAttachedFeatureDataSourceIds([]);
-    setLoadedDataSourceLabels({});
-    setLoadedFeatureDataSourceLabels({});
     setLoadedNonce(n => n + 1);
   }, []);
 
@@ -538,8 +528,6 @@ export function useWorkspace(instanceId: string): UseWorkspaceReturn {
     dataSourceAccesses,
     loadedAttachedDataSourceIds,
     loadedAttachedFeatureDataSourceIds,
-    loadedDataSourceLabels,
-    loadedFeatureDataSourceLabels,
     loadedNonce,
   };
 }