@the-open-engine/zeroshot 6.45.0 → 6.46.0

package/src/cluster/validators.ts

@@ -7,6 +7,8 @@ import {
 import type { ClusterMethod, GraphProfile, GraphSpec, ServerCapabilities } from './generated/protocol.js';
 import { CLUSTER_PROTOCOL_SCHEMA } from './generated/protocol-schema.js';
 import { ClusterProtocolError, ClusterRequestError } from './errors.js';
+import { isRecord } from './frames.js';
+import type { FrameRecord } from './frames.js';
 
 const ajv = new Ajv2020({
   allErrors: true,
@@ -26,6 +28,41 @@ function validatorFor(definition: string): ValidateFunction {
   return validator;
 }
 
+const RUN_PROJECTION_DEFINITIONS = new Set([
+  'RunForceResult',
+  'RunStatusResult',
+  'RunWatchEventNotification',
+]);
+
+function normalizeRunProjection(value: unknown): unknown {
+  if (!isRecord(value)) return value;
+  const size = value.size === 'tiny'
+    ? 'small'
+    : value.size === 'standard'
+      ? 'medium'
+      : value.size;
+  return size === value.size ? value : { ...value, size };
+}
+
+/** Normalize deserialize-only aliases before validating an inbound protocol value. */
+function normalizeInboundDefinition(definition: string, value: unknown): unknown {
+  if (RUN_PROJECTION_DEFINITIONS.has(definition)) return normalizeRunProjection(value);
+  if (definition !== 'RunListResult' || !isRecord(value)) return value;
+  const existingRuns = value.runs;
+  if (!Array.isArray(existingRuns)) return value;
+  const runs = existingRuns.map(normalizeRunProjection);
+  return runs.some((run, index) => run !== existingRuns[index]) ? { ...value, runs } : value;
+}
+
+export function normalizeInboundSubscriptionFrame(
+  kind: string,
+  method: string,
+  frame: FrameRecord,
+): FrameRecord {
+  if (kind !== 'run/watch' || method !== 'event' || !isRecord(frame.params)) return frame;
+  return { ...frame, params: normalizeInboundDefinition('RunWatchEventNotification', frame.params) };
+}
+
 export function assertDefinition(definition: string, value: unknown): void {
   const validate = validatorFor(definition);
   if (validate(value)) return;
@@ -38,9 +75,12 @@ export function assertDefinition(definition: string, value: unknown): void {
   );
 }
 
-export function assertMethodResult(method: ClusterMethod, value: unknown): void {
+export function assertMethodResult(method: ClusterMethod, value: unknown): unknown {
   try {
-    assertDefinition(METHOD_RESULT_DEFINITIONS[method], value);
+    const definition = METHOD_RESULT_DEFINITIONS[method];
+    const normalized = normalizeInboundDefinition(definition, value);
+    assertDefinition(definition, normalized);
+    return normalized;
   } catch (error) {
     const receivedVersion = method === 'initialize' && value !== null && typeof value === 'object'
       ? (value as Readonly<Record<string, unknown>>).protocolVersion