@d-ai/pi 0.2.2 → 0.2.3

package/README.md

@@ -21,7 +21,7 @@ Set a D.AI API key:
 export DAI_API_KEY=ddsk_...
 ```
 
-Or in Pi run `/login`, choose **D.AI**, and paste `ddsk_...`. That discovers chat models from `GET /v1/models` and media models from the public catalog. Then pick `dai/<model>` in `/model`.
+Or in Pi run `/login`, choose **D.AI**, and paste `ddsk_...`. That discovers chat models from `GET /v1/models` and media models from the public catalog. Then pick `dai/<model>` in `/model`. GPT-5.6 / GPT-6 expose thinking levels (`low` … `max`); `gpt-6-astra` cannot turn thinking off.
 
 `/dai key` also writes `~/.pi/agent/dai.json`. Optional `baseUrl` defaults to `https://dai.itbug.shop`.
 

package/dist/index.js

@@ -1017,6 +1017,10 @@ function toUserError(error) {
       return new DaiPluginError("Invalid D.AI API key.");
     }
     if (error.status === 402 || /余额不足|insufficient_quota/iu.test(error.message)) {
+      const text = sanitize(publicMessage(error));
+      if (text.includes("\u9884\u6263")) {
+        return new DaiPluginError(text);
+      }
       return new DaiPluginError("D.AI \u4F59\u989D\u4E0D\u8DB3\u3002");
     }
     if (error.status >= 400 && error.status < 500) {
@@ -1137,6 +1141,115 @@ function normalizeOrigin(baseUrl) {
   return baseUrl.replace(/\/+$/u, "").replace(/\/v1$/u, "");
 }
 
+// src/thinking.ts
+var GPT_5_6 = {
+  off: "none",
+  minimal: null,
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: "xhigh",
+  max: "max"
+};
+var GPT_6 = {
+  off: null,
+  minimal: null,
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: "xhigh",
+  max: "max"
+};
+var GPT_5_4 = {
+  off: "none",
+  minimal: null,
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: "xhigh",
+  max: null
+};
+var GPT_5_1 = {
+  off: "none",
+  minimal: null,
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: null,
+  max: null
+};
+var GPT_5 = {
+  off: null,
+  minimal: "minimal",
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: null,
+  max: null
+};
+var O_SERIES = {
+  off: null,
+  minimal: null,
+  low: "low",
+  medium: "medium",
+  high: "high",
+  xhigh: null,
+  max: null
+};
+var KNOWN = {
+  "gpt-6-astra": GPT_6,
+  "gpt-5.6-luna": GPT_5_6,
+  "gpt-5.6-sol": GPT_5_6,
+  "gpt-5.6-terra": GPT_5_6,
+  "gpt-5.5": GPT_5_4,
+  "gpt-5.5-pro": { ...GPT_5_4, off: null, low: null },
+  "gpt-5.4": GPT_5_4,
+  "gpt-5.4-mini": GPT_5_4,
+  "gpt-5.4-nano": GPT_5_4,
+  "gpt-5.4-pro": { ...GPT_5_4, off: null, low: null },
+  "gpt-5.2": GPT_5_4,
+  "gpt-5.1": GPT_5_1,
+  "gpt-5": GPT_5,
+  "gpt-5-mini": GPT_5,
+  "gpt-5-nano": GPT_5,
+  "gpt-5-pro": { off: null, minimal: null, low: null, medium: null, high: "high", xhigh: null, max: null }
+};
+var NO_REASONING = /* @__PURE__ */ new Set(["gpt-5-chat-latest", "gpt-5.3-chat-latest"]);
+function chatModelThinking(id) {
+  if (NO_REASONING.has(id)) {
+    return { reasoning: false, input: ["text", "image"] };
+  }
+  const known = KNOWN[id];
+  if (known) {
+    return { reasoning: true, thinkingLevelMap: known, input: visionInput(id) };
+  }
+  if (/^gpt-6(?:[-.]|$)/iu.test(id)) {
+    return { reasoning: true, thinkingLevelMap: GPT_6, input: ["text", "image"] };
+  }
+  if (/^gpt-5\.6/iu.test(id)) {
+    return { reasoning: true, thinkingLevelMap: GPT_5_6, input: ["text", "image"] };
+  }
+  if (/^gpt-5\.(?:2|3|4|5)/iu.test(id)) {
+    return { reasoning: true, thinkingLevelMap: GPT_5_4, input: ["text", "image"] };
+  }
+  if (/^gpt-5\.1/iu.test(id)) {
+    return { reasoning: true, thinkingLevelMap: GPT_5_1, input: ["text", "image"] };
+  }
+  if (/^gpt-5(?:[-.]|$)/iu.test(id)) {
+    return { reasoning: true, thinkingLevelMap: GPT_5, input: ["text", "image"] };
+  }
+  if (/^(?:o1|o3|o4)(?:[-.]|$)/iu.test(id)) {
+    return { reasoning: true, thinkingLevelMap: O_SERIES, input: ["text"] };
+  }
+  if (/reason/iu.test(id)) {
+    return { reasoning: true, input: ["text"] };
+  }
+  return { reasoning: false, input: ["text"] };
+}
+function visionInput(id) {
+  return /^(?:gpt-5|gpt-6)/iu.test(id) ? ["text", "image"] : ["text"];
+}
+
 // src/discover.ts
 async function discoverDaiCatalog(options = {}) {
   const origin = normalizeOrigin(options.origin ?? DEFAULT_BASE_URL2);
@@ -1223,12 +1336,14 @@ function toChatModel(id, model) {
   const path = model.integration_docs?.path ?? "";
   const responses = path.includes("/responses");
   const rate = model.token_rate;
+  const thinking = chatModelThinking(id);
   return {
     id,
     name: model.display_name?.trim() || id,
     api: responses ? "openai-responses" : "openai-completions",
-    reasoning: /gpt-5|o1|o3|o4|reason/iu.test(id),
-    input: ["text"],
+    reasoning: thinking.reasoning,
+    ...thinking.thinkingLevelMap === void 0 ? {} : { thinkingLevelMap: thinking.thinkingLevelMap },
+    input: thinking.input,
     cost: {
       input: rate?.input_usd_per_1m_tokens ?? 0,
       output: rate?.output_usd_per_1m_tokens ?? 0,
@@ -1600,6 +1715,7 @@ function toProviderModels(models, baseUrl) {
     api: model.api,
     baseUrl,
     reasoning: model.reasoning,
+    ...model.thinkingLevelMap === void 0 ? {} : { thinkingLevelMap: model.thinkingLevelMap },
     input: model.input,
     cost: model.cost,
     contextWindow: model.contextWindow,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@d-ai/pi",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Pi extension for D.AI image, video, and music generation",
   "license": "MIT",
   "type": "module",