File size: 3,029 Bytes
4e23b01 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | import Ajv, { type ErrorObject, type ValidateFunction } from 'ajv';
import Ajv2019 from 'ajv/dist/2019';
import Ajv2020 from 'ajv/dist/2020';
import addFormats from 'ajv-formats';
const DRAFT_07_AJV = new Ajv({ strict: false, allErrors: true });
addFormats(DRAFT_07_AJV);
const DRAFT_2019_AJV = new Ajv2019({ strict: false, allErrors: true });
addFormats(DRAFT_2019_AJV);
const DRAFT_2020_AJV = new Ajv2020({ strict: false, allErrors: true });
addFormats(DRAFT_2020_AJV);
const DRAFT_2019_KEYWORDS = new Set([
'dependentRequired',
'dependentSchemas',
'maxContains',
'minContains',
'unevaluatedItems',
'unevaluatedProperties',
'$recursiveAnchor',
'$recursiveRef',
]);
const DRAFT_2020_KEYWORDS = new Set(['prefixItems', '$dynamicAnchor', '$dynamicRef']);
function ajvFor(schema: Record<string, unknown>): Ajv | Ajv2019 | Ajv2020 {
const $schema = schema['$schema'];
if (typeof $schema === 'string') {
if ($schema.includes('2020-12')) return DRAFT_2020_AJV;
if ($schema.includes('2019-09')) return DRAFT_2019_AJV;
return DRAFT_07_AJV;
}
if (containsSchemaKeyword(schema, DRAFT_2020_KEYWORDS)) return DRAFT_2020_AJV;
if (containsSchemaKeyword(schema, DRAFT_2019_KEYWORDS)) return DRAFT_2019_AJV;
return DRAFT_07_AJV;
}
function containsSchemaKeyword(value: unknown, keywords: ReadonlySet<string>): boolean {
if (Array.isArray(value)) {
return value.some((item) => containsSchemaKeyword(item, keywords));
}
if (typeof value !== 'object' || value === null) return false;
for (const [key, child] of Object.entries(value)) {
if (keywords.has(key)) return true;
if (containsSchemaKeyword(child, keywords)) return true;
}
return false;
}
export type JsonType = null | number | string | boolean | JsonArray | JsonObject;
export interface JsonArray extends Array<JsonType> {}
export interface JsonObject extends Record<string, JsonType> {}
export type ToolArgsValidator = ValidateFunction<JsonType>;
function formatValidationError(error: ErrorObject): string {
if (error.keyword === 'required' && 'missingProperty' in error.params) {
return `must have required property '${String(error.params['missingProperty'])}'`;
}
if (error.keyword === 'additionalProperties' && 'additionalProperty' in error.params) {
return `must NOT have additional property '${String(error.params['additionalProperty'])}'`;
}
const path = error.instancePath ? `${error.instancePath} ` : '';
return `${path}${error.message ?? 'is invalid'}`;
}
export function compileToolArgsValidator(schema: Record<string, unknown>): ToolArgsValidator {
return ajvFor(schema).compile(schema) as ToolArgsValidator;
}
export function validateToolArgs(validator: ToolArgsValidator, args: JsonType): string | null {
const valid = validator(args);
if (valid) {
return null;
}
const errors = validator.errors ?? [];
if (errors.length === 0) {
return 'Tool parameter validation failed';
}
return errors.map((error) => formatValidationError(error)).join('; ');
}
|