@opengeni/runtime 4.3.0-canary.36844267522001 → 4.3.0

package/dist/{chunk-XFDVXBB7.js → chunk-22M75RJF.js}

@@ -9474,161 +9474,273 @@ function isModelCallFetch(input) {
 // src/replayable-json-body.ts
 import OpenAI6 from "openai";
 import { types as utilTypes } from "util";
-
-// src/model-preparation-diagnostics.ts
-import { AsyncLocalStorage as AsyncLocalStorage3 } from "async_hooks";
-
-// src/model-request-capture.ts
-import { AsyncLocalStorage as AsyncLocalStorage2 } from "async_hooks";
-var modelRequestCapture = new AsyncLocalStorage2();
-var captureIndices = /* @__PURE__ */ new WeakMap();
-var modelCallLifecycle = new AsyncLocalStorage2();
-function withModelCallLifecycle(lifecycle, fn) {
-  return modelCallLifecycle.run(lifecycle, fn);
-}
-async function beforeModelRequest() {
-  await modelCallLifecycle.getStore()?.beforeModelRequest?.();
-}
-function modelResponseSettlement(event) {
-  if (event.type !== "response_done") return void 0;
-  return modelCallLifecycle.getStore()?.onModelResponse?.(event);
-}
-function nextModelContextCaptureIndex(agent) {
-  const index = (captureIndices.get(agent) ?? 0) + 1;
-  captureIndices.set(agent, index);
-  return index;
-}
-function withModelRequestCapture(capture, fn) {
-  return capture ? modelRequestCapture.run(capture, fn) : fn();
-}
-function captureProviderRequestBody(provider, input, init) {
-  const context = modelRequestCapture.getStore();
-  const callback = context?.onProviderRequest;
-  if (!callback) return { init, captured: Promise.resolve(), cancel: () => {
-  } };
-  const index = context.nextProviderRequestIndex?.();
-  const observer = (providerId, body2, reason) => callback(providerId, body2, reason, index);
-  let cancel = () => {
-  };
-  let body = init?.body;
-  let nextInit = init;
-  if (body instanceof ReadableStream) {
-    const [sent, inspected] = body.tee();
-    nextInit = { ...init, body: sent };
-    body = inspected;
-  } else if (body == null && input instanceof Request) {
-    body = input.clone().body;
+var TARGET_CHUNK_CHARS = 64 * 1024;
+var LARGE_STRING_SOURCE_CHARS = 32 * 1024;
+var STRING_SLICE_CODE_UNITS = 10 * 1024;
+var REPLAYABLE_REQUEST_BODY_FACTORY = Symbol.for(
+  "opengeni.replayable-request-body-factory"
+);
+var ReplayableJsonBody = class {
+  source;
+  constructor(source) {
+    this.source = source;
   }
-  const captured = (async () => {
-    const limit = 4 * 1024 * 1024;
-    let text2;
-    if (typeof body === "string") {
-      if (Buffer.byteLength(body, "utf8") > limit) {
-        await observer(provider, null, "Request exceeds the 4 MiB capture limit.");
-        return;
+  async *[Symbol.asyncIterator]() {
+    const encoder = new TextEncoder();
+    for (const chunk of jsonTextChunks(this.source)) {
+      yield encoder.encode(chunk);
+    }
+  }
+  createStream() {
+    const iterator = this[Symbol.asyncIterator]();
+    return new ReadableStream({
+      async pull(controller) {
+        try {
+          const next = await iterator.next();
+          if (next.done) controller.close();
+          else controller.enqueue(next.value);
+        } catch (error) {
+          controller.error(error);
+        }
+      },
+      async cancel(reason) {
+        await iterator.return?.(reason).catch(() => void 0);
       }
-      text2 = body;
-    } else if (body instanceof ReadableStream) {
-      const reader = body.getReader();
-      let cancelled = false;
-      cancel = () => {
-        cancelled = true;
-        void reader.cancel().catch(() => void 0);
-      };
-      const signal = init?.signal ?? (input instanceof Request ? input.signal : void 0);
-      signal?.addEventListener("abort", cancel, { once: true });
-      if (signal?.aborted) cancel();
-      const decoder = new TextDecoder("utf-8", { fatal: true });
-      const chunks = [];
-      let bytes = 0;
-      try {
-        while (true) {
-          const chunk = await reader.read();
-          if (chunk.done) break;
-          bytes += chunk.value.byteLength;
-          if (bytes > limit) {
-            void reader.cancel().catch(() => void 0);
-            await observer(provider, null, "Request exceeds the 4 MiB capture limit.");
-            return;
-          }
-          chunks.push(decoder.decode(chunk.value, { stream: true }));
+    });
+  }
+};
+var ReplayableJsonOpenAI = class extends OpenAI6 {
+  modelRequestPolicy;
+  constructor(options, hooks = {}) {
+    super(options);
+    this.modelRequestPolicy = hooks.modelRequestPolicy;
+  }
+  post(path, opts) {
+    if (!isModelRequestPath(path) || opts === void 0) {
+      return super.post(path, opts);
+    }
+    const wrapped = Promise.resolve(opts).then((resolved) => {
+      const body = resolved.body;
+      let prepared = resolved;
+      if (this.modelRequestPolicy && isJsonRecordBody(body)) {
+        const policy = this.modelRequestPolicy({ path, body });
+        if (policy?.body || policy?.headers) {
+          prepared = {
+            ...resolved,
+            ...policy.body ? { body: policy.body } : {},
+            ...policy.headers ? { headers: mergePolicyHeaders(resolved.headers, policy.headers) } : {}
+          };
         }
-        if (cancelled) return;
-        chunks.push(decoder.decode());
-        text2 = chunks.join("");
-      } catch {
-        void reader.cancel().catch(() => void 0);
-        if (!cancelled)
-          await observer(provider, null, "The provider body could not be read as UTF-8.");
-        return;
-      } finally {
-        signal?.removeEventListener("abort", cancel);
-        reader.releaseLock();
-        cancel = () => {
-        };
       }
+      return wrapJsonRequestOptions(prepared);
+    });
+    return super.post(path, wrapped);
+  }
+  async buildRequest(inputOptions, context) {
+    const replayable = inputOptions.body instanceof ReplayableJsonBody ? inputOptions.body : void 0;
+    const priorStreamingMarker = inputOptions.__metadata?.hasStreamingBody;
+    const built = await super.buildRequest(inputOptions, context);
+    if (!replayable) return built;
+    if (priorStreamingMarker === void 0) {
+      if (inputOptions.__metadata) delete inputOptions.__metadata.hasStreamingBody;
     } else {
-      await observer(provider, null, "This transport body cannot be inspected.");
-      return;
+      inputOptions.__metadata = {
+        ...inputOptions.__metadata,
+        hasStreamingBody: priorStreamingMarker
+      };
     }
-    await observer(provider, text2);
-  })().catch(() => void 0);
-  return { init: nextInit, captured, cancel: () => cancel() };
-}
-async function notifyModelRequestCapture(request) {
-  const capture = modelRequestCapture.getStore();
-  if (!capture) return;
-  try {
-    await capture(snapshotModelRequestPrefix(request));
-  } catch {
+    const request = built.req;
+    request[REPLAYABLE_REQUEST_BODY_FACTORY] = () => replayable.createStream();
+    return built;
   }
+};
+function isModelRequestPath(path) {
+  const pathname = path.split("?", 1)[0] ?? path;
+  return pathname.endsWith("/responses") || pathname.endsWith("/chat/completions");
 }
-function snapshotModelRequestPrefix(request) {
+function wrapJsonRequestOptions(options) {
+  const body = options.body;
+  if (!isJsonObjectBody(body) || body instanceof ReplayableJsonBody) return options;
   return {
-    ...request,
-    tools: structuredClone(request.tools)
+    ...options,
+    body: new ReplayableJsonBody(body),
+    headers: mergePolicyHeaders(options.headers, { "content-type": "application/json" })
   };
 }
-var ModelRequestCaptureModel = class {
-  constructor(inner) {
-    this.inner = inner;
+function isJsonObjectBody(value) {
+  if (!value || typeof value !== "object") return false;
+  if (value instanceof ArrayBuffer || ArrayBuffer.isView(value)) return false;
+  if (value instanceof Blob || value instanceof FormData || value instanceof URLSearchParams) {
+    return false;
+  }
+  if (value instanceof ReadableStream) return false;
+  return !(Symbol.asyncIterator in value) && !(Symbol.iterator in value);
+}
+function isJsonRecordBody(value) {
+  return isJsonObjectBody(value) && !Array.isArray(value);
+}
+function mergePolicyHeaders(headers, additions) {
+  const replacedNames = new Set(Object.keys(additions).map((name) => name.toLowerCase()));
+  if (headers instanceof Headers) {
+    const merged2 = new Headers(headers);
+    for (const [name, value] of Object.entries(additions)) merged2.set(name, value);
+    return merged2;
+  }
+  if (headers && typeof headers === "object" && "values" in headers && "nulls" in headers && headers.values instanceof Headers && headers.nulls instanceof Set) {
+    const resolved = headers;
+    const rows = [
+      ...resolved.values.entries(),
+      ...[...resolved.nulls].map((name) => [name, null])
+    ];
+    return mergePolicyHeaders(rows, additions);
+  }
+  if (Array.isArray(headers)) {
+    return [
+      ...headers.filter(
+        (row) => !Array.isArray(row) || typeof row[0] !== "string" || !replacedNames.has(row[0].toLowerCase())
+      ),
+      ...Object.entries(additions)
+    ];
+  }
+  const merged = { ...headers ?? {} };
+  for (const name of Object.keys(merged)) {
+    if (replacedNames.has(name.toLowerCase())) delete merged[name];
+  }
+  return { ...merged, ...additions };
+}
+function* jsonTextChunks(value) {
+  const fragments = jsonValueFragments(value, "", /* @__PURE__ */ new Set(), 0);
+  if (!fragments) throw new TypeError("Model request body is not JSON serializable");
+  let buffered = "";
+  for (const fragment of fragments) {
+    if (buffered.length > 0 && buffered.length + fragment.length > TARGET_CHUNK_CHARS) {
+      yield buffered;
+      buffered = "";
+    }
+    if (fragment.length >= TARGET_CHUNK_CHARS) yield fragment;
+    else buffered += fragment;
+  }
+  if (buffered.length > 0) yield buffered;
+}
+function jsonValueFragments(source, key, ancestors, depth) {
+  if (typeof source === "string" && source.length > LARGE_STRING_SOURCE_CHARS) {
+    return jsonStringFragments(source);
+  }
+  if (depth >= 2 || depth > 0 && !Array.isArray(source)) {
+    return nativeJsonPropertyFragments(source, key);
+  }
+  const value = prepareJsonValue(source, key);
+  if (value === void 0 || typeof value === "function" || typeof value === "symbol") {
+    return null;
+  }
+  if (typeof value === "bigint") {
+    throw new TypeError("Do not know how to serialize a BigInt");
+  }
+  if (typeof value === "string") {
+    return jsonStringFragments(value);
+  }
+  if (value === null || typeof value !== "object") {
+    return [JSON.stringify(value)];
+  }
+  const rawJson = JSON;
+  if (rawJson.isRawJSON?.(value)) return [JSON.stringify(value)];
+  if (ancestors.has(value)) throw new TypeError("cyclic object value");
+  return Array.isArray(value) ? jsonArrayFragments(value, ancestors, depth) : jsonObjectFragments(value, ancestors, depth);
+}
+function* jsonArrayFragments(value, ancestors, depth) {
+  ancestors.add(value);
+  try {
+    const length = value.length;
+    yield "[";
+    for (let index = 0; index < length; index += 1) {
+      if (index > 0) yield ",";
+      const child = jsonValueFragments(value[index], String(index), ancestors, depth + 1);
+      if (child) yield* child;
+      else yield "null";
+    }
+    yield "]";
+  } finally {
+    ancestors.delete(value);
+  }
+}
+function* jsonObjectFragments(value, ancestors, depth) {
+  ancestors.add(value);
+  try {
+    const keys = Object.keys(value);
+    let emitted = false;
+    yield "{";
+    for (const key of keys) {
+      const child = jsonValueFragments(value[key], key, ancestors, depth + 1);
+      if (!child) continue;
+      if (emitted) yield ",";
+      emitted = true;
+      yield* jsonStringFragments(key);
+      yield ":";
+      yield* child;
+    }
+    yield "}";
+  } finally {
+    ancestors.delete(value);
   }
-  async getResponse(request) {
-    await beforeModelRequest();
-    rememberPreparedModelRequest(request);
-    void notifyModelRequestCapture(request);
-    return this.inner.getResponse(request);
+}
+function nativeJsonPropertyFragments(source, key) {
+  const holder = /* @__PURE__ */ Object.create(null);
+  holder[key] = source;
+  const serialized = JSON.stringify(holder);
+  if (serialized === "{}") return null;
+  const prefix = `{${JSON.stringify(key)}:`;
+  if (!serialized.startsWith(prefix) || !serialized.endsWith("}")) {
+    throw new TypeError("Model request body is not JSON serializable");
   }
-  async *getStreamedResponse(request) {
-    await beforeModelRequest();
-    rememberPreparedModelRequest(request);
-    void notifyModelRequestCapture(request);
-    for await (const event of this.inner.getStreamedResponse(request)) {
-      const settlement = modelResponseSettlement(event);
-      void settlement?.catch(() => void 0);
-      yield event;
+  return boundedNativeJsonFragments(serialized.slice(prefix.length, -1));
+}
+function* boundedNativeJsonFragments(serialized) {
+  for (let start = 0; start < serialized.length; ) {
+    let end = Math.min(serialized.length, start + TARGET_CHUNK_CHARS);
+    if (end < serialized.length && serialized.charCodeAt(end - 1) >= 55296 && serialized.charCodeAt(end - 1) <= 56319 && serialized.charCodeAt(end) >= 56320 && serialized.charCodeAt(end) <= 57343) {
+      end -= 1;
     }
+    yield serialized.slice(start, end);
+    start = end;
   }
-  getRetryAdvice(args) {
-    return this.inner.getRetryAdvice?.(args);
+}
+function prepareJsonValue(source, key) {
+  let value = source;
+  const sourceType = typeof value;
+  if (value !== null && (sourceType === "object" || sourceType === "function" || sourceType === "bigint")) {
+    const toJSON = value.toJSON;
+    if (typeof toJSON === "function") value = toJSON.call(value, key);
   }
-};
-var ModelRequestCaptureProvider = class {
-  constructor(inner) {
-    this.inner = inner;
+  if (value !== null && typeof value === "object") {
+    if (utilTypes.isNumberObject(value)) return Number.prototype.valueOf.call(value);
+    if (utilTypes.isStringObject(value)) return String.prototype.valueOf.call(value);
+    if (utilTypes.isBooleanObject(value)) return Boolean.prototype.valueOf.call(value);
+    if (utilTypes.isBigIntObject(value)) return BigInt.prototype.valueOf.call(value);
   }
-  async getModel(modelName) {
-    const model = await this.inner.getModel(modelName);
-    if (model instanceof ModelRequestCaptureModel) return model;
-    return new ModelRequestCaptureModel(model);
+  return value;
+}
+function* jsonStringFragments(value) {
+  if (value.length <= LARGE_STRING_SOURCE_CHARS) {
+    yield JSON.stringify(value);
+    return;
   }
-};
+  yield '"';
+  for (let start = 0; start < value.length; ) {
+    let end = Math.min(value.length, start + STRING_SLICE_CODE_UNITS);
+    if (end < value.length && value.charCodeAt(end - 1) >= 55296 && value.charCodeAt(end - 1) <= 56319 && value.charCodeAt(end) >= 56320 && value.charCodeAt(end) <= 57343) {
+      end -= 1;
+    }
+    const encoded = JSON.stringify(value.slice(start, end));
+    yield encoded.slice(1, -1);
+    start = end;
+  }
+  yield '"';
+}
 
 // src/model-preparation-diagnostics.ts
-var modelPreparationObserver = new AsyncLocalStorage3();
-var modelTransportStartedObserver = new AsyncLocalStorage3();
-var modelTransportAdmission = new AsyncLocalStorage3();
-var modelTransportRefusals = /* @__PURE__ */ new WeakMap();
+import { AsyncLocalStorage as AsyncLocalStorage2 } from "async_hooks";
+var modelPreparationObserver = new AsyncLocalStorage2();
+var modelTransportStartedObserver = new AsyncLocalStorage2();
 var ModelPreparationTraceProcessor = class {
   async onTraceStart(_trace) {
   }
@@ -9669,38 +9781,7 @@ function withModelTransportStartedObserver(observer, callback) {
   return observer ? modelTransportStartedObserver.run(observer, callback) : callback();
 }
 async function recordModelTransportStarted() {
-  try {
-    await beforeModelRequest();
-    await modelTransportStartedObserver.getStore()?.();
-  } catch (error) {
-    const admission = modelTransportAdmission.getStore();
-    if (admission) admission.refusal = { error };
-    throw error;
-  }
-}
-function sdkModelTransportAdmissionFetch(inner) {
-  return async (input, init) => {
-    const admission = {};
-    return modelTransportAdmission.run(admission, async () => {
-      try {
-        return await inner(input, init);
-      } catch (error) {
-        if (!admission.refusal || admission.refusal.error !== error) throw error;
-        const response = new Response(null, {
-          status: 400,
-          headers: { "x-should-retry": "false" }
-        });
-        modelTransportRefusals.set(response.headers, admission.refusal);
-        return response;
-      }
-    });
-  };
-}
-function rethrowModelTransportAdmissionRefusal(headers) {
-  const refusal = modelTransportRefusals.get(headers);
-  if (!refusal) return;
-  modelTransportRefusals.delete(headers);
-  throw refusal.error;
+  await modelTransportStartedObserver.getStore()?.();
 }
 function markModelPreparationFirstSandboxOperation(durationSeconds) {
   try {
@@ -9865,300 +9946,158 @@ function withModelPreparationClientDiagnostics(client) {
             durationSeconds: (performance.now() - startedAt) / 1e3
           });
         }
-      };
-    }
-  });
-}
-function recordModelPreparationManifestInventory(phase, manifest) {
-  const startedAt = performance.now();
-  let outcome = "completed";
-  let count = 0;
-  try {
-    if (manifest && typeof manifest === "object" && "iterEntries" in manifest && typeof manifest.iterEntries === "function") {
-      for (const _entry of manifest.iterEntries()) count += 1;
-    }
-  } catch {
-    outcome = "failed";
-  } finally {
-    recordModelPreparationMeasurement({
-      phase,
-      outcome,
-      durationSeconds: (performance.now() - startedAt) / 1e3,
-      count
-    });
-  }
-}
-
-// src/replayable-json-body.ts
-var TARGET_CHUNK_CHARS = 64 * 1024;
-var LARGE_STRING_SOURCE_CHARS = 32 * 1024;
-var STRING_SLICE_CODE_UNITS = 10 * 1024;
-var REPLAYABLE_REQUEST_BODY_FACTORY = Symbol.for(
-  "opengeni.replayable-request-body-factory"
-);
-var ReplayableJsonBody = class {
-  source;
-  constructor(source) {
-    this.source = source;
-  }
-  async *[Symbol.asyncIterator]() {
-    const encoder = new TextEncoder();
-    for (const chunk of jsonTextChunks(this.source)) {
-      yield encoder.encode(chunk);
-    }
-  }
-  createStream() {
-    const iterator = this[Symbol.asyncIterator]();
-    return new ReadableStream({
-      async pull(controller) {
-        try {
-          const next = await iterator.next();
-          if (next.done) controller.close();
-          else controller.enqueue(next.value);
-        } catch (error) {
-          controller.error(error);
-        }
-      },
-      async cancel(reason) {
-        await iterator.return?.(reason).catch(() => void 0);
-      }
-    });
-  }
-};
-var ReplayableJsonOpenAI = class extends OpenAI6 {
-  modelRequestPolicy;
-  constructor(options, hooks = {}) {
-    super({
-      ...options,
-      fetch: sdkModelTransportAdmissionFetch(options?.fetch ?? globalThis.fetch)
-    });
-    this.modelRequestPolicy = hooks.modelRequestPolicy;
-  }
-  makeStatusError(status, error, message, headers) {
-    rethrowModelTransportAdmissionRefusal(headers);
-    return super.makeStatusError(status, error, message, headers);
-  }
-  post(path, opts) {
-    if (!isModelRequestPath(path) || opts === void 0) {
-      return super.post(path, opts);
-    }
-    const wrapped = Promise.resolve(opts).then((resolved) => {
-      const body = resolved.body;
-      let prepared = resolved;
-      if (this.modelRequestPolicy && isJsonRecordBody(body)) {
-        const policy = this.modelRequestPolicy({ path, body });
-        if (policy?.body || policy?.headers) {
-          prepared = {
-            ...resolved,
-            ...policy.body ? { body: policy.body } : {},
-            ...policy.headers ? { headers: mergePolicyHeaders(resolved.headers, policy.headers) } : {}
-          };
-        }
-      }
-      return wrapJsonRequestOptions(prepared);
-    });
-    return super.post(path, wrapped);
-  }
-  async buildRequest(inputOptions, context) {
-    const replayable = inputOptions.body instanceof ReplayableJsonBody ? inputOptions.body : void 0;
-    const priorStreamingMarker = inputOptions.__metadata?.hasStreamingBody;
-    const built = await super.buildRequest(inputOptions, context);
-    if (!replayable) return built;
-    if (priorStreamingMarker === void 0) {
-      if (inputOptions.__metadata) delete inputOptions.__metadata.hasStreamingBody;
-    } else {
-      inputOptions.__metadata = {
-        ...inputOptions.__metadata,
-        hasStreamingBody: priorStreamingMarker
-      };
-    }
-    const request = built.req;
-    request[REPLAYABLE_REQUEST_BODY_FACTORY] = () => replayable.createStream();
-    return built;
-  }
-};
-function isModelRequestPath(path) {
-  const pathname = path.split("?", 1)[0] ?? path;
-  return pathname.endsWith("/responses") || pathname.endsWith("/chat/completions");
-}
-function wrapJsonRequestOptions(options) {
-  const body = options.body;
-  if (!isJsonObjectBody(body) || body instanceof ReplayableJsonBody) return options;
-  return {
-    ...options,
-    body: new ReplayableJsonBody(body),
-    headers: mergePolicyHeaders(options.headers, { "content-type": "application/json" })
-  };
-}
-function isJsonObjectBody(value) {
-  if (!value || typeof value !== "object") return false;
-  if (value instanceof ArrayBuffer || ArrayBuffer.isView(value)) return false;
-  if (value instanceof Blob || value instanceof FormData || value instanceof URLSearchParams) {
-    return false;
-  }
-  if (value instanceof ReadableStream) return false;
-  return !(Symbol.asyncIterator in value) && !(Symbol.iterator in value);
-}
-function isJsonRecordBody(value) {
-  return isJsonObjectBody(value) && !Array.isArray(value);
-}
-function mergePolicyHeaders(headers, additions) {
-  const replacedNames = new Set(Object.keys(additions).map((name) => name.toLowerCase()));
-  if (headers instanceof Headers) {
-    const merged2 = new Headers(headers);
-    for (const [name, value] of Object.entries(additions)) merged2.set(name, value);
-    return merged2;
-  }
-  if (headers && typeof headers === "object" && "values" in headers && "nulls" in headers && headers.values instanceof Headers && headers.nulls instanceof Set) {
-    const resolved = headers;
-    const rows = [
-      ...resolved.values.entries(),
-      ...[...resolved.nulls].map((name) => [name, null])
-    ];
-    return mergePolicyHeaders(rows, additions);
-  }
-  if (Array.isArray(headers)) {
-    return [
-      ...headers.filter(
-        (row) => !Array.isArray(row) || typeof row[0] !== "string" || !replacedNames.has(row[0].toLowerCase())
-      ),
-      ...Object.entries(additions)
-    ];
-  }
-  const merged = { ...headers ?? {} };
-  for (const name of Object.keys(merged)) {
-    if (replacedNames.has(name.toLowerCase())) delete merged[name];
-  }
-  return { ...merged, ...additions };
-}
-function* jsonTextChunks(value) {
-  const fragments = jsonValueFragments(value, "", /* @__PURE__ */ new Set(), 0);
-  if (!fragments) throw new TypeError("Model request body is not JSON serializable");
-  let buffered = "";
-  for (const fragment of fragments) {
-    if (buffered.length > 0 && buffered.length + fragment.length > TARGET_CHUNK_CHARS) {
-      yield buffered;
-      buffered = "";
+      };
     }
-    if (fragment.length >= TARGET_CHUNK_CHARS) yield fragment;
-    else buffered += fragment;
-  }
-  if (buffered.length > 0) yield buffered;
-}
-function jsonValueFragments(source, key, ancestors, depth) {
-  if (typeof source === "string" && source.length > LARGE_STRING_SOURCE_CHARS) {
-    return jsonStringFragments(source);
-  }
-  if (depth >= 2 || depth > 0 && !Array.isArray(source)) {
-    return nativeJsonPropertyFragments(source, key);
-  }
-  const value = prepareJsonValue(source, key);
-  if (value === void 0 || typeof value === "function" || typeof value === "symbol") {
-    return null;
-  }
-  if (typeof value === "bigint") {
-    throw new TypeError("Do not know how to serialize a BigInt");
-  }
-  if (typeof value === "string") {
-    return jsonStringFragments(value);
-  }
-  if (value === null || typeof value !== "object") {
-    return [JSON.stringify(value)];
-  }
-  const rawJson = JSON;
-  if (rawJson.isRawJSON?.(value)) return [JSON.stringify(value)];
-  if (ancestors.has(value)) throw new TypeError("cyclic object value");
-  return Array.isArray(value) ? jsonArrayFragments(value, ancestors, depth) : jsonObjectFragments(value, ancestors, depth);
+  });
 }
-function* jsonArrayFragments(value, ancestors, depth) {
-  ancestors.add(value);
+function recordModelPreparationManifestInventory(phase, manifest) {
+  const startedAt = performance.now();
+  let outcome = "completed";
+  let count = 0;
   try {
-    const length = value.length;
-    yield "[";
-    for (let index = 0; index < length; index += 1) {
-      if (index > 0) yield ",";
-      const child = jsonValueFragments(value[index], String(index), ancestors, depth + 1);
-      if (child) yield* child;
-      else yield "null";
+    if (manifest && typeof manifest === "object" && "iterEntries" in manifest && typeof manifest.iterEntries === "function") {
+      for (const _entry of manifest.iterEntries()) count += 1;
     }
-    yield "]";
+  } catch {
+    outcome = "failed";
   } finally {
-    ancestors.delete(value);
+    recordModelPreparationMeasurement({
+      phase,
+      outcome,
+      durationSeconds: (performance.now() - startedAt) / 1e3,
+      count
+    });
   }
 }
-function* jsonObjectFragments(value, ancestors, depth) {
-  ancestors.add(value);
-  try {
-    const keys = Object.keys(value);
-    let emitted = false;
-    yield "{";
-    for (const key of keys) {
-      const child = jsonValueFragments(value[key], key, ancestors, depth + 1);
-      if (!child) continue;
-      if (emitted) yield ",";
-      emitted = true;
-      yield* jsonStringFragments(key);
-      yield ":";
-      yield* child;
-    }
-    yield "}";
-  } finally {
-    ancestors.delete(value);
-  }
+
+// src/model-request-capture.ts
+import { AsyncLocalStorage as AsyncLocalStorage3 } from "async_hooks";
+var modelRequestCapture = new AsyncLocalStorage3();
+var captureIndices = /* @__PURE__ */ new WeakMap();
+function nextModelContextCaptureIndex(agent) {
+  const index = (captureIndices.get(agent) ?? 0) + 1;
+  captureIndices.set(agent, index);
+  return index;
 }
-function nativeJsonPropertyFragments(source, key) {
-  const holder = /* @__PURE__ */ Object.create(null);
-  holder[key] = source;
-  const serialized = JSON.stringify(holder);
-  if (serialized === "{}") return null;
-  const prefix = `{${JSON.stringify(key)}:`;
-  if (!serialized.startsWith(prefix) || !serialized.endsWith("}")) {
-    throw new TypeError("Model request body is not JSON serializable");
-  }
-  return boundedNativeJsonFragments(serialized.slice(prefix.length, -1));
+function withModelRequestCapture(capture, fn) {
+  return capture ? modelRequestCapture.run(capture, fn) : fn();
 }
-function* boundedNativeJsonFragments(serialized) {
-  for (let start = 0; start < serialized.length; ) {
-    let end = Math.min(serialized.length, start + TARGET_CHUNK_CHARS);
-    if (end < serialized.length && serialized.charCodeAt(end - 1) >= 55296 && serialized.charCodeAt(end - 1) <= 56319 && serialized.charCodeAt(end) >= 56320 && serialized.charCodeAt(end) <= 57343) {
-      end -= 1;
+function captureProviderRequestBody(provider, input, init) {
+  const context = modelRequestCapture.getStore();
+  const callback = context?.onProviderRequest;
+  if (!callback) return { init, captured: Promise.resolve(), cancel: () => {
+  } };
+  const index = context.nextProviderRequestIndex?.();
+  const observer = (providerId, body2, reason) => callback(providerId, body2, reason, index);
+  let cancel = () => {
+  };
+  let body = init?.body;
+  let nextInit = init;
+  if (body instanceof ReadableStream) {
+    const [sent, inspected] = body.tee();
+    nextInit = { ...init, body: sent };
+    body = inspected;
+  } else if (body == null && input instanceof Request) {
+    body = input.clone().body;
+  }
+  const captured = (async () => {
+    const limit = 4 * 1024 * 1024;
+    let text2;
+    if (typeof body === "string") {
+      if (Buffer.byteLength(body, "utf8") > limit) {
+        await observer(provider, null, "Request exceeds the 4 MiB capture limit.");
+        return;
+      }
+      text2 = body;
+    } else if (body instanceof ReadableStream) {
+      const reader = body.getReader();
+      let cancelled = false;
+      cancel = () => {
+        cancelled = true;
+        void reader.cancel().catch(() => void 0);
+      };
+      const signal = init?.signal ?? (input instanceof Request ? input.signal : void 0);
+      signal?.addEventListener("abort", cancel, { once: true });
+      if (signal?.aborted) cancel();
+      const decoder = new TextDecoder("utf-8", { fatal: true });
+      const chunks = [];
+      let bytes = 0;
+      try {
+        while (true) {
+          const chunk = await reader.read();
+          if (chunk.done) break;
+          bytes += chunk.value.byteLength;
+          if (bytes > limit) {
+            void reader.cancel().catch(() => void 0);
+            await observer(provider, null, "Request exceeds the 4 MiB capture limit.");
+            return;
+          }
+          chunks.push(decoder.decode(chunk.value, { stream: true }));
+        }
+        if (cancelled) return;
+        chunks.push(decoder.decode());
+        text2 = chunks.join("");
+      } catch {
+        void reader.cancel().catch(() => void 0);
+        if (!cancelled)
+          await observer(provider, null, "The provider body could not be read as UTF-8.");
+        return;
+      } finally {
+        signal?.removeEventListener("abort", cancel);
+        reader.releaseLock();
+        cancel = () => {
+        };
+      }
+    } else {
+      await observer(provider, null, "This transport body cannot be inspected.");
+      return;
     }
-    yield serialized.slice(start, end);
-    start = end;
+    await observer(provider, text2);
+  })().catch(() => void 0);
+  return { init: nextInit, captured, cancel: () => cancel() };
+}
+async function notifyModelRequestCapture(request) {
+  const capture = modelRequestCapture.getStore();
+  if (!capture) return;
+  try {
+    await capture(snapshotModelRequestPrefix(request));
+  } catch {
   }
 }
-function prepareJsonValue(source, key) {
-  let value = source;
-  const sourceType = typeof value;
-  if (value !== null && (sourceType === "object" || sourceType === "function" || sourceType === "bigint")) {
-    const toJSON = value.toJSON;
-    if (typeof toJSON === "function") value = toJSON.call(value, key);
+function snapshotModelRequestPrefix(request) {
+  return {
+    ...request,
+    tools: structuredClone(request.tools)
+  };
+}
+var ModelRequestCaptureModel = class {
+  constructor(inner) {
+    this.inner = inner;
   }
-  if (value !== null && typeof value === "object") {
-    if (utilTypes.isNumberObject(value)) return Number.prototype.valueOf.call(value);
-    if (utilTypes.isStringObject(value)) return String.prototype.valueOf.call(value);
-    if (utilTypes.isBooleanObject(value)) return Boolean.prototype.valueOf.call(value);
-    if (utilTypes.isBigIntObject(value)) return BigInt.prototype.valueOf.call(value);
+  async getResponse(request) {
+    rememberPreparedModelRequest(request);
+    void notifyModelRequestCapture(request);
+    return this.inner.getResponse(request);
   }
-  return value;
-}
-function* jsonStringFragments(value) {
-  if (value.length <= LARGE_STRING_SOURCE_CHARS) {
-    yield JSON.stringify(value);
-    return;
+  async *getStreamedResponse(request) {
+    rememberPreparedModelRequest(request);
+    void notifyModelRequestCapture(request);
+    yield* this.inner.getStreamedResponse(request);
   }
-  yield '"';
-  for (let start = 0; start < value.length; ) {
-    let end = Math.min(value.length, start + STRING_SLICE_CODE_UNITS);
-    if (end < value.length && value.charCodeAt(end - 1) >= 55296 && value.charCodeAt(end - 1) <= 56319 && value.charCodeAt(end) >= 56320 && value.charCodeAt(end) <= 57343) {
-      end -= 1;
-    }
-    const encoded = JSON.stringify(value.slice(start, end));
-    yield encoded.slice(1, -1);
-    start = end;
+  getRetryAdvice(args) {
+    return this.inner.getRetryAdvice?.(args);
   }
-  yield '"';
-}
+};
+var ModelRequestCaptureProvider = class {
+  constructor(inner) {
+    this.inner = inner;
+  }
+  async getModel(modelName) {
+    const model = await this.inner.getModel(modelName);
+    if (model instanceof ModelRequestCaptureModel) return model;
+    return new ModelRequestCaptureModel(model);
+  }
+};
 
 // src/provider-quota.ts
 import { APIError } from "openai";
@@ -10410,19 +10349,12 @@ function withoutQuotaExhaustedRetries(inner) {
 // src/claude-subscription-usage.ts
 import { AsyncLocalStorage as AsyncLocalStorage4 } from "async_hooks";
 var observers = new AsyncLocalStorage4();
-function withClaudeUsageObserver(observer, run2, prepare) {
-  return observers.run({ observe: observer, ...prepare ? { prepare } : {} }, run2);
-}
-async function prepareClaudeSubscriptionRequest(providerId, input, init) {
-  const prepare = observers.getStore()?.prepare;
-  if (!prepare) return init;
-  const headers = new Headers(input instanceof Request ? input.headers : void 0);
-  new Headers(init?.headers).forEach((value, key) => headers.set(key, value));
-  return { ...init, headers: await prepare(providerId, headers) };
+function withClaudeUsageObserver(observer, run2) {
+  return observers.run(observer, run2);
 }
 function observeClaudeUsageResponse(providerId, response) {
   try {
-    observers.getStore()?.observe(providerId, response);
+    observers.getStore()?.(providerId, response);
   } catch {
   }
 }
@@ -10695,7 +10627,6 @@ function instrumentedModelFetch(provider, inner) {
     if (!isModelCallFetch(input)) {
       return await inner(input, init);
     }
-    init = await prepareClaudeSubscriptionRequest(provider, input, init);
     await recordModelTransportStarted();
     const capture = captureProviderRequestBody(provider, input, init);
     const started = performance.now();
@@ -10762,7 +10693,7 @@ var RunMcpCredentials = class {
   #closed = false;
   #onAbort = () => this.close();
   constructor(targets, options = {}) {
-    this.#targets = targets.filter((target) => !target.connectionRef).map(({ id, url }) => ({ id, url }));
+    this.#targets = targets.map(({ id, url }) => ({ id, url }));
     this.#signal = options.signal;
     this.#now = options.now ?? Date.now;
     this.#signal?.addEventListener("abort", this.#onAbort, { once: true });
@@ -10839,11 +10770,11 @@ var RunMcpCredentials = class {
   replace(material) {
     this.prepare(material)();
   }
-  /** Refuse credentials for native, omitted, local, or rewritten server routes. */
+  /** Refuse credentials for omitted, local, or rewritten server routes. */
   assertRemoteTargets(targets) {
     this.#remoteTargets = this.#targets.filter(
       (original) => targets.some(
-        (target) => !target.connectionRef && target.id === original.id && normalizedUrl(target.url) !== null && normalizedUrl(target.url) === normalizedUrl(original.url)
+        (target) => target.id === original.id && normalizedUrl(target.url) !== null && normalizedUrl(target.url) === normalizedUrl(original.url)
       )
     );
     this.#dropExcluded();
@@ -10886,10 +10817,6 @@ var RunMcpCredentials = class {
   /** Called at the literal fetch boundary, including POST, SSE GET and DELETE. */
   requestInit(target, input, init) {
     this.#assertOpen();
-    if (target.connectionRef) {
-      this.excludeLocalTarget(target.id);
-      return init;
-    }
     this.assertAvailable(target.id);
     const entry = this.#entries.get(target.id);
     if (!entry) return init;
@@ -12089,7 +12016,6 @@ var LazyToolModel = class {
     this.runtime = runtime;
   }
   async getResponse(request) {
-    await beforeModelRequest();
     const prepared = prepareLazyToolRequest(request, this.runtime);
     rememberPreparedModelRequest(prepared);
     void notifyModelRequestCapture(prepared);
@@ -12100,7 +12026,6 @@ var LazyToolModel = class {
     return this.runtime.transport === "generic_dispatch" ? transformGenericDispatchResponse(response, this.runtime) : response;
   }
   async *getStreamedResponse(request) {
-    await beforeModelRequest();
     const prepared = prepareLazyToolRequest(request, this.runtime);
     rememberPreparedModelRequest(prepared);
     void notifyModelRequestCapture(prepared);
@@ -12110,18 +12035,19 @@ var LazyToolModel = class {
           await this.runtime.ensurePrepared();
         }
       }
-      const delivered = this.runtime.transport === "generic_dispatch" && event.type === "response_done" ? {
-        ...event,
-        response: {
-          ...event.response,
-          output: event.response.output.flatMap(
-            (item) => transformGenericDispatchCall(item, this.runtime)
-          )
-        }
-      } : event;
-      const settlement = modelResponseSettlement(delivered);
-      void settlement?.catch(() => void 0);
-      yield delivered;
+      if (this.runtime.transport === "generic_dispatch" && event.type === "response_done") {
+        yield {
+          ...event,
+          response: {
+            ...event.response,
+            output: event.response.output.flatMap(
+              (item) => transformGenericDispatchCall(item, this.runtime)
+            )
+          }
+        };
+      } else {
+        yield event;
+      }
     }
   }
   getRetryAdvice(args) {
@@ -13771,7 +13697,7 @@ import { existsSync, readdirSync } from "fs";
 // ../sdk/package.json
 var package_default = {
   name: "@opengeni/sdk",
-  version: "7.4.0-canary.36844267522001",
+  version: "7.4.0",
   description: "Framework-agnostic TypeScript SDK for the OpenGeni API: typed client, session lifecycle, SSE event streaming with reconnect + replay-by-sequence, and proxy re-streaming helpers.",
   license: "Apache-2.0",
   repository: {
@@ -13793,10 +13719,6 @@ var package_default = {
   module: "./src/index.ts",
   types: "./src/index.ts",
   exports: {
-    "./allowance-refusal": {
-      types: "./src/allowance-refusal.ts",
-      default: "./src/allowance-refusal.ts"
-    },
     ".": {
       types: "./src/index.ts",
       default: "./src/index.ts"
@@ -13897,10 +13819,6 @@ var package_default = {
       types: "./src/chat/index.ts",
       default: "./src/chat/index.ts"
     },
-    "./session-proxy": {
-      types: "./src/session-proxy.ts",
-      default: "./src/session-proxy.ts"
-    },
     "./next": {
       types: "./src/adapters/next.ts",
       default: "./src/adapters/next.ts"
@@ -13947,7 +13865,7 @@ var package_default = {
 // ../react/package.json
 var package_default2 = {
   name: "@opengeni/react",
-  version: "7.4.0-canary.36844267522001",
+  version: "7.4.0",
   description: "React hooks and styled components for OpenGeni: live session streaming, chat composer, message timeline, session status, and fleet views \u2014 token-themed (CSS variables), dark-first, built on Tailwind v4 + Radix + Motion.",
   license: "Apache-2.0",
   repository: {
@@ -14204,7 +14122,7 @@ var package_default2 = {
 // ../codemode/package.json
 var package_default3 = {
   name: "@opengeni/codemode",
-  version: "0.6.5-canary.36844267522001",
+  version: "0.6.5",
   description: "Attempt-frozen programmatic tool catalog and execution authority for OpenGeni.",
   license: "Apache-2.0",
   repository: {
@@ -14254,7 +14172,7 @@ var package_default3 = {
 // ../ogtool/package.json
 var package_default4 = {
   name: "@opengeni/ogtool",
-  version: "0.3.47-canary.36844267522001",
+  version: "0.3.47",
   description: "Codemode CLI and typed client entrypoint for an OpenGeni execution attempt.",
   license: "Apache-2.0",
   repository: {
@@ -20755,7 +20673,7 @@ async function prepareAgentTools(settings, tools, options = {}) {
   options = { ...options, sessionAttachedRemoteMcpTargets: identityTargets };
   options.runMcpCredentials?.assertRemoteTargets(
     settings.mcpServers.filter(
-      (config) => tools.some((tool2) => tool2.id === config.id) && !config.connectionRef && !localRegistry.has(config.id) && !isFirstPartyMcpServer(settings, config) && !isCodexAppsMcpServer(config)
+      (config) => tools.some((tool2) => tool2.id === config.id) && !localRegistry.has(config.id) && !isFirstPartyMcpServer(settings, config) && !isCodexAppsMcpServer(config)
     )
   );
   const aggregateToolBudget = new McpAggregateToolListBudget();
@@ -21226,7 +21144,7 @@ function selectedSessionRemoteMcpTargets(settings, attachments, tools, localServ
     );
     if (configs.length !== 1) return [];
     const config = configs[0];
-    if (config.connectionRef || isFirstPartyMcpServer(settings, config) || isCodexAppsMcpServer(config) || BUILT_IN_MCP_BRIDGE_ADAPTERS.some(
+    if (isFirstPartyMcpServer(settings, config) || isCodexAppsMcpServer(config) || BUILT_IN_MCP_BRIDGE_ADAPTERS.some(
       (adapter) => adapter.matches({
         url: config.url,
         ...config.connectionRef ? { connectionRef: config.connectionRef } : {}
@@ -23414,15 +23332,9 @@ async function runAgentStream(agent, input, settings, overrides = {}) {
   const scope = gate?.beginStream(overrides.signal);
   try {
     if (scope) agent.toolUseBehavior = scope.toolUseBehavior;
-    const stream = await withModelCallLifecycle(
-      {
-        ...overrides.beforeModelRequest ? { beforeModelRequest: overrides.beforeModelRequest } : {},
-        ...overrides.onModelResponse ? { onModelResponse: overrides.onModelResponse } : {}
-      },
-      () => withPreparedCompactionRequest(
-        agent,
-        () => runAgentStreamInternal(agent, input, settings, overrides, scope)
-      )
+    const stream = await withPreparedCompactionRequest(
+      agent,
+      () => runAgentStreamInternal(agent, input, settings, overrides, scope)
     );
     if (scope) {
       const finish = () => scope.endStream(stream.cancelled);
@@ -26365,7 +26277,6 @@ export {
   normalizeVercelGatewayRequestBody,
   modelRequestPolicyForProvider,
   vercelGatewayRoutingFetch,
-  beforeModelRequest,
   markModelPreparationFirstSandboxOperation,
   recordModelPreparationMeasurement,
   PROVIDER_QUOTA_EXHAUSTED_CODE,
@@ -26572,4 +26483,4 @@ export {
   runAzureCliLoginHook,
   renderSessionGoalContext2 as renderSessionGoalContext
 };
-//# sourceMappingURL=chunk-XFDVXBB7.js.map
+//# sourceMappingURL=chunk-22M75RJF.js.map