gensokyo-devandAkshad Jogi
9085222ca1
Add opt-in bounded output-schema advertisements while preserving the full default, canonical server validation and complete structured content. Verify real stdio discovery under a 1 MiB receive cap, all 5,367 libcrypto export coordinates, complete Evidence parity, whole-task serialization estimates and resource bounds. Fixes #2512. Integrates #7054 by Akshad Jogi with original authorship retained. Co-authored-by: Akshad Jogi <akshadjogi@gmail.com>
81 lines
2.8 KiB
TypeScript
81 lines
2.8 KiB
TypeScript
import {
|
|
COMPACT_SCHEMA_MAX_DEPTH,
|
|
hasGatewayIncompatibleShape,
|
|
inlineWithinBudget,
|
|
reduceToShallowProperties,
|
|
schemaDepth,
|
|
serializedBytes,
|
|
stripAnnotations,
|
|
typeArrayItemUnions,
|
|
} from "./compactJsonSchema.js";
|
|
|
|
// Kimi expands schema-local references before sending tools to Moonshot.
|
|
// Compact advertisements must fit after expansion, independently of client
|
|
// reference handling. Canonical server-side validation remains unchanged.
|
|
|
|
export type AdvertisedInputSchemaPresentation = "inline" | "reduced";
|
|
|
|
/** Budget selected by the caller for compact MCP input-schema presentation. */
|
|
export interface CompactInputSchemaPresentation {
|
|
readonly budgetBytes: number;
|
|
}
|
|
|
|
export interface CompactedInputSchema {
|
|
readonly schema: Record<string, unknown>;
|
|
readonly presentation: AdvertisedInputSchemaPresentation;
|
|
}
|
|
|
|
const INPUT_REDUCED_ANNOTATION =
|
|
"Advertised in reduced form by the compact input schema profile; the server still validates complete canonical inputs.";
|
|
|
|
/**
|
|
* Render an advertised input schema within a per-schema byte budget for
|
|
* providers that validate the literal serialized schema size. Validation
|
|
* keywords and the canonical contract are preserved; only
|
|
* annotation prose and, for structurally oversized schemas, nested
|
|
* validation detail leave the advertised form.
|
|
*/
|
|
export const compactAdvertisedInputSchema = (
|
|
schema: Record<string, unknown>,
|
|
budgetBytes: number,
|
|
): CompactedInputSchema => {
|
|
const expanded = inlineWithinBudget(stripAnnotations(schema), budgetBytes);
|
|
const inline =
|
|
expanded === undefined ? undefined : typeArrayItemUnions(expanded);
|
|
if (
|
|
inline !== undefined &&
|
|
serializedBytes(inline) <= budgetBytes &&
|
|
schemaDepth(inline) <= COMPACT_SCHEMA_MAX_DEPTH &&
|
|
!hasGatewayIncompatibleShape(inline)
|
|
)
|
|
return { schema: inline, presentation: "inline" };
|
|
const reduced = reduceToShallowProperties(schema, INPUT_REDUCED_ANNOTATION);
|
|
if (serializedBytes(reduced) <= budgetBytes)
|
|
return { schema: reduced, presentation: "reduced" };
|
|
// A future contract with an extreme root-property count still advertises
|
|
// an honest, budget-fitting shape instead of an oversized one.
|
|
return {
|
|
schema: reduceToShallowProperties(
|
|
{
|
|
type: "object",
|
|
description:
|
|
(typeof schema.description === "string"
|
|
? `${schema.description} `
|
|
: "") + INPUT_REDUCED_ANNOTATION,
|
|
},
|
|
INPUT_REDUCED_ANNOTATION,
|
|
),
|
|
presentation: "reduced",
|
|
};
|
|
};
|
|
|
|
/** Return the JSON schema clients receive for the selected input presentation. */
|
|
export const presentAdvertisedInputSchema = (
|
|
schema: Record<string, unknown>,
|
|
compactPresentation: CompactInputSchemaPresentation | undefined,
|
|
): Record<string, unknown> =>
|
|
compactPresentation === undefined
|
|
? schema
|
|
: compactAdvertisedInputSchema(schema, compactPresentation.budgetBytes)
|
|
.schema;
|