@contractspec/module.ai-chat 4.0.2 → 4.1.0

package/dist/browser/presentation/hooks/index.js

@@ -8,8 +8,8 @@ var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require
 
 // src/presentation/hooks/useChat.tsx
 import * as React from "react";
-import { tool } from "ai";
-import { z } from "zod";
+import { tool as tool4 } from "ai";
+import { z as z4 } from "zod";
 
 // src/core/chat-service.ts
 import { generateText, streamText } from "ai";
@@ -91,11 +91,65 @@ class InMemoryConversationStore {
     if (options?.status) {
       results = results.filter((c) => c.status === options.status);
     }
+    if (options?.projectId) {
+      results = results.filter((c) => c.projectId === options.projectId);
+    }
+    if (options?.tags && options.tags.length > 0) {
+      const tagSet = new Set(options.tags);
+      results = results.filter((c) => c.tags && c.tags.some((t) => tagSet.has(t)));
+    }
     results.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
     const offset = options?.offset ?? 0;
     const limit = options?.limit ?? 100;
     return results.slice(offset, offset + limit);
   }
+  async fork(conversationId, upToMessageId) {
+    const source = this.conversations.get(conversationId);
+    if (!source) {
+      throw new Error(`Conversation ${conversationId} not found`);
+    }
+    let messagesToCopy = source.messages;
+    if (upToMessageId) {
+      const idx = source.messages.findIndex((m) => m.id === upToMessageId);
+      if (idx === -1) {
+        throw new Error(`Message ${upToMessageId} not found`);
+      }
+      messagesToCopy = source.messages.slice(0, idx + 1);
+    }
+    const now = new Date;
+    const forkedMessages = messagesToCopy.map((m) => ({
+      ...m,
+      id: generateId("msg"),
+      conversationId: "",
+      createdAt: new Date(m.createdAt),
+      updatedAt: new Date(m.updatedAt)
+    }));
+    const forked = {
+      ...source,
+      id: generateId("conv"),
+      title: source.title ? `${source.title} (fork)` : undefined,
+      forkedFromId: source.id,
+      createdAt: now,
+      updatedAt: now,
+      messages: forkedMessages
+    };
+    for (const m of forked.messages) {
+      m.conversationId = forked.id;
+    }
+    this.conversations.set(forked.id, forked);
+    return forked;
+  }
+  async truncateAfter(conversationId, messageId) {
+    const conv = this.conversations.get(conversationId);
+    if (!conv)
+      return null;
+    const idx = conv.messages.findIndex((m) => m.id === messageId);
+    if (idx === -1)
+      return null;
+    conv.messages = conv.messages.slice(0, idx + 1);
+    conv.updatedAt = new Date;
+    return conv;
+  }
   async search(query, limit = 20) {
     const lowerQuery = query.toLowerCase();
     const results = [];
@@ -121,7 +175,504 @@ function createInMemoryConversationStore() {
   return new InMemoryConversationStore;
 }
 
+// src/core/thinking-levels.ts
+var THINKING_LEVEL_LABELS = {
+  instant: "Instant",
+  thinking: "Thinking",
+  extra_thinking: "Extra Thinking",
+  max: "Max"
+};
+var THINKING_LEVEL_DESCRIPTIONS = {
+  instant: "Fast responses, minimal reasoning",
+  thinking: "Standard reasoning depth",
+  extra_thinking: "More thorough reasoning",
+  max: "Maximum reasoning depth"
+};
+function getProviderOptions(level, providerName) {
+  if (!level || level === "instant") {
+    return {};
+  }
+  switch (providerName) {
+    case "anthropic": {
+      const budgetMap = {
+        thinking: 8000,
+        extra_thinking: 16000,
+        max: 32000
+      };
+      return {
+        anthropic: {
+          thinking: { type: "enabled", budgetTokens: budgetMap[level] }
+        }
+      };
+    }
+    case "openai": {
+      const effortMap = {
+        thinking: "low",
+        extra_thinking: "medium",
+        max: "high"
+      };
+      return {
+        openai: {
+          reasoningEffort: effortMap[level]
+        }
+      };
+    }
+    case "ollama":
+    case "mistral":
+    case "gemini":
+      return {};
+    default:
+      return {};
+  }
+}
+
+// src/core/workflow-tools.ts
+import { tool } from "ai";
+import { z } from "zod";
+import {
+  WorkflowComposer,
+  validateExtension
+} from "@contractspec/lib.workflow-composer";
+var StepTypeSchema = z.enum(["human", "automation", "decision"]);
+var StepActionSchema = z.object({
+  operation: z.object({
+    name: z.string(),
+    version: z.number()
+  }).optional(),
+  form: z.object({
+    key: z.string(),
+    version: z.number()
+  }).optional()
+}).optional();
+var StepSchema = z.object({
+  id: z.string(),
+  type: StepTypeSchema,
+  label: z.string(),
+  description: z.string().optional(),
+  action: StepActionSchema
+});
+var StepInjectionSchema = z.object({
+  after: z.string().optional(),
+  before: z.string().optional(),
+  inject: StepSchema,
+  transitionTo: z.string().optional(),
+  transitionFrom: z.string().optional(),
+  when: z.string().optional()
+});
+var WorkflowExtensionInputSchema = z.object({
+  workflow: z.string(),
+  tenantId: z.string().optional(),
+  role: z.string().optional(),
+  priority: z.number().optional(),
+  customSteps: z.array(StepInjectionSchema).optional(),
+  hiddenSteps: z.array(z.string()).optional()
+});
+function createWorkflowTools(config) {
+  const { baseWorkflows, composer } = config;
+  const baseByKey = new Map(baseWorkflows.map((b) => [b.meta.key, b]));
+  const createWorkflowExtensionTool = tool({
+    description: "Create or validate a workflow extension. Use when the user asks to add steps, modify a workflow, or create a tenant-specific extension. The extension targets an existing base workflow.",
+    inputSchema: WorkflowExtensionInputSchema,
+    execute: async (input) => {
+      const extension = {
+        workflow: input.workflow,
+        tenantId: input.tenantId,
+        role: input.role,
+        priority: input.priority,
+        customSteps: input.customSteps,
+        hiddenSteps: input.hiddenSteps
+      };
+      const base = baseByKey.get(input.workflow);
+      if (!base) {
+        return {
+          success: false,
+          error: `Base workflow "${input.workflow}" not found. Available: ${Array.from(baseByKey.keys()).join(", ")}`,
+          extension
+        };
+      }
+      try {
+        validateExtension(extension, base);
+        return {
+          success: true,
+          message: "Extension validated successfully",
+          extension
+        };
+      } catch (err) {
+        return {
+          success: false,
+          error: err instanceof Error ? err.message : String(err),
+          extension
+        };
+      }
+    }
+  });
+  const composeWorkflowInputSchema = z.object({
+    workflowKey: z.string().describe("Base workflow meta.key"),
+    tenantId: z.string().optional(),
+    role: z.string().optional(),
+    extensions: z.array(WorkflowExtensionInputSchema).optional().describe("Extensions to register before composing")
+  });
+  const composeWorkflowTool = tool({
+    description: "Compose a workflow by applying registered extensions to a base workflow. Returns the composed WorkflowSpec.",
+    inputSchema: composeWorkflowInputSchema,
+    execute: async (input) => {
+      const base = baseByKey.get(input.workflowKey);
+      if (!base) {
+        return {
+          success: false,
+          error: `Base workflow "${input.workflowKey}" not found. Available: ${Array.from(baseByKey.keys()).join(", ")}`
+        };
+      }
+      const comp = composer ?? new WorkflowComposer;
+      if (input.extensions?.length) {
+        for (const ext of input.extensions) {
+          comp.register({
+            workflow: ext.workflow,
+            tenantId: ext.tenantId,
+            role: ext.role,
+            priority: ext.priority,
+            customSteps: ext.customSteps,
+            hiddenSteps: ext.hiddenSteps
+          });
+        }
+      }
+      try {
+        const composed = comp.compose({
+          base,
+          tenantId: input.tenantId,
+          role: input.role
+        });
+        return {
+          success: true,
+          workflow: composed,
+          meta: composed.meta,
+          stepIds: composed.definition.steps.map((s) => s.id)
+        };
+      } catch (err) {
+        return {
+          success: false,
+          error: err instanceof Error ? err.message : String(err)
+        };
+      }
+    }
+  });
+  const generateWorkflowSpecCodeInputSchema = z.object({
+    workflowKey: z.string().describe("Workflow meta.key"),
+    composedSteps: z.array(z.object({
+      id: z.string(),
+      type: z.enum(["human", "automation", "decision"]),
+      label: z.string(),
+      description: z.string().optional()
+    })).optional().describe("Steps to include; if omitted, uses the base workflow")
+  });
+  const generateWorkflowSpecCodeTool = tool({
+    description: "Generate TypeScript code for a workflow spec. Use after composing a workflow to output the spec as code the user can save.",
+    inputSchema: generateWorkflowSpecCodeInputSchema,
+    execute: async (input) => {
+      const base = baseByKey.get(input.workflowKey);
+      if (!base) {
+        return {
+          success: false,
+          error: `Base workflow "${input.workflowKey}" not found. Available: ${Array.from(baseByKey.keys()).join(", ")}`,
+          code: null
+        };
+      }
+      const steps = input.composedSteps ?? base.definition.steps;
+      const specVarName = toPascalCase((base.meta.key.split(".").pop() ?? "Workflow") + "") + "Workflow";
+      const stepsCode = steps.map((s) => `    {
+      id: '${s.id}',
+      type: '${s.type}',
+      label: '${escapeString(s.label)}',${s.description ? `
+      description: '${escapeString(s.description)}',` : ""}
+    }`).join(`,
+`);
+      const meta = base.meta;
+      const transitionsJson = JSON.stringify(base.definition.transitions, null, 6);
+      const code = `import type { WorkflowSpec } from '@contractspec/lib.contracts-spec/workflow';
+
+/**
+ * Workflow: ${base.meta.key}
+ * Generated via AI chat workflow tools.
+ */
+export const ${specVarName}: WorkflowSpec = {
+  meta: {
+    key: '${base.meta.key}',
+    version: '${String(base.meta.version)}',
+    title: '${escapeString(meta.title ?? base.meta.key)}',
+    description: '${escapeString(meta.description ?? "")}',
+  },
+  definition: {
+    entryStepId: '${base.definition.entryStepId ?? base.definition.steps[0]?.id ?? ""}',
+    steps: [
+${stepsCode}
+    ],
+    transitions: ${transitionsJson},
+  },
+};
+`;
+      return {
+        success: true,
+        code,
+        workflowKey: input.workflowKey
+      };
+    }
+  });
+  return {
+    create_workflow_extension: createWorkflowExtensionTool,
+    compose_workflow: composeWorkflowTool,
+    generate_workflow_spec_code: generateWorkflowSpecCodeTool
+  };
+}
+function toPascalCase(value) {
+  return value.split(/[-_.]/).filter(Boolean).map((part) => part.charAt(0).toUpperCase() + part.slice(1)).join("");
+}
+function escapeString(value) {
+  return value.replace(/\\/g, "\\\\").replace(/'/g, "\\'");
+}
+
+// src/core/contracts-context.ts
+function buildContractsContextPrompt(config) {
+  const parts = [];
+  if (!config.agentSpecs?.length && !config.dataViewSpecs?.length && !config.formSpecs?.length && !config.presentationSpecs?.length && !config.operationRefs?.length) {
+    return "";
+  }
+  parts.push(`
+
+## Available resources`);
+  if (config.agentSpecs?.length) {
+    parts.push(`
+### Agent tools`);
+    for (const agent of config.agentSpecs) {
+      const toolNames = agent.tools?.map((t) => t.name).join(", ") ?? "none";
+      parts.push(`- **${agent.key}**: tools: ${toolNames}`);
+    }
+  }
+  if (config.dataViewSpecs?.length) {
+    parts.push(`
+### Data views`);
+    for (const dv of config.dataViewSpecs) {
+      parts.push(`- **${dv.key}**: ${dv.meta.title ?? dv.key}`);
+    }
+  }
+  if (config.formSpecs?.length) {
+    parts.push(`
+### Forms`);
+    for (const form of config.formSpecs) {
+      parts.push(`- **${form.key}**: ${form.meta.title ?? form.key}`);
+    }
+  }
+  if (config.presentationSpecs?.length) {
+    parts.push(`
+### Presentations`);
+    for (const pres of config.presentationSpecs) {
+      parts.push(`- **${pres.key}**: ${pres.meta.title ?? pres.key} (targets: ${pres.targets?.join(", ") ?? "react"})`);
+    }
+  }
+  if (config.operationRefs?.length) {
+    parts.push(`
+### Operations`);
+    for (const op of config.operationRefs) {
+      parts.push(`- **${op.key}@${op.version}**`);
+    }
+  }
+  parts.push(`
+Use the available tools to invoke operations, query data views, or propose surface changes when appropriate.`);
+  return parts.join(`
+`);
+}
+
+// src/core/agent-tools-adapter.ts
+import { tool as tool2 } from "ai";
+import { z as z2 } from "zod";
+function getInputSchema(_schema) {
+  return z2.object({}).passthrough();
+}
+function agentToolConfigsToToolSet(configs, handlers) {
+  const result = {};
+  for (const config of configs) {
+    const handler = handlers?.[config.name];
+    const inputSchema = getInputSchema(config.schema);
+    result[config.name] = tool2({
+      description: config.description ?? config.name,
+      inputSchema,
+      execute: async (input) => {
+        if (!handler) {
+          return {
+            status: "unimplemented",
+            message: "Wire handler in host",
+            toolName: config.name
+          };
+        }
+        try {
+          const output = await Promise.resolve(handler(input));
+          return typeof output === "string" ? output : output;
+        } catch (err) {
+          return {
+            status: "error",
+            error: err instanceof Error ? err.message : String(err),
+            toolName: config.name
+          };
+        }
+      }
+    });
+  }
+  return result;
+}
+
+// src/core/surface-planner-tools.ts
+import { tool as tool3 } from "ai";
+import { z as z3 } from "zod";
+import { validatePatchProposal } from "@contractspec/lib.surface-runtime/spec/validate-surface-patch";
+import { buildSurfacePatchProposal } from "@contractspec/lib.surface-runtime/runtime/planner-tools";
+var VALID_OPS = [
+  "insert-node",
+  "replace-node",
+  "remove-node",
+  "move-node",
+  "resize-panel",
+  "set-layout",
+  "reveal-field",
+  "hide-field",
+  "promote-action",
+  "set-focus"
+];
+var DEFAULT_NODE_KINDS = [
+  "entity-section",
+  "entity-card",
+  "data-view",
+  "assistant-panel",
+  "chat-thread",
+  "action-bar",
+  "timeline",
+  "table",
+  "rich-doc",
+  "form",
+  "chart",
+  "custom-widget"
+];
+function collectSlotIdsFromRegion(node) {
+  const ids = [];
+  if (node.type === "slot") {
+    ids.push(node.slotId);
+  }
+  if (node.type === "panel-group" || node.type === "stack") {
+    for (const child of node.children) {
+      ids.push(...collectSlotIdsFromRegion(child));
+    }
+  }
+  if (node.type === "tabs") {
+    for (const tab of node.tabs) {
+      ids.push(...collectSlotIdsFromRegion(tab.child));
+    }
+  }
+  if (node.type === "floating") {
+    ids.push(node.anchorSlotId);
+    ids.push(...collectSlotIdsFromRegion(node.child));
+  }
+  return ids;
+}
+function deriveConstraints(plan) {
+  const slotIds = collectSlotIdsFromRegion(plan.layoutRoot);
+  const uniqueSlots = [...new Set(slotIds)];
+  return {
+    allowedOps: VALID_OPS,
+    allowedSlots: uniqueSlots.length > 0 ? uniqueSlots : ["assistant", "primary"],
+    allowedNodeKinds: DEFAULT_NODE_KINDS
+  };
+}
+var ProposePatchInputSchema = z3.object({
+  proposalId: z3.string().describe("Unique proposal identifier"),
+  ops: z3.array(z3.object({
+    op: z3.enum([
+      "insert-node",
+      "replace-node",
+      "remove-node",
+      "move-node",
+      "resize-panel",
+      "set-layout",
+      "reveal-field",
+      "hide-field",
+      "promote-action",
+      "set-focus"
+    ]),
+    slotId: z3.string().optional(),
+    nodeId: z3.string().optional(),
+    toSlotId: z3.string().optional(),
+    index: z3.number().optional(),
+    node: z3.object({
+      nodeId: z3.string(),
+      kind: z3.string(),
+      title: z3.string().optional(),
+      props: z3.record(z3.string(), z3.unknown()).optional(),
+      children: z3.array(z3.unknown()).optional()
+    }).optional(),
+    persistKey: z3.string().optional(),
+    sizes: z3.array(z3.number()).optional(),
+    layoutId: z3.string().optional(),
+    fieldId: z3.string().optional(),
+    actionId: z3.string().optional(),
+    placement: z3.enum(["header", "inline", "context", "assistant"]).optional(),
+    targetId: z3.string().optional()
+  }))
+});
+function createSurfacePlannerTools(config) {
+  const { plan, constraints, onPatchProposal } = config;
+  const resolvedConstraints = constraints ?? deriveConstraints(plan);
+  const proposePatchTool = tool3({
+    description: "Propose surface patches (layout changes, node insertions, etc.) for user approval. " + "Only use allowed ops, slots, and node kinds from the planner context.",
+    inputSchema: ProposePatchInputSchema,
+    execute: async (input) => {
+      const ops = input.ops;
+      try {
+        validatePatchProposal(ops, resolvedConstraints);
+        const proposal = buildSurfacePatchProposal(input.proposalId, ops);
+        onPatchProposal?.(proposal);
+        return {
+          success: true,
+          proposalId: proposal.proposalId,
+          opsCount: proposal.ops.length,
+          message: "Patch proposal validated; awaiting user approval"
+        };
+      } catch (err) {
+        return {
+          success: false,
+          error: err instanceof Error ? err.message : String(err),
+          proposalId: input.proposalId
+        };
+      }
+    }
+  });
+  return {
+    "propose-patch": proposePatchTool
+  };
+}
+function buildPlannerPromptInput(plan) {
+  const constraints = deriveConstraints(plan);
+  return {
+    bundleMeta: {
+      key: plan.bundleKey,
+      version: "0.0.0",
+      title: plan.bundleKey
+    },
+    surfaceId: plan.surfaceId,
+    allowedPatchOps: constraints.allowedOps,
+    allowedSlots: [...constraints.allowedSlots],
+    allowedNodeKinds: [...constraints.allowedNodeKinds],
+    actions: plan.actions.map((a) => ({ actionId: a.actionId, title: a.title })),
+    preferences: {
+      guidance: "hints",
+      density: "standard",
+      dataDepth: "detailed",
+      control: "standard",
+      media: "text",
+      pace: "balanced",
+      narrative: "top-down"
+    }
+  };
+}
+
 // src/core/chat-service.ts
+import { compilePlannerPrompt } from "@contractspec/lib.surface-runtime/runtime/planner-prompt";
 var DEFAULT_SYSTEM_PROMPT = `You are ContractSpec AI, an expert coding assistant specialized in ContractSpec development.
 
 Your capabilities:
@@ -136,6 +687,9 @@ Guidelines:
 - Reference relevant ContractSpec concepts and patterns
 - Ask clarifying questions when the user's intent is unclear
 - When suggesting code changes, explain the rationale`;
+var WORKFLOW_TOOLS_PROMPT = `
+
+Workflow creation: You can create and modify workflows. Use create_workflow_extension when the user asks to add steps, change a workflow, or create a tenant-specific extension. Use compose_workflow to apply extensions to a base workflow. Use generate_workflow_spec_code to output TypeScript for the user to save.`;
 
 class ChatService {
   provider;
@@ -145,19 +699,93 @@ class ChatService {
   maxHistoryMessages;
   onUsage;
   tools;
+  thinkingLevel;
   sendReasoning;
   sendSources;
+  modelSelector;
   constructor(config) {
     this.provider = config.provider;
     this.context = config.context;
     this.store = config.store ?? new InMemoryConversationStore;
-    this.systemPrompt = config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+    this.systemPrompt = this.buildSystemPrompt(config);
     this.maxHistoryMessages = config.maxHistoryMessages ?? 20;
     this.onUsage = config.onUsage;
-    this.tools = config.tools;
-    this.sendReasoning = config.sendReasoning ?? false;
+    this.tools = this.mergeTools(config);
+    this.thinkingLevel = config.thinkingLevel;
+    this.modelSelector = config.modelSelector;
+    this.sendReasoning = config.sendReasoning ?? (config.thinkingLevel != null && config.thinkingLevel !== "instant");
     this.sendSources = config.sendSources ?? false;
   }
+  buildSystemPrompt(config) {
+    let base = config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+    if (config.workflowToolsConfig?.baseWorkflows?.length) {
+      base += WORKFLOW_TOOLS_PROMPT;
+    }
+    const contractsPrompt = buildContractsContextPrompt(config.contractsContext ?? {});
+    if (contractsPrompt) {
+      base += contractsPrompt;
+    }
+    if (config.surfacePlanConfig?.plan) {
+      const plannerInput = buildPlannerPromptInput(config.surfacePlanConfig.plan);
+      base += `
+
+` + compilePlannerPrompt(plannerInput);
+    }
+    return base;
+  }
+  mergeTools(config) {
+    let merged = config.tools ?? {};
+    const wfConfig = config.workflowToolsConfig;
+    if (wfConfig?.baseWorkflows?.length) {
+      const workflowTools = createWorkflowTools({
+        baseWorkflows: wfConfig.baseWorkflows,
+        composer: wfConfig.composer
+      });
+      merged = { ...merged, ...workflowTools };
+    }
+    const contractsCtx = config.contractsContext;
+    if (contractsCtx?.agentSpecs?.length) {
+      const allTools = [];
+      for (const agent of contractsCtx.agentSpecs) {
+        if (agent.tools?.length)
+          allTools.push(...agent.tools);
+      }
+      if (allTools.length > 0) {
+        const agentTools = agentToolConfigsToToolSet(allTools);
+        merged = { ...merged, ...agentTools };
+      }
+    }
+    const surfaceConfig = config.surfacePlanConfig;
+    if (surfaceConfig?.plan) {
+      const plannerTools = createSurfacePlannerTools({
+        plan: surfaceConfig.plan,
+        onPatchProposal: surfaceConfig.onPatchProposal
+      });
+      merged = { ...merged, ...plannerTools };
+    }
+    if (config.mcpTools && Object.keys(config.mcpTools).length > 0) {
+      merged = { ...merged, ...config.mcpTools };
+    }
+    return Object.keys(merged).length > 0 ? merged : undefined;
+  }
+  async resolveModel() {
+    if (this.modelSelector) {
+      const dimension = this.thinkingLevelToDimension(this.thinkingLevel);
+      const { model, selection } = await this.modelSelector.selectAndCreate({
+        taskDimension: dimension
+      });
+      return { model, providerName: selection.providerKey };
+    }
+    return {
+      model: this.provider.getModel(),
+      providerName: this.provider.name
+    };
+  }
+  thinkingLevelToDimension(level) {
+    if (!level || level === "instant")
+      return "latency";
+    return "reasoning";
+  }
   async send(options) {
     let conversation;
     if (options.conversationId) {
@@ -175,20 +803,25 @@ class ChatService {
         workspacePath: this.context?.workspacePath
       });
     }
-    await this.store.appendMessage(conversation.id, {
-      role: "user",
-      content: options.content,
-      status: "completed",
-      attachments: options.attachments
-    });
+    if (!options.skipUserAppend) {
+      await this.store.appendMessage(conversation.id, {
+        role: "user",
+        content: options.content,
+        status: "completed",
+        attachments: options.attachments
+      });
+    }
+    conversation = await this.store.get(conversation.id) ?? conversation;
     const messages = this.buildMessages(conversation, options);
-    const model = this.provider.getModel();
+    const { model, providerName } = await this.resolveModel();
+    const providerOptions = getProviderOptions(this.thinkingLevel, providerName);
     try {
       const result = await generateText({
         model,
         messages,
         system: this.systemPrompt,
-        tools: this.tools
+        tools: this.tools,
+        providerOptions: Object.keys(providerOptions).length > 0 ? providerOptions : undefined
       });
       const assistantMessage = await this.store.appendMessage(conversation.id, {
         role: "assistant",
@@ -233,23 +866,27 @@ class ChatService {
         workspacePath: this.context?.workspacePath
       });
     }
-    await this.store.appendMessage(conversation.id, {
-      role: "user",
-      content: options.content,
-      status: "completed",
-      attachments: options.attachments
-    });
+    if (!options.skipUserAppend) {
+      await this.store.appendMessage(conversation.id, {
+        role: "user",
+        content: options.content,
+        status: "completed",
+        attachments: options.attachments
+      });
+    }
+    conversation = await this.store.get(conversation.id) ?? conversation;
     const assistantMessage = await this.store.appendMessage(conversation.id, {
       role: "assistant",
       content: "",
       status: "streaming"
     });
     const messages = this.buildMessages(conversation, options);
-    const model = this.provider.getModel();
+    const { model, providerName } = await this.resolveModel();
     const systemPrompt = this.systemPrompt;
     const tools = this.tools;
     const store = this.store;
     const onUsage = this.onUsage;
+    const streamProviderOptions = getProviderOptions(this.thinkingLevel, providerName);
     async function* streamGenerator() {
       let fullContent = "";
       let fullReasoning = "";
@@ -260,7 +897,8 @@ class ChatService {
           model,
           messages,
           system: systemPrompt,
-          tools
+          tools,
+          providerOptions: Object.keys(streamProviderOptions).length > 0 ? streamProviderOptions : undefined
         });
         for await (const part of result.fullStream) {
           if (part.type === "text-delta") {
@@ -375,6 +1013,18 @@ class ChatService {
       ...options
     });
   }
+  async updateConversation(conversationId, updates) {
+    return this.store.update(conversationId, updates);
+  }
+  async forkConversation(conversationId, upToMessageId) {
+    return this.store.fork(conversationId, upToMessageId);
+  }
+  async updateMessage(conversationId, messageId, updates) {
+    return this.store.updateMessage(conversationId, messageId, updates);
+  }
+  async truncateAfter(conversationId, messageId) {
+    return this.store.truncateAfter(conversationId, messageId);
+  }
   async deleteConversation(conversationId) {
     return this.store.delete(conversationId);
   }
@@ -445,9 +1095,9 @@ import {
 function toolsToToolSet(defs) {
   const result = {};
   for (const def of defs) {
-    result[def.name] = tool({
+    result[def.name] = tool4({
       description: def.description ?? def.name,
-      inputSchema: z.object({}).passthrough(),
+      inputSchema: z4.object({}).passthrough(),
       execute: async () => ({})
     });
   }
@@ -461,21 +1111,63 @@ function useChat(options = {}) {
     apiKey,
     proxyUrl,
     conversationId: initialConversationId,
+    store,
     systemPrompt,
     streaming = true,
     onSend,
     onResponse,
     onError,
     onUsage,
-    tools: toolsDefs
+    tools: toolsDefs,
+    thinkingLevel,
+    workflowToolsConfig,
+    modelSelector,
+    contractsContext,
+    surfacePlanConfig,
+    mcpServers,
+    agentMode
   } = options;
   const [messages, setMessages] = React.useState([]);
+  const [mcpTools, setMcpTools] = React.useState(null);
+  const mcpCleanupRef = React.useRef(null);
   const [conversation, setConversation] = React.useState(null);
   const [isLoading, setIsLoading] = React.useState(false);
   const [error, setError] = React.useState(null);
   const [conversationId, setConversationId] = React.useState(initialConversationId ?? null);
   const abortControllerRef = React.useRef(null);
   const chatServiceRef = React.useRef(null);
+  React.useEffect(() => {
+    if (!mcpServers?.length) {
+      setMcpTools(null);
+      return;
+    }
+    let cancelled = false;
+    import("@contractspec/lib.ai-agent/tools/mcp-client").then(({ createMcpToolsets }) => {
+      createMcpToolsets(mcpServers).then(({ tools, cleanup }) => {
+        if (!cancelled) {
+          setMcpTools(tools);
+          mcpCleanupRef.current = cleanup;
+        } else {
+          cleanup().catch(() => {
+            return;
+          });
+        }
+      }).catch(() => {
+        if (!cancelled)
+          setMcpTools(null);
+      });
+    });
+    return () => {
+      cancelled = true;
+      const cleanup = mcpCleanupRef.current;
+      mcpCleanupRef.current = null;
+      if (cleanup)
+        cleanup().catch(() => {
+          return;
+        });
+      setMcpTools(null);
+    };
+  }, [mcpServers]);
   React.useEffect(() => {
     const chatProvider = createProvider({
       provider,
@@ -485,9 +1177,16 @@ function useChat(options = {}) {
     });
     chatServiceRef.current = new ChatService({
       provider: chatProvider,
+      store,
       systemPrompt,
       onUsage,
-      tools: toolsDefs?.length ? toolsToToolSet(toolsDefs) : undefined
+      tools: toolsDefs?.length ? toolsToToolSet(toolsDefs) : undefined,
+      thinkingLevel,
+      workflowToolsConfig,
+      modelSelector,
+      contractsContext,
+      surfacePlanConfig,
+      mcpTools
     });
   }, [
     provider,
@@ -495,9 +1194,16 @@ function useChat(options = {}) {
     model,
     apiKey,
     proxyUrl,
+    store,
     systemPrompt,
     onUsage,
-    toolsDefs
+    toolsDefs,
+    thinkingLevel,
+    workflowToolsConfig,
+    modelSelector,
+    contractsContext,
+    surfacePlanConfig,
+    mcpTools
   ]);
   React.useEffect(() => {
     if (!conversationId || !chatServiceRef.current)
@@ -513,7 +1219,90 @@ function useChat(options = {}) {
     };
     loadConversation().catch(console.error);
   }, [conversationId]);
-  const sendMessage = React.useCallback(async (content, attachments) => {
+  const sendMessage = React.useCallback(async (content, attachments, opts) => {
+    if (agentMode?.agent) {
+      setIsLoading(true);
+      setError(null);
+      abortControllerRef.current = new AbortController;
+      try {
+        if (!opts?.skipUserAppend) {
+          const userMessage = {
+            id: `msg_${Date.now()}`,
+            conversationId: conversationId ?? "",
+            role: "user",
+            content,
+            status: "completed",
+            createdAt: new Date,
+            updatedAt: new Date,
+            attachments
+          };
+          setMessages((prev) => [...prev, userMessage]);
+          onSend?.(userMessage);
+        }
+        const result = await agentMode.agent.generate({
+          prompt: content,
+          signal: abortControllerRef.current.signal
+        });
+        const toolCallsMap = new Map;
+        for (const tc of result.toolCalls ?? []) {
+          const tr = result.toolResults?.find((r) => r.toolCallId === tc.toolCallId);
+          toolCallsMap.set(tc.toolCallId, {
+            id: tc.toolCallId,
+            name: tc.toolName,
+            args: tc.args ?? {},
+            result: tr?.output,
+            status: "completed"
+          });
+        }
+        const assistantMessage = {
+          id: `msg_${Date.now()}_a`,
+          conversationId: conversationId ?? "",
+          role: "assistant",
+          content: result.text,
+          status: "completed",
+          createdAt: new Date,
+          updatedAt: new Date,
+          toolCalls: toolCallsMap.size ? Array.from(toolCallsMap.values()) : undefined,
+          usage: result.usage
+        };
+        setMessages((prev) => [...prev, assistantMessage]);
+        onResponse?.(assistantMessage);
+        onUsage?.(result.usage ?? { inputTokens: 0, outputTokens: 0 });
+        if (store && !conversationId) {
+          const conv = await store.create({
+            status: "active",
+            provider: "agent",
+            model: "agent",
+            messages: []
+          });
+          if (!opts?.skipUserAppend) {
+            await store.appendMessage(conv.id, {
+              role: "user",
+              content,
+              status: "completed",
+              attachments
+            });
+          }
+          await store.appendMessage(conv.id, {
+            role: "assistant",
+            content: result.text,
+            status: "completed",
+            toolCalls: assistantMessage.toolCalls,
+            usage: result.usage
+          });
+          const updated = await store.get(conv.id);
+          if (updated)
+            setConversation(updated);
+          setConversationId(conv.id);
+        }
+      } catch (err) {
+        setError(err instanceof Error ? err : new Error(String(err)));
+        onError?.(err instanceof Error ? err : new Error(String(err)));
+      } finally {
+        setIsLoading(false);
+      }
+      return;
+    }
     if (!chatServiceRef.current) {
       throw new Error("Chat service not initialized");
     }
@@ -521,25 +1310,28 @@ function useChat(options = {}) {
     setError(null);
     abortControllerRef.current = new AbortController;
     try {
-      const userMessage = {
-        id: `msg_${Date.now()}`,
-        conversationId: conversationId ?? "",
-        role: "user",
-        content,
-        status: "completed",
-        createdAt: new Date,
-        updatedAt: new Date,
-        attachments
-      };
-      setMessages((prev) => [...prev, userMessage]);
-      onSend?.(userMessage);
+      if (!opts?.skipUserAppend) {
+        const userMessage = {
+          id: `msg_${Date.now()}`,
+          conversationId: conversationId ?? "",
+          role: "user",
+          content,
+          status: "completed",
+          createdAt: new Date,
+          updatedAt: new Date,
+          attachments
+        };
+        setMessages((prev) => [...prev, userMessage]);
+        onSend?.(userMessage);
+      }
       if (streaming) {
         const result = await chatServiceRef.current.stream({
           conversationId: conversationId ?? undefined,
           content,
-          attachments
+          attachments,
+          skipUserAppend: opts?.skipUserAppend
         });
-        if (!conversationId) {
+        if (!conversationId && !opts?.skipUserAppend) {
           setConversationId(result.conversationId);
         }
         const assistantMessage = {
@@ -620,7 +1412,8 @@ function useChat(options = {}) {
         const result = await chatServiceRef.current.send({
           conversationId: conversationId ?? undefined,
           content,
-          attachments
+          attachments,
+          skipUserAppend: opts?.skipUserAppend
         });
         setConversation(result.conversation);
         setMessages(result.conversation.messages);
@@ -637,7 +1430,7 @@ function useChat(options = {}) {
       setIsLoading(false);
       abortControllerRef.current = null;
     }
-  }, [conversationId, streaming, onSend, onResponse, onError, messages]);
+  }, [conversationId, streaming, onSend, onResponse, onError, onUsage, messages, agentMode, store]);
   const clearConversation = React.useCallback(() => {
     setMessages([]);
     setConversation(null);
@@ -658,6 +1451,44 @@ function useChat(options = {}) {
     abortControllerRef.current?.abort();
     setIsLoading(false);
   }, []);
+  const createNewConversation = clearConversation;
+  const editMessage = React.useCallback(async (messageId, newContent) => {
+    if (!chatServiceRef.current || !conversationId)
+      return;
+    const msg = messages.find((m) => m.id === messageId);
+    if (!msg || msg.role !== "user")
+      return;
+    await chatServiceRef.current.updateMessage(conversationId, messageId, { content: newContent });
+    const truncated = await chatServiceRef.current.truncateAfter(conversationId, messageId);
+    if (truncated) {
+      setMessages(truncated.messages);
+    }
+    await sendMessage(newContent, undefined, { skipUserAppend: true });
+  }, [conversationId, messages, sendMessage]);
+  const forkConversation = React.useCallback(async (upToMessageId) => {
+    if (!chatServiceRef.current)
+      return null;
+    const idToFork = conversationId ?? conversation?.id;
+    if (!idToFork)
+      return null;
+    try {
+      const forked = await chatServiceRef.current.forkConversation(idToFork, upToMessageId);
+      setConversationId(forked.id);
+      setConversation(forked);
+      setMessages(forked.messages);
+      return forked.id;
+    } catch {
+      return null;
+    }
+  }, [conversationId, conversation]);
+  const updateConversationFn = React.useCallback(async (updates) => {
+    if (!chatServiceRef.current || !conversationId)
+      return null;
+    const updated = await chatServiceRef.current.updateConversation(conversationId, updates);
+    if (updated)
+      setConversation(updated);
+    return updated;
+  }, [conversationId]);
   const addToolApprovalResponse = React.useCallback((_toolCallId, _result) => {
     throw new Error(`addToolApprovalResponse: Tool approval requires server route with toUIMessageStreamResponse. ` + `Use createChatRoute and @ai-sdk/react useChat for tools with requireApproval.`);
   }, []);
@@ -672,6 +1503,10 @@ function useChat(options = {}) {
     setConversationId,
     regenerate,
     stop,
+    createNewConversation,
+    editMessage,
+    forkConversation,
+    updateConversation: updateConversationFn,
     ...hasApprovalTools && { addToolApprovalResponse }
   };
 }
@@ -715,11 +1550,94 @@ function useProviders() {
     refresh: loadProviders
   };
 }
+// src/presentation/hooks/useMessageSelection.ts
+import * as React3 from "react";
+"use client";
+function useMessageSelection(messageIds) {
+  const [selectedIds, setSelectedIds] = React3.useState(() => new Set);
+  const idSet = React3.useMemo(() => new Set(messageIds), [messageIds.join(",")]);
+  React3.useEffect(() => {
+    setSelectedIds((prev) => {
+      const next = new Set;
+      for (const id of prev) {
+        if (idSet.has(id))
+          next.add(id);
+      }
+      return next.size === prev.size ? prev : next;
+    });
+  }, [idSet]);
+  const toggle = React3.useCallback((id) => {
+    setSelectedIds((prev) => {
+      const next = new Set(prev);
+      if (next.has(id))
+        next.delete(id);
+      else
+        next.add(id);
+      return next;
+    });
+  }, []);
+  const selectAll = React3.useCallback(() => {
+    setSelectedIds(new Set(messageIds));
+  }, [messageIds.join(",")]);
+  const clearSelection = React3.useCallback(() => {
+    setSelectedIds(new Set);
+  }, []);
+  const isSelected = React3.useCallback((id) => selectedIds.has(id), [selectedIds]);
+  const selectedCount = selectedIds.size;
+  return {
+    selectedIds,
+    toggle,
+    selectAll,
+    clearSelection,
+    isSelected,
+    selectedCount
+  };
+}
+// src/presentation/hooks/useConversations.ts
+import * as React4 from "react";
+"use client";
+function useConversations(options) {
+  const { store, projectId, tags, limit = 50 } = options;
+  const [conversations, setConversations] = React4.useState([]);
+  const [isLoading, setIsLoading] = React4.useState(true);
+  const refresh = React4.useCallback(async () => {
+    setIsLoading(true);
+    try {
+      const list = await store.list({
+        status: "active",
+        projectId,
+        tags,
+        limit
+      });
+      setConversations(list);
+    } finally {
+      setIsLoading(false);
+    }
+  }, [store, projectId, tags, limit]);
+  React4.useEffect(() => {
+    refresh();
+  }, [refresh]);
+  const deleteConversation = React4.useCallback(async (id) => {
+    const ok = await store.delete(id);
+    if (ok) {
+      setConversations((prev) => prev.filter((c) => c.id !== id));
+    }
+    return ok;
+  }, [store]);
+  return {
+    conversations,
+    isLoading,
+    refresh,
+    deleteConversation
+  };
+}
 
 // src/presentation/hooks/index.ts
 import { useCompletion } from "@ai-sdk/react";
 export {
   useProviders,
+  useMessageSelection,
+  useConversations,
   useCompletion,
   useChat
 };