npm - @bastani/atomic - Versions diffs - 0.9.24 → 0.9.25-alpha.2 - Mend

@bastani/atomic 0.9.24 → 0.9.25-alpha.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (430) hide show

package/dist/core/extensions/tool-types.d.ts CHANGED Viewed

@@ -6,7 +6,7 @@ import type { Theme } from "../../modes/interactive/theme/theme.js";
 import type { SourceInfo } from "../source-info.ts";
 import type { ToolConcurrency } from "../tools/tool-concurrency.ts";
 export type { ToolConcurrency, ToolConcurrencyMode } from "../tools/tool-concurrency.ts";
-import type { ExtensionContext } from "./context-types.ts";
+import type { ExtensionToolContext } from "./context-types.ts";
 /** Rendering options for tool results */
 export interface ToolRenderResultOptions {
     /** Whether the result view is expanded */
@@ -47,6 +47,29 @@ type ToolRenderCall<TParams extends TSchema, TState> = {
 type ToolRenderResult<TParams extends TSchema, TDetails, TState> = {
     bivarianceHack(result: AgentToolResult<TDetails>, options: ToolRenderResultOptions, theme: Theme, context: ToolRenderContext<TState, Static<TParams>>): Component;
 }["bivarianceHack"];
+/** How the model reaches a registered tool. */
+export type ToolExposure = "direct" | "model-only" | "codemode" | "deferred" | "hidden";
+export interface ToolAnnotations {
+    readOnlyHint?: boolean;
+    destructiveHint?: boolean;
+    idempotentHint?: boolean;
+    openWorldHint?: boolean;
+}
+export interface ToolNamespace {
+    name: string;
+    description?: string;
+}
+export interface ToolLoadout {
+    readonly declared: readonly import("@earendil-works/pi-agent-core").AgentTool[];
+    readonly callable: readonly import("@earendil-works/pi-agent-core").AgentTool[];
+    readonly registered: readonly import("@earendil-works/pi-agent-core").AgentTool[];
+    getExposure(name: string): ToolExposure;
+    getNamespace(name: string): ToolNamespace | undefined;
+}
+export interface ToolLoadoutChanges {
+    descriptions?: Readonly<Record<string, string>>;
+    hiddenDeclarations?: readonly string[];
+}
 /**
  * Tool definition for registerTool().
  */
@@ -73,6 +96,14 @@ export interface ToolDefinition<TParams extends TSchema = TSchema, TDetails = un
     renderShell?: "default" | "self";
     /** Optional compatibility shim to prepare raw tool call arguments before schema validation. Must return an object conforming to TParams. */
     prepareArguments?: (args: unknown) => Static<TParams>;
+    /** Schema of successful machine-readable results. */
+    outputSchema?: TSchema;
+    /** Direct and model-only tools activate by default; other exposures do not. */
+    exposure?: ToolExposure;
+    namespace?: ToolNamespace;
+    annotations?: ToolAnnotations;
+    defaultActive?: boolean;
+    prepareLoadout?: (loadout: ToolLoadout) => ToolLoadoutChanges | undefined;
     /**
      * Per-tool execution mode override.
      * - "sequential": this tool must execute one at a time with other tool calls.
@@ -89,7 +120,7 @@ export interface ToolDefinition<TParams extends TSchema = TSchema, TDetails = un
      */
     concurrency?: ToolConcurrency<Static<TParams>>;
     /** Execute the tool. */
-    execute(toolCallId: string, params: Static<TParams>, signal: AbortSignal | undefined, onUpdate: AgentToolUpdateCallback<TDetails> | undefined, ctx: ExtensionContext): Promise<AgentToolResult<TDetails>>;
+    execute(toolCallId: string, params: Static<TParams>, signal: AbortSignal | undefined, onUpdate: AgentToolUpdateCallback<TDetails> | undefined, ctx: ExtensionToolContext): Promise<AgentToolResult<TDetails>>;
     /** Custom rendering for tool call display */
     renderCall?: ToolRenderCall<TParams, TState>;
     /** Custom rendering for tool result display */
@@ -107,7 +138,7 @@ export declare function defineTool<TParams extends TSchema, TDetails = unknown,
 /** Public constrained-sampling shape inherited from pi-ai. */
 export type { ConstrainedSamplingConfig } from "@bastani/pi-ai/compat";
 /** Tool info with name, description, parameter schema, constraint, and source metadata */
-export type ToolInfo = Pick<ToolDefinition, "name" | "description" | "parameters" | "constrainedSampling" | "promptGuidelines"> & {
+export type ToolInfo = Pick<ToolDefinition, "name" | "description" | "parameters" | "constrainedSampling" | "promptGuidelines" | "exposure" | "namespace" | "annotations"> & {
     sourceInfo: SourceInfo;
 };
 //# sourceMappingURL=tool-types.d.ts.map

package/dist/core/extensions/tool-types.d.ts.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"tool-types.d.ts","sourceRoot":"","sources":["../../../src/core/extensions/tool-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,KAAK,EAAE,eAAe,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AACjH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wCAAwC,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAEpE,YAAY,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AAEzF,OAAO,KAAK,EAAE,~~gBAAgB~~,EAAE,MAAM,oBAAoB,CAAC;~~AAE3D~~,yCAAyC;AACzC,MAAM,WAAW,uBAAuB;IACvC,0CAA0C;IAC1C,QAAQ,EAAE,OAAO,CAAC;IAClB,iDAAiD;IACjD,SAAS,EAAE,OAAO,CAAC;CACnB;AAED,wCAAwC;AACxC,MAAM,WAAW,iBAAiB,CAAC,MAAM,GAAG,OAAO,EAAE,KAAK,GAAG,OAAO;IACnE,6FAA6F;IAC7F,IAAI,EAAE,KAAK,CAAC;IACZ,mGAAmG;IACnG,UAAU,EAAE,MAAM,CAAC;IACnB,gEAAgE;IAChE,UAAU,EAAE,MAAM,IAAI,CAAC;IACvB,kEAAkE;IAClE,aAAa,EAAE,SAAS,GAAG,SAAS,CAAC;IACrC,iFAAiF;IACjF,KAAK,EAAE,MAAM,CAAC;IACd,iDAAiD;IACjD,GAAG,EAAE,MAAM,CAAC;IACZ,8CAA8C;IAC9C,gBAAgB,EAAE,OAAO,CAAC;IAC1B,oDAAoD;IACpD,YAAY,EAAE,OAAO,CAAC;IACtB,oDAAoD;IACpD,SAAS,EAAE,OAAO,CAAC;IACnB,2CAA2C;IAC3C,QAAQ,EAAE,OAAO,CAAC;IAClB,4DAA4D;IAC5D,UAAU,EAAE,OAAO,CAAC;IACpB,8CAA8C;IAC9C,OAAO,EAAE,OAAO,CAAC;CACjB;AAED,KAAK,cAAc,CAAC,OAAO,SAAS,OAAO,EAAE,MAAM,IAAI;IACtD,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;CACpH,CAAC,gBAAgB,CAAC,CAAC;AAEpB,KAAK,gBAAgB,CAAC,OAAO,SAAS,OAAO,EAAE,QAAQ,EAAE,MAAM,IAAI;IAClE,cAAc,CACb,MAAM,EAAE,eAAe,CAAC,QAAQ,CAAC,EACjC,OAAO,EAAE,uBAAuB,EAChC,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,GACjD,SAAS,CAAC;CACb,CAAC,gBAAgB,CAAC,CAAC;AAEpB;;GAEG;AACH,MAAM,WAAW,cAAc,CAAC,OAAO,SAAS,OAAO,GAAG,OAAO,EAAE,QAAQ,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO;IACtG,yCAAyC;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,kCAAkC;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,0BAA0B;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,oKAAoK;IACpK,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oHAAoH;IACpH,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,iCAAiC;IACjC,UAAU,EAAE,OAAO,CAAC;IACpB,2FAA2F;IAC3F,mBAAmB,CAAC,EAAE,KAAK,GAAG,yBAAyB,CAAC;IACxD,4HAA4H;IAC5H,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,4KAA4K;IAC5K,gBAAgB,CAAC,EAAE,IAAI,CAAC;IACxB,sHAAsH;IACtH,WAAW,CAAC,EAAE,SAAS,GAAG,MAAM,CAAC;IAEjC,4IAA4I;IAC5I,gBAAgB,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC;~~IAEtD~~;;;;;;OAMG;IACH,aAAa,CAAC,EAAE,iBAAiB,CAAC;IAElC;;;;;OAKG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAE/C,wBAAwB;IACxB,OAAO,CACN,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,EACvB,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,QAAQ,EAAE,uBAAuB,CAAC,QAAQ,CAAC,GAAG,SAAS,EACvD,GAAG,EAAE,~~gBAAgB~~,~~GACnB~~,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEtC,6CAA6C;IAC7C,UAAU,CAAC,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAE7C,+CAA+C;IAC/C,YAAY,CAAC,EAAE,gBAAgB,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;CAC3D;AAED,KAAK,iBAAiB,GAAG,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAEnE;;;;;;GAMG;AACH,wBAAgB,UAAU,CAAC,OAAO,SAAS,OAAO,EAAE,QAAQ,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,EACvF,IAAI,EAAE,cAAc,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,GAC7C,cAAc,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,GAAG,iBAAiB,CAE/D;AAED,8DAA8D;AAC9D,YAAY,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAEvE,0FAA0F;AAC1F,MAAM,MAAM,QAAQ,GAAG,IAAI,CAC1B,cAAc,~~EACd~~,MAAM,~~GAAG~~,aAAa,~~GAAG~~,YAAY,~~GAAG~~,qBAAqB,~~GAAG~~,kBAAkB,~~CAClF~~,GAAG;IACH,UAAU,EAAE,UAAU,CAAC;CACvB,CAAC"}
1	+ {"version":3,"file":"tool-types.d.ts","sourceRoot":"","sources":["../../../src/core/extensions/tool-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,KAAK,EAAE,eAAe,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AACjH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,wCAAwC,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAEpE,YAAY,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AAEzF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAE/D,yCAAyC;AACzC,MAAM,WAAW,uBAAuB;IACvC,0CAA0C;IAC1C,QAAQ,EAAE,OAAO,CAAC;IAClB,iDAAiD;IACjD,SAAS,EAAE,OAAO,CAAC;CACnB;AAED,wCAAwC;AACxC,MAAM,WAAW,iBAAiB,CAAC,MAAM,GAAG,OAAO,EAAE,KAAK,GAAG,OAAO;IACnE,6FAA6F;IAC7F,IAAI,EAAE,KAAK,CAAC;IACZ,mGAAmG;IACnG,UAAU,EAAE,MAAM,CAAC;IACnB,gEAAgE;IAChE,UAAU,EAAE,MAAM,IAAI,CAAC;IACvB,kEAAkE;IAClE,aAAa,EAAE,SAAS,GAAG,SAAS,CAAC;IACrC,iFAAiF;IACjF,KAAK,EAAE,MAAM,CAAC;IACd,iDAAiD;IACjD,GAAG,EAAE,MAAM,CAAC;IACZ,8CAA8C;IAC9C,gBAAgB,EAAE,OAAO,CAAC;IAC1B,oDAAoD;IACpD,YAAY,EAAE,OAAO,CAAC;IACtB,oDAAoD;IACpD,SAAS,EAAE,OAAO,CAAC;IACnB,2CAA2C;IAC3C,QAAQ,EAAE,OAAO,CAAC;IAClB,4DAA4D;IAC5D,UAAU,EAAE,OAAO,CAAC;IACpB,8CAA8C;IAC9C,OAAO,EAAE,OAAO,CAAC;CACjB;AAED,KAAK,cAAc,CAAC,OAAO,SAAS,OAAO,EAAE,MAAM,IAAI;IACtD,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;CACpH,CAAC,gBAAgB,CAAC,CAAC;AAEpB,KAAK,gBAAgB,CAAC,OAAO,SAAS,OAAO,EAAE,QAAQ,EAAE,MAAM,IAAI;IAClE,cAAc,CACb,MAAM,EAAE,eAAe,CAAC,QAAQ,CAAC,EACjC,OAAO,EAAE,uBAAuB,EAChC,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,GACjD,SAAS,CAAC;CACb,CAAC,gBAAgB,CAAC,CAAC;AAEpB,+CAA+C;AAC/C,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,YAAY,GAAG,UAAU,GAAG,UAAU,GAAG,QAAQ,CAAC;AAExF,MAAM,WAAW,eAAe;IAC/B,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,aAAa,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,aAAa;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC3B,QAAQ,CAAC,QAAQ,EAAE,SAAS,OAAO,+BAA+B,EAAE,SAAS,EAAE,CAAC;IAChF,QAAQ,CAAC,QAAQ,EAAE,SAAS,OAAO,+BAA+B,EAAE,SAAS,EAAE,CAAC;IAChF,QAAQ,CAAC,UAAU,EAAE,SAAS,OAAO,+BAA+B,EAAE,SAAS,EAAE,CAAC;IAClF,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,YAAY,CAAC;IACxC,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS,CAAC;CACtD;AAED,MAAM,WAAW,kBAAkB;IAClC,YAAY,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAChD,kBAAkB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;CACvC;AAED;;GAEG;AACH,MAAM,WAAW,cAAc,CAAC,OAAO,SAAS,OAAO,GAAG,OAAO,EAAE,QAAQ,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO;IACtG,yCAAyC;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,kCAAkC;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,0BAA0B;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,oKAAoK;IACpK,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oHAAoH;IACpH,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,iCAAiC;IACjC,UAAU,EAAE,OAAO,CAAC;IACpB,2FAA2F;IAC3F,mBAAmB,CAAC,EAAE,KAAK,GAAG,yBAAyB,CAAC;IACxD,4HAA4H;IAC5H,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,4KAA4K;IAC5K,gBAAgB,CAAC,EAAE,IAAI,CAAC;IACxB,sHAAsH;IACtH,WAAW,CAAC,EAAE,SAAS,GAAG,MAAM,CAAC;IAEjC,4IAA4I;IAC5I,gBAAgB,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC;IACtD,qDAAqD;IACrD,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,+EAA+E;IAC/E,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,SAAS,CAAC,EAAE,aAAa,CAAC;IAC1B,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,kBAAkB,GAAG,SAAS,CAAC;IAE1E;;;;;;OAMG;IACH,aAAa,CAAC,EAAE,iBAAiB,CAAC;IAElC;;;;;OAKG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAE/C,wBAAwB;IACxB,OAAO,CACN,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,EACvB,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,QAAQ,EAAE,uBAAuB,CAAC,QAAQ,CAAC,GAAG,SAAS,EACvD,GAAG,EAAE,oBAAoB,GACvB,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEtC,6CAA6C;IAC7C,UAAU,CAAC,EAAE,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAE7C,+CAA+C;IAC/C,YAAY,CAAC,EAAE,gBAAgB,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;CAC3D;AAED,KAAK,iBAAiB,GAAG,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAEnE;;;;;;GAMG;AACH,wBAAgB,UAAU,CAAC,OAAO,SAAS,OAAO,EAAE,QAAQ,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,EACvF,IAAI,EAAE,cAAc,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,GAC7C,cAAc,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,GAAG,iBAAiB,CAE/D;AAED,8DAA8D;AAC9D,YAAY,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAEvE,0FAA0F;AAC1F,MAAM,MAAM,QAAQ,GAAG,IAAI,CAC1B,cAAc,EACZ,MAAM,GACN,aAAa,GACb,YAAY,GACZ,qBAAqB,GACrB,kBAAkB,GAClB,UAAU,GACV,WAAW,GACX,aAAa,CACf,GAAG;IACH,UAAU,EAAE,UAAU,CAAC;CACvB,CAAC"}

package/dist/core/extensions/tool-types.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"tool-types.js","sourceRoot":"","sources":["../../../src/core/extensions/tool-types.ts"],"names":[],"mappings":"~~AA4HA~~;;;;;;GAMG;AACH,MAAM,UAAU,UAAU,CACzB,IAA+C;IAE/C,OAAO,IAAqE,CAAC;AAC9E,CAAC","sourcesContent":["import type { ConstrainedSamplingConfig } from \"@bastani/pi-ai/compat\";\nimport type { AgentToolResult, AgentToolUpdateCallback, ToolExecutionMode } from \"@earendil-works/pi-agent-core\";\nimport type { Component } from \"@earendil-works/pi-tui\";\nimport type { Static, TSchema } from \"typebox\";\nimport type { Theme } from \"../../modes/interactive/theme/theme.js\";\nimport type { SourceInfo } from \"../source-info.ts\";\nimport type { ToolConcurrency } from \"../tools/tool-concurrency.ts\";\n\nexport type { ToolConcurrency, ToolConcurrencyMode } from \"../tools/tool-concurrency.ts\";\n\nimport type { ~~ExtensionContext~~ } from \"./context-types.ts\";\n\n/** Rendering options for tool results /\nexport interface ToolRenderResultOptions {\n\t/* Whether the result view is expanded /\n\texpanded: boolean;\n\t/* Whether this is a partial/streaming result /\n\tisPartial: boolean;\n}\n\n/* Context passed to tool renderers. /\nexport interface ToolRenderContext<TState = unknown, TArgs = unknown> {\n\t/* Current tool call arguments. Shared across call/result renders for the same tool call. /\n\targs: TArgs;\n\t/* Unique id for this tool execution. Stable across call/result renders for the same tool call. /\n\ttoolCallId: string;\n\t/* Invalidate just this tool execution component for redraw. /\n\tinvalidate: () => void;\n\t/* Previously returned component for this render slot, if any. /\n\tlastComponent: Component \| undefined;\n\t/* Shared renderer state for this tool row. Initialized by tool-execution.ts. /\n\tstate: TState;\n\t/* Working directory for this tool execution. /\n\tcwd: string;\n\t/* Whether the tool execution has started. /\n\texecutionStarted: boolean;\n\t/* Whether the tool call arguments are complete. /\n\targsComplete: boolean;\n\t/* Whether the tool result is partial/streaming. /\n\tisPartial: boolean;\n\t/* Whether the result view is expanded. /\n\texpanded: boolean;\n\t/* Whether inline images are currently shown in the TUI. /\n\tshowImages: boolean;\n\t/* Whether the current result is an error. /\n\tisError: boolean;\n}\n\ntype ToolRenderCall<TParams extends TSchema, TState> = {\n\tbivarianceHack(args: Static<TParams>, theme: Theme, context: ToolRenderContext<TState, Static<TParams>>): Component;\n}[\"bivarianceHack\"];\n\ntype ToolRenderResult<TParams extends TSchema, TDetails, TState> = {\n\tbivarianceHack(\n\t\tresult: AgentToolResult<TDetails>,\n\t\toptions: ToolRenderResultOptions,\n\t\ttheme: Theme,\n\t\tcontext: ToolRenderContext<TState, Static<TParams>>,\n\t): Component;\n}[\"bivarianceHack\"];\n\n/\n Tool definition for registerTool().\n /\nexport interface ToolDefinition<TParams extends TSchema = TSchema, TDetails = unknown, TState = unknown> {\n\t/* Tool name (used in LLM tool calls) /\n\tname: string;\n\t/* Human-readable label for UI /\n\tlabel: string;\n\t/* Description for LLM /\n\tdescription: string;\n\t/* Optional one-line snippet for the Available tools section in the default system prompt. Custom tools are omitted from that section when this is not provided. /\n\tpromptSnippet?: string;\n\t/* Optional guideline bullets appended to the default system prompt Guidelines section when this tool is active. /\n\tpromptGuidelines?: string[];\n\t/* Parameter schema (TypeBox) /\n\tparameters: TParams;\n\t/* Optional provider-side constrained sampling request. `false` explicitly disables it. /\n\tconstrainedSampling?: false \| ConstrainedSamplingConfig;\n\t/* Optional per-tool character cap for model-visible result persistence. Use Infinity to opt out for self-bounded tools. /\n\tmaxResultSizeChars?: number;\n\t/* Marks a terminating tool created by createStructuredOutputTool() so print mode can emit custom-named final JSON without treating every terminating tool as printable. /\n\tstructuredOutput?: true;\n\t/* Controls whether ToolExecutionComponent renders the standard colored shell or the tool renders its own framing. /\n\trenderShell?: \"default\" \| \"self\";\n\n\t/* Optional compatibility shim to prepare raw tool call arguments before schema validation. Must return an object conforming to TParams. /\n\tprepareArguments?: (args: unknown) => Static<TParams>;\n\n\t/\n\t Per-tool execution mode override.\n\t * - \"sequential\": this tool must execute one at a time with other tool calls.\n\t * - \"parallel\": this tool can execute concurrently with other tool calls.\n\t \n\t If omitted, the default execution mode applies.\n\t /\n\texecutionMode?: ToolExecutionMode;\n\n\t/\n\t Ordering against the other calls of the same assistant message.\n\t * - \"shared\": may overlap other shared calls, but starts after the latest earlier exclusive call (default)\n\t * - \"exclusive\": starts after every earlier call finishes; later calls start after it finishes\n\t * - function: resolved per call from its arguments; a throwing resolver means \"exclusive\"\n\t /\n\tconcurrency?: ToolConcurrency<Static<TParams>>;\n\n\t/* Execute the tool. /\n\texecute(\n\t\ttoolCallId: string,\n\t\tparams: Static<TParams>,\n\t\tsignal: AbortSignal \| undefined,\n\t\tonUpdate: AgentToolUpdateCallback<TDetails> \| undefined,\n\t\tctx: ~~ExtensionContext~~,\n\t): Promise<AgentToolResult<TDetails>>;\n\n\t/* Custom rendering for tool call display /\n\trenderCall?: ToolRenderCall<TParams, TState>;\n\n\t/* Custom rendering for tool result display /\n\trenderResult?: ToolRenderResult<TParams, TDetails, TState>;\n}\n\ntype AnyToolDefinition = ToolDefinition<TSchema, unknown, unknown>;\n\n/\n Preserve parameter inference for standalone tool definitions.\n \n Use this when assigning a tool to a variable or passing it through arrays such\n * as `customTools`, where contextual typing would otherwise widen params to\n * `unknown`.\n /\nexport function defineTool<TParams extends TSchema, TDetails = unknown, TState = unknown>(\n\ttool: ToolDefinition<TParams, TDetails, TState>,\n): ToolDefinition<TParams, TDetails, TState> & AnyToolDefinition {\n\treturn tool as ToolDefinition<TParams, TDetails, TState> & AnyToolDefinition;\n}\n\n/* Public constrained-sampling shape inherited from pi-ai. /\nexport type { ConstrainedSamplingConfig } from \"@bastani/pi-ai/compat\";\n\n/* Tool info with name, description, parameter schema, constraint, and source metadata */\nexport type ToolInfo = Pick<\n\tToolDefinition,\n\t\"name\" \| \"description\" \| \"parameters\" \| \"constrainedSampling\" \| \"promptGuidelines\"\n> & {\n\tsourceInfo: SourceInfo;\n};\n"]}
1	+ {"version":3,"file":"tool-types.js","sourceRoot":"","sources":["../../../src/core/extensions/tool-types.ts"],"names":[],"mappings":"AAgKA;;;;;;GAMG;AACH,MAAM,UAAU,UAAU,CACzB,IAA+C;IAE/C,OAAO,IAAqE,CAAC;AAC9E,CAAC","sourcesContent":["import type { ConstrainedSamplingConfig } from \"@bastani/pi-ai/compat\";\nimport type { AgentToolResult, AgentToolUpdateCallback, ToolExecutionMode } from \"@earendil-works/pi-agent-core\";\nimport type { Component } from \"@earendil-works/pi-tui\";\nimport type { Static, TSchema } from \"typebox\";\nimport type { Theme } from \"../../modes/interactive/theme/theme.js\";\nimport type { SourceInfo } from \"../source-info.ts\";\nimport type { ToolConcurrency } from \"../tools/tool-concurrency.ts\";\n\nexport type { ToolConcurrency, ToolConcurrencyMode } from \"../tools/tool-concurrency.ts\";\n\nimport type { ExtensionToolContext } from \"./context-types.ts\";\n\n/** Rendering options for tool results /\nexport interface ToolRenderResultOptions {\n\t/* Whether the result view is expanded /\n\texpanded: boolean;\n\t/* Whether this is a partial/streaming result /\n\tisPartial: boolean;\n}\n\n/* Context passed to tool renderers. /\nexport interface ToolRenderContext<TState = unknown, TArgs = unknown> {\n\t/* Current tool call arguments. Shared across call/result renders for the same tool call. /\n\targs: TArgs;\n\t/* Unique id for this tool execution. Stable across call/result renders for the same tool call. /\n\ttoolCallId: string;\n\t/* Invalidate just this tool execution component for redraw. /\n\tinvalidate: () => void;\n\t/* Previously returned component for this render slot, if any. /\n\tlastComponent: Component \| undefined;\n\t/* Shared renderer state for this tool row. Initialized by tool-execution.ts. /\n\tstate: TState;\n\t/* Working directory for this tool execution. /\n\tcwd: string;\n\t/* Whether the tool execution has started. /\n\texecutionStarted: boolean;\n\t/* Whether the tool call arguments are complete. /\n\targsComplete: boolean;\n\t/* Whether the tool result is partial/streaming. /\n\tisPartial: boolean;\n\t/* Whether the result view is expanded. /\n\texpanded: boolean;\n\t/* Whether inline images are currently shown in the TUI. /\n\tshowImages: boolean;\n\t/* Whether the current result is an error. /\n\tisError: boolean;\n}\n\ntype ToolRenderCall<TParams extends TSchema, TState> = {\n\tbivarianceHack(args: Static<TParams>, theme: Theme, context: ToolRenderContext<TState, Static<TParams>>): Component;\n}[\"bivarianceHack\"];\n\ntype ToolRenderResult<TParams extends TSchema, TDetails, TState> = {\n\tbivarianceHack(\n\t\tresult: AgentToolResult<TDetails>,\n\t\toptions: ToolRenderResultOptions,\n\t\ttheme: Theme,\n\t\tcontext: ToolRenderContext<TState, Static<TParams>>,\n\t): Component;\n}[\"bivarianceHack\"];\n\n/* How the model reaches a registered tool. /\nexport type ToolExposure = \"direct\" \| \"model-only\" \| \"codemode\" \| \"deferred\" \| \"hidden\";\n\nexport interface ToolAnnotations {\n\treadOnlyHint?: boolean;\n\tdestructiveHint?: boolean;\n\tidempotentHint?: boolean;\n\topenWorldHint?: boolean;\n}\n\nexport interface ToolNamespace {\n\tname: string;\n\tdescription?: string;\n}\n\nexport interface ToolLoadout {\n\treadonly declared: readonly import(\"@earendil-works/pi-agent-core\").AgentTool[];\n\treadonly callable: readonly import(\"@earendil-works/pi-agent-core\").AgentTool[];\n\treadonly registered: readonly import(\"@earendil-works/pi-agent-core\").AgentTool[];\n\tgetExposure(name: string): ToolExposure;\n\tgetNamespace(name: string): ToolNamespace \| undefined;\n}\n\nexport interface ToolLoadoutChanges {\n\tdescriptions?: Readonly<Record<string, string>>;\n\thiddenDeclarations?: readonly string[];\n}\n\n/\n Tool definition for registerTool().\n /\nexport interface ToolDefinition<TParams extends TSchema = TSchema, TDetails = unknown, TState = unknown> {\n\t/* Tool name (used in LLM tool calls) /\n\tname: string;\n\t/* Human-readable label for UI /\n\tlabel: string;\n\t/* Description for LLM /\n\tdescription: string;\n\t/* Optional one-line snippet for the Available tools section in the default system prompt. Custom tools are omitted from that section when this is not provided. /\n\tpromptSnippet?: string;\n\t/* Optional guideline bullets appended to the default system prompt Guidelines section when this tool is active. /\n\tpromptGuidelines?: string[];\n\t/* Parameter schema (TypeBox) /\n\tparameters: TParams;\n\t/* Optional provider-side constrained sampling request. `false` explicitly disables it. /\n\tconstrainedSampling?: false \| ConstrainedSamplingConfig;\n\t/* Optional per-tool character cap for model-visible result persistence. Use Infinity to opt out for self-bounded tools. /\n\tmaxResultSizeChars?: number;\n\t/* Marks a terminating tool created by createStructuredOutputTool() so print mode can emit custom-named final JSON without treating every terminating tool as printable. /\n\tstructuredOutput?: true;\n\t/* Controls whether ToolExecutionComponent renders the standard colored shell or the tool renders its own framing. /\n\trenderShell?: \"default\" \| \"self\";\n\n\t/* Optional compatibility shim to prepare raw tool call arguments before schema validation. Must return an object conforming to TParams. /\n\tprepareArguments?: (args: unknown) => Static<TParams>;\n\t/* Schema of successful machine-readable results. /\n\toutputSchema?: TSchema;\n\t/* Direct and model-only tools activate by default; other exposures do not. /\n\texposure?: ToolExposure;\n\tnamespace?: ToolNamespace;\n\tannotations?: ToolAnnotations;\n\tdefaultActive?: boolean;\n\tprepareLoadout?: (loadout: ToolLoadout) => ToolLoadoutChanges \| undefined;\n\n\t/\n\t Per-tool execution mode override.\n\t * - \"sequential\": this tool must execute one at a time with other tool calls.\n\t * - \"parallel\": this tool can execute concurrently with other tool calls.\n\t \n\t If omitted, the default execution mode applies.\n\t /\n\texecutionMode?: ToolExecutionMode;\n\n\t/\n\t Ordering against the other calls of the same assistant message.\n\t * - \"shared\": may overlap other shared calls, but starts after the latest earlier exclusive call (default)\n\t * - \"exclusive\": starts after every earlier call finishes; later calls start after it finishes\n\t * - function: resolved per call from its arguments; a throwing resolver means \"exclusive\"\n\t /\n\tconcurrency?: ToolConcurrency<Static<TParams>>;\n\n\t/* Execute the tool. /\n\texecute(\n\t\ttoolCallId: string,\n\t\tparams: Static<TParams>,\n\t\tsignal: AbortSignal \| undefined,\n\t\tonUpdate: AgentToolUpdateCallback<TDetails> \| undefined,\n\t\tctx: ExtensionToolContext,\n\t): Promise<AgentToolResult<TDetails>>;\n\n\t/* Custom rendering for tool call display /\n\trenderCall?: ToolRenderCall<TParams, TState>;\n\n\t/* Custom rendering for tool result display /\n\trenderResult?: ToolRenderResult<TParams, TDetails, TState>;\n}\n\ntype AnyToolDefinition = ToolDefinition<TSchema, unknown, unknown>;\n\n/\n Preserve parameter inference for standalone tool definitions.\n \n Use this when assigning a tool to a variable or passing it through arrays such\n * as `customTools`, where contextual typing would otherwise widen params to\n * `unknown`.\n /\nexport function defineTool<TParams extends TSchema, TDetails = unknown, TState = unknown>(\n\ttool: ToolDefinition<TParams, TDetails, TState>,\n): ToolDefinition<TParams, TDetails, TState> & AnyToolDefinition {\n\treturn tool as ToolDefinition<TParams, TDetails, TState> & AnyToolDefinition;\n}\n\n/* Public constrained-sampling shape inherited from pi-ai. /\nexport type { ConstrainedSamplingConfig } from \"@bastani/pi-ai/compat\";\n\n/* Tool info with name, description, parameter schema, constraint, and source metadata */\nexport type ToolInfo = Pick<\n\tToolDefinition,\n\t\| \"name\"\n\t\| \"description\"\n\t\| \"parameters\"\n\t\| \"constrainedSampling\"\n\t\| \"promptGuidelines\"\n\t\| \"exposure\"\n\t\| \"namespace\"\n\t\| \"annotations\"\n> & {\n\tsourceInfo: SourceInfo;\n};\n"]}

package/dist/core/extensions/wrapper.d.ts.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"wrapper.d.ts","sourceRoot":"","sources":["../../../src/core/extensions/wrapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAG/D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,cAAc,EAAE,cAAc,EAAE,MAAM,EAAE,eAAe,GAAG,SAAS,~~CAGrG~~;AAED;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,eAAe,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,eAAe,GAAG,SAAS,EAAE,CAE3G"}
1	+ {"version":3,"file":"wrapper.d.ts","sourceRoot":"","sources":["../../../src/core/extensions/wrapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAG/D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,cAAc,EAAE,cAAc,EAAE,MAAM,EAAE,eAAe,GAAG,SAAS,CAerG;AAED;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,eAAe,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,eAAe,GAAG,SAAS,EAAE,CAE3G"}

package/dist/core/extensions/wrapper.js CHANGED Viewed

@@ -11,8 +11,20 @@ import { wrapToolDefinition } from "../tools/tool-definition-wrapper.js";
  * Uses the runner's createContext() for consistent context across tools and event handlers.
  */
 export function wrapRegisteredTool(registeredTool, runner) {
-    const tool = wrapToolDefinition(registeredTool.definition, () => runner.createContext());
-    return { ...tool, execute: (...args) => withBuiltinDiagnostics(runner, () => tool.execute(...args)) };
+    return {
+        ...wrapToolDefinition(registeredTool.definition),
+        execute: (id, args, signal, update) => withBuiltinDiagnostics(runner, async () => {
+            const lifetime = new AbortController();
+            const callSignal = signal ? AbortSignal.any([signal, lifetime.signal]) : lifetime.signal;
+            const tool = wrapToolDefinition(registeredTool.definition, () => runner.createToolContext(id, callSignal));
+            try {
+                return await tool.execute(id, args, callSignal, update);
+            }
+            finally {
+                lifetime.abort();
+            }
+        }),
+    };
 }
 /**
  * Wrap all registered tools into AgentTools.

package/dist/core/extensions/wrapper.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/core/extensions/wrapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,qCAAqC,CAAC;AAIzE;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,cAA8B,EAAE,MAAuB;IACzF,~~MAAM~~,~~IAAI,~~GAAG,kBAAkB,CAAC,cAAc,CAAC,UAAU,~~EAAE~~,~~GAAG~~,EAAE,CAAC,MAAM,~~CAAC~~,~~aAAa~~,EAAE,~~CAAC~~,CAAC~~;IACzF~~,~~OAAO~~,EAAE,~~GAAG~~,IAAI,EAAE,~~OAAO~~,EAAE,CAAC,GAAG,~~IAAI~~,~~EAAE~~,EAAE,CAAC,~~sBAAsB~~,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,~~GAAG~~,IAAI,CAAC,CAAC,EAAE,CAAC;~~AACvG~~,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,eAAiC,EAAE,MAAuB;IAC7F,OAAO,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;AACxE,CAAC","sourcesContent":["/*\n Tool wrappers for extension-registered tools.\n \n These wrappers only adapt tool execution so extension tools receive the runner context.\n * Tool call and tool result interception is handled by AgentSession via agent-core hooks.\n /\n\nimport type { AgentTool } from \"@earendil-works/pi-agent-core\";\nimport { withBuiltinDiagnostics } from \"../builtin-diagnostics.ts\";\nimport { wrapToolDefinition } from \"../tools/tool-definition-wrapper.ts\";\nimport type { ExtensionRunner } from \"./runner.ts\";\nimport type { RegisteredTool } from \"./types.ts\";\n\n/\n Wrap a RegisteredTool into an AgentTool.\n * Uses the runner's createContext() for consistent context across tools and event handlers.\n /\nexport function wrapRegisteredTool(registeredTool: RegisteredTool, runner: ExtensionRunner): AgentTool {\n\tconst tool = wrapToolDefinition(registeredTool.definition, () => runner.~~createContext~~());\n\~~treturn~~ { ~~...~~tool, ~~execute: (...~~args) ~~=> withBuiltinDiagnostics(runner~~, () => ~~tool~~.~~execute~~(~~...args~~)) };\n}\n\n/\n Wrap all registered tools into AgentTools.\n * Uses the runner's createContext() for consistent context across tools and event handlers.\n */\nexport function wrapRegisteredTools(registeredTools: RegisteredTool[], runner: ExtensionRunner): AgentTool[] {\n\treturn registeredTools.map((tool) => wrapRegisteredTool(tool, runner));\n}\n"]}
1	+ {"version":3,"file":"wrapper.js","sourceRoot":"","sources":["../../../src/core/extensions/wrapper.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,qCAAqC,CAAC;AAIzE;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,cAA8B,EAAE,MAAuB;IACzF,OAAO;QACN,GAAG,kBAAkB,CAAC,cAAc,CAAC,UAAU,CAAC;QAChD,OAAO,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,CACrC,sBAAsB,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YACzF,MAAM,IAAI,GAAG,kBAAkB,CAAC,cAAc,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC;YAC3G,IAAI,CAAC;gBACJ,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YACzD,CAAC;oBAAS,CAAC;gBACV,QAAQ,CAAC,KAAK,EAAE,CAAC;YAClB,CAAC;QACF,CAAC,CAAC;KACH,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,eAAiC,EAAE,MAAuB;IAC7F,OAAO,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;AACxE,CAAC","sourcesContent":["/*\n Tool wrappers for extension-registered tools.\n \n These wrappers only adapt tool execution so extension tools receive the runner context.\n * Tool call and tool result interception is handled by AgentSession via agent-core hooks.\n /\n\nimport type { AgentTool } from \"@earendil-works/pi-agent-core\";\nimport { withBuiltinDiagnostics } from \"../builtin-diagnostics.ts\";\nimport { wrapToolDefinition } from \"../tools/tool-definition-wrapper.ts\";\nimport type { ExtensionRunner } from \"./runner.ts\";\nimport type { RegisteredTool } from \"./types.ts\";\n\n/\n Wrap a RegisteredTool into an AgentTool.\n * Uses the runner's createContext() for consistent context across tools and event handlers.\n /\nexport function wrapRegisteredTool(registeredTool: RegisteredTool, runner: ExtensionRunner): AgentTool {\n\treturn {\n\t\t...wrapToolDefinition(registeredTool.definition),\n\t\texecute: (id, args, signal, update) =>\n\t\t\twithBuiltinDiagnostics(runner, async () => {\n\t\t\t\tconst lifetime = new AbortController();\n\t\t\t\tconst callSignal = signal ? AbortSignal.any([signal, lifetime.signal]) : lifetime.signal;\n\t\t\t\tconst tool = wrapToolDefinition(registeredTool.definition, () => runner.createToolContext(id, callSignal));\n\t\t\t\ttry {\n\t\t\t\t\treturn await tool.execute(id, args, callSignal, update);\n\t\t\t\t} finally {\n\t\t\t\t\tlifetime.abort();\n\t\t\t\t}\n\t\t\t}),\n\t};\n}\n\n/\n Wrap all registered tools into AgentTools.\n * Uses the runner's createContext() for consistent context across tools and event handlers.\n */\nexport function wrapRegisteredTools(registeredTools: RegisteredTool[], runner: ExtensionRunner): AgentTool[] {\n\treturn registeredTools.map((tool) => wrapRegisteredTool(tool, runner));\n}\n"]}

package/dist/core/fast-model-routing-transport.d.ts.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"fast-model-routing-transport.d.ts","sourceRoot":"","sources":["../../src/core/fast-model-routing-transport.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,2BAA2B,iBAAiB,CAAC;AAC1D,eAAO,MAAM,uBAAuB,yBAAyB,CAAC;AAkB9D,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,OAAO,CAG5E;AACD,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,OAAO,CAK/E;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,OAAO,CAOhF;~~AAaD~~,wBAAgB,yBAAyB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,GAAG,IAAI,CAIzG;AAED,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,GAAG,IAAI,CAI7F;AAED,wBAAgB,6BAA6B,CAC5C,SAAS,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,EACjC,WAAW,EAAE,WAAW,GAAG,SAAS,GAClC,OAAO,CAWT;AAED,wBAAgB,uBAAuB,CAAC,mBAAmB,EAAE,OAAO,UAAU,CAAC,KAAK,GAAG,OAAO,UAAU,CAAC,KAAK,CAc7G;AAUD,wBAAgB,2BAA2B,CAC1C,uBAAuB,EAAE,OAAO,UAAU,CAAC,SAAS,GAClD,OAAO,UAAU,CAAC,SAAS,CA6B7B;AAED,wBAAgB,sCAAsC,IAAI,IAAI,CAG7D"}
1	+ {"version":3,"file":"fast-model-routing-transport.d.ts","sourceRoot":"","sources":["../../src/core/fast-model-routing-transport.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,2BAA2B,iBAAiB,CAAC;AAC1D,eAAO,MAAM,uBAAuB,yBAAyB,CAAC;AAkB9D,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,OAAO,CAG5E;AACD,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,OAAO,CAK/E;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,GAAG,OAAO,CAOhF;AAcD,wBAAgB,yBAAyB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,GAAG,IAAI,CAIzG;AAED,wBAAgB,gCAAgC,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,GAAG,IAAI,CAI7F;AAED,wBAAgB,6BAA6B,CAC5C,SAAS,EAAE,MAAM,GAAG,GAAG,GAAG,OAAO,EACjC,WAAW,EAAE,WAAW,GAAG,SAAS,GAClC,OAAO,CAWT;AAED,wBAAgB,uBAAuB,CAAC,mBAAmB,EAAE,OAAO,UAAU,CAAC,KAAK,GAAG,OAAO,UAAU,CAAC,KAAK,CAc7G;AAUD,wBAAgB,2BAA2B,CAC1C,uBAAuB,EAAE,OAAO,UAAU,CAAC,SAAS,GAClD,OAAO,UAAU,CAAC,SAAS,CA6B7B;AAED,wBAAgB,sCAAsC,IAAI,IAAI,CAG7D"}

package/dist/core/fast-model-routing-transport.js CHANGED Viewed

@@ -34,7 +34,8 @@ export function isFirstPartyCodexEndpoint(input) {
         url.pathname.replace(/\/+$/u, "") === "/backend-api/codex/responses");
 }
 function hasCodexPriorityRoutingHint(headers) {
-    return headers.get(CODEX_FAST_ROUTE_HEADER)?.endsWith(";tier=priority") === true;
+    const hint = headers.get(CODEX_FAST_ROUTE_HEADER);
+    return hint?.endsWith(";tier=priority") === true || hint?.endsWith(";tier=ultrafast") === true;
 }
 function codexFastRouteRequestMarker(headers) {
     const marker = headers.get(CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER);

package/dist/core/fast-model-routing-transport.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"fast-model-routing-transport.js","sourceRoot":"","sources":["../../src/core/fast-model-routing-transport.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,2BAA2B,GAAG,cAAc,CAAC;AAC1D,MAAM,CAAC,MAAM,uBAAuB,GAAG,sBAAsB,CAAC;AAC9D,MAAM,sCAAsC,GAAG,2BAA2B,CAAC;AAC3E,MAAM,sCAAsC,GAAG,UAAU,CAAC;AAC1D,MAAM,oCAAoC,GAAG,QAAQ,CAAC;AAEtD,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC;AACxE,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAA4D,CAAC;AAClG,MAAM,0BAA0B,GAAG,IAAI,OAAO,EAA+B,CAAC;AAE9E,SAAS,UAAU,CAAC,KAA6B;IAChD,IAAI,CAAC;QACJ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,YAAY,GAAG;YAAE,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7E,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAA6B;IAClE,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,OAAO,GAAG,KAAK,SAAS,IAAI,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;AACjF,CAAC;AACD,MAAM,UAAU,wBAAwB,CAAC,KAA6B;IACrE,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,IAAI,GAAG,KAAK,SAAS,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IACnE,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC/C,OAAO,IAAI,KAAK,cAAc,IAAI,IAAI,KAAK,oBAAoB,IAAI,IAAI,KAAK,8BAA8B,CAAC;AAC5G,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,KAA6B;IACtE,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,OAAO,CACN,GAAG,KAAK,SAAS;QACjB,qBAAqB,CAAC,GAAG,CAAC;QAC1B,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,8BAA8B,CACpE,CAAC;AACH,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAgB;IACpD,OAAO,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,EAAE,QAAQ,CAAC,gBAAgB,CAAC,KAAK,IAAI,CAAC;AAClF,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAgB;IACpD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IACnE,OAAO,MAAM,KAAK,sCAAsC,IAAI,MAAM,KAAK,oCAAoC;QAC1G,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,SAAS,CAAC;AACd,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,OAAsC,EAAE,QAAiB;IAClG,OAAO,CAAC,sCAAsC,CAAC,GAAG,QAAQ;QACzD,CAAC,CAAC,sCAAsC;QACxC,CAAC,CAAC,oCAAoC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,gCAAgC,CAAC,OAAsC;IACtF,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACzC,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK,sCAAsC;YAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,CAAC;AACF,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC5C,SAAiC,EACjC,WAAoC;IAEpC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC;IACzC,MAAM,MAAM,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;IACpD,OAAO,CAAC,MAAM,CAAC,sCAAsC,CAAC,CAAC;IACvD,IAAI,MAAM,KAAK,UAAU,IAAI,2BAA2B,CAAC,OAAO,CAAC,EAAE,CAAC;QACnE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,2BAA2B,CAAC,CAAC;IACxD,CAAC;SAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;QACxC,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,2BAA2B;YAAE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAChG,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,mBAA4C;IACnF,OAAO,IAAI,KAAK,CAAC,mBAAmB,EAAE;QACrC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,IAAyC;YAC/D,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,IAAI,EAAE,OAAO;gBAAE,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YAChE,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE;gBACrC,KAAK;gBACL;oBACC,GAAG,IAAI;oBACP,OAAO,EAAE,6BAA6B,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC;iBAC3D;aACD,CAAC,CAAC;QACJ,CAAC;KACD,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAwB;IAC1D,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,SAAS,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;IACrF,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,2BAA2B,CAC1C,uBAAoD;IAEpD,IAAI,0BAA0B,CAAC,GAAG,CAAC,uBAAuB,CAAC;QAAE,OAAO,uBAAuB,CAAC;IAC5F,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IAChE,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,QAAQ,CAAC;IAE5C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,uBAAuB,EAAE;QAClD,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS;YAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,YAAY,GAAG,yBAAyB,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,CAAC,QAAQ,YAAY,GAAG,CAAC,CAAC,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAChG,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAA8B,CAAC;YACzE,MAAM,gBAAgB,GAAG,IAAI,OAAO,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC/D,IAAI,2BAA2B,CAAC,gBAAgB,CAAC,KAAK,SAAS,EAAE,CAAC;gBACjE,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,OAAO,GAAG,gBAAgB,CAAC;YACjC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC3B,QAAQ,CAAC,YAAY,CAAC,GAAG;gBACxB,GAAG,OAAO;gBACV,OAAO,EAAE,6BAA6B,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC;aACjE,CAAC;YACF,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC;KACD,CAAgC,CAAC;IAClC,iBAAiB,CAAC,GAAG,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;IACxD,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACxC,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,sCAAsC;IACrD,IAAI,OAAO,UAAU,CAAC,SAAS,KAAK,UAAU;QAAE,OAAO;IACvD,UAAU,CAAC,SAAS,GAAG,2BAA2B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;AAC1E,CAAC","sourcesContent":["export const CODEX_FAST_ROUTE_ORIGINATOR = \"codex_cli_rs\";\nexport const CODEX_FAST_ROUTE_HEADER = \"x-codex-routing-hint\";\nconst CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER = \"x-atomic-codex-fast-route\";\nconst CODEX_FAST_ROUTE_PRIORITY_MARKER_VALUE = \"priority\";\nconst CODEX_FAST_ROUTE_NORMAL_MARKER_VALUE = \"normal\";\n\nconst CODEX_BACKEND_HOSTS = new Set([\"chatgpt.com\", \"chat.openai.com\"]);\nconst wrappedWebSockets = new WeakMap<typeof globalThis.WebSocket, typeof globalThis.WebSocket>();\nconst installedWebSocketWrappers = new WeakSet<typeof globalThis.WebSocket>();\n\nfunction requestUrl(input: string \| URL \| Request): URL \| undefined {\n\ttry {\n\t\tif (typeof input === \"string\" \|\| input instanceof URL) return new URL(input);\n\t\treturn new URL(input.url);\n\t} catch {\n\t\treturn undefined;\n\t}\n}\n\nexport function isFirstPartyCodexHost(input: string \| URL \| Request): boolean {\n\tconst url = requestUrl(input);\n\treturn url !== undefined && CODEX_BACKEND_HOSTS.has(url.hostname.toLowerCase());\n}\nexport function isFirstPartyCodexBaseUrl(input: string \| URL \| Request): boolean {\n\tconst url = requestUrl(input);\n\tif (url === undefined \|\| !isFirstPartyCodexHost(url)) return false;\n\tconst path = url.pathname.replace(/\\/+$/u, \"\");\n\treturn path === \"/backend-api\" \|\| path === \"/backend-api/codex\" \|\| path === \"/backend-api/codex/responses\";\n}\n\nexport function isFirstPartyCodexEndpoint(input: string \| URL \| Request): boolean {\n\tconst url = requestUrl(input);\n\treturn (\n\t\turl !== undefined &&\n\t\tisFirstPartyCodexHost(url) &&\n\t\turl.pathname.replace(/\\/+$/u, \"\") === \"/backend-api/codex/responses\"\n\t);\n}\n\nfunction hasCodexPriorityRoutingHint(headers: Headers): boolean {\n\treturn headers.get(CODEX_FAST_ROUTE_HEADER)?.endsWith(\";tier=priority\") === true;\n}\n\nfunction codexFastRouteRequestMarker(headers: Headers): \"normal\" \| \"priority\" \| undefined {\n\tconst marker = headers.get(CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER);\n\treturn marker === CODEX_FAST_ROUTE_PRIORITY_MARKER_VALUE \|\| marker === CODEX_FAST_ROUTE_NORMAL_MARKER_VALUE\n\t\t? marker\n\t\t: undefined;\n}\n\nexport function markCodexFastRouteRequest(headers: Record<string, string \| null>, priority: boolean): void {\n\theaders[CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER] = priority\n\t\t? CODEX_FAST_ROUTE_PRIORITY_MARKER_VALUE\n\t\t: CODEX_FAST_ROUTE_NORMAL_MARKER_VALUE;\n}\n\nexport function clearCodexFastRouteRequestMarker(headers: Record<string, string \| null>): void {\n\tfor (const name of Object.keys(headers)) {\n\t\tif (name.toLowerCase() === CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER) delete headers[name];\n\t}\n}\n\nexport function forceCodexFastRouteOriginator(\n\t_endpoint: string \| URL \| Request,\n\theadersInit: HeadersInit \| undefined,\n): Headers {\n\tconst headers = new Headers(headersInit);\n\tconst marker = codexFastRouteRequestMarker(headers);\n\theaders.delete(CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER);\n\tif (marker === \"priority\" && hasCodexPriorityRoutingHint(headers)) {\n\t\theaders.set(\"originator\", CODEX_FAST_ROUTE_ORIGINATOR);\n\t} else if (marker !== undefined) {\n\t\theaders.delete(CODEX_FAST_ROUTE_HEADER);\n\t\tif (headers.get(\"originator\") === CODEX_FAST_ROUTE_ORIGINATOR) headers.set(\"originator\", \"pi\");\n\t}\n\treturn headers;\n}\n\nexport function wrapCodexFastRouteFetch(fetchImplementation: typeof globalThis.fetch): typeof globalThis.fetch {\n\treturn new Proxy(fetchImplementation, {\n\t\tapply(target, thisArg, args: Parameters<typeof globalThis.fetch>) {\n\t\t\tconst [input, init] = args;\n\t\t\tif (!init?.headers) return Reflect.apply(target, thisArg, args);\n\t\t\treturn Reflect.apply(target, thisArg, [\n\t\t\t\tinput,\n\t\t\t\t{\n\t\t\t\t\t...init,\n\t\t\t\t\theaders: forceCodexFastRouteOriginator(input, init.headers),\n\t\t\t\t},\n\t\t\t]);\n\t\t},\n\t});\n}\n\nfunction findWebSocketOptionsIndex(args: readonly unknown[]): number \| undefined {\n\tfor (let index = 1; index < args.length; index += 1) {\n\t\tconst value = args[index];\n\t\tif (typeof value === \"object\" && value !== null && \"headers\" in value) return index;\n\t}\n\treturn undefined;\n}\n\nexport function wrapCodexFastRouteWebSocket(\n\tWebSocketImplementation: typeof globalThis.WebSocket,\n): typeof globalThis.WebSocket {\n\tif (installedWebSocketWrappers.has(WebSocketImplementation)) return WebSocketImplementation;\n\tconst existing = wrappedWebSockets.get(WebSocketImplementation);\n\tif (existing !== undefined) return existing;\n\n\tconst wrapped = new Proxy(WebSocketImplementation, {\n\t\tconstruct(target, args, newTarget) {\n\t\t\tconst endpoint = args[0];\n\t\t\tconst optionsIndex = findWebSocketOptionsIndex(args);\n\t\t\tif ((typeof endpoint !== \"string\" && !(endpoint instanceof URL)) \|\| optionsIndex === undefined) {\n\t\t\t\treturn Reflect.construct(target, args, newTarget);\n\t\t\t}\n\t\t\tconst candidateOptions = args[optionsIndex] as { headers?: HeadersInit };\n\t\t\tconst candidateHeaders = new Headers(candidateOptions.headers);\n\t\t\tif (codexFastRouteRequestMarker(candidateHeaders) === undefined) {\n\t\t\t\treturn Reflect.construct(target, args, newTarget);\n\t\t\t}\n\t\t\tconst options = candidateOptions;\n\t\t\tconst nextArgs = [...args];\n\t\t\tnextArgs[optionsIndex] = {\n\t\t\t\t...options,\n\t\t\t\theaders: forceCodexFastRouteOriginator(endpoint, options.headers),\n\t\t\t};\n\t\t\treturn Reflect.construct(target, nextArgs, newTarget);\n\t\t},\n\t}) as typeof globalThis.WebSocket;\n\twrappedWebSockets.set(WebSocketImplementation, wrapped);\n\tinstalledWebSocketWrappers.add(wrapped);\n\treturn wrapped;\n}\n\nexport function installCodexFastRouteWebSocketIdentity(): void {\n\tif (typeof globalThis.WebSocket !== \"function\") return;\n\tglobalThis.WebSocket = wrapCodexFastRouteWebSocket(globalThis.WebSocket);\n}\n"]}
1	+ {"version":3,"file":"fast-model-routing-transport.js","sourceRoot":"","sources":["../../src/core/fast-model-routing-transport.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,2BAA2B,GAAG,cAAc,CAAC;AAC1D,MAAM,CAAC,MAAM,uBAAuB,GAAG,sBAAsB,CAAC;AAC9D,MAAM,sCAAsC,GAAG,2BAA2B,CAAC;AAC3E,MAAM,sCAAsC,GAAG,UAAU,CAAC;AAC1D,MAAM,oCAAoC,GAAG,QAAQ,CAAC;AAEtD,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC;AACxE,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAA4D,CAAC;AAClG,MAAM,0BAA0B,GAAG,IAAI,OAAO,EAA+B,CAAC;AAE9E,SAAS,UAAU,CAAC,KAA6B;IAChD,IAAI,CAAC;QACJ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,YAAY,GAAG;YAAE,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7E,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAA6B;IAClE,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,OAAO,GAAG,KAAK,SAAS,IAAI,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;AACjF,CAAC;AACD,MAAM,UAAU,wBAAwB,CAAC,KAA6B;IACrE,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,IAAI,GAAG,KAAK,SAAS,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IACnE,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC/C,OAAO,IAAI,KAAK,cAAc,IAAI,IAAI,KAAK,oBAAoB,IAAI,IAAI,KAAK,8BAA8B,CAAC;AAC5G,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,KAA6B;IACtE,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAC9B,OAAO,CACN,GAAG,KAAK,SAAS;QACjB,qBAAqB,CAAC,GAAG,CAAC;QAC1B,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,8BAA8B,CACpE,CAAC;AACH,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAgB;IACpD,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IAClD,OAAO,IAAI,EAAE,QAAQ,CAAC,gBAAgB,CAAC,KAAK,IAAI,IAAI,IAAI,EAAE,QAAQ,CAAC,iBAAiB,CAAC,KAAK,IAAI,CAAC;AAChG,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAgB;IACpD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IACnE,OAAO,MAAM,KAAK,sCAAsC,IAAI,MAAM,KAAK,oCAAoC;QAC1G,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,SAAS,CAAC;AACd,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,OAAsC,EAAE,QAAiB;IAClG,OAAO,CAAC,sCAAsC,CAAC,GAAG,QAAQ;QACzD,CAAC,CAAC,sCAAsC;QACxC,CAAC,CAAC,oCAAoC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,gCAAgC,CAAC,OAAsC;IACtF,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACzC,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK,sCAAsC;YAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,CAAC;AACF,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC5C,SAAiC,EACjC,WAAoC;IAEpC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC;IACzC,MAAM,MAAM,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;IACpD,OAAO,CAAC,MAAM,CAAC,sCAAsC,CAAC,CAAC;IACvD,IAAI,MAAM,KAAK,UAAU,IAAI,2BAA2B,CAAC,OAAO,CAAC,EAAE,CAAC;QACnE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,2BAA2B,CAAC,CAAC;IACxD,CAAC;SAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;QACxC,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,2BAA2B;YAAE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAChG,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,mBAA4C;IACnF,OAAO,IAAI,KAAK,CAAC,mBAAmB,EAAE;QACrC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,IAAyC;YAC/D,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,IAAI,EAAE,OAAO;gBAAE,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YAChE,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE;gBACrC,KAAK;gBACL;oBACC,GAAG,IAAI;oBACP,OAAO,EAAE,6BAA6B,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC;iBAC3D;aACD,CAAC,CAAC;QACJ,CAAC;KACD,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAwB;IAC1D,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,SAAS,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;IACrF,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,2BAA2B,CAC1C,uBAAoD;IAEpD,IAAI,0BAA0B,CAAC,GAAG,CAAC,uBAAuB,CAAC;QAAE,OAAO,uBAAuB,CAAC;IAC5F,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IAChE,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,QAAQ,CAAC;IAE5C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,uBAAuB,EAAE;QAClD,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS;YAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,YAAY,GAAG,yBAAyB,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,CAAC,QAAQ,YAAY,GAAG,CAAC,CAAC,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAChG,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAA8B,CAAC;YACzE,MAAM,gBAAgB,GAAG,IAAI,OAAO,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC/D,IAAI,2BAA2B,CAAC,gBAAgB,CAAC,KAAK,SAAS,EAAE,CAAC;gBACjE,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,OAAO,GAAG,gBAAgB,CAAC;YACjC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC3B,QAAQ,CAAC,YAAY,CAAC,GAAG;gBACxB,GAAG,OAAO;gBACV,OAAO,EAAE,6BAA6B,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC;aACjE,CAAC;YACF,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC;KACD,CAAgC,CAAC;IAClC,iBAAiB,CAAC,GAAG,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC;IACxD,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACxC,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,sCAAsC;IACrD,IAAI,OAAO,UAAU,CAAC,SAAS,KAAK,UAAU;QAAE,OAAO;IACvD,UAAU,CAAC,SAAS,GAAG,2BAA2B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;AAC1E,CAAC","sourcesContent":["export const CODEX_FAST_ROUTE_ORIGINATOR = \"codex_cli_rs\";\nexport const CODEX_FAST_ROUTE_HEADER = \"x-codex-routing-hint\";\nconst CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER = \"x-atomic-codex-fast-route\";\nconst CODEX_FAST_ROUTE_PRIORITY_MARKER_VALUE = \"priority\";\nconst CODEX_FAST_ROUTE_NORMAL_MARKER_VALUE = \"normal\";\n\nconst CODEX_BACKEND_HOSTS = new Set([\"chatgpt.com\", \"chat.openai.com\"]);\nconst wrappedWebSockets = new WeakMap<typeof globalThis.WebSocket, typeof globalThis.WebSocket>();\nconst installedWebSocketWrappers = new WeakSet<typeof globalThis.WebSocket>();\n\nfunction requestUrl(input: string \| URL \| Request): URL \| undefined {\n\ttry {\n\t\tif (typeof input === \"string\" \|\| input instanceof URL) return new URL(input);\n\t\treturn new URL(input.url);\n\t} catch {\n\t\treturn undefined;\n\t}\n}\n\nexport function isFirstPartyCodexHost(input: string \| URL \| Request): boolean {\n\tconst url = requestUrl(input);\n\treturn url !== undefined && CODEX_BACKEND_HOSTS.has(url.hostname.toLowerCase());\n}\nexport function isFirstPartyCodexBaseUrl(input: string \| URL \| Request): boolean {\n\tconst url = requestUrl(input);\n\tif (url === undefined \|\| !isFirstPartyCodexHost(url)) return false;\n\tconst path = url.pathname.replace(/\\/+$/u, \"\");\n\treturn path === \"/backend-api\" \|\| path === \"/backend-api/codex\" \|\| path === \"/backend-api/codex/responses\";\n}\n\nexport function isFirstPartyCodexEndpoint(input: string \| URL \| Request): boolean {\n\tconst url = requestUrl(input);\n\treturn (\n\t\turl !== undefined &&\n\t\tisFirstPartyCodexHost(url) &&\n\t\turl.pathname.replace(/\\/+$/u, \"\") === \"/backend-api/codex/responses\"\n\t);\n}\n\nfunction hasCodexPriorityRoutingHint(headers: Headers): boolean {\n\tconst hint = headers.get(CODEX_FAST_ROUTE_HEADER);\n\treturn hint?.endsWith(\";tier=priority\") === true \|\| hint?.endsWith(\";tier=ultrafast\") === true;\n}\n\nfunction codexFastRouteRequestMarker(headers: Headers): \"normal\" \| \"priority\" \| undefined {\n\tconst marker = headers.get(CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER);\n\treturn marker === CODEX_FAST_ROUTE_PRIORITY_MARKER_VALUE \|\| marker === CODEX_FAST_ROUTE_NORMAL_MARKER_VALUE\n\t\t? marker\n\t\t: undefined;\n}\n\nexport function markCodexFastRouteRequest(headers: Record<string, string \| null>, priority: boolean): void {\n\theaders[CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER] = priority\n\t\t? CODEX_FAST_ROUTE_PRIORITY_MARKER_VALUE\n\t\t: CODEX_FAST_ROUTE_NORMAL_MARKER_VALUE;\n}\n\nexport function clearCodexFastRouteRequestMarker(headers: Record<string, string \| null>): void {\n\tfor (const name of Object.keys(headers)) {\n\t\tif (name.toLowerCase() === CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER) delete headers[name];\n\t}\n}\n\nexport function forceCodexFastRouteOriginator(\n\t_endpoint: string \| URL \| Request,\n\theadersInit: HeadersInit \| undefined,\n): Headers {\n\tconst headers = new Headers(headersInit);\n\tconst marker = codexFastRouteRequestMarker(headers);\n\theaders.delete(CODEX_FAST_ROUTE_REQUEST_MARKER_HEADER);\n\tif (marker === \"priority\" && hasCodexPriorityRoutingHint(headers)) {\n\t\theaders.set(\"originator\", CODEX_FAST_ROUTE_ORIGINATOR);\n\t} else if (marker !== undefined) {\n\t\theaders.delete(CODEX_FAST_ROUTE_HEADER);\n\t\tif (headers.get(\"originator\") === CODEX_FAST_ROUTE_ORIGINATOR) headers.set(\"originator\", \"pi\");\n\t}\n\treturn headers;\n}\n\nexport function wrapCodexFastRouteFetch(fetchImplementation: typeof globalThis.fetch): typeof globalThis.fetch {\n\treturn new Proxy(fetchImplementation, {\n\t\tapply(target, thisArg, args: Parameters<typeof globalThis.fetch>) {\n\t\t\tconst [input, init] = args;\n\t\t\tif (!init?.headers) return Reflect.apply(target, thisArg, args);\n\t\t\treturn Reflect.apply(target, thisArg, [\n\t\t\t\tinput,\n\t\t\t\t{\n\t\t\t\t\t...init,\n\t\t\t\t\theaders: forceCodexFastRouteOriginator(input, init.headers),\n\t\t\t\t},\n\t\t\t]);\n\t\t},\n\t});\n}\n\nfunction findWebSocketOptionsIndex(args: readonly unknown[]): number \| undefined {\n\tfor (let index = 1; index < args.length; index += 1) {\n\t\tconst value = args[index];\n\t\tif (typeof value === \"object\" && value !== null && \"headers\" in value) return index;\n\t}\n\treturn undefined;\n}\n\nexport function wrapCodexFastRouteWebSocket(\n\tWebSocketImplementation: typeof globalThis.WebSocket,\n): typeof globalThis.WebSocket {\n\tif (installedWebSocketWrappers.has(WebSocketImplementation)) return WebSocketImplementation;\n\tconst existing = wrappedWebSockets.get(WebSocketImplementation);\n\tif (existing !== undefined) return existing;\n\n\tconst wrapped = new Proxy(WebSocketImplementation, {\n\t\tconstruct(target, args, newTarget) {\n\t\t\tconst endpoint = args[0];\n\t\t\tconst optionsIndex = findWebSocketOptionsIndex(args);\n\t\t\tif ((typeof endpoint !== \"string\" && !(endpoint instanceof URL)) \|\| optionsIndex === undefined) {\n\t\t\t\treturn Reflect.construct(target, args, newTarget);\n\t\t\t}\n\t\t\tconst candidateOptions = args[optionsIndex] as { headers?: HeadersInit };\n\t\t\tconst candidateHeaders = new Headers(candidateOptions.headers);\n\t\t\tif (codexFastRouteRequestMarker(candidateHeaders) === undefined) {\n\t\t\t\treturn Reflect.construct(target, args, newTarget);\n\t\t\t}\n\t\t\tconst options = candidateOptions;\n\t\t\tconst nextArgs = [...args];\n\t\t\tnextArgs[optionsIndex] = {\n\t\t\t\t...options,\n\t\t\t\theaders: forceCodexFastRouteOriginator(endpoint, options.headers),\n\t\t\t};\n\t\t\treturn Reflect.construct(target, nextArgs, newTarget);\n\t\t},\n\t}) as typeof globalThis.WebSocket;\n\twrappedWebSockets.set(WebSocketImplementation, wrapped);\n\tinstalledWebSocketWrappers.add(wrapped);\n\treturn wrapped;\n}\n\nexport function installCodexFastRouteWebSocketIdentity(): void {\n\tif (typeof globalThis.WebSocket !== \"function\") return;\n\tglobalThis.WebSocket = wrapCodexFastRouteWebSocket(globalThis.WebSocket);\n}\n"]}

package/dist/core/fast-model-routing.d.ts CHANGED Viewed

@@ -1,8 +1,7 @@
 import { closeOpenAICodexWebSocketSessions } from "@bastani/pi-ai/api/openai-codex-responses";
 import { type Api, type AssistantMessageEventStream, type Context, type Model, type ModelFastRoute, type OpenAICodexResponsesOptions, type OpenAIResponsesOptions, type ProviderHeaders, type SimpleStreamOptions, type StreamOptions, type ThinkingLevel, type TranscriptContext } from "@bastani/pi-ai/compat";
-import { FAST_MODEL_SERVICE_TIER } from "./fast-model-variants.ts";
 export interface FastRouteStreamOptions extends SimpleStreamOptions {
-    serviceTier?: typeof FAST_MODEL_SERVICE_TIER;
+    serviceTier?: ModelFastRoute["serviceTier"];
 }
 export interface FastRouteStreamers {
     streamSimple: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;

package/dist/core/fast-model-routing.d.ts.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"fast-model-routing.d.ts","sourceRoot":"","sources":["../../src/core/fast-model-routing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iCAAiC,EAAE,MAAM,2CAA2C,CAAC;AAC9F,OAAO,EACN,KAAK,GAAG,EACR,KAAK,2BAA2B,EAChC,KAAK,OAAO,EAEZ,KAAK,KAAK,EACV,KAAK,cAAc,EAEnB,KAAK,2BAA2B,EAChC,KAAK,sBAAsB,EAC3B,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,aAAa,EAIlB,KAAK,aAAa,EAClB,KAAK,iBAAiB,EACtB,MAAM,uBAAuB,CAAC;~~AAU~~/B,~~OAAO,EAAE,uBAAuB,EAAwB,~~MAAM,~~0BAA0B,CAAC;AAEzF,MAAM,~~WAAW,sBAAuB,SAAQ,mBAAmB;IAClE,WAAW,CAAC,EAAE,~~OAAO~~,~~uBAAuB~~,CAAC;~~CAC7C~~;AAED,MAAM,WAAW,kBAAkB;IAClC,YAAY,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,mBAAmB,KAAK,2BAA2B,CAAC;IAClH,qBAAqB,EAAE,CACtB,KAAK,EAAE,KAAK,CAAC,kBAAkB,CAAC,EAChC,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,sBAAsB,KAC5B,2BAA2B,CAAC;IACjC,0BAA0B,EAAE,CAC3B,KAAK,EAAE,KAAK,CAAC,wBAAwB,CAAC,EACtC,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,2BAA2B,KACjC,2BAA2B,CAAC;CACjC;AAWD;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,cAAc,GAAG,SAAS,CAElG;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,WAAW,GAAG,IAAI,CAAC,GAAG,MAAM,CAE1F;AAED;;;GAGG;AACH,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC,GAAG,OAAO,CAEnG;AAED,gGAAgG;AAChG,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,GAAG,OAAO,CAEjF;AAaD;;;;;;;;GAQG;AACH,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,GAAG,WAAW,GAAG,IAAI,GAAG,UAAU,CAAC,EACpE,OAAO,EAAE,eAAe,GAAG,SAAS,GAClC,eAAe,GAAG,SAAS,~~CAY7B~~;AAED,sEAAsE;AACtE,wBAAgB,0BAA0B,CACzC,SAAS,EAAE,cAAc,GAAG,SAAS,EACrC,OAAO,EAAE,mBAAmB,GAAG,SAAS,GACtC,sBAAsB,GAAG,SAAS,CAGpC;AAED,8FAA8F;AAC9F,wBAAgB,wBAAwB,CACvC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,EAC9B,OAAO,EAAE,sBAAsB,GAAG,SAAS,GACzC,OAAO,~~CAET~~;AAYD,KAAK,2BAA2B,GAAG,OAAO,iCAAiC,CAAC;AAwB5E,wBAAgB,gCAAgC,CAAC,QAAQ,SAAS,aAAa,EAC9E,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,QAAQ,EACjB,sBAAsB,GAAE,2BAA+D,GACrF,QAAQ,~~CA2BV~~;AAqCD,wBAAgB,2BAA2B,CAC1C,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,SAAS,EAAE,aAAa,GAAG,SAAS,GAClC,aAAa,GAAG,SAAS,CAG3B;AAED,wBAAgB,oCAAoC,CACnD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,sBAAsB,GAAG,SAAS,GACzC,sBAAsB,CAMxB;AAED,wBAAgB,yCAAyC,CACxD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,sBAAsB,GAAG,SAAS,GACzC,2BAA2B,CAM7B;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,CAClC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,sBAAsB,GAAG,SAAS,EAC3C,SAAS,GAAE,kBAAiD,GAC1D,2BAA2B,CAyB7B"}
1	+ {"version":3,"file":"fast-model-routing.d.ts","sourceRoot":"","sources":["../../src/core/fast-model-routing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iCAAiC,EAAE,MAAM,2CAA2C,CAAC;AAC9F,OAAO,EACN,KAAK,GAAG,EACR,KAAK,2BAA2B,EAChC,KAAK,OAAO,EAEZ,KAAK,KAAK,EACV,KAAK,cAAc,EAEnB,KAAK,2BAA2B,EAChC,KAAK,sBAAsB,EAC3B,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,aAAa,EAIlB,KAAK,aAAa,EAClB,KAAK,iBAAiB,EACtB,MAAM,uBAAuB,CAAC;AAY/B,MAAM,WAAW,sBAAuB,SAAQ,mBAAmB;IAClE,WAAW,CAAC,EAAE,cAAc,CAAC,aAAa,CAAC,CAAC;CAC5C;AAED,MAAM,WAAW,kBAAkB;IAClC,YAAY,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,mBAAmB,KAAK,2BAA2B,CAAC;IAClH,qBAAqB,EAAE,CACtB,KAAK,EAAE,KAAK,CAAC,kBAAkB,CAAC,EAChC,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,sBAAsB,KAC5B,2BAA2B,CAAC;IACjC,0BAA0B,EAAE,CAC3B,KAAK,EAAE,KAAK,CAAC,wBAAwB,CAAC,EACtC,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,2BAA2B,KACjC,2BAA2B,CAAC;CACjC;AAWD;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,WAAW,CAAC,GAAG,cAAc,GAAG,SAAS,CAElG;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,WAAW,GAAG,IAAI,CAAC,GAAG,MAAM,CAE1F;AAED;;;GAGG;AACH,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC,GAAG,OAAO,CAEnG;AAED,gGAAgG;AAChG,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,GAAG,OAAO,CAEjF;AAaD;;;;;;;;GAQG;AACH,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,GAAG,WAAW,GAAG,IAAI,GAAG,UAAU,CAAC,EACpE,OAAO,EAAE,eAAe,GAAG,SAAS,GAClC,eAAe,GAAG,SAAS,CAS7B;AAED,sEAAsE;AACtE,wBAAgB,0BAA0B,CACzC,SAAS,EAAE,cAAc,GAAG,SAAS,EACrC,OAAO,EAAE,mBAAmB,GAAG,SAAS,GACtC,sBAAsB,GAAG,SAAS,CAGpC;AAED,8FAA8F;AAC9F,wBAAgB,wBAAwB,CACvC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,EAC9B,OAAO,EAAE,sBAAsB,GAAG,SAAS,GACzC,OAAO,CAKT;AAYD,KAAK,2BAA2B,GAAG,OAAO,iCAAiC,CAAC;AAwB5E,wBAAgB,gCAAgC,CAAC,QAAQ,SAAS,aAAa,EAC9E,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,QAAQ,EACjB,sBAAsB,GAAE,2BAA+D,GACrF,QAAQ,CA0BV;AAqCD,wBAAgB,2BAA2B,CAC1C,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,SAAS,EAAE,aAAa,GAAG,SAAS,GAClC,aAAa,GAAG,SAAS,CAG3B;AAED,wBAAgB,oCAAoC,CACnD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,sBAAsB,GAAG,SAAS,GACzC,sBAAsB,CAMxB;AAED,wBAAgB,yCAAyC,CACxD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,sBAAsB,GAAG,SAAS,GACzC,2BAA2B,CAM7B;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,CAClC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EACjB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,sBAAsB,GAAG,SAAS,EAC3C,SAAS,GAAE,kBAAiD,GAC1D,2BAA2B,CAyB7B"}

package/dist/core/fast-model-routing.js CHANGED Viewed

@@ -1,7 +1,7 @@
 import { closeOpenAICodexWebSocketSessions } from "@bastani/pi-ai/api/openai-codex-responses";
 import { clampThinkingLevel, normalizeContext, streamOpenAICodexResponses, streamOpenAIResponses, streamSimple, } from "@bastani/pi-ai/compat";
 import { CODEX_FAST_ROUTE_HEADER, CODEX_FAST_ROUTE_ORIGINATOR, clearCodexFastRouteRequestMarker, installCodexFastRouteWebSocketIdentity, isFirstPartyCodexBaseUrl, markCodexFastRouteRequest, wrapCodexFastRouteFetch, } from "./fast-model-routing-transport.js";
-import { FAST_MODEL_SERVICE_TIER, isNativeFastRouteApi } from "./fast-model-variants.js";
+import { FAST_MODEL_SERVICE_TIER, isNativeFastRouteApi, ULTRAFAST_MODEL_SERVICE_TIER } from "./fast-model-variants.js";
 const DEFAULT_FAST_ROUTE_STREAMERS = {
     streamSimple,
     streamOpenAIResponses,
@@ -56,12 +56,13 @@ function setHeader(headers, name, value) {
  * flag, so a model without a route can never be granted this identity.
  */
 export function withCodexFastRouteHeaders(model, headers) {
-    if (getModelFastRoute(model)?.serviceTier !== FAST_MODEL_SERVICE_TIER || !usesFirstPartyCodexRouting(model)) {
+    const tier = getModelFastRoute(model)?.serviceTier;
+    if (tier === undefined || !usesFirstPartyCodexRouting(model)) {
         return headers;
     }
     const fastHeaders = { ...(headers ?? {}) };
     setHeader(fastHeaders, "originator", CODEX_FAST_ROUTE_ORIGINATOR);
-    setHeader(fastHeaders, CODEX_FAST_ROUTE_HEADER, `model=${resolveUpstreamModelId(model)};tier=${FAST_MODEL_SERVICE_TIER}`);
+    setHeader(fastHeaders, CODEX_FAST_ROUTE_HEADER, `model=${resolveUpstreamModelId(model)};tier=${tier}`);
     return fastHeaders;
 }
 /** Attach the route's service tier to the outgoing stream options. */
@@ -72,7 +73,8 @@ export function withFastRouteStreamOptions(fastRoute, options) {
 }
 /** A service-tier route needs pi-ai's native adapters; `streamSimple` drops `serviceTier`. */
 export function shouldUseNativeFastRoute(model, options) {
-    return isNativeFastRouteApi(model.api) && options?.serviceTier === FAST_MODEL_SERVICE_TIER;
+    return (isNativeFastRouteApi(model.api) &&
+        (options?.serviceTier === FAST_MODEL_SERVICE_TIER || options?.serviceTier === ULTRAFAST_MODEL_SERVICE_TIER));
 }
 function codexEnvironmentScope(env) {
     if (!env)
@@ -84,10 +86,10 @@ function codexEnvironmentScope(env) {
     }
     return (hash >>> 0).toString(16);
 }
-function updateCodexWebSocketRoutingState(model, routedModel, options, priority, closeWebSocketSessions) {
+function updateCodexWebSocketRoutingState(model, routedModel, options, tier, closeWebSocketSessions) {
     if (options.transport === "sse" || !options.sessionId)
         return;
-    const signature = `${model.provider}\0${model.baseUrl}\0${codexEnvironmentScope(options.env)}\0${priority ? `priority:${routedModel}` : "normal"}`;
+    const signature = `${model.provider}\0${model.baseUrl}\0${codexEnvironmentScope(options.env)}\0${tier ? `${tier}:${routedModel}` : "normal"}`;
     const previous = codexWebSocketRoutingState.get(options.sessionId);
     if (previous?.signature !== undefined && previous.signature !== signature) {
         closeWebSocketSessions(options.sessionId);
@@ -105,7 +107,7 @@ function updateCodexWebSocketRoutingState(model, routedModel, options, priority,
 export function withChatGptCodexTransportRouting(model, options, closeWebSocketSessions = closeOpenAICodexWebSocketSessions) {
     const headers = { ...(options.headers ?? {}) };
     const onPayload = options.onPayload;
-    const priority = getModelFastRoute(model)?.serviceTier === FAST_MODEL_SERVICE_TIER && usesFirstPartyCodexRouting(model);
+    const tier = usesFirstPartyCodexRouting(model) ? getModelFastRoute(model)?.serviceTier : undefined;
     const routedModel = resolveUpstreamModelId(model);
     return {
         ...options,
@@ -116,8 +118,8 @@ export function withChatGptCodexTransportRouting(model, options, closeWebSocketS
             const finalPayload = replacement === undefined ? payload : replacement;
             deleteHeader(headers, CODEX_FAST_ROUTE_HEADER);
             clearCodexFastRouteRequestMarker(headers);
-            if (priority) {
-                setHeader(headers, CODEX_FAST_ROUTE_HEADER, `model=${routedModel};tier=${FAST_MODEL_SERVICE_TIER}`);
+            if (tier) {
+                setHeader(headers, CODEX_FAST_ROUTE_HEADER, `model=${routedModel};tier=${tier}`);
             }
             else {
                 for (const [name, value] of Object.entries(headers)) {
@@ -125,8 +127,8 @@ export function withChatGptCodexTransportRouting(model, options, closeWebSocketS
                         delete headers[name];
                 }
             }
-            markCodexFastRouteRequest(headers, priority);
-            updateCodexWebSocketRoutingState(model, routedModel, options, priority, closeWebSocketSessions);
+            markCodexFastRouteRequest(headers, tier !== undefined);
+            updateCodexWebSocketRoutingState(model, routedModel, options, tier, closeWebSocketSessions);
             return finalPayload;
         },
     };

package/dist/core/fast-model-routing.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"fast-model-routing.js","sourceRoot":"","sources":["../../src/core/fast-model-routing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iCAAiC,EAAE,MAAM,2CAA2C,CAAC;AAC9F,OAAO,EAIN,kBAAkB,EAGlB,gBAAgB,EAMhB,0BAA0B,EAC1B,qBAAqB,EACrB,YAAY,GAGZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACN,uBAAuB,EACvB,2BAA2B,EAC3B,gCAAgC,EAChC,sCAAsC,EACtC,wBAAwB,EACxB,yBAAyB,EACzB,uBAAuB,GACvB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAoBzF,MAAM,4BAA4B,GAAuB;IACxD,YAAY;IACZ,qBAAqB;IACrB,0BAA0B;CAC1B,CAAC;AAEF,MAAM,qCAAqC,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AAC5D,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAwE,CAAC;AAEnH;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAoC;IACrE,OAAO,KAAK,CAAC,SAAS,CAAC;AACxB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CAAC,KAA2C;IACjF,OAAO,KAAK,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,CAAC,EAAE,CAAC;AACrD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,0BAA0B,CAAC,KAA+C;IACzF,OAAO,KAAK,CAAC,QAAQ,KAAK,cAAc,IAAI,wBAAwB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AACrF,CAAC;AAED,gGAAgG;AAChG,MAAM,UAAU,yBAAyB,CAAC,KAA8B;IACvE,OAAO,KAAK,CAAC,GAAG,KAAK,wBAAwB,CAAC;AAC/C,CAAC;AAED,SAAS,YAAY,CAAC,OAAwB,EAAE,IAAY;IAC3D,KAAK,MAAM,YAAY,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACjD,IAAI,YAAY,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,WAAW,EAAE;YAAE,OAAO,OAAO,CAAC,YAAY,CAAC,CAAC;IACrF,CAAC;AACF,CAAC;AAED,SAAS,SAAS,CAAC,OAAwB,EAAE,IAAY,EAAE,KAAa;IACvE,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5B,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;AACvB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,yBAAyB,CACxC,KAAoE,EACpE,OAAoC;IAEpC,IAAI,iBAAiB,CAAC,KAAK,CAAC,EAAE,WAAW,KAAK,uBAAuB,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7G,OAAO,OAAO,CAAC;IAChB,CAAC;IACD,MAAM,WAAW,GAAoB,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;IAC5D,SAAS,CAAC,WAAW,EAAE,YAAY,EAAE,2BAA2B,CAAC,CAAC;IAClE,SAAS,CACR,WAAW,EACX,uBAAuB,EACvB,SAAS,sBAAsB,CAAC,KAAK,CAAC,SAAS,uBAAuB,EAAE,CACxE,CAAC;IACF,OAAO,WAAW,CAAC;AACpB,CAAC;AAED,sEAAsE;AACtE,MAAM,UAAU,0BAA0B,CACzC,SAAqC,EACrC,OAAwC;IAExC,IAAI,SAAS,EAAE,WAAW,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACzD,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,CAAC,WAAW,EAAE,CAAC;AAC9F,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,wBAAwB,CACvC,KAA8B,EAC9B,OAA2C;IAE3C,OAAO,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,OAAO,EAAE,WAAW,KAAK,uBAAuB,CAAC;AAC5F,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAyB;IACvD,IAAI,CAAC,GAAG;QAAE,OAAO,MAAM,CAAC;IACxB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5G,IAAI,IAAI,GAAG,UAAU,CAAC;IACtB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC3D,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;AAClC,CAAC;AAGD,SAAS,gCAAgC,CACxC,KAA+C,EAC/C,WAAmB,EACnB,OAA+D,EAC/D,QAAiB,EACjB,sBAAmD;IAEnD,IAAI,OAAO,CAAC,SAAS,KAAK,KAAK,IAAI,CAAC,OAAO,CAAC,SAAS;QAAE,OAAO;IAC9D,MAAM,SAAS,GAAG,GAAG,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,OAAO,KAAK,qBAAqB,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,WAAW,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IACnJ,MAAM,QAAQ,GAAG,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACnE,IAAI,QAAQ,EAAE,SAAS,KAAK,SAAS,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QAC3E,sBAAsB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IACD,IAAI,QAAQ;QAAE,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,EAAE;QAC9B,IAAI,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,SAAU,CAAC,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;YACjF,0BAA0B,CAAC,MAAM,CAAC,OAAO,CAAC,SAAU,CAAC,CAAC;QACvD,CAAC;IACF,CAAC,EAAE,qCAAqC,CAAC,CAAC;IAC1C,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;IACjB,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC/C,KAAiB,EACjB,OAAiB,EACjB,sBAAsB,GAAgC,iCAAiC;IAEvF,MAAM,OAAO,GAAoB,EAAE,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;IAChE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACpC,MAAM,QAAQ,GACb,iBAAiB,CAAC,KAAK,CAAC,EAAE,WAAW,KAAK,uBAAuB,IAAI,0BAA0B,CAAC,KAAK,CAAC,CAAC;IACxG,MAAM,WAAW,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO;QACN,GAAG,OAAO;QACV,OAAO;QACP,KAAK,EAAE,uBAAuB,CAAC,OAAO,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,CAAC;QACjE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE;YAC1C,MAAM,WAAW,GAAG,MAAM,SAAS,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAC7D,MAAM,YAAY,GAAG,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC;YACvE,YAAY,CAAC,OAAO,EAAE,uBAAuB,CAAC,CAAC;YAC/C,gCAAgC,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,QAAQ,EAAE,CAAC;gBACd,SAAS,CAAC,OAAO,EAAE,uBAAuB,EAAE,SAAS,WAAW,SAAS,uBAAuB,EAAE,CAAC,CAAC;YACrG,CAAC;iBAAM,CAAC;gBACP,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;oBACrD,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK,YAAY,IAAI,KAAK,KAAK,2BAA2B;wBAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;gBACxG,CAAC;YACF,CAAC;YACD,yBAAyB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC7C,gCAAgC,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,sBAAsB,CAAC,CAAC;YAChG,OAAO,YAAY,CAAC;QACrB,CAAC;KACW,CAAC;AACf,CAAC;AAED,SAAS,iCAAiC,CACzC,KAAiB,EACjB,OAA2C;IAE3C,4EAA4E;IAC5E,MAAM,cAAc,GACnB,KAAK,CAAC,cAAc,IAAI,OAAO,EAAE,cAAc;QAC9C,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,cAAc,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE;QACzD,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,eAAe,GAAkB;QACtC,cAAc;QACd,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,MAAM,EAAE,OAAO,EAAE,MAAM;QACvB,MAAM,EAAE,OAAO,EAAE,MAAM;QACvB,KAAK,EAAE,OAAO,EAAE,KAAK;QACrB,GAAG,EAAE,OAAO,EAAE,GAAG;QACjB,gBAAgB,EAAE,OAAO,EAAE,gBAAgB;QAC3C,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,UAAU,EAAE,OAAO,EAAE,UAAU;QAC/B,qBAAqB,EAAE,OAAO,EAAE,qBAAqB;QACrD,OAAO,EAAE,OAAO,EAAE,OAAO;QACzB,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,yBAAyB,EAAE,OAAO,EAAE,yBAAyB;QAC7D,gBAAgB,EAAE,OAAO,EAAE,gBAAgB;QAC3C,UAAU,EAAE,OAAO,EAAE,UAAU;QAC/B,eAAe,EAAE,OAAO,EAAE,eAAe;QACzC,QAAQ,EAAE,OAAO,EAAE,QAAQ;KAC3B,CAAC;IACF,OAAO,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,gCAAgC,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC;AACtH,CAAC;AAED,MAAM,UAAU,2BAA2B,CAC1C,KAAiB,EACjB,SAAoC;IAEpC,MAAM,gBAAgB,GAAG,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtF,OAAO,gBAAgB,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,gBAAgB,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,oCAAoC,CACnD,KAAiB,EACjB,OAA2C;IAE3C,OAAO;QACN,GAAG,iCAAiC,CAAC,KAAK,EAAE,OAAO,CAAC;QACpD,eAAe,EAAE,2BAA2B,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC;QACvE,WAAW,EAAE,OAAO,EAAE,WAAW;KACjC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,yCAAyC,CACxD,KAAiB,EACjB,OAA2C;IAE3C,OAAO;QACN,GAAG,iCAAiC,CAAC,KAAK,EAAE,OAAO,CAAC;QACpD,eAAe,EAAE,2BAA2B,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC;QACvE,WAAW,EAAE,OAAO,EAAE,WAAW;KACjC,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAClC,KAAiB,EACjB,OAAgB,EAChB,OAA2C,EAC3C,SAAS,GAAuB,4BAA4B;IAE5D,IAAI,wBAAwB,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC;QAC9C,IAAI,KAAK,CAAC,GAAG,KAAK,kBAAkB,EAAE,CAAC;YACtC,OAAO,SAAS,CAAC,qBAAqB,CACrC,KAAkC,EAClC,gBAAgB,CAAC,OAAO,CAAC,EACzB,oCAAoC,CAAC,KAAK,EAAE,OAAO,CAAC,CACpD,CAAC;QACH,CAAC;QAED,2EAA2E;QAC3E,0EAA0E;QAC1E,kEAAkE;QAClE,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,IAAI,yBAAyB,CAAC,KAAK,CAAC,EAAE,CAAC;YACtE,sCAAsC,EAAE,CAAC;QAC1C,CAAC;QAED,OAAO,SAAS,CAAC,0BAA0B,CAC1C,KAAwC,EACxC,gBAAgB,CAAC,OAAO,CAAC,EACzB,yCAAyC,CAAC,KAAK,EAAE,OAAO,CAAC,CACzD,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import { closeOpenAICodexWebSocketSessions } from \"@bastani/pi-ai/api/openai-codex-responses\";\nimport {\n\ttype Api,\n\ttype AssistantMessageEventStream,\n\ttype Context,\n\tclampThinkingLevel,\n\ttype Model,\n\ttype ModelFastRoute,\n\tnormalizeContext,\n\ttype OpenAICodexResponsesOptions,\n\ttype OpenAIResponsesOptions,\n\ttype ProviderHeaders,\n\ttype SimpleStreamOptions,\n\ttype StreamOptions,\n\tstreamOpenAICodexResponses,\n\tstreamOpenAIResponses,\n\tstreamSimple,\n\ttype ThinkingLevel,\n\ttype TranscriptContext,\n} from \"@bastani/pi-ai/compat\";\nimport {\n\tCODEX_FAST_ROUTE_HEADER,\n\tCODEX_FAST_ROUTE_ORIGINATOR,\n\tclearCodexFastRouteRequestMarker,\n\tinstallCodexFastRouteWebSocketIdentity,\n\tisFirstPartyCodexBaseUrl,\n\tmarkCodexFastRouteRequest,\n\twrapCodexFastRouteFetch,\n} from \"./fast-model-routing-transport.ts\";\nimport { FAST_MODEL_SERVICE_TIER, isNativeFastRouteApi } from \"./fast-model-variants.ts\";\n\nexport interface FastRouteStreamOptions extends SimpleStreamOptions {\n\tserviceTier?: typeof FAST_MODEL_SERVICE_TIER;\n}\n\nexport interface FastRouteStreamers {\n\tstreamSimple: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;\n\tstreamOpenAIResponses: (\n\t\tmodel: Model<\"openai-responses\">,\n\t\tcontext: TranscriptContext,\n\t\toptions?: OpenAIResponsesOptions,\n\t) => AssistantMessageEventStream;\n\tstreamOpenAICodexResponses: (\n\t\tmodel: Model<\"openai-codex-responses\">,\n\t\tcontext: TranscriptContext,\n\t\toptions?: OpenAICodexResponsesOptions,\n\t) => AssistantMessageEventStream;\n}\n\nconst DEFAULT_FAST_ROUTE_STREAMERS: FastRouteStreamers = {\n\tstreamSimple,\n\tstreamOpenAIResponses,\n\tstreamOpenAICodexResponses,\n};\n\nconst CODEX_WEBSOCKET_ROUTING_STATE_IDLE_MS = 6 * 60 * 1000;\nconst codexWebSocketRoutingState = new Map<string, { signature: string; expiry: ReturnType<typeof setTimeout> }>();\n\n/*\n Explicit route metadata is the only source of fast semantics. A model whose ID merely ends in\n * `-fast` — a provider, `models.json`, or extension model that owns that exact ID — routes normally.\n /\nexport function getModelFastRoute(model: Pick<Model<Api>, \"fastRoute\">): ModelFastRoute \| undefined {\n\treturn model.fastRoute;\n}\n\n/\n The model ID this request sends upstream. An OpenAI-style fast route sends the base model and\n * carries a priority service tier instead; a provider with real fast siblings sends its own\n * advertised ID, which is already the canonical one. The canonical `-fast` identity always stays on\n * the model object Atomic selected, persisted, and records — only the serialized request differs.\n /\nexport function resolveUpstreamModelId(model: Pick<Model<Api>, \"fastRoute\" \| \"id\">): string {\n\treturn model.fastRoute?.upstreamModelId ?? model.id;\n}\n\n/\n Whether a model uses the first-party OpenAI Codex provider identity and endpoint.\n * A renamed provider on the same protocol remains non-first-party.\n /\nexport function usesFirstPartyCodexRouting(model: Pick<Model<Api>, \"baseUrl\" \| \"provider\">): boolean {\n\treturn model.provider === \"openai-codex\" && isFirstPartyCodexBaseUrl(model.baseUrl);\n}\n\n/* A provider alias or proxy still uses the shared ChatGPT Codex protocol when its API does. /\nexport function usesChatGptCodexTransport(model: Pick<Model<Api>, \"api\">): boolean {\n\treturn model.api === \"openai-codex-responses\";\n}\n\nfunction deleteHeader(headers: ProviderHeaders, name: string): void {\n\tfor (const existingName of Object.keys(headers)) {\n\t\tif (existingName.toLowerCase() === name.toLowerCase()) delete headers[existingName];\n\t}\n}\n\nfunction setHeader(headers: ProviderHeaders, name: string, value: string): void {\n\tdeleteHeader(headers, name);\n\theaders[name] = value;\n}\n\n/\n Eager first-party routing headers for a model that carries an OpenAI-style fast route.\n \n Derived entirely from the model: the headers are added only when the model's own `fastRoute`\n * declares the priority tier and the model uses first-party Codex routing, and the hint names the\n * route's upstream model ID — the same value the request serializes and\n * `withChatGptCodexTransportRouting` reads from route metadata. There is no caller-supplied enable\n * flag, so a model without a route can never be granted this identity.\n /\nexport function withCodexFastRouteHeaders(\n\tmodel: Pick<Model<Api>, \"baseUrl\" \| \"fastRoute\" \| \"id\" \| \"provider\">,\n\theaders: ProviderHeaders \| undefined,\n): ProviderHeaders \| undefined {\n\tif (getModelFastRoute(model)?.serviceTier !== FAST_MODEL_SERVICE_TIER \|\| !usesFirstPartyCodexRouting(model)) {\n\t\treturn headers;\n\t}\n\tconst fastHeaders: ProviderHeaders = { ...(headers ?? {}) };\n\tsetHeader(fastHeaders, \"originator\", CODEX_FAST_ROUTE_ORIGINATOR);\n\tsetHeader(\n\t\tfastHeaders,\n\t\tCODEX_FAST_ROUTE_HEADER,\n\t\t`model=${resolveUpstreamModelId(model)};tier=${FAST_MODEL_SERVICE_TIER}`,\n\t);\n\treturn fastHeaders;\n}\n\n/* Attach the route's service tier to the outgoing stream options. /\nexport function withFastRouteStreamOptions(\n\tfastRoute: ModelFastRoute \| undefined,\n\toptions: SimpleStreamOptions \| undefined,\n): FastRouteStreamOptions \| undefined {\n\tif (fastRoute?.serviceTier === undefined) return options;\n\treturn { ...(options ?? {}), headers: options?.headers, serviceTier: fastRoute.serviceTier };\n}\n\n/* A service-tier route needs pi-ai's native adapters; `streamSimple` drops `serviceTier`. /\nexport function shouldUseNativeFastRoute(\n\tmodel: Pick<Model<Api>, \"api\">,\n\toptions: FastRouteStreamOptions \| undefined,\n): boolean {\n\treturn isNativeFastRouteApi(model.api) && options?.serviceTier === FAST_MODEL_SERVICE_TIER;\n}\n\nfunction codexEnvironmentScope(env: StreamOptions[\"env\"]): string {\n\tif (!env) return \"none\";\n\tconst serialized = JSON.stringify(Object.entries(env).sort(([left], [right]) => left.localeCompare(right)));\n\tlet hash = 2166136261;\n\tfor (let index = 0; index < serialized.length; index += 1) {\n\t\thash = Math.imul(hash ^ serialized.charCodeAt(index), 16777619);\n\t}\n\treturn (hash >>> 0).toString(16);\n}\n\ntype CloseCodexWebSocketSessions = typeof closeOpenAICodexWebSocketSessions;\nfunction updateCodexWebSocketRoutingState(\n\tmodel: Pick<Model<Api>, \"baseUrl\" \| \"provider\">,\n\troutedModel: string,\n\toptions: Pick<StreamOptions, \"env\" \| \"sessionId\" \| \"transport\">,\n\tpriority: boolean,\n\tcloseWebSocketSessions: CloseCodexWebSocketSessions,\n): void {\n\tif (options.transport === \"sse\" \|\| !options.sessionId) return;\n\tconst signature = `${model.provider}\\0${model.baseUrl}\\0${codexEnvironmentScope(options.env)}\\0${priority ? `priority:${routedModel}` : \"normal\"}`;\n\tconst previous = codexWebSocketRoutingState.get(options.sessionId);\n\tif (previous?.signature !== undefined && previous.signature !== signature) {\n\t\tcloseWebSocketSessions(options.sessionId);\n\t}\n\tif (previous) clearTimeout(previous.expiry);\n\tconst expiry = setTimeout(() => {\n\t\tif (codexWebSocketRoutingState.get(options.sessionId!)?.signature === signature) {\n\t\t\tcodexWebSocketRoutingState.delete(options.sessionId!);\n\t\t}\n\t}, CODEX_WEBSOCKET_ROUTING_STATE_IDLE_MS);\n\texpiry.unref?.();\n\tcodexWebSocketRoutingState.set(options.sessionId, { signature, expiry });\n}\n\nexport function withChatGptCodexTransportRouting<TOptions extends StreamOptions>(\n\tmodel: Model<Api>,\n\toptions: TOptions,\n\tcloseWebSocketSessions: CloseCodexWebSocketSessions = closeOpenAICodexWebSocketSessions,\n): TOptions {\n\tconst headers: ProviderHeaders = { ...(options.headers ?? {}) };\n\tconst onPayload = options.onPayload;\n\tconst priority =\n\t\tgetModelFastRoute(model)?.serviceTier === FAST_MODEL_SERVICE_TIER && usesFirstPartyCodexRouting(model);\n\tconst routedModel = resolveUpstreamModelId(model);\n\treturn {\n\t\t...options,\n\t\theaders,\n\t\tfetch: wrapCodexFastRouteFetch(options.fetch ?? globalThis.fetch),\n\t\tonPayload: async (payload, requestModel) => {\n\t\t\tconst replacement = await onPayload?.(payload, requestModel);\n\t\t\tconst finalPayload = replacement === undefined ? payload : replacement;\n\t\t\tdeleteHeader(headers, CODEX_FAST_ROUTE_HEADER);\n\t\t\tclearCodexFastRouteRequestMarker(headers);\n\t\t\tif (priority) {\n\t\t\t\tsetHeader(headers, CODEX_FAST_ROUTE_HEADER, `model=${routedModel};tier=${FAST_MODEL_SERVICE_TIER}`);\n\t\t\t} else {\n\t\t\t\tfor (const [name, value] of Object.entries(headers)) {\n\t\t\t\t\tif (name.toLowerCase() === \"originator\" && value === CODEX_FAST_ROUTE_ORIGINATOR) delete headers[name];\n\t\t\t\t}\n\t\t\t}\n\t\t\tmarkCodexFastRouteRequest(headers, priority);\n\t\t\tupdateCodexWebSocketRoutingState(model, routedModel, options, priority, closeWebSocketSessions);\n\t\t\treturn finalPayload;\n\t\t},\n\t} as TOptions;\n}\n\nfunction buildFastRouteBaseProviderOptions(\n\tmodel: Model<Api>,\n\toptions: FastRouteStreamOptions \| undefined,\n): StreamOptions {\n\t// Native fast-route streams bypass pi-ai's streamSimple base-options merge.\n\tconst samplingParams =\n\t\tmodel.samplingParams \|\| options?.samplingParams\n\t\t\t? { ...model.samplingParams, ...options?.samplingParams }\n\t\t\t: undefined;\n\tconst providerOptions: StreamOptions = {\n\t\tsamplingParams,\n\t\ttemperature: options?.temperature,\n\t\tmaxTokens: options?.maxTokens,\n\t\tsignal: options?.signal,\n\t\tapiKey: options?.apiKey,\n\t\tfetch: options?.fetch,\n\t\tenv: options?.env,\n\t\ttelemetryContext: options?.telemetryContext,\n\t\ttransport: options?.transport,\n\t\tcacheRetention: options?.cacheRetention,\n\t\tsessionId: options?.sessionId,\n\t\tonPayload: options?.onPayload,\n\t\tonResponse: options?.onResponse,\n\t\tonProviderStreamEvent: options?.onProviderStreamEvent,\n\t\theaders: options?.headers,\n\t\ttimeoutMs: options?.timeoutMs,\n\t\twebsocketConnectTimeoutMs: options?.websocketConnectTimeoutMs,\n\t\tstreamDeadlineMs: options?.streamDeadlineMs,\n\t\tmaxRetries: options?.maxRetries,\n\t\tmaxRetryDelayMs: options?.maxRetryDelayMs,\n\t\tmetadata: options?.metadata,\n\t};\n\treturn usesChatGptCodexTransport(model) ? withChatGptCodexTransportRouting(model, providerOptions) : providerOptions;\n}\n\nexport function mapFastRouteReasoningEffort(\n\tmodel: Model<Api>,\n\treasoning: ThinkingLevel \| undefined,\n): ThinkingLevel \| undefined {\n\tconst clampedReasoning = reasoning ? clampThinkingLevel(model, reasoning) : undefined;\n\treturn clampedReasoning === \"off\" ? undefined : clampedReasoning;\n}\n\nexport function buildOpenAIResponsesFastRouteOptions(\n\tmodel: Model<Api>,\n\toptions: FastRouteStreamOptions \| undefined,\n): OpenAIResponsesOptions {\n\treturn {\n\t\t...buildFastRouteBaseProviderOptions(model, options),\n\t\treasoningEffort: mapFastRouteReasoningEffort(model, options?.reasoning),\n\t\tserviceTier: options?.serviceTier,\n\t};\n}\n\nexport function buildOpenAICodexResponsesFastRouteOptions(\n\tmodel: Model<Api>,\n\toptions: FastRouteStreamOptions \| undefined,\n): OpenAICodexResponsesOptions {\n\treturn {\n\t\t...buildFastRouteBaseProviderOptions(model, options),\n\t\treasoningEffort: mapFastRouteReasoningEffort(model, options?.reasoning),\n\t\tserviceTier: options?.serviceTier,\n\t};\n}\n\n/\n Stream a request whose canonical selected model carries fast-route metadata. The provider adapter\n * translates only the serialized request to the route's upstream ID; the model object keeps its\n * canonical `-fast` identity.\n */\nexport function streamWithFastRoute(\n\tmodel: Model<Api>,\n\tcontext: Context,\n\toptions: FastRouteStreamOptions \| undefined,\n\tstreamers: FastRouteStreamers = DEFAULT_FAST_ROUTE_STREAMERS,\n): AssistantMessageEventStream {\n\tif (shouldUseNativeFastRoute(model, options)) {\n\t\tif (model.api === \"openai-responses\") {\n\t\t\treturn streamers.streamOpenAIResponses(\n\t\t\t\tmodel as Model<\"openai-responses\">,\n\t\t\t\tnormalizeContext(context),\n\t\t\t\tbuildOpenAIResponsesFastRouteOptions(model, options),\n\t\t\t);\n\t\t}\n\n\t\t// The WebSocket handshake builds its headers inside pi-ai, and the runtime\n\t\t// constructor can be cached before the first turn, so repair the identity\n\t\t// through the global constructor rather than per-request options.\n\t\tif (options?.transport !== \"sse\" && usesChatGptCodexTransport(model)) {\n\t\t\tinstallCodexFastRouteWebSocketIdentity();\n\t\t}\n\n\t\treturn streamers.streamOpenAICodexResponses(\n\t\t\tmodel as Model<\"openai-codex-responses\">,\n\t\t\tnormalizeContext(context),\n\t\t\tbuildOpenAICodexResponsesFastRouteOptions(model, options),\n\t\t);\n\t}\n\n\treturn streamers.streamSimple(model, context, options);\n}\n"]}
1	+ {"version":3,"file":"fast-model-routing.js","sourceRoot":"","sources":["../../src/core/fast-model-routing.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iCAAiC,EAAE,MAAM,2CAA2C,CAAC;AAC9F,OAAO,EAIN,kBAAkB,EAGlB,gBAAgB,EAMhB,0BAA0B,EAC1B,qBAAqB,EACrB,YAAY,GAGZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACN,uBAAuB,EACvB,2BAA2B,EAC3B,gCAAgC,EAChC,sCAAsC,EACtC,wBAAwB,EACxB,yBAAyB,EACzB,uBAAuB,GACvB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AAoBvH,MAAM,4BAA4B,GAAuB;IACxD,YAAY;IACZ,qBAAqB;IACrB,0BAA0B;CAC1B,CAAC;AAEF,MAAM,qCAAqC,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AAC5D,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAwE,CAAC;AAEnH;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAoC;IACrE,OAAO,KAAK,CAAC,SAAS,CAAC;AACxB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CAAC,KAA2C;IACjF,OAAO,KAAK,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,CAAC,EAAE,CAAC;AACrD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,0BAA0B,CAAC,KAA+C;IACzF,OAAO,KAAK,CAAC,QAAQ,KAAK,cAAc,IAAI,wBAAwB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AACrF,CAAC;AAED,gGAAgG;AAChG,MAAM,UAAU,yBAAyB,CAAC,KAA8B;IACvE,OAAO,KAAK,CAAC,GAAG,KAAK,wBAAwB,CAAC;AAC/C,CAAC;AAED,SAAS,YAAY,CAAC,OAAwB,EAAE,IAAY;IAC3D,KAAK,MAAM,YAAY,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACjD,IAAI,YAAY,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,WAAW,EAAE;YAAE,OAAO,OAAO,CAAC,YAAY,CAAC,CAAC;IACrF,CAAC;AACF,CAAC;AAED,SAAS,SAAS,CAAC,OAAwB,EAAE,IAAY,EAAE,KAAa;IACvE,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5B,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;AACvB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,yBAAyB,CACxC,KAAoE,EACpE,OAAoC;IAEpC,MAAM,IAAI,GAAG,iBAAiB,CAAC,KAAK,CAAC,EAAE,WAAW,CAAC;IACnD,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9D,OAAO,OAAO,CAAC;IAChB,CAAC;IACD,MAAM,WAAW,GAAoB,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;IAC5D,SAAS,CAAC,WAAW,EAAE,YAAY,EAAE,2BAA2B,CAAC,CAAC;IAClE,SAAS,CAAC,WAAW,EAAE,uBAAuB,EAAE,SAAS,sBAAsB,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;IACvG,OAAO,WAAW,CAAC;AACpB,CAAC;AAED,sEAAsE;AACtE,MAAM,UAAU,0BAA0B,CACzC,SAAqC,EACrC,OAAwC;IAExC,IAAI,SAAS,EAAE,WAAW,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACzD,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,SAAS,CAAC,WAAW,EAAE,CAAC;AAC9F,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,wBAAwB,CACvC,KAA8B,EAC9B,OAA2C;IAE3C,OAAO,CACN,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC;QAC/B,CAAC,OAAO,EAAE,WAAW,KAAK,uBAAuB,IAAI,OAAO,EAAE,WAAW,KAAK,4BAA4B,CAAC,CAC3G,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAyB;IACvD,IAAI,CAAC,GAAG;QAAE,OAAO,MAAM,CAAC;IACxB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5G,IAAI,IAAI,GAAG,UAAU,CAAC;IACtB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC3D,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;AAClC,CAAC;AAGD,SAAS,gCAAgC,CACxC,KAA+C,EAC/C,WAAmB,EACnB,OAA+D,EAC/D,IAAmC,EACnC,sBAAmD;IAEnD,IAAI,OAAO,CAAC,SAAS,KAAK,KAAK,IAAI,CAAC,OAAO,CAAC,SAAS;QAAE,OAAO;IAC9D,MAAM,SAAS,GAAG,GAAG,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,OAAO,KAAK,qBAAqB,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,WAAW,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC9I,MAAM,QAAQ,GAAG,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACnE,IAAI,QAAQ,EAAE,SAAS,KAAK,SAAS,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QAC3E,sBAAsB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IACD,IAAI,QAAQ;QAAE,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,EAAE;QAC9B,IAAI,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,SAAU,CAAC,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;YACjF,0BAA0B,CAAC,MAAM,CAAC,OAAO,CAAC,SAAU,CAAC,CAAC;QACvD,CAAC;IACF,CAAC,EAAE,qCAAqC,CAAC,CAAC;IAC1C,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;IACjB,0BAA0B,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC/C,KAAiB,EACjB,OAAiB,EACjB,sBAAsB,GAAgC,iCAAiC;IAEvF,MAAM,OAAO,GAAoB,EAAE,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;IAChE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACpC,MAAM,IAAI,GAAG,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;IACnG,MAAM,WAAW,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO;QACN,GAAG,OAAO;QACV,OAAO;QACP,KAAK,EAAE,uBAAuB,CAAC,OAAO,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,CAAC;QACjE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE;YAC1C,MAAM,WAAW,GAAG,MAAM,SAAS,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAC7D,MAAM,YAAY,GAAG,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC;YACvE,YAAY,CAAC,OAAO,EAAE,uBAAuB,CAAC,CAAC;YAC/C,gCAAgC,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,IAAI,EAAE,CAAC;gBACV,SAAS,CAAC,OAAO,EAAE,uBAAuB,EAAE,SAAS,WAAW,SAAS,IAAI,EAAE,CAAC,CAAC;YAClF,CAAC;iBAAM,CAAC;gBACP,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;oBACrD,IAAI,IAAI,CAAC,WAAW,EAAE,KAAK,YAAY,IAAI,KAAK,KAAK,2BAA2B;wBAAE,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;gBACxG,CAAC;YACF,CAAC;YACD,yBAAyB,CAAC,OAAO,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC;YACvD,gCAAgC,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,sBAAsB,CAAC,CAAC;YAC5F,OAAO,YAAY,CAAC;QACrB,CAAC;KACW,CAAC;AACf,CAAC;AAED,SAAS,iCAAiC,CACzC,KAAiB,EACjB,OAA2C;IAE3C,4EAA4E;IAC5E,MAAM,cAAc,GACnB,KAAK,CAAC,cAAc,IAAI,OAAO,EAAE,cAAc;QAC9C,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,cAAc,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE;QACzD,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,eAAe,GAAkB;QACtC,cAAc;QACd,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,MAAM,EAAE,OAAO,EAAE,MAAM;QACvB,MAAM,EAAE,OAAO,EAAE,MAAM;QACvB,KAAK,EAAE,OAAO,EAAE,KAAK;QACrB,GAAG,EAAE,OAAO,EAAE,GAAG;QACjB,gBAAgB,EAAE,OAAO,EAAE,gBAAgB;QAC3C,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,UAAU,EAAE,OAAO,EAAE,UAAU;QAC/B,qBAAqB,EAAE,OAAO,EAAE,qBAAqB;QACrD,OAAO,EAAE,OAAO,EAAE,OAAO;QACzB,SAAS,EAAE,OAAO,EAAE,SAAS;QAC7B,yBAAyB,EAAE,OAAO,EAAE,yBAAyB;QAC7D,gBAAgB,EAAE,OAAO,EAAE,gBAAgB;QAC3C,UAAU,EAAE,OAAO,EAAE,UAAU;QAC/B,eAAe,EAAE,OAAO,EAAE,eAAe;QACzC,QAAQ,EAAE,OAAO,EAAE,QAAQ;KAC3B,CAAC;IACF,OAAO,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,gCAAgC,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC;AACtH,CAAC;AAED,MAAM,UAAU,2BAA2B,CAC1C,KAAiB,EACjB,SAAoC;IAEpC,MAAM,gBAAgB,GAAG,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtF,OAAO,gBAAgB,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,gBAAgB,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,oCAAoC,CACnD,KAAiB,EACjB,OAA2C;IAE3C,OAAO;QACN,GAAG,iCAAiC,CAAC,KAAK,EAAE,OAAO,CAAC;QACpD,eAAe,EAAE,2BAA2B,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC;QACvE,WAAW,EAAE,OAAO,EAAE,WAAW;KACjC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,yCAAyC,CACxD,KAAiB,EACjB,OAA2C;IAE3C,OAAO;QACN,GAAG,iCAAiC,CAAC,KAAK,EAAE,OAAO,CAAC;QACpD,eAAe,EAAE,2BAA2B,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC;QACvE,WAAW,EAAE,OAAO,EAAE,WAAW;KACjC,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CAClC,KAAiB,EACjB,OAAgB,EAChB,OAA2C,EAC3C,SAAS,GAAuB,4BAA4B;IAE5D,IAAI,wBAAwB,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC;QAC9C,IAAI,KAAK,CAAC,GAAG,KAAK,kBAAkB,EAAE,CAAC;YACtC,OAAO,SAAS,CAAC,qBAAqB,CACrC,KAAkC,EAClC,gBAAgB,CAAC,OAAO,CAAC,EACzB,oCAAoC,CAAC,KAAK,EAAE,OAAO,CAAC,CACpD,CAAC;QACH,CAAC;QAED,2EAA2E;QAC3E,0EAA0E;QAC1E,kEAAkE;QAClE,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,IAAI,yBAAyB,CAAC,KAAK,CAAC,EAAE,CAAC;YACtE,sCAAsC,EAAE,CAAC;QAC1C,CAAC;QAED,OAAO,SAAS,CAAC,0BAA0B,CAC1C,KAAwC,EACxC,gBAAgB,CAAC,OAAO,CAAC,EACzB,yCAAyC,CAAC,KAAK,EAAE,OAAO,CAAC,CACzD,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import { closeOpenAICodexWebSocketSessions } from \"@bastani/pi-ai/api/openai-codex-responses\";\nimport {\n\ttype Api,\n\ttype AssistantMessageEventStream,\n\ttype Context,\n\tclampThinkingLevel,\n\ttype Model,\n\ttype ModelFastRoute,\n\tnormalizeContext,\n\ttype OpenAICodexResponsesOptions,\n\ttype OpenAIResponsesOptions,\n\ttype ProviderHeaders,\n\ttype SimpleStreamOptions,\n\ttype StreamOptions,\n\tstreamOpenAICodexResponses,\n\tstreamOpenAIResponses,\n\tstreamSimple,\n\ttype ThinkingLevel,\n\ttype TranscriptContext,\n} from \"@bastani/pi-ai/compat\";\nimport {\n\tCODEX_FAST_ROUTE_HEADER,\n\tCODEX_FAST_ROUTE_ORIGINATOR,\n\tclearCodexFastRouteRequestMarker,\n\tinstallCodexFastRouteWebSocketIdentity,\n\tisFirstPartyCodexBaseUrl,\n\tmarkCodexFastRouteRequest,\n\twrapCodexFastRouteFetch,\n} from \"./fast-model-routing-transport.ts\";\nimport { FAST_MODEL_SERVICE_TIER, isNativeFastRouteApi, ULTRAFAST_MODEL_SERVICE_TIER } from \"./fast-model-variants.ts\";\n\nexport interface FastRouteStreamOptions extends SimpleStreamOptions {\n\tserviceTier?: ModelFastRoute[\"serviceTier\"];\n}\n\nexport interface FastRouteStreamers {\n\tstreamSimple: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;\n\tstreamOpenAIResponses: (\n\t\tmodel: Model<\"openai-responses\">,\n\t\tcontext: TranscriptContext,\n\t\toptions?: OpenAIResponsesOptions,\n\t) => AssistantMessageEventStream;\n\tstreamOpenAICodexResponses: (\n\t\tmodel: Model<\"openai-codex-responses\">,\n\t\tcontext: TranscriptContext,\n\t\toptions?: OpenAICodexResponsesOptions,\n\t) => AssistantMessageEventStream;\n}\n\nconst DEFAULT_FAST_ROUTE_STREAMERS: FastRouteStreamers = {\n\tstreamSimple,\n\tstreamOpenAIResponses,\n\tstreamOpenAICodexResponses,\n};\n\nconst CODEX_WEBSOCKET_ROUTING_STATE_IDLE_MS = 6 * 60 * 1000;\nconst codexWebSocketRoutingState = new Map<string, { signature: string; expiry: ReturnType<typeof setTimeout> }>();\n\n/*\n Explicit route metadata is the only source of fast semantics. A model whose ID merely ends in\n * `-fast` — a provider, `models.json`, or extension model that owns that exact ID — routes normally.\n /\nexport function getModelFastRoute(model: Pick<Model<Api>, \"fastRoute\">): ModelFastRoute \| undefined {\n\treturn model.fastRoute;\n}\n\n/\n The model ID this request sends upstream. An OpenAI-style fast route sends the base model and\n * carries a priority service tier instead; a provider with real fast siblings sends its own\n * advertised ID, which is already the canonical one. The canonical `-fast` identity always stays on\n * the model object Atomic selected, persisted, and records — only the serialized request differs.\n /\nexport function resolveUpstreamModelId(model: Pick<Model<Api>, \"fastRoute\" \| \"id\">): string {\n\treturn model.fastRoute?.upstreamModelId ?? model.id;\n}\n\n/\n Whether a model uses the first-party OpenAI Codex provider identity and endpoint.\n * A renamed provider on the same protocol remains non-first-party.\n /\nexport function usesFirstPartyCodexRouting(model: Pick<Model<Api>, \"baseUrl\" \| \"provider\">): boolean {\n\treturn model.provider === \"openai-codex\" && isFirstPartyCodexBaseUrl(model.baseUrl);\n}\n\n/* A provider alias or proxy still uses the shared ChatGPT Codex protocol when its API does. /\nexport function usesChatGptCodexTransport(model: Pick<Model<Api>, \"api\">): boolean {\n\treturn model.api === \"openai-codex-responses\";\n}\n\nfunction deleteHeader(headers: ProviderHeaders, name: string): void {\n\tfor (const existingName of Object.keys(headers)) {\n\t\tif (existingName.toLowerCase() === name.toLowerCase()) delete headers[existingName];\n\t}\n}\n\nfunction setHeader(headers: ProviderHeaders, name: string, value: string): void {\n\tdeleteHeader(headers, name);\n\theaders[name] = value;\n}\n\n/\n Eager first-party routing headers for a model that carries an OpenAI-style fast route.\n \n Derived entirely from the model: the headers are added only when the model's own `fastRoute`\n * declares the priority tier and the model uses first-party Codex routing, and the hint names the\n * route's upstream model ID — the same value the request serializes and\n * `withChatGptCodexTransportRouting` reads from route metadata. There is no caller-supplied enable\n * flag, so a model without a route can never be granted this identity.\n /\nexport function withCodexFastRouteHeaders(\n\tmodel: Pick<Model<Api>, \"baseUrl\" \| \"fastRoute\" \| \"id\" \| \"provider\">,\n\theaders: ProviderHeaders \| undefined,\n): ProviderHeaders \| undefined {\n\tconst tier = getModelFastRoute(model)?.serviceTier;\n\tif (tier === undefined \|\| !usesFirstPartyCodexRouting(model)) {\n\t\treturn headers;\n\t}\n\tconst fastHeaders: ProviderHeaders = { ...(headers ?? {}) };\n\tsetHeader(fastHeaders, \"originator\", CODEX_FAST_ROUTE_ORIGINATOR);\n\tsetHeader(fastHeaders, CODEX_FAST_ROUTE_HEADER, `model=${resolveUpstreamModelId(model)};tier=${tier}`);\n\treturn fastHeaders;\n}\n\n/* Attach the route's service tier to the outgoing stream options. /\nexport function withFastRouteStreamOptions(\n\tfastRoute: ModelFastRoute \| undefined,\n\toptions: SimpleStreamOptions \| undefined,\n): FastRouteStreamOptions \| undefined {\n\tif (fastRoute?.serviceTier === undefined) return options;\n\treturn { ...(options ?? {}), headers: options?.headers, serviceTier: fastRoute.serviceTier };\n}\n\n/* A service-tier route needs pi-ai's native adapters; `streamSimple` drops `serviceTier`. /\nexport function shouldUseNativeFastRoute(\n\tmodel: Pick<Model<Api>, \"api\">,\n\toptions: FastRouteStreamOptions \| undefined,\n): boolean {\n\treturn (\n\t\tisNativeFastRouteApi(model.api) &&\n\t\t(options?.serviceTier === FAST_MODEL_SERVICE_TIER \|\| options?.serviceTier === ULTRAFAST_MODEL_SERVICE_TIER)\n\t);\n}\n\nfunction codexEnvironmentScope(env: StreamOptions[\"env\"]): string {\n\tif (!env) return \"none\";\n\tconst serialized = JSON.stringify(Object.entries(env).sort(([left], [right]) => left.localeCompare(right)));\n\tlet hash = 2166136261;\n\tfor (let index = 0; index < serialized.length; index += 1) {\n\t\thash = Math.imul(hash ^ serialized.charCodeAt(index), 16777619);\n\t}\n\treturn (hash >>> 0).toString(16);\n}\n\ntype CloseCodexWebSocketSessions = typeof closeOpenAICodexWebSocketSessions;\nfunction updateCodexWebSocketRoutingState(\n\tmodel: Pick<Model<Api>, \"baseUrl\" \| \"provider\">,\n\troutedModel: string,\n\toptions: Pick<StreamOptions, \"env\" \| \"sessionId\" \| \"transport\">,\n\ttier: ModelFastRoute[\"serviceTier\"],\n\tcloseWebSocketSessions: CloseCodexWebSocketSessions,\n): void {\n\tif (options.transport === \"sse\" \|\| !options.sessionId) return;\n\tconst signature = `${model.provider}\\0${model.baseUrl}\\0${codexEnvironmentScope(options.env)}\\0${tier ? `${tier}:${routedModel}` : \"normal\"}`;\n\tconst previous = codexWebSocketRoutingState.get(options.sessionId);\n\tif (previous?.signature !== undefined && previous.signature !== signature) {\n\t\tcloseWebSocketSessions(options.sessionId);\n\t}\n\tif (previous) clearTimeout(previous.expiry);\n\tconst expiry = setTimeout(() => {\n\t\tif (codexWebSocketRoutingState.get(options.sessionId!)?.signature === signature) {\n\t\t\tcodexWebSocketRoutingState.delete(options.sessionId!);\n\t\t}\n\t}, CODEX_WEBSOCKET_ROUTING_STATE_IDLE_MS);\n\texpiry.unref?.();\n\tcodexWebSocketRoutingState.set(options.sessionId, { signature, expiry });\n}\n\nexport function withChatGptCodexTransportRouting<TOptions extends StreamOptions>(\n\tmodel: Model<Api>,\n\toptions: TOptions,\n\tcloseWebSocketSessions: CloseCodexWebSocketSessions = closeOpenAICodexWebSocketSessions,\n): TOptions {\n\tconst headers: ProviderHeaders = { ...(options.headers ?? {}) };\n\tconst onPayload = options.onPayload;\n\tconst tier = usesFirstPartyCodexRouting(model) ? getModelFastRoute(model)?.serviceTier : undefined;\n\tconst routedModel = resolveUpstreamModelId(model);\n\treturn {\n\t\t...options,\n\t\theaders,\n\t\tfetch: wrapCodexFastRouteFetch(options.fetch ?? globalThis.fetch),\n\t\tonPayload: async (payload, requestModel) => {\n\t\t\tconst replacement = await onPayload?.(payload, requestModel);\n\t\t\tconst finalPayload = replacement === undefined ? payload : replacement;\n\t\t\tdeleteHeader(headers, CODEX_FAST_ROUTE_HEADER);\n\t\t\tclearCodexFastRouteRequestMarker(headers);\n\t\t\tif (tier) {\n\t\t\t\tsetHeader(headers, CODEX_FAST_ROUTE_HEADER, `model=${routedModel};tier=${tier}`);\n\t\t\t} else {\n\t\t\t\tfor (const [name, value] of Object.entries(headers)) {\n\t\t\t\t\tif (name.toLowerCase() === \"originator\" && value === CODEX_FAST_ROUTE_ORIGINATOR) delete headers[name];\n\t\t\t\t}\n\t\t\t}\n\t\t\tmarkCodexFastRouteRequest(headers, tier !== undefined);\n\t\t\tupdateCodexWebSocketRoutingState(model, routedModel, options, tier, closeWebSocketSessions);\n\t\t\treturn finalPayload;\n\t\t},\n\t} as TOptions;\n}\n\nfunction buildFastRouteBaseProviderOptions(\n\tmodel: Model<Api>,\n\toptions: FastRouteStreamOptions \| undefined,\n): StreamOptions {\n\t// Native fast-route streams bypass pi-ai's streamSimple base-options merge.\n\tconst samplingParams =\n\t\tmodel.samplingParams \|\| options?.samplingParams\n\t\t\t? { ...model.samplingParams, ...options?.samplingParams }\n\t\t\t: undefined;\n\tconst providerOptions: StreamOptions = {\n\t\tsamplingParams,\n\t\ttemperature: options?.temperature,\n\t\tmaxTokens: options?.maxTokens,\n\t\tsignal: options?.signal,\n\t\tapiKey: options?.apiKey,\n\t\tfetch: options?.fetch,\n\t\tenv: options?.env,\n\t\ttelemetryContext: options?.telemetryContext,\n\t\ttransport: options?.transport,\n\t\tcacheRetention: options?.cacheRetention,\n\t\tsessionId: options?.sessionId,\n\t\tonPayload: options?.onPayload,\n\t\tonResponse: options?.onResponse,\n\t\tonProviderStreamEvent: options?.onProviderStreamEvent,\n\t\theaders: options?.headers,\n\t\ttimeoutMs: options?.timeoutMs,\n\t\twebsocketConnectTimeoutMs: options?.websocketConnectTimeoutMs,\n\t\tstreamDeadlineMs: options?.streamDeadlineMs,\n\t\tmaxRetries: options?.maxRetries,\n\t\tmaxRetryDelayMs: options?.maxRetryDelayMs,\n\t\tmetadata: options?.metadata,\n\t};\n\treturn usesChatGptCodexTransport(model) ? withChatGptCodexTransportRouting(model, providerOptions) : providerOptions;\n}\n\nexport function mapFastRouteReasoningEffort(\n\tmodel: Model<Api>,\n\treasoning: ThinkingLevel \| undefined,\n): ThinkingLevel \| undefined {\n\tconst clampedReasoning = reasoning ? clampThinkingLevel(model, reasoning) : undefined;\n\treturn clampedReasoning === \"off\" ? undefined : clampedReasoning;\n}\n\nexport function buildOpenAIResponsesFastRouteOptions(\n\tmodel: Model<Api>,\n\toptions: FastRouteStreamOptions \| undefined,\n): OpenAIResponsesOptions {\n\treturn {\n\t\t...buildFastRouteBaseProviderOptions(model, options),\n\t\treasoningEffort: mapFastRouteReasoningEffort(model, options?.reasoning),\n\t\tserviceTier: options?.serviceTier,\n\t};\n}\n\nexport function buildOpenAICodexResponsesFastRouteOptions(\n\tmodel: Model<Api>,\n\toptions: FastRouteStreamOptions \| undefined,\n): OpenAICodexResponsesOptions {\n\treturn {\n\t\t...buildFastRouteBaseProviderOptions(model, options),\n\t\treasoningEffort: mapFastRouteReasoningEffort(model, options?.reasoning),\n\t\tserviceTier: options?.serviceTier,\n\t};\n}\n\n/\n Stream a request whose canonical selected model carries fast-route metadata. The provider adapter\n * translates only the serialized request to the route's upstream ID; the model object keeps its\n * canonical `-fast` identity.\n */\nexport function streamWithFastRoute(\n\tmodel: Model<Api>,\n\tcontext: Context,\n\toptions: FastRouteStreamOptions \| undefined,\n\tstreamers: FastRouteStreamers = DEFAULT_FAST_ROUTE_STREAMERS,\n): AssistantMessageEventStream {\n\tif (shouldUseNativeFastRoute(model, options)) {\n\t\tif (model.api === \"openai-responses\") {\n\t\t\treturn streamers.streamOpenAIResponses(\n\t\t\t\tmodel as Model<\"openai-responses\">,\n\t\t\t\tnormalizeContext(context),\n\t\t\t\tbuildOpenAIResponsesFastRouteOptions(model, options),\n\t\t\t);\n\t\t}\n\n\t\t// The WebSocket handshake builds its headers inside pi-ai, and the runtime\n\t\t// constructor can be cached before the first turn, so repair the identity\n\t\t// through the global constructor rather than per-request options.\n\t\tif (options?.transport !== \"sse\" && usesChatGptCodexTransport(model)) {\n\t\t\tinstallCodexFastRouteWebSocketIdentity();\n\t\t}\n\n\t\treturn streamers.streamOpenAICodexResponses(\n\t\t\tmodel as Model<\"openai-codex-responses\">,\n\t\t\tnormalizeContext(context),\n\t\t\tbuildOpenAICodexResponsesFastRouteOptions(model, options),\n\t\t);\n\t}\n\n\treturn streamers.streamSimple(model, context, options);\n}\n"]}

package/dist/core/fast-model-variants.d.ts CHANGED Viewed

@@ -1,9 +1,12 @@
-import type { Api, Credential, Model, Provider } from "@bastani/pi-ai";
+import { type Api, type Credential, type Model, type Provider } from "@bastani/pi-ai";
 import type { ModelsJsonModelOverride } from "./model-config.ts";
 /** Canonical suffix every selectable fast model variant carries. */
 export declare const FAST_MODEL_ID_SUFFIX = "-fast";
 /** Service tier OpenAI-style providers use to route fast traffic. */
 export declare const FAST_MODEL_SERVICE_TIER: "priority";
+/** Codex's Ultrafast tier, offered only for models that advertise it; access depends on the account. */
+export declare const ULTRAFAST_MODEL_ID_SUFFIX = "-ultrafast";
+export declare const ULTRAFAST_MODEL_SERVICE_TIER: "ultrafast";
 /** Inference speed Anthropic's Messages API uses to route fast traffic. */
 export declare const FAST_MODEL_SPEED: "fast";
 /**

package/dist/core/fast-model-variants.d.ts.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"file":"fast-model-variants.d.ts","sourceRoot":"","sources":["../../src/core/fast-model-variants.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,~~EAAE,~~GAAG,~~EAAE~~,UAAU,~~EAAE~~,KAAK,~~EAAkB~~,QAAQ,~~EAAE~~,MAAM,gBAAgB,CAAC;~~AACvF~~,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAGjE,oEAAoE;AACpE,eAAO,MAAM,oBAAoB,UAAU,CAAC;AAE5C,qEAAqE;AACrE,eAAO,MAAM,uBAAuB,EAAG,UAAmB,CAAC;AAE3D,2EAA2E;AAC3E,eAAO,MAAM,gBAAgB,EAAG,MAAe,CAAC;AAEhD;;;;GAIG;AACH,eAAO,MAAM,6BAA6B,EAAE,WAAW,CAAC,MAAM,CAI5D,CAAC;AAEH,oGAAoG;AACpG,MAAM,WAAW,0BAA0B;IAC1C,QAAQ,EAAE,MAAM,CAAC;IACjB,mDAAmD;IACnD,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,0BAA0B;IAC1C,MAAM,EAAE,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IAC9B,WAAW,EAAE,SAAS,0BAA0B,EAAE,CAAC;CACnD;AAED,MAAM,WAAW,8BAA8B;IAC9C,6FAA6F;IAC7F,mBAAmB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IACxC,8GAA8G;IAC9G,cAAc,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC,CAAC;IACnE,0GAA0G;IAC1G,kBAAkB,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC;IACtC,2GAA2G;IAC3G,cAAc,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,wBAAwB;IACxC;;;OAGG;IACH,sBAAsB,CAAC,EAAE,MAAM,SAAS,MAAM,EAAE,GAAG,SAAS,CAAC;IAC7D,oGAAoG;IACpG,iBAAiB,CAAC,EAAE,MAAM,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC,GAAG,SAAS,CAAC;IACxF,wFAAwF;IACxF,iBAAiB,CAAC,EAAE,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;IAC9C;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,WAAW,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;IAC3D,8FAA8F;IAC9F,aAAa,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,0BAA0B,EAAE,KAAK,IAAI,CAAC;CACjG;AAED,oEAAoE;AACpE,wBAAgB,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,kBAAkB,GAAG,wBAAwB,CAEnG;AAED;;;;;;;;GAQG;AACH,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,GAAG,OAAO,CAE9F;AAED;;;;;;;;GAQG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,GAAG,OAAO,CAE3F;AAED,uGAAuG;AACvG,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,CAAC,GAAG,OAAO,CAEjF;AAED;;;;;;GAMG;AACH,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,GAAG,UAAU,CAAC,GAAG,OAAO,CAMjG;AAOD,yFAAyF;AACzF,wBAAgB,6BAA6B,CAAC,UAAU,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,MAAM,EAAE,GAAG,SAAS,CAG/G;AAED;;;;;;;GAOG;AACH,wBAAgB,wBAAwB,CAAC,UAAU,EAAE,UAAU,GAAG,SAAS,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAMrG;~~AAqBD~~;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,uBAAuB,CACtC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,EAC7B,OAAO,GAAE,8BAAmC,GAC1C,0BAA0B,~~CAmC5B~~;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,GAAE,wBAA6B,GAAG,QAAQ,CAgB1G"}
1	+ {"version":3,"file":"fast-model-variants.d.ts","sourceRoot":"","sources":["../../src/core/fast-model-variants.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,GAAG,EACR,KAAK,UAAU,EAEf,KAAK,KAAK,EAEV,KAAK,QAAQ,EACb,MAAM,gBAAgB,CAAC;AACxB,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAGjE,oEAAoE;AACpE,eAAO,MAAM,oBAAoB,UAAU,CAAC;AAE5C,qEAAqE;AACrE,eAAO,MAAM,uBAAuB,EAAG,UAAmB,CAAC;AAE3D,wGAAwG;AACxG,eAAO,MAAM,yBAAyB,eAAe,CAAC;AACtD,eAAO,MAAM,4BAA4B,EAAG,WAAoB,CAAC;AAEjE,2EAA2E;AAC3E,eAAO,MAAM,gBAAgB,EAAG,MAAe,CAAC;AAEhD;;;;GAIG;AACH,eAAO,MAAM,6BAA6B,EAAE,WAAW,CAAC,MAAM,CAI5D,CAAC;AAEH,oGAAoG;AACpG,MAAM,WAAW,0BAA0B;IAC1C,QAAQ,EAAE,MAAM,CAAC;IACjB,mDAAmD;IACnD,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,0BAA0B;IAC1C,MAAM,EAAE,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IAC9B,WAAW,EAAE,SAAS,0BAA0B,EAAE,CAAC;CACnD;AAED,MAAM,WAAW,8BAA8B;IAC9C,6FAA6F;IAC7F,mBAAmB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IACxC,8GAA8G;IAC9G,cAAc,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC,CAAC;IACnE,0GAA0G;IAC1G,kBAAkB,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC;IACtC,2GAA2G;IAC3G,cAAc,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,wBAAwB;IACxC;;;OAGG;IACH,sBAAsB,CAAC,EAAE,MAAM,SAAS,MAAM,EAAE,GAAG,SAAS,CAAC;IAC7D,oGAAoG;IACpG,iBAAiB,CAAC,EAAE,MAAM,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,uBAAuB,CAAC,CAAC,GAAG,SAAS,CAAC;IACxF,wFAAwF;IACxF,iBAAiB,CAAC,EAAE,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;IAC9C;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,WAAW,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;IAC3D,8FAA8F;IAC9F,aAAa,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,0BAA0B,EAAE,KAAK,IAAI,CAAC;CACjG;AAED,oEAAoE;AACpE,wBAAgB,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,kBAAkB,GAAG,wBAAwB,CAEnG;AAED;;;;;;;;GAQG;AACH,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,GAAG,OAAO,CAE9F;AAED;;;;;;;;GAQG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,GAAG,OAAO,CAE3F;AAED,uGAAuG;AACvG,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,CAAC,GAAG,OAAO,CAEjF;AAED;;;;;;GAMG;AACH,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,GAAG,UAAU,CAAC,GAAG,OAAO,CAMjG;AAOD,yFAAyF;AACzF,wBAAgB,6BAA6B,CAAC,UAAU,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,MAAM,EAAE,GAAG,SAAS,CAG/G;AAED;;;;;;;GAOG;AACH,wBAAgB,wBAAwB,CAAC,UAAU,EAAE,UAAU,GAAG,SAAS,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAMrG;AA+BD;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,uBAAuB,CACtC,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,EAC7B,OAAO,GAAE,8BAAmC,GAC1C,0BAA0B,CAkD5B;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,GAAE,wBAA6B,GAAG,QAAQ,CAgB1G"}

package/dist/core/fast-model-variants.js CHANGED Viewed

@@ -1,8 +1,12 @@
+import { getServiceTierCost, } from "@bastani/pi-ai";
 import { applyModelOverride } from "./provider-composer-internal.js";
 /** Canonical suffix every selectable fast model variant carries. */
 export const FAST_MODEL_ID_SUFFIX = "-fast";
 /** Service tier OpenAI-style providers use to route fast traffic. */
 export const FAST_MODEL_SERVICE_TIER = "priority";
+/** Codex's Ultrafast tier, offered only for models that advertise it; access depends on the account. */
+export const ULTRAFAST_MODEL_ID_SUFFIX = "-ultrafast";
+export const ULTRAFAST_MODEL_SERVICE_TIER = "ultrafast";
 /** Inference speed Anthropic's Messages API uses to route fast traffic. */
 export const FAST_MODEL_SPEED = "fast";
 /**
@@ -92,8 +96,17 @@ export function copilotAdvertisesModelId(credential, modelId) {
     return (advertisedModelIds(credential.availableModelIds)?.includes(modelId) === true ||
         advertisedModelIds(credential.fastModelIds)?.includes(modelId) === true);
 }
+/** A model without tier metadata keeps offering Fast; one that lists its tiers must advertise it, as in Codex. */
+function advertisesFastServiceTier(model) {
+    return model.serviceTiers === undefined || getServiceTierCost(model, FAST_MODEL_SERVICE_TIER) !== undefined;
+}
 function fastRouteForBaseModel(model, entitledCopilotFastModelIds) {
-    if (usesOpenAIFastServiceTier(model) || usesXaiFastServiceTier(model)) {
+    if (usesOpenAIFastServiceTier(model)) {
+        return advertisesFastServiceTier(model)
+            ? { baseModelId: model.id, upstreamModelId: model.id, serviceTier: FAST_MODEL_SERVICE_TIER }
+            : undefined;
+    }
+    if (usesXaiFastServiceTier(model)) {
         return { baseModelId: model.id, upstreamModelId: model.id, serviceTier: FAST_MODEL_SERVICE_TIER };
     }
     if (usesAnthropicFastMode(model)) {
@@ -133,7 +146,9 @@ export function deriveFastModelVariants(providerId, models, options = {}) {
     let changed = false;
     for (const model of models) {
         derived.push(model);
-        if (model.fastRoute !== undefined || model.id.endsWith(FAST_MODEL_ID_SUFFIX))
+        if (model.fastRoute !== undefined ||
+            model.id.endsWith(FAST_MODEL_ID_SUFFIX) ||
+            model.id.endsWith(ULTRAFAST_MODEL_ID_SUFFIX))
             continue;
         if (options.customModelIds?.has(model.id))
             continue;
@@ -143,23 +158,33 @@ export function deriveFastModelVariants(providerId, models, options = {}) {
         if (options.extensionOwnedApis?.has(model.api) === true)
             continue;
         const fastRoute = fastRouteForBaseModel(model, entitledCopilotFastModelIds);
-        if (!fastRoute)
-            continue;
-        const variantId = fastModelId(model.id);
-        if (ownedModelIds.has(variantId)) {
-            diagnostics.push({
-                provider: providerId,
-                modelId: variantId,
-                message: `Model "${providerId}/${variantId}" is already defined by the provider, models.json, or an extension, ` +
-                    `so Atomic did not derive a fast variant of "${providerId}/${model.id}". ` +
-                    `That model routes exactly as declared. Rename or remove it to get the derived fast variant instead.`,
+        const routes = [];
+        if (fastRoute)
+            routes.push({ suffix: FAST_MODEL_ID_SUFFIX, label: "fast", route: fastRoute });
+        if (usesOpenAIFastServiceTier(model) && getServiceTierCost(model, ULTRAFAST_MODEL_SERVICE_TIER)) {
+            routes.push({
+                suffix: ULTRAFAST_MODEL_ID_SUFFIX,
+                label: "ultrafast",
+                route: { baseModelId: model.id, upstreamModelId: model.id, serviceTier: ULTRAFAST_MODEL_SERVICE_TIER },
             });
-            continue;
         }
-        const override = options.modelOverrides?.[variantId];
-        const variant = { ...model, id: variantId, name: `${model.name} (fast)`, fastRoute };
-        derived.push(override ? applyModelOverride(variant, override) : variant);
-        changed = true;
+        for (const { suffix, label, route } of routes) {
+            const variantId = `${model.id}${suffix}`;
+            if (ownedModelIds.has(variantId)) {
+                diagnostics.push({
+                    provider: providerId,
+                    modelId: variantId,
+                    message: `Model "${providerId}/${variantId}" is already defined by the provider, models.json, or an extension, ` +
+                        `so Atomic did not derive a ${label} variant of "${providerId}/${model.id}". ` +
+                        `That model routes exactly as declared. Rename or remove it to get the derived ${label} variant instead.`,
+                });
+                continue;
+            }
+            const override = options.modelOverrides?.[variantId];
+            const variant = { ...model, id: variantId, name: `${model.name} (${label})`, fastRoute: route };
+            derived.push(override ? applyModelOverride(variant, override) : variant);
+            changed = true;
+        }
     }
     if (!changed && diagnostics.length === 0)
         return { models, diagnostics };