pi-openai-toolkit 0.18.0 → 0.20.0

package/src/supported-environment.ts

@@ -1,14 +1,19 @@
 export {
 	buildCompactUrl,
 	buildResponsesUrl,
+	VIRTUAL_MODEL_API,
 	getNativeCompactionRuntime,
 	getRuntimeModelDescriptor,
+	resolveAzureDeploymentName,
+	resolveEffectiveModel,
+	registerPhysicalModelObserver,
 	isResponsesCompatiblePayload,
 	isSupportedApi,
 	normalizeBaseUrl,
 	parseModelSpec,
 	resolveNativeCompactionEnvironment,
 	resolveRemoteCompactionExecution,
+	type EffectiveModelResolution,
 	type NativeCompactionEnvironmentFailure,
 	type NativeCompactionEnvironmentResolution,
 	type NativeCompactionEnvironmentSuccess,

package/src/types.ts

@@ -13,7 +13,11 @@ export const REDACTED_VALUE = "[REDACTED]";
  * APIs the extension knows how to build a `/responses/compact` URL for.
  * `compaction.responsesApis` in config.json may only narrow this set.
  */
-export const RESPONSES_COMPACT_CAPABLE_APIS = ["openai-responses", "openai-codex-responses"] as const;
+export const RESPONSES_COMPACT_CAPABLE_APIS = [
+	"openai-responses",
+	"openai-codex-responses",
+	"azure-openai-responses",
+] as const;
 /**
  * Bare model ID used by the independent Astra effort compatibility layer.
  * Remote Context eligibility is API/provider based with an exact gateway opt-in;
@@ -602,7 +606,7 @@ export const DEFAULT_WEB_SEARCH_CONFIG: WebSearchConfig = {
 };
 
 /** Single source of truth for the hosted image model used when nothing is configured. */
-export const DEFAULT_IMAGE_GENERATION_MODEL = "gpt-image-2.5";
+export const DEFAULT_IMAGE_GENERATION_MODEL = "grok-imagine-image-2.0";
 export const DEFAULT_IMAGE_GENERATION_MODELS: readonly string[] = [DEFAULT_IMAGE_GENERATION_MODEL];
 
 export const DEFAULT_IMAGE_GENERATION_CONFIG: ImageGenerationConfig = {

package/src/web-search/extension.ts

@@ -1,7 +1,7 @@
 import type { ExtensionAPI, ExtensionContext, ToolDefinition } from "@earendil-works/pi-coding-agent";
 import { assertConfigValid, loadToolkitConfig, resolveToolkitConfig, type ResolvedToolkitConfig } from "../config";
 import { notifyConfigIssues } from "../config/notifications";
-import { resolveResponsesEnvironment, type ResponsesEnvironmentResolution } from "../runtime";
+import { registerPhysicalModelObserver, resolveResponsesEnvironment, type ResponsesEnvironmentResolution } from "../runtime";
 import {
 	requestAlphaSearch,
 	normalizeStandaloneWebRunCommands,
@@ -18,7 +18,7 @@ import {
 	LOCAL_WEB_SEARCH_TOOL_NAME,
 	resolveWebSearchRoute,
 	WEB_RUN_TOOL_NAME,
-	WEB_SEARCH_CAPABLE_APIS,
+	WEB_SEARCH_STANDALONE_APIS,
 	type WebSearchModel,
 	type WebSearchRouteResolution,
 } from "./types";
@@ -245,7 +245,7 @@ export async function executeStandaloneWebRun(args: {
 
 	const runtime = await resolveRuntime(args.ctx, {
 		enabled: config.webSearch.enabled,
-		responsesApis: WEB_SEARCH_CAPABLE_APIS,
+		responsesApis: WEB_SEARCH_STANDALONE_APIS,
 		codexGatewayModels: resolved.gatewayModelKeys,
 	});
 	if (!runtime.ok) throw standaloneRuntimeFailure(runtime);
@@ -379,6 +379,7 @@ export function registerWebSearchExtension(
 ): void {
 	if (registeredApis.has(pi)) return;
 	registeredApis.add(pi);
+	registerPhysicalModelObserver(pi);
 
 	const states = { webSearch: cloneToolState(), webRun: cloneToolState() };
 	let standaloneRegistrationSucceeded = false;

package/src/web-search/types.ts

@@ -1,13 +1,21 @@
+import { VIRTUAL_MODEL_API } from "../runtime";
 import { getExactModelKey, isExactModelAllowed } from "../model-scope";
 import type { WebSearchConfig, WebSearchRoute } from "../types";
 
-export const WEB_SEARCH_CAPABLE_APIS = ["openai-responses", "openai-codex-responses"] as const;
+export const WEB_SEARCH_CAPABLE_APIS = [
+	"openai-responses",
+	"openai-codex-responses",
+	"azure-openai-responses",
+] as const;
+/** The provider-relative alpha endpoint is only verified for OpenAI/Codex routes. */
+export const WEB_SEARCH_STANDALONE_APIS = ["openai-responses", "openai-codex-responses"] as const;
 export const WEB_SEARCH_SOURCE_INCLUDE = "web_search_call.action.sources";
 export const WEB_SEARCH_PROMPT_MARKER = "<!-- pi-openai-toolkit:web-search -->";
 export const WEB_RUN_TOOL_NAME = "web_run";
 export const LOCAL_WEB_SEARCH_TOOL_NAME = "web_search";
 
 export type WebSearchCapableApi = (typeof WEB_SEARCH_CAPABLE_APIS)[number];
+export type WebSearchStandaloneApi = (typeof WEB_SEARCH_STANDALONE_APIS)[number];
 export type WebSearchRouteSource = "exact" | "default" | "legacy" | "none";
 export type WebSearchRouteUnavailableReason = "missing-model" | "unsupported-api" | "invalid-route";
 
@@ -76,7 +84,13 @@ function selectedRoute(args: {
 			reason: "missing-model",
 		};
 	}
-	if (args.route !== "local" && !isWebSearchCapableApi(args.model?.api)) {
+	if (
+		args.route !== "local" &&
+		args.model?.api !== VIRTUAL_MODEL_API &&
+		(args.route === "standalone-alpha"
+			? !isWebSearchStandaloneCapableApi(args.model?.api)
+			: !isWebSearchCapableApi(args.model?.api))
+	) {
 		return {
 			route: args.route,
 			source: args.source,
@@ -97,6 +111,10 @@ export function isWebSearchCapableApi(api: unknown): api is WebSearchCapableApi
 	return typeof api === "string" && (WEB_SEARCH_CAPABLE_APIS as readonly string[]).includes(api);
 }
 
+export function isWebSearchStandaloneCapableApi(api: unknown): api is WebSearchStandaloneApi {
+	return typeof api === "string" && (WEB_SEARCH_STANDALONE_APIS as readonly string[]).includes(api);
+}
+
 /**
  * Resolve exactly one configured Web Search route. The legacy model allowlist is
  * deliberately consulted only after explicit route fields so old configs retain
@@ -127,7 +145,11 @@ export function resolveWebSearchRoute(args: {
 		return selectedRoute({ route: config.defaultRoute, source: "default", model, modelKey });
 	}
 
-	if (modelKey && isExactModelAllowed(model, config.models) && isWebSearchCapableApi(model?.api)) {
+	if (
+		modelKey &&
+		isExactModelAllowed(model, config.models) &&
+		(model?.api === VIRTUAL_MODEL_API || isWebSearchCapableApi(model?.api))
+	) {
 		return selectedRoute({ route: "hosted", source: "legacy", model, modelKey });
 	}
 

package/tsconfig.check.json

@@ -33,6 +33,8 @@
     "src/auto-mode/policy.ts",
     "src/auto-mode/reviewer.ts",
     "src/auto-mode/types.ts",
+    "src/responses-websocket/extension.ts",
+    "src/responses-websocket/transport.ts",
     "src/model-scope.ts",
     "src/native-fallback.ts",
     "src/payload-rewrite.ts",