@inkeep/agents-manage-ui 0.18.1 → 0.19.0

package/src/features/{graph → agent}/domain/__tests__/serialize.test.ts

@@ -1,16 +1,16 @@
 import type { Edge, Node } from '@xyflow/react';
-import { EdgeType } from '@/components/graph/configuration/edge-types';
-import type { AgentNodeData, MCPNodeData } from '@/components/graph/configuration/node-types';
-import { NodeType } from '@/components/graph/configuration/node-types';
-import { serializeGraphData } from '../serialize';
+import { EdgeType } from '@/components/agent/configuration/edge-types';
+import type { AgentNodeData, MCPNodeData } from '@/components/agent/configuration/node-types';
+import { NodeType } from '@/components/agent/configuration/node-types';
+import { serializeAgentData } from '../serialize';
 
-describe('serializeGraphData', () => {
+describe('serializeAgentData', () => {
   describe('models object processing', () => {
     it('should set models to undefined when models object has only empty values', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -26,7 +26,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toBeUndefined();
     });
@@ -35,7 +35,7 @@ describe('serializeGraphData', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -51,7 +51,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toBeUndefined();
     });
@@ -60,7 +60,7 @@ describe('serializeGraphData', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -76,7 +76,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toEqual({
         base: { model: 'gpt-4' },
@@ -89,7 +89,7 @@ describe('serializeGraphData', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -105,7 +105,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toEqual({
         base: undefined,
@@ -118,7 +118,7 @@ describe('serializeGraphData', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -134,7 +134,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toEqual({
         base: undefined,
@@ -147,7 +147,7 @@ describe('serializeGraphData', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -163,7 +163,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toEqual({
         base: { model: 'gpt-4' },
@@ -176,7 +176,7 @@ describe('serializeGraphData', () => {
       const nodes: Node<AgentNodeData>[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -188,7 +188,7 @@ describe('serializeGraphData', () => {
       ];
       const edges: Edge[] = [];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).models).toBeUndefined();
     });
@@ -199,7 +199,7 @@ describe('serializeGraphData', () => {
       const nodes: Node[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -228,7 +228,7 @@ describe('serializeGraphData', () => {
         },
       ];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).canUse).toBeDefined();
       expect((result.subAgents.agent1 as any).canUse).toHaveLength(1);
@@ -243,7 +243,7 @@ describe('serializeGraphData', () => {
       const nodes: Node[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -273,7 +273,7 @@ describe('serializeGraphData', () => {
         },
       ];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       // When tempSelectedTools is null, all tools should be selected (toolSelection: null)
       expect((result.subAgents.agent1 as any).canUse).toBeDefined();
@@ -289,7 +289,7 @@ describe('serializeGraphData', () => {
       const nodes: Node[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -318,7 +318,7 @@ describe('serializeGraphData', () => {
         },
       ];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       expect((result.subAgents.agent1 as any).canUse).toBeDefined();
       expect((result.subAgents.agent1 as any).canUse).toHaveLength(1);
@@ -333,7 +333,7 @@ describe('serializeGraphData', () => {
       const nodes: Node[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -362,7 +362,7 @@ describe('serializeGraphData', () => {
         },
       ];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       // selectedTools should not be created if tempSelectedTools is undefined
       expect((result.subAgents.agent1 as any).selectedTools).toBeUndefined();
@@ -372,7 +372,7 @@ describe('serializeGraphData', () => {
       const nodes: Node[] = [
         {
           id: 'agent1',
-          type: NodeType.Agent,
+          type: NodeType.SubAgent,
           position: { x: 0, y: 0 },
           data: {
             id: 'agent1',
@@ -403,7 +403,7 @@ describe('serializeGraphData', () => {
         },
       ];
 
-      const result = serializeGraphData(nodes, edges, undefined, {}, {}, {});
+      const result = serializeAgentData(nodes, edges, undefined, {}, {}, {});
 
       // When tempSelectedTools is undefined and there's an edge to MCP tool,
       // the toolSelection will be null (all tools selected by default)

package/src/features/agent/domain/agent-initializer-helpers.ts

@@ -0,0 +1,61 @@
+import type { ContextConfig, AgentMetadata } from '@/components/agent/configuration/agent-types';
+import type { FullAgentDefinition } from '@/lib/types/agent-full';
+import { formatJsonField } from '@/lib/utils';
+
+export type ExtendedFullAgentDefinition = FullAgentDefinition & {
+  contextConfig?: Partial<Pick<ContextConfig, 'id'>> & {
+    contextVariables?: Record<string, any>;
+    headersSchema?: Record<string, any>;
+  };
+};
+
+/**
+ * Extracts and formats agent metadata from a FullAgentDefinition object.
+ * This helper function handles the complex transformation of the agent data
+ * into the format expected by the AgentMetadata type, including proper
+ * JSON field formatting for form compatibility.
+ */
+export function extractAgentMetadata(
+  agent: ExtendedFullAgentDefinition | null | undefined
+): AgentMetadata {
+  return {
+    id: agent?.id,
+    name: agent?.name ?? '',
+    description: agent?.description ?? '',
+    agentPrompt: agent?.agentPrompt,
+    models: agent?.models
+      ? {
+          base: agent.models.base
+            ? {
+                model: agent.models.base.model,
+                providerOptions: formatJsonField(agent.models.base.providerOptions),
+              }
+            : undefined,
+          structuredOutput: agent.models.structuredOutput
+            ? {
+                model: agent.models.structuredOutput.model,
+                providerOptions: formatJsonField(agent.models.structuredOutput.providerOptions),
+              }
+            : undefined,
+          summarizer: agent.models.summarizer
+            ? {
+                model: agent.models.summarizer.model,
+                providerOptions: formatJsonField(agent.models.summarizer.providerOptions),
+              }
+            : undefined,
+        }
+      : undefined,
+    stopWhen: agent?.stopWhen,
+    statusUpdates: agent?.statusUpdates
+      ? {
+          ...agent.statusUpdates,
+          statusComponents: formatJsonField(agent.statusUpdates.statusComponents) || '',
+        }
+      : undefined,
+    contextConfig: {
+      id: agent?.contextConfig?.id ?? '',
+      contextVariables: formatJsonField(agent?.contextConfig?.contextVariables) || '',
+      headersSchema: formatJsonField(agent?.contextConfig?.headersSchema) || '',
+    },
+  };
+}

package/src/features/{graph → agent}/domain/deserialize.ts

@@ -1,7 +1,7 @@
 import type { Edge, Node } from '@xyflow/react';
 import * as dagre from 'dagre';
 import { nanoid } from 'nanoid';
-import { EdgeType } from '@/components/graph/configuration/edge-types';
+import { EdgeType } from '@/components/agent/configuration/edge-types';
 import {
   agentNodeSourceHandleId,
   agentNodeTargetHandleId,
@@ -9,12 +9,12 @@ import {
   functionToolNodeHandleId,
   mcpNodeHandleId,
   NodeType,
-} from '@/components/graph/configuration/node-types';
+} from '@/components/agent/configuration/node-types';
 import type {
   ExternalAgentDefinition,
-  FullGraphDefinition,
+  FullAgentDefinition,
   InternalAgentDefinition,
-} from '@/lib/types/graph-full';
+} from '@/lib/types/agent-full';
 import { formatJsonField } from '@/lib/utils';
 
 interface TransformResult {
@@ -31,7 +31,7 @@ function calculateNodeHeight(node: Node): number {
   let height = MIN_NODE_HEIGHT;
 
   // Agent and External Agent nodes have dynamic height
-  if (node.type === NodeType.Agent || node.type === NodeType.ExternalAgent) {
+  if (node.type === NodeType.SubAgent || node.type === NodeType.ExternalAgent) {
     const data = node.data as any;
 
     // Add height for description if it exists
@@ -101,7 +101,7 @@ export function applyDagreLayout(nodes: Node[], edges: Edge[]): Node[] {
   });
 }
 
-export function deserializeGraphData(data: FullGraphDefinition): TransformResult {
+export function deserializeAgentData(data: FullAgentDefinition): TransformResult {
   const nodes: Node[] = [];
   const edges: Edge[] = [];
 
@@ -111,7 +111,7 @@ export function deserializeGraphData(data: FullGraphDefinition): TransformResult
     const isDefault = subAgentId === data.defaultSubAgentId;
     const isExternal = agent.type === 'external';
 
-    const nodeType = isExternal ? NodeType.ExternalAgent : NodeType.Agent;
+    const nodeType = isExternal ? NodeType.ExternalAgent : NodeType.SubAgent;
     const agentNodeData = isExternal
       ? {
           id: agent.id,

package/src/features/{graph → agent}/domain/index.ts

@@ -1,3 +1,3 @@
 export * from './deserialize';
-export * from './graph-initializer-helpers';
+export * from './agent-initializer-helpers';
 export * from './serialize';

package/src/features/{graph → agent}/domain/serialize.ts

@@ -1,13 +1,13 @@
 import type { Edge, Node } from '@xyflow/react';
 import { nanoid } from 'nanoid';
-import type { A2AEdgeData } from '@/components/graph/configuration/edge-types';
-import { EdgeType } from '@/components/graph/configuration/edge-types';
-import type { GraphMetadata } from '@/components/graph/configuration/graph-types';
-import { NodeType } from '@/components/graph/configuration/node-types';
-import type { AgentToolConfigLookup } from '@/components/graph/graph';
+import type { AgentToolConfigLookup } from '@/components/agent/agent';
+import type { AgentMetadata } from '@/components/agent/configuration/agent-types';
+import type { A2AEdgeData } from '@/components/agent/configuration/edge-types';
+import { EdgeType } from '@/components/agent/configuration/edge-types';
+import { NodeType } from '@/components/agent/configuration/node-types';
 import type { ArtifactComponent } from '@/lib/api/artifact-components';
 import type { DataComponent } from '@/lib/api/data-components';
-import type { FullGraphDefinition, InternalAgentDefinition } from '@/lib/types/graph-full';
+import type { FullAgentDefinition, InternalAgentDefinition } from '@/lib/types/agent-full';
 
 // Use the exported InternalAgentDefinition from core
 type InternalAgent = InternalAgentDefinition;
@@ -26,19 +26,19 @@ export type ExtendedAgent =
   | (InternalAgent & {
       dataComponents: string[];
       artifactComponents: string[];
-      models?: GraphMetadata['models'];
+      models?: AgentMetadata['models'];
       type: 'internal';
     })
   | ExternalAgent;
 
 // Type guard to check if an agent is an internal agent
 function isInternalAgent(
-  agent: ExtendedAgent | FullGraphDefinition['subAgents'][string]
+  agent: ExtendedAgent | FullAgentDefinition['subAgents'][string]
 ): agent is InternalAgent {
   return agent.type === 'internal' && 'canUse' in agent;
 }
 
-// Note: Tools are now project-scoped, not part of FullGraphDefinition
+// Note: Tools are now project-scoped, not part of FullAgentDefinition
 
 /**
  * Safely parse a JSON string, returning undefined if parsing fails or input is falsy
@@ -53,7 +53,7 @@ function safeJsonParse(jsonString: string | undefined | null): any {
     return undefined;
   }
 }
-function processModels(modelsData: GraphMetadata['models']): GraphMetadata['models'] | undefined {
+function processModels(modelsData: AgentMetadata['models']): AgentMetadata['models'] | undefined {
   if (modelsData && typeof modelsData === 'object') {
     const hasNonEmptyValue = Object.values(modelsData).some(
       (value) => value !== null && value !== undefined && String(value).trim() !== ''
@@ -87,24 +87,24 @@ function processModels(modelsData: GraphMetadata['models']): GraphMetadata['mode
 /**
  * Transforms React Flow nodes and edges back into the API data structure
  */
-export function serializeGraphData(
+export function serializeAgentData(
   nodes: Node[],
   edges: Edge[],
-  metadata?: GraphMetadata,
+  metadata?: AgentMetadata,
   dataComponentLookup?: Record<string, DataComponent>,
   artifactComponentLookup?: Record<string, ArtifactComponent>,
   agentToolConfigLookup?: AgentToolConfigLookup
-): FullGraphDefinition {
+): FullAgentDefinition {
   const subAgents: Record<string, ExtendedAgent> = {};
   const functionTools: Record<string, any> = {};
   const functions: Record<string, any> = {};
-  // Note: Tools are now project-scoped and not included in graph serialization
+  // Note: Tools are now project-scoped and not included in agent serialization
   const usedDataComponents = new Set<string>();
   const usedArtifactComponents = new Set<string>();
   let defaultSubAgentId = '';
 
   for (const node of nodes) {
-    if (node.type === NodeType.Agent) {
+    if (node.type === NodeType.SubAgent) {
       const subAgentId = (node.data.id as string) || node.id;
       const subAgentDataComponents = (node.data.dataComponents as string[]) || [];
       const subAgentArtifactComponents = (node.data.artifactComponents as string[]) || [];
@@ -116,7 +116,7 @@ export function serializeGraphData(
         usedArtifactComponents.add(componentId);
       });
       // Process models - only include if it has non-empty, non-whitespace values
-      const modelsData = node.data.models as GraphMetadata['models'] | undefined;
+      const modelsData = node.data.models as AgentMetadata['models'] | undefined;
       const processedModels = processModels(modelsData);
 
       const stopWhen = (node.data as any).stopWhen;
@@ -383,20 +383,20 @@ export function serializeGraphData(
     });
   }
 
-  const result: FullGraphDefinition = {
+  const result: FullAgentDefinition = {
     id: metadata?.id || nanoid(),
-    name: metadata?.name || 'Untitled Graph',
+    name: metadata?.name || 'Untitled Agent',
     description: metadata?.description || undefined,
     defaultSubAgentId,
     subAgents: subAgents,
     ...(Object.keys(functionTools).length > 0 && { functionTools }),
     ...(Object.keys(functions).length > 0 && { functions }),
-    // Note: Tools are now project-scoped and not included in FullGraphDefinition
+    // Note: Tools are now project-scoped and not included in FullAgentDefinition
     // ...(Object.keys(dataComponents).length > 0 && { dataComponents }),
     // ...(Object.keys(artifactComponents).length > 0 && { artifactComponents }),
   };
 
-  // Add new graph-level fields
+  // Add new agent-level fields
   if (metadata?.models) {
     (result as any).models = {
       base: metadata.models.base
@@ -424,8 +424,8 @@ export function serializeGraphData(
     (result as any).stopWhen = metadata.stopWhen;
   }
 
-  if (metadata?.graphPrompt) {
-    (result as any).graphPrompt = metadata.graphPrompt;
+  if (metadata?.agentPrompt) {
+    (result as any).agentPrompt = metadata.agentPrompt;
   }
 
   if (metadata?.statusUpdates) {
@@ -459,7 +459,7 @@ interface StructuredValidationError {
 }
 
 export function validateSerializedData(
-  data: FullGraphDefinition,
+  data: FullAgentDefinition,
   functionToolNodeMap?: Map<string, string>
 ): StructuredValidationError[] {
   const errors: StructuredValidationError[] = [];

package/src/features/{graph/state/use-graph-store.ts → agent/state/use-agent-store.ts}

@@ -3,20 +3,20 @@ import { addEdge, applyEdgeChanges, applyNodeChanges } from '@xyflow/react';
 import { create } from 'zustand';
 import { devtools } from 'zustand/middleware';
 import { useShallow } from 'zustand/react/shallow';
-import type { GraphMetadata } from '@/components/graph/configuration/graph-types';
-import { mcpNodeHandleId, NodeType } from '@/components/graph/configuration/node-types';
-import type { AgentToolConfigLookup } from '@/components/graph/graph';
+import type { AgentMetadata } from '@/components/agent/configuration/agent-types';
+import { mcpNodeHandleId, NodeType } from '@/components/agent/configuration/node-types';
+import type { AgentToolConfigLookup } from '@/components/agent/agent';
 import type { ArtifactComponent } from '@/lib/api/artifact-components';
 import type { DataComponent } from '@/lib/api/data-components';
 import type { MCPTool } from '@/lib/types/tools';
-import type { GraphErrorSummary } from '@/lib/utils/graph-error-parser';
+import type { AgentErrorSummary } from '@/lib/utils/agent-error-parser';
 
 type HistoryEntry = { nodes: Node[]; edges: Edge[] };
 
-type GraphStateData = {
+type AgentStateData = {
   nodes: Node[];
   edges: Edge[];
-  metadata: GraphMetadata;
+  metadata: AgentMetadata;
   dataComponentLookup: Record<string, DataComponent>;
   artifactComponentLookup: Record<string, ArtifactComponent>;
   toolLookup: Record<string, MCPTool>;
@@ -24,15 +24,15 @@ type GraphStateData = {
   dirty: boolean;
   history: HistoryEntry[];
   future: HistoryEntry[];
-  errors: GraphErrorSummary | null;
+  errors: AgentErrorSummary | null;
   showErrors: boolean;
 };
 
-type GraphActions = {
+type AgentActions = {
   setInitial(
     nodes: Node[],
     edges: Edge[],
-    metadata: GraphMetadata,
+    metadata: AgentMetadata,
     dataComponentLookup?: Record<string, DataComponent>,
     artifactComponentLookup?: Record<string, ArtifactComponent>,
     toolLookup?: Record<string, MCPTool>,
@@ -47,7 +47,7 @@ type GraphActions = {
   onNodesChange(changes: NodeChange[]): void;
   onEdgesChange(changes: EdgeChange[]): void;
   onConnect(connection: Connection): void;
-  setMetadata<K extends keyof GraphMetadata>(field: K, value: GraphMetadata[K]): void;
+  setMetadata<K extends keyof AgentMetadata>(field: K, value: AgentMetadata[K]): void;
   push(nodes: Node[], edges: Edge[]): void;
   undo(): void;
   redo(): void;
@@ -55,19 +55,19 @@ type GraphActions = {
   markUnsaved(): void;
   clearSelection(): void;
   deleteSelected(): void;
-  setErrors(errors: GraphErrorSummary | null): void;
+  setErrors(errors: AgentErrorSummary | null): void;
   clearErrors(): void;
   setShowErrors(show: boolean): void;
   hasErrors(): boolean;
-  getNodeErrors(nodeId: string): GraphErrorSummary['allErrors'];
-  getEdgeErrors(edgeId: string): GraphErrorSummary['allErrors'];
+  getNodeErrors(nodeId: string): AgentErrorSummary['allErrors'];
+  getEdgeErrors(edgeId: string): AgentErrorSummary['allErrors'];
 };
 
-type GraphState = GraphStateData & {
-  actions: GraphActions;
+type AgentState = AgentStateData & {
+  actions: AgentActions;
 };
 
-export const graphStore = create<GraphState>()(
+export const agentStore = create<AgentState>()(
   devtools((set, get) => ({
     nodes: [],
     edges: [],
@@ -81,7 +81,7 @@ export const graphStore = create<GraphState>()(
       },
       models: undefined,
       stopWhen: undefined,
-      graphPrompt: undefined,
+      agentPrompt: undefined,
       statusUpdates: undefined,
     },
     dataComponentLookup: {},
@@ -144,7 +144,7 @@ export const graphStore = create<GraphState>()(
         }));
       },
       onNodesChange(changes) {
-        // Check if any change type would modify the graph (not just selection changes)
+        // Check if any change type would modify the agent (not just selection changes)
         const hasModifyingChange = changes.some(
           // Don't trigger `position` as modified change, since when the nodes are repositioned,
           // they'll be re-laid out during the initial load anyway
@@ -158,7 +158,7 @@ export const graphStore = create<GraphState>()(
         }));
       },
       onEdgesChange(changes) {
-        // Check if any change type would modify the graph (not just selection changes)
+        // Check if any change type would modify the agent (not just selection changes)
         const hasModifyingChange = changes.some(
           (change) => change.type === 'remove' || change.type === 'add' || change.type === 'replace'
         );
@@ -284,15 +284,15 @@ export const graphStore = create<GraphState>()(
  *
  * @see https://tkdodo.eu/blog/working-with-zustand#separate-actions-from-state
  */
-export const useGraphActions = () => graphStore((state) => state.actions);
+export const useAgentActions = () => agentStore((state) => state.actions);
 
 /**
- * Select values from the graph store (excluding actions).
+ * Select values from the agent store (excluding actions).
  *
- * We explicitly use `GraphStateData` instead of `GraphState`,
- * which includes actions, to encourage using `useGraphActions`
+ * We explicitly use `AgentStateData` instead of `AgentState`,
+ * which includes actions, to encourage using `useAgentActions`
  * when accessing or calling actions.
  */
-export function useGraphStore<T>(selector: (state: GraphStateData) => T): T {
-  return graphStore(useShallow(selector));
+export function useAgentStore<T>(selector: (state: AgentStateData) => T): T {
+  return agentStore(useShallow(selector));
 }

package/src/features/{graph/ui/use-graph-shortcuts.ts → agent/ui/use-agent-shortcuts.ts}

@@ -1,8 +1,8 @@
 import { useCallback, useEffect } from 'react';
-import { useGraphActions } from '@/features/graph/state/use-graph-store';
+import { useAgentActions } from '@/features/agent/state/use-agent-store';
 
-export function useGraphShortcuts() {
-  const { undo, redo, deleteSelected } = useGraphActions();
+export function useAgentShortcuts() {
+  const { undo, redo, deleteSelected } = useAgentActions();
 
   const onKeyDown = useCallback(
     (e: KeyboardEvent) => {

package/src/hooks/{use-graph-errors.ts → use-agent-errors.ts}

@@ -1,9 +1,9 @@
 /**
- * Hook for accessing graph error state and utilities
+ * Hook for accessing agent error state and utilities
  */
 
 import { useCallback } from 'react';
-import { useGraphStore, useGraphActions } from '@/features/graph/state/use-graph-store';
+import { useAgentStore, useAgentActions } from '@/features/agent/state/use-agent-store';
 
 export interface ErrorHelpers {
   hasFieldError: (fieldName: string) => boolean;
@@ -11,13 +11,13 @@ export interface ErrorHelpers {
   getFirstErrorField: () => string | undefined;
 }
 
-export function useGraphErrors() {
-  const { errors, showErrors } = useGraphStore((state) => ({
+export function useAgentErrors() {
+  const { errors, showErrors } = useAgentStore((state) => ({
     errors: state.errors,
     showErrors: state.showErrors,
   }));
   const { hasErrors, getNodeErrors, getEdgeErrors, setErrors, clearErrors, setShowErrors } =
-    useGraphActions();
+    useAgentActions();
 
   const getNodeErrorCount = (nodeId: string): number => {
     return getNodeErrors(nodeId).length;