Hugging Face's logo

SaylorTwift
/
kimi-code

agent
coding-agent
cli
llm
tool-use
Model card Files
xet
 Community
kimi-code / packages /klient /src /contract /agent /schemas.ts
SaylorTwift's picture
SaylorTwift HF Staff
Add files using upload-large-folder tool
4e23b01 verified
Raw History Blame Contribute Delete
6.53 kB
/**
* Shared agent-scope wire schemas — the payload/result vocabulary reused by
* the per-domain contracts in `agent/services.ts` and pinned against the
* engine types by `test/contract-parity.ts`. `PromptPayload.input` mirrors the
* `PromptPart` subset of `ContentPart` (text / image_url / video_url) from
* `agent-core-v2/llm-adapter/contract/message.ts`. Task wire shapes mirror the
* `TaskInfo` union in `protocol/src/events.ts`.
*/
import { z } from 'zod';
// ── prompt parts ────────────────────────────────────────────────────────────
const textPartSchema = z.object({
type: z.literal('text'),
text: z.string(),
});
const imageUrlPartSchema = z.object({
type: z.literal('image_url'),
imageUrl: z.object({ url: z.string(), id: z.string().optional() }),
});
const videoUrlPartSchema = z.object({
type: z.literal('video_url'),
videoUrl: z.object({ url: z.string(), id: z.string().optional() }),
});
/** `PromptPart = Extract<ContentPart, { type: 'text' | 'image_url' | 'video_url' }>`. */
export const promptPartSchema = z.discriminatedUnion('type', [
textPartSchema,
imageUrlPartSchema,
videoUrlPartSchema,
]);
// ── payloads / results ──────────────────────────────────────────────────────
export const emptyPayloadSchema = z.object({});
export const promptPayloadSchema = z.object({
input: z.array(promptPartSchema),
// Mirrors `PromptPayload.promptId` in the engine (client-chosen prompt
// record id, echoed on the consuming turn's `turn.started`).
promptId: z.string().min(1).optional(),
});
/** Same shape as `PromptSkillActivation` in the engine. */
export const promptSkillActivationSchema = z.object({
name: z.string(),
args: z.string().optional(),
});
/** Same shape as `PromptWithSkillsInput` in the engine. */
export const promptWithSkillsPayloadSchema = promptPayloadSchema.extend({
skills: z.array(promptSkillActivationSchema).min(1),
});
/** Same shape as `PromptWithSkillsResult` in the engine. */
export const promptWithSkillsResultSchema = z.object({
turn_id: z.number().optional(),
prompt_id: z.string(),
created_at: z.string(),
state: z.enum(['running', 'queued', 'blocked']),
});
/** Same shape as `SteerPayload` in the engine. */
export const steerPayloadSchema = z.object({
input: z.array(promptPartSchema),
});
/** Same shape as `SkillActivationInput`'s wire subset in the engine. */
export const activateSkillPayloadSchema = z.object({
name: z.string(),
args: z.string().optional(),
});
export const promptLaunchResultSchema = z.object({
turn_id: z.number(),
});
export const cancelPayloadSchema = z.object({
turnId: z.number().optional(),
});
export const runShellCommandPayloadSchema = z.object({
command: z.string(),
commandId: z.string().optional(),
});
export const shellCommandResultSchema = z.object({
stdout: z.string(),
stderr: z.string(),
isError: z.boolean().optional(),
backgrounded: z.boolean().optional(),
});
export const cancelShellCommandPayloadSchema = z.object({
commandId: z.string(),
});
export const setModelPayloadSchema = z.object({
model: z.string(),
});
export const setModelResultSchema = z.object({
model: z.string(),
providerName: z.string().optional(),
});
export const runtimeBindingSchema = z.object({
workspaceId: z.string(),
runtimeId: z.string(),
});
export const permissionModeSchema = z.enum(['manual', 'yolo', 'auto']);
export const setPermissionPayloadSchema = z.object({
mode: permissionModeSchema,
});
export const tokenUsageSchema = z.object({
inputOther: z.number(),
output: z.number(),
inputCacheRead: z.number(),
inputCacheCreation: z.number(),
});
export const usageStatusSchema = z.object({
byModel: z.record(z.string(), tokenUsageSchema).optional(),
currentTurn: tokenUsageSchema.optional(),
total: tokenUsageSchema.optional(),
});
/**
* `AgentContextData` — `history` items are full `ContextMessage`s (deep
* `Message` / `Tool` / `PromptOrigin` unions); mirrored as `unknown` entries.
*/
export const agentContextDataSchema = z.object({
history: z.array(z.unknown()),
tokenCount: z.number(),
});
/** `AgentCommandInfo` (`agent-core-v2/agent/command/agentCommand.ts`). */
export const agentCommandInfoSchema = z.object({
name: z.string(),
description: z.string().optional(),
source: z.string(),
});
/** The facade's `runCommand` input shape. */
export const runCommandPayloadSchema = z.object({
name: z.string(),
args: z.string().optional(),
});
/** `PlanData = null | { id, content, path }` — null is JSON-representable. */
export const planDataSchema = z.union([
z.null(),
z.object({
id: z.string(),
content: z.string(),
path: z.string(),
}),
]);
export const cancelPlanPayloadSchema = z.object({
id: z.string().optional(),
});
export const getTasksPayloadSchema = z.object({
activeOnly: z.boolean().optional(),
limit: z.number().optional(),
});
const taskLifecycleStatusSchema = z.enum([
'running',
'completed',
'failed',
'timed_out',
'killed',
'lost',
]);
const taskInfoBaseFields = {
taskId: z.string(),
description: z.string(),
status: taskLifecycleStatusSchema,
detached: z.boolean().optional(),
startedAt: z.number(),
endedAt: z.union([z.number(), z.null()]),
stopReason: z.string().optional(),
terminalNotificationSuppressed: z.boolean().optional(),
timeoutMs: z.number().optional(),
} as const;
/** Protocol `TaskInfo` union (`protocol/src/events.ts`). */
export const agentTaskInfoSchema = z.discriminatedUnion('kind', [
z.object({
kind: z.literal('process'),
command: z.string(),
pid: z.number(),
exitCode: z.union([z.number(), z.null()]),
...taskInfoBaseFields,
}),
z.object({
kind: z.literal('agent'),
agentId: z.string().optional(),
subagentType: z.string().optional(),
model: z.string().optional(),
thinkingEffort: z.string().optional(),
...taskInfoBaseFields,
}),
z.object({
kind: z.literal('question'),
questionCount: z.number(),
toolCallId: z.string().optional(),
...taskInfoBaseFields,
}),
]);
export const stopTaskPayloadSchema = z.object({
taskId: z.string(),
reason: z.string().optional(),
});
export const getTaskOutputPayloadSchema = z.object({
taskId: z.string(),
tail: z.number().optional(),
});