@springbrand/agent-runtime 0.3.0-alpha.23 → 0.3.0-alpha.24

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/agent-runtime",
-  "version": "0.3.0-alpha.23",
+  "version": "0.3.0-alpha.24",
   "type": "module",
   "files": [
     "src",
@@ -35,7 +35,8 @@
     "lodash-es": "^4.18.1",
     "zod": "^4.4.3",
     "@earendil-works/pi-durable": "1.1.0",
-    "@earendil-works/chord": "1.1.0"
+    "@earendil-works/chord": "1.1.0",
+    "@springbrand/model-client": "0.1.0-alpha.1"
   },
   "peerDependencies": {
     "agents": ">=0.28.1 <0.29.0",

package/src/config/ports.ts

@@ -1,5 +1,6 @@
 export type { RuntimeSchedulePreparation } from "../submission/contracts";
 import type { RuntimeModelInputModality } from "./profile";
+import type { OpenRouterGenerationProtocol } from "@springbrand/model-client";
 import type { SkillSource } from "agents/skills";
 import type { ScheduleSpec, RuntimeSchedulePreparation } from "../submission/contracts";
 import type { RuntimeActivityProjection } from "../conversation/state";
@@ -949,9 +950,7 @@ export interface RuntimeBrowserPort {
  */
 export type RuntimeModelProtocol =
   | "openai-chat"
-  | "openrouter-messages"
-  | "openrouter-chat"
-  | "openrouter-responses"
+  | OpenRouterGenerationProtocol
   | "anthropic-messages"
   | "google-generative-ai"
   | "openai-codex-responses";

package/src/index.ts

@@ -145,11 +145,3 @@ export {
   modelRequestUrl,
   resolvePiModel,
 } from "./models";
-export {
-  openRouterRequestUrl,
-  requestOpenRouter,
-} from "./model-protocol/openrouter";
-export type { OpenRouterRequestOptions } from "./model-protocol/openrouter";
-export type {
-  OpenRouterProtocol,
-} from "./models/openrouter/protocol-response";

package/src/models/index.ts

@@ -1,4 +1,9 @@
 import type { StreamFn } from "@earendil-works/pi-agent-core";
+import {
+  openRouterRequestUrl,
+  withOpenRouterStructuredOutput,
+  type OpenRouterGenerationProtocol,
+} from "@springbrand/model-client";
 import {
   createModels,
   createProvider,
@@ -148,18 +153,23 @@ type ProviderBoundStream = (
   options?: SimpleStreamOptions,
 ) => ReturnType<StreamFn>;
 
+function openRouterProtocolForModel(model: Model<Api>): OpenRouterGenerationProtocol | undefined {
+  if (model.api === "anthropic-messages" &&
+    model.provider.startsWith("model-api-openrouter-messages-")) return "openrouter-messages";
+  if (model.api === "openai-completions" &&
+    model.provider.startsWith("model-api-openrouter-chat-")) return "openrouter-chat";
+  if (model.api === "openai-responses" &&
+    model.provider.startsWith("model-api-openrouter-responses-")) return "openrouter-responses";
+  return undefined;
+}
+
 export function structuredOutputStream(
   streamFn: StreamFn,
   model: Model<Api>,
   schema: Record<string, unknown>,
 ): StreamFn | undefined {
-  const messages = model.api === "anthropic-messages" &&
-    model.provider.startsWith("model-api-openrouter-messages-");
-  const chat = model.api === "openai-completions" &&
-    model.provider.startsWith("model-api-openrouter-chat-");
-  const responses = model.api === "openai-responses" &&
-    model.provider.startsWith("model-api-openrouter-responses-");
-  if (!messages && !chat && !responses) return undefined;
+  const protocol = openRouterProtocolForModel(model);
+  if (!protocol) return undefined;
 
   return (activeModel, context, options) =>
     streamFn(activeModel, context, {
@@ -169,38 +179,9 @@ export function structuredOutputStream(
         if (!value || typeof value !== "object" || Array.isArray(value)) {
           throw new Error("Structured model payload must be an object");
         }
-        const body = value as Record<string, unknown>;
-        if (messages) {
-          const current = body.output_config;
-          return {
-            ...body,
-            output_config: {
-              ...(current && typeof current === "object" && !Array.isArray(current)
-                ? current
-                : {}),
-              format: { type: "json_schema", schema },
-            },
-          };
-        }
-        if (chat) {
-          return {
-            ...body,
-            response_format: {
-              type: "json_schema",
-              json_schema: { name: "agent_output", strict: true, schema },
-            },
-          };
-        }
-        const current = body.text;
-        return {
-          ...body,
-          text: {
-            ...(current && typeof current === "object" && !Array.isArray(current)
-              ? current
-              : {}),
-            format: { type: "json_schema", name: "agent_output", strict: true, schema },
-          },
-        };
+        return withOpenRouterStructuredOutput(
+          value as Record<string, unknown>, protocol, schema, "agent_output",
+        );
       },
     });
 }
@@ -506,6 +487,8 @@ function trimTrailingSlash(value: string): string {
  * Codex 走 `resolveCodexUrl`。不要在这里"顺手规范化"路径，否则日志会与真实请求脱节，反而掩盖配置错误。
  */
 export function modelRequestUrl(model: Model<Api>): string {
+  const protocol = openRouterProtocolForModel(model);
+  if (protocol) return openRouterRequestUrl({ protocol, baseURL: model.baseUrl ?? "" });
   const base = trimTrailingSlash(model.baseUrl ?? "");
   switch (model.api) {
     case "openai-completions":
@@ -766,7 +749,7 @@ function piApiFor(endpoint: RuntimeModelEndpoint): ProviderStreams {
     }
   })();
   const api = protocol.startsWith("openrouter-")
-    ? withModelProtocolResponse(implementation, protocol as "openrouter-messages" | "openrouter-chat" | "openrouter-responses")
+    ? withModelProtocolResponse(implementation, protocol as OpenRouterGenerationProtocol)
     : implementation;
   const contextFor = (
     model: Model<Api>,

package/src/models/openrouter/messages.ts

@@ -5,6 +5,7 @@ import type {
   TranscriptContext,
 } from "@earendil-works/pi-ai";
 import { getCurrentTools } from "@earendil-works/pi-ai";
+import { createOpenRouterFetch, openRouterHeaders, withOpenRouterRouting } from "@springbrand/model-client";
 import {
   anthropicMessagesApi,
 } from "@earendil-works/pi-ai/api/anthropic-messages.lazy";
@@ -60,13 +61,7 @@ function decoratePayload(
   }
 
   return {
-    ...body,
-    ...(sessionId ? { session_id: sessionId } : {}),
-    provider: {
-      ...record(body.provider),
-      ...model.compat?.openRouterRouting,
-      require_parameters: true,
-    },
+    ...withOpenRouterRouting(body, { ...model.compat?.openRouterRouting }, sessionId),
     ...(deferred.length > 0
       ? {
           tools: [
@@ -96,12 +91,14 @@ function openRouterOptions(
   return {
     ...options,
     apiKey: undefined,
+    fetch: createOpenRouterFetch({ apiKey, fetch: options?.fetch }),
     headers: {
+      // Pi uses null to suppress SDK/model defaults before building native headers.
       ...headers,
+      ...Object.fromEntries(openRouterHeaders(apiKey)),
       ...(model.compat?.supportsStrictTools === true
         ? { "x-anthropic-beta": STRUCTURED_OUTPUTS_BETA }
         : {}),
-      ...(apiKey ? { authorization: `Bearer ${apiKey}` } : {}),
       "x-api-key": null,
     },
     onPayload: async (payload, activeModel) => {

package/src/models/openrouter/openai.ts

@@ -6,6 +6,7 @@ import type {
   TranscriptContext,
 } from "@earendil-works/pi-ai";
 import { getCurrentTools } from "@earendil-works/pi-ai";
+import { createOpenRouterFetch, withOpenRouterRouting } from "@springbrand/model-client";
 import {
   openAICompletionsApi,
 } from "@earendil-works/pi-ai/api/openai-completions.lazy";
@@ -27,6 +28,7 @@ function openRouterOptions(
 ): StreamOptions {
   return {
     ...options,
+    fetch: createOpenRouterFetch({ apiKey: options?.apiKey, fetch: options?.fetch }),
     onPayload: async (payload, activeModel) => {
       const callerPayload = await options?.onPayload?.(payload, activeModel) ?? payload;
       const body = record(callerPayload);
@@ -48,18 +50,12 @@ function openRouterOptions(
       const reasoning = record(replayed.reasoning);
       const include = Array.isArray(replayed.include) ? replayed.include : [];
       return {
-        ...replayed,
+        ...withOpenRouterRouting(replayed, compat?.openRouterRouting, options?.sessionId),
         ...(model.api === "openai-responses" && reasoning?.effort !== undefined &&
           reasoning.effort !== null && reasoning.effort !== "none" &&
           !include.includes("reasoning.encrypted_content")
           ? { include: [...include, "reasoning.encrypted_content"] }
           : {}),
-        ...(options?.sessionId ? { session_id: options.sessionId } : {}),
-        provider: {
-          ...record(replayed.provider),
-          ...compat?.openRouterRouting,
-          require_parameters: true,
-        },
       };
     },
   };

package/src/models/openrouter/protocol-response.ts

@@ -7,12 +7,10 @@ import {
   type ProviderStreams,
   type StreamOptions,
 } from "@earendil-works/pi-ai";
-import type { RuntimeModelProtocol } from "../../config/ports";
+import type { OpenRouterGenerationProtocol } from "@springbrand/model-client";
 import { overlayProviderStreams } from "../provider-streams";
 import type { ModelProtocolResponse } from "../../conversation/canonical-context";
 
-export type OpenRouterProtocol = Extract<RuntimeModelProtocol, `openrouter-${string}`>;
-
 function record(value: unknown): Record<string, unknown> | undefined {
   return value !== null && typeof value === "object" && !Array.isArray(value)
     ? value as Record<string, unknown>
@@ -63,7 +61,7 @@ function streamWithProtocolResponse<T extends StreamOptions>(
 
 export function withModelProtocolResponse(
   api: ProviderStreams,
-  protocol: OpenRouterProtocol,
+  protocol: OpenRouterGenerationProtocol,
 ): ProviderStreams {
   if (protocol !== "openrouter-responses") return api;
   return overlayProviderStreams(api, {

package/src/tools/web/search/api.ts

@@ -6,6 +6,7 @@
  */
 
 import type { RuntimeModelEndpoint } from "../../../config/ports";
+import { createOpenRouterFetch, openRouterRequestUrl } from "@springbrand/model-client";
 
 export interface WebSearchSource {
   title: string;
@@ -295,6 +296,9 @@ function isOpenRouterModel(model: NativeWebSearchModel): boolean {
 }
 
 function resolveOpenAIResponsesUrl(model: NativeWebSearchModel): string {
+  if (isOpenRouterModel(model)) {
+    return openRouterRequestUrl({ protocol: "openrouter-responses", baseURL: model.baseUrl });
+  }
   const base = trimTrailingSlash(model.baseUrl);
   if (!isOpenAICodexModel(model)) return `${base}/responses`;
   if (base.endsWith("/codex/responses")) return base;
@@ -733,7 +737,8 @@ async function callOpenAIStream(
     requestBody.parallel_tool_calls = true;
   }
 
-  const response = await fetch(resolveOpenAIResponsesUrl(model), {
+  const fetcher = isOpenRouter ? createOpenRouterFetch({ apiKey: auth.apiKey }) : fetch;
+  const response = await fetcher(resolveOpenAIResponsesUrl(model), {
     method: "POST",
     headers: requestHeaders,
     body: JSON.stringify(requestBody),

package/src/model-protocol/openrouter.ts

@@ -1,41 +0,0 @@
-import type { RuntimeModelEndpoint } from "../config/ports";
-
-export interface OpenRouterRequestOptions {
-  readonly headers?: HeadersInit;
-  readonly signal?: AbortSignal;
-  readonly fetch?: typeof fetch;
-}
-
-export function openRouterRequestUrl(
-  endpoint: Pick<RuntimeModelEndpoint, "protocol" | "baseURL">,
-): string {
-  const baseURL = endpoint.baseURL.replace(/\/+$/, "");
-  switch (endpoint.protocol) {
-    case "openrouter-messages":
-      return `${baseURL}/v1/messages`;
-    case "openrouter-chat":
-      return `${baseURL}/chat/completions`;
-    case "openrouter-responses":
-      return `${baseURL}/responses`;
-    default:
-      throw new Error(`Not an OpenRouter protocol: ${endpoint.protocol}`);
-  }
-}
-
-export function requestOpenRouter(
-  endpoint: RuntimeModelEndpoint,
-  body: Readonly<Record<string, unknown>>,
-  options: OpenRouterRequestOptions = {},
-): Promise<Response> {
-  const url = openRouterRequestUrl(endpoint);
-  const headers = new Headers(endpoint.headers);
-  new Headers(options.headers).forEach((value, name) => headers.set(name, value));
-  headers.set("content-type", "application/json");
-  headers.set("authorization", `Bearer ${endpoint.apiKey}`);
-  return (options.fetch ?? globalThis.fetch)(url, {
-    method: "POST",
-    headers,
-    body: JSON.stringify(body),
-    ...(options.signal ? { signal: options.signal } : {}),
-  });
-}