Files
rea/src/contracts/advertisedInputExamples.ts
T
Kaoru0822andClaude Opus 5.5 a53689c55d fix(mcp): stop advertising complete Evidence in input examples (#1560)
Hosts forward every tool's input schema to the model on each request.
Seventeen tools advertised a canonical example that embedded a complete
producer Evidence record, up to 22 KB each, accounting for 116 KB of the
450 KB of advertised input schemas.

Advertised examples now show an embedded Evidence record as its exact
retained-evidence reference when the canonical input accepts one, and
omit an example that can only carry inline Evidence. Canonical contract
examples, their executable tests, and validation are unchanged.

Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-10 07:10:38 -07:00

76 lines
2.8 KiB
TypeScript

import type { ToolContract } from "./toolContractTypes.js";
import { evidenceSchema } from "../domain/evidence.js";
import type { JsonValue } from "../domain/jsonValue.js";
// Hosts forward every advertised input schema to the model on each turn. A
// canonical example that embeds complete producer Evidence repeats a prior
// result's bytes in that context while showing nothing about argument choice
// that the schema and descriptions do not already state; agents pass Evidence
// they received rather than composing it.
interface RetainedProjection {
readonly value: JsonValue;
readonly embedsEvidence: boolean;
}
/** The Evidence identifier of a complete Evidence record, or undefined. */
const evidenceRecordId = (value: {
readonly [key: string]: JsonValue;
}): string | undefined =>
typeof value.evidence_id === "string" &&
Object.hasOwn(value, "normalized_result") &&
evidenceSchema.safeParse(value).success
? value.evidence_id
: undefined;
/** Replace each embedded complete Evidence record with its exact retained reference. */
const referenceEmbeddedEvidence = (value: JsonValue): RetainedProjection => {
if (Array.isArray(value)) {
const items = value.map(referenceEmbeddedEvidence);
return {
value: items.map((item) => item.value),
embedsEvidence: items.some((item) => item.embedsEvidence),
};
}
if (value === null || typeof value !== "object")
return { value, embedsEvidence: false };
const evidenceId = evidenceRecordId(value);
if (evidenceId !== undefined)
return {
value: { kind: "retained-evidence", evidence_id: evidenceId },
embedsEvidence: true,
};
const entries = Object.entries(value).map(
([key, child]) => [key, referenceEmbeddedEvidence(child)] as const,
);
return {
value: Object.fromEntries(
entries.map(([key, child]) => [key, child.value]),
),
embedsEvidence: entries.some(([, child]) => child.embedsEvidence),
};
};
/**
* Examples advertised in a tool's input schema. Embedded complete Evidence is
* shown as its exact same-session retained reference when the canonical input
* schema accepts that form; an example that can only carry inline Evidence
* stays in the canonical contract and its executable tests but leaves the
* advertisement. Every advertised example is a distinct valid canonical input.
*/
export const advertisedInputExamples = (
contract: Pick<ToolContract, "inputSchema" | "examples">,
): readonly JsonValue[] => {
const advertised = new Map<string, JsonValue>();
for (const { input } of contract.examples) {
const projection = referenceEmbeddedEvidence(input);
if (
projection.embedsEvidence &&
!contract.inputSchema.safeParse(projection.value).success
)
continue;
advertised.set(JSON.stringify(projection.value), projection.value);
}
return [...advertised.values()];
};