xiaoju
99f40c2488
CI / check (pull_request) Successful in 2m28s
- Add Usage type to protocol (turns, inputTokens, outputTokens, duration) - Add usage to StepRecord, StepNodePayload, StepEntry, STEP_NODE_SCHEMA - Thread usage through util-agent extract pipeline (writeStepNode → persistStep → createAgent) - All adapters return usage: null as placeholder (mock, hermes, claude-code, builtin) - 746 tests pass, no breaking changes (usage not in schema required array) Fixes #74 Refs #68
113 lines
2.8 KiB
TypeScript
113 lines
2.8 KiB
TypeScript
import type { JSONSchema } from "@ocas/core";
|
|
|
|
const ROLE_DEFINITION: JSONSchema = {
|
|
type: "object",
|
|
required: ["description", "goal", "capabilities", "procedure", "output", "frontmatter"],
|
|
properties: {
|
|
description: { type: "string" },
|
|
goal: { type: "string" },
|
|
capabilities: { type: "array", items: { type: "string" } },
|
|
procedure: { type: "string" },
|
|
output: { type: "string" },
|
|
frontmatter: { type: "string", format: "ocas_ref" },
|
|
},
|
|
additionalProperties: false,
|
|
};
|
|
|
|
const TARGET: JSONSchema = {
|
|
type: "object",
|
|
required: ["role", "prompt"],
|
|
properties: {
|
|
role: { type: "string", description: "Role name or pseudo-role ($END, $SUSPEND)" },
|
|
prompt: { type: "string" },
|
|
location: {
|
|
anyOf: [{ type: "string" }, { type: "null" }],
|
|
},
|
|
},
|
|
additionalProperties: false,
|
|
};
|
|
|
|
export const WORKFLOW_SCHEMA: JSONSchema = {
|
|
title: "Workflow",
|
|
type: "object",
|
|
required: ["name", "description", "roles", "graph"],
|
|
properties: {
|
|
name: { type: "string" },
|
|
description: { type: "string" },
|
|
roles: {
|
|
type: "object",
|
|
additionalProperties: ROLE_DEFINITION,
|
|
},
|
|
graph: {
|
|
type: "object",
|
|
additionalProperties: {
|
|
type: "object",
|
|
additionalProperties: TARGET,
|
|
},
|
|
},
|
|
},
|
|
additionalProperties: false,
|
|
};
|
|
|
|
export const START_NODE_SCHEMA: JSONSchema = {
|
|
title: "StartNode",
|
|
type: "object",
|
|
required: ["workflow", "prompt", "cwd"],
|
|
properties: {
|
|
workflow: { type: "string", format: "ocas_ref" },
|
|
prompt: { type: "string" },
|
|
cwd: { type: "string" },
|
|
},
|
|
additionalProperties: false,
|
|
};
|
|
|
|
export const STEP_NODE_SCHEMA: JSONSchema = {
|
|
title: "StepNode",
|
|
type: "object",
|
|
required: [
|
|
"start",
|
|
"prev",
|
|
"role",
|
|
"output",
|
|
"detail",
|
|
"agent",
|
|
"startedAtMs",
|
|
"completedAtMs",
|
|
"cwd",
|
|
],
|
|
properties: {
|
|
start: { type: "string", format: "ocas_ref" },
|
|
prev: {
|
|
anyOf: [{ type: "string", format: "ocas_ref" }, { type: "null" }],
|
|
},
|
|
role: { type: "string" },
|
|
output: { type: "string", format: "ocas_ref" },
|
|
detail: { type: "string", format: "ocas_ref" },
|
|
agent: { type: "string" },
|
|
edgePrompt: { type: "string" },
|
|
startedAtMs: { type: "integer" },
|
|
completedAtMs: { type: "integer" },
|
|
cwd: { type: "string" },
|
|
assembledPrompt: {
|
|
anyOf: [{ type: "string", format: "ocas_ref" }, { type: "null" }],
|
|
},
|
|
usage: {
|
|
anyOf: [
|
|
{
|
|
type: "object",
|
|
required: ["turns", "inputTokens", "outputTokens", "duration"],
|
|
properties: {
|
|
turns: { type: "integer" },
|
|
inputTokens: { type: "integer" },
|
|
outputTokens: { type: "integer" },
|
|
duration: { type: "number" },
|
|
},
|
|
additionalProperties: false,
|
|
},
|
|
{ type: "null" },
|
|
],
|
|
},
|
|
},
|
|
additionalProperties: false,
|
|
};
|