diff --git a/ui/web/src/pages/memory/hooks/use-knowledge-graph.ts b/ui/web/src/pages/memory/hooks/use-knowledge-graph.ts index db27cc97..5ac1d33d 100644 --- a/ui/web/src/pages/memory/hooks/use-knowledge-graph.ts +++ b/ui/web/src/pages/memory/hooks/use-knowledge-graph.ts @@ -141,7 +141,7 @@ export function useKGGraph(agentId: string, userId?: string) { queryKey: queryKeys.kg.graph(agentId, userId), queryFn: async () => { if (!agentId) return { entities: [], relations: [] }; - const params: Record = { limit: "200" }; + const params: Record = { limit: "50" }; if (userId) params.user_id = userId; return http.get(`/v1/agents/${agentId}/kg/graph`, params); }, diff --git a/ui/web/src/pages/memory/kg-graph-view.tsx b/ui/web/src/pages/memory/kg-graph-view.tsx index d2ab87d3..1940330b 100644 --- a/ui/web/src/pages/memory/kg-graph-view.tsx +++ b/ui/web/src/pages/memory/kg-graph-view.tsx @@ -179,9 +179,10 @@ function KGGraphViewInner({ entities: allEntities, relations: allRelations, onEn useEffect(() => { if (dataKey === layoutKey) { // Theme change only — update node colors without re-layout + const freshMap = new Map(rawNodes.map((r) => [r.id, r.data])); setNodes((prev) => prev.map((n) => { - const fresh = rawNodes.find((r) => r.id === n.id); - return fresh ? { ...n, data: fresh.data } : n; + const data = freshMap.get(n.id); + return data ? { ...n, data } : n; })); return; } @@ -207,7 +208,7 @@ function KGGraphViewInner({ entities: allEntities, relations: allRelations, onEn if (!selectedNodeId) { setEdges((eds) => eds.map((e) => ({ ...e, label: undefined, animated: false, - style: { stroke: "#334155", strokeWidth: 1, opacity: 0.4 }, + style: { stroke: "#64748b", strokeWidth: 2, opacity: 0.6 }, labelStyle: undefined, labelBgStyle: undefined, labelBgPadding: undefined, labelShowBg: undefined, }))); return; @@ -261,7 +262,7 @@ function KGGraphViewInner({ entities: allEntities, relations: allRelations, onEn onNodesChange={onNodesChange} onEdgesChange={onEdgesChange} onNodeClick={handleNodeClick} onPaneClick={handlePaneClick} nodeTypes={nodeTypes} colorMode={colorMode} - fitView minZoom={0.1} maxZoom={3} + minZoom={0.1} maxZoom={3} nodesConnectable={false} nodesDraggable={true} elementsSelectable={true} proOptions={{ hideAttribution: true }} >