custom-provider-pi 0.1.3 → 0.1.4

package/README.md

@@ -217,13 +217,19 @@ pi install local:/path/to/custom-provider
 
 ### 多模态（图片输入）
 
-所有模型默认 `input: ["text", "image"]`，pi 会将图片附件发送给模型。纯文本模型不受影响（用户不发图片时不触发任何副作用）。
+默认 `input: ["text"]`（纯文本）。pi 会自动识别已知的多模态模型并开启图片输入：
 
-如需某个模型只支持纯文本，通过 `--overrides` 显式设置：
+1. **OpenRouter 目录**：`architecture.input_modalities` 含 `image` 时自动标记
+2. **内置模式匹配**：`gpt-4o` / `claude-*` / `gemini-*` / `grok-4+` / `glm-4v` / `qwen-vl` / `minimax-m*` 及 ID 含 `vision` 的模型
+3. **显式覆盖**：模型对象写 `input: ["text", "image"]` 或 `["text"]`
+
+**为什么默认不开**：向不支持图片的模型发图会收到上游 404（如 deepseek-v4-flash-0731 非 vision 版）。未知模型保守处理为纯文本；已知视觉模型自动开启，无需手动配置。
+
+**强制指定**：
 
 ```bash
-/custom-provider add pure-text --base-url ... \
-    --overrides '{"mymodel":{"input":["text"]}}'
+/custom-provider add my --base-url ... \
+    --overrides '{"mymodel":{"input":["text","image"]}}'
 ```
 
 ### 上下文窗口与规格推断

package/custom-provider.ts

@@ -233,7 +233,7 @@ async function refreshRemoteSpecs(): Promise<void> {
 }
 
 // 查询模型规格：远程实时表 → 本地已知预设 → undefined
-function lookupModelSpec(modelId: string): { contextWindow: number; maxTokens: number } | undefined {
+function lookupModelSpec(modelId: string): { contextWindow: number; maxTokens: number; vision?: boolean } | undefined {
   if (remoteSpecStore && remoteSpecStore.size > 0) {
     // 候选链逐形态命中：deepseek-v4-flash-free → deepseek-v4-flash → deepseek-v4 …
     for (const candidate of normalizeModelIdCandidates(modelId)) {
@@ -416,16 +416,17 @@ function stripBasePath(baseUrl: string): string {
 function normalizeBaseUrl(baseUrl: string, api: string): string {
   const root = stripBasePath(baseUrl);
 
+  // 已经包含版本号后缀（/v1、/v4、/v1beta 等）或特殊版本路径段 → 不再追加
+  // 覆盖智谱 /api/paas/v4、Google /v1beta 等风格
+  if (/\/v\d+(?:beta)?(?:\/|$)/i.test(root)) return root;
+
   switch (api) {
     case "openai-completions":
     case "openai-responses":
     case "anthropic-messages":
-      // OpenAI/Anthropic 兼容协议要求 /v1 前缀；只在"不以 /v1 结尾"时补，
-      // 避免路径中段含 /v1/ 的历史误判
       return /\/v1$/i.test(root) ? root : `${root}/v1`;
 
     case "google-generative-ai":
-      // Google Generative AI 需要 /v1beta 或 /v1 前缀
       if (!/(\/v1|\/v1beta)(\/|$)/i.test(root)) return `${root}/v1`;
       return root;
 
@@ -562,12 +563,19 @@ async function fetchModels(
   headers?: Record<string, string>
 ): Promise<string[]> {
   const cleanBase = stripBasePath(baseUrl);
-  const v1Base = /\/v1$/i.test(cleanBase) ? cleanBase : `${cleanBase}/v1`;
+  // 已包含版本段（/v1、/v4、/api/paas/v4）则不追加 /v1
+  const hasVersion = /\/v\d+(?:beta)?(?:\/|$)/i.test(cleanBase);
+  const v1Base = hasVersion ? cleanBase : `${cleanBase}/v1`;
 
-  // 尝试多种端点路径（google / anthropic 也走标准 /models 列表端点）
+  // 尝试多种端点路径
   const endpoints = api === "google-generative-ai"
     ? [`${v1Base}/models`]
-    : [ `${v1Base}/models`, `${cleanBase}/models`, `${cleanBase}/api/models` ];
+    : [
+        `${v1Base}/models`,     // 标准 /v1/models
+        `${cleanBase}/models`,  // 已含版本或原始路径
+        `${cleanBase}/v4/models`, // 智谱 /api/paas/v4 后追加
+        `${cleanBase}/api/models`,
+      ];
 
   // 按协议组装认证头：anthropic 用 x-api-key，google 用 x-goog-api-key，其余 Bearer
   const requestHeaders: Record<string, string> = { ...resolveHeaders(headers) };
@@ -667,10 +675,11 @@ async function probeEndpoint(
   const base = url.replace(/\/+$/, "");
   const resolvedKey = apiKey ? resolveValue(apiKey) : "";
 
-  // 按常见端点路径尝试（OpenAI 兼容最多，其次 Anthropic）
+  // 按常见端点路径尝试（OpenAI 兼容最多，其次 Anthropic/Google）
   const attempts = [
     { ep: `${base}/v1/models`, api: "openai-completions" },
     { ep: `${base}/models`, api: "openai-completions" },
+    { ep: `${base}/v4/models`, api: "openai-completions" }, // 智谱风格 /api/paas/v4
     { ep: `${base}/v1/models`, api: "anthropic-messages", useXApiKey: true },
   ];
 
@@ -1162,7 +1171,7 @@ export default function customProviderExtension(pi: ExtensionAPI) {
       );
       if (!addMode) return;
 
-      // ================= 简单添加路径（3 步完成）=================
+      // ================= 简单添加路径（4 步完成）=================
       if (addMode.startsWith("简单")) {
         // 1. URL
         const simpleUrl = await ctx.ui.input(
@@ -1171,9 +1180,15 @@ export default function customProviderExtension(pi: ExtensionAPI) {
         );
         if (!simpleUrl) return;
 
-        // 2. 探测
+        // 2. API Key（留空 = local 无认证）
+        const simpleKeyInput = await ctx.ui.input(
+          "API Key（留空 = 无认证本地服务，支持 $ENV 环境变量）",
+          "$DEEPSEEK_API_KEY"
+        );
+        const simpleApiKey = simpleKeyInput?.trim() || "local";
+
+        // 3. 探测
         ctx.ui.notify("正在探测端点…", "info");
-        let simpleApiKey = "local";
         const probeResult = await probeEndpoint(simpleUrl, simpleApiKey);
 
         let simpleModels: (string | IModel)[] = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "custom-provider-pi",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "pi 扩展：统一管理第三方模型 Provider。子命令体系 /custom-provider add|remove|refresh|list|test|config|enable|disable|prune，支持交互向导、flags/JSON 非交互添加、双协议混用（OpenAI + Anthropic）、模型关键字过滤与修剪",
   "keywords": [
     "pi-package",