@wuyaos/pi-model-config 0.1.2 → 0.1.4

package/src/transport.ts

@@ -1,6 +1,6 @@
 import { ProxyAgent, fetch as undiciFetch } from "undici";
 import type { FetchFunction, ProviderEnv, ProviderHeaders, Transport } from "@earendil-works/pi-ai";
-import type { ProviderDraft } from "./types.ts";
+import { IMAGE_API_NAMES, type ProviderDraft } from "./types.ts";
 import { validateHttpUrl } from "./config.ts";
 import { mcError } from "./i18n.ts";
 
@@ -26,7 +26,7 @@ export const CLASSIFIER_APIS = new Set([
   "cloudflare-workers-ai-system-one",
   "openai-decisions",
 ]);
-export const IMAGE_APIS = new Set(["openrouter-images"]);
+export const IMAGE_APIS = new Set(IMAGE_API_NAMES);
 
 export function assertSupportedProtocol(draft: ProviderDraft, type: "chat" | "classifier" | "image"): void {
   const api = type === "chat" ? draft.chatApi : type === "classifier" ? draft.classifierApi : draft.imageApi;
@@ -87,6 +87,7 @@ export function applyProviderTransport<T extends TransportOptions>(
       ...next.headers,
       authorization: null,
       "x-api-key": null,
+      "x-goog-api-key": null,
       "cf-aig-authorization": null,
     };
   }

package/src/types.ts

@@ -1,7 +1,9 @@
 import type { AnyModel } from "@earendil-works/pi-ai";
 
 export const MODEL_TYPES = ["chat", "classifier", "image"] as const;
+export const IMAGE_API_NAMES = ["openrouter-images", "openai-images", "gemini-images", "xai-images", "dashscope-images"] as const;
 export type ManagedModelType = (typeof MODEL_TYPES)[number];
+export type ImageApiName = (typeof IMAGE_API_NAMES)[number];
 export type ModelIdentity = `${string}\u0000${ManagedModelType}\u0000${string}`;
 
 export type AuthMode = "models-json" | "reuse-pi" | "anonymous";
@@ -64,7 +66,7 @@ export interface ProviderConnectionConfig {
   proxyUrl?: string;
   chatApi?: string;
   classifierApi?: string;
-  imageApi?: string;
+  imageApi?: ImageApiName;
   defaultType?: ManagedModelType;
   defaultContextWindow?: number;
   defaultMaxTokens?: number;
@@ -111,7 +113,7 @@ export interface ProviderDraft {
   proxyUrl?: string;
   chatApi: string;
   classifierApi: string;
-  imageApi: string;
+  imageApi: ImageApiName;
   defaultType: ManagedModelType;
 }