packages/klient/src/contract/session/approval.ts

/**
 * `sessionApprovalService` — session-scope approval broker. Mirrors
 * `agent-core-v2/agent/interaction/approval.ts`. `ApprovalRequest.display` is
 * the protocol `ToolInputDisplay` union (huge); it crosses the wire
 * uninspected, so it is `z.unknown()` here.
 */

import { z } from 'zod';

import { noResult } from '../helpers.js';
import type { ServiceContract } from '../types.js';

export const approvalRequestSchema = z.object({
  id: z.string().optional(),
  sessionId: z.string().optional(),
  agentId: z.string().optional(),
  turnId: z.number().optional(),
  toolCallId: z.string().optional(),
  toolName: z.string(),
  action: z.string(),
  /** Protocol `ToolInputDisplay` — mirrored as `unknown` (see file header). */
  display: z.unknown(),
});

export const approvalResponseSchema = z.object({
  decision: z.enum(['approved', 'rejected', 'cancelled']),
  scope: z.literal('session').optional(),
  feedback: z.string().optional(),
  selectedLabel: z.string().optional(),
});

export const sessionApprovalContract = {
  listPending: { input: z.tuple([]), output: z.array(approvalRequestSchema) },
  decide: { input: z.tuple([z.string(), approvalResponseSchema]), output: noResult },
} satisfies ServiceContract;