@d-ai/pi 0.2.1 → 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) {
@@ -1042,6 +1046,34 @@ function sanitize(message) {
 function configPath() {
   return join(homedir(), ".pi", "agent", "dai.json");
 }
+function authJsonPath() {
+  return join(homedir(), ".pi", "agent", "auth.json");
+}
+async function loadAuthJsonKey(path = authJsonPath()) {
+  try {
+    const raw = JSON.parse(await readFile(path, "utf8"));
+    if (typeof raw !== "object" || raw === null || Array.isArray(raw)) {
+      return void 0;
+    }
+    const dai = raw.dai;
+    if (typeof dai !== "object" || dai === null || Array.isArray(dai)) {
+      return void 0;
+    }
+    const record = dai;
+    for (const field of ["key", "access"]) {
+      const value = record[field];
+      if (typeof value === "string" && value.trim()) {
+        return value.trim();
+      }
+    }
+    return void 0;
+  } catch (error) {
+    if (isMissingFile(error)) {
+      return void 0;
+    }
+    throw error;
+  }
+}
 async function loadFileConfig(path = configPath()) {
   try {
     const raw = JSON.parse(await readFile(path, "utf8"));
@@ -1071,17 +1103,24 @@ async function saveFileConfig(config, path = configPath()) {
   await writeFile(`${path}`, `${JSON.stringify(next, null, 2)}
 `, { mode: 384 });
 }
-async function resolveConfig(env = process.env, path = configPath()) {
+async function resolveConfig(env = process.env, path = configPath(), authPath = authJsonPath()) {
   const file = await loadFileConfig(path);
-  const apiKey = firstNonEmpty(file.apiKey, env.DAI_API_KEY);
+  const fromAuth = await loadAuthJsonKey(authPath);
+  const apiKey = firstNonEmpty(file.apiKey, env.DAI_API_KEY, fromAuth);
   if (!apiKey) {
     throw missingApiKeyError();
   }
+  if (!file.apiKey && fromAuth) {
+    await saveFileConfig({ apiKey: fromAuth }, path);
+  }
   const baseUrl = firstNonEmpty(file.baseUrl, env.DAI_BASE_URL);
   return baseUrl === void 0 ? { apiKey } : { apiKey, baseUrl };
 }
-function hasApiKey(config, env = process.env) {
-  return Boolean(firstNonEmpty(config.apiKey, env.DAI_API_KEY));
+async function hasApiKey(config, env = process.env, authPath = authJsonPath()) {
+  if (firstNonEmpty(config.apiKey, env.DAI_API_KEY)) {
+    return true;
+  }
+  return Boolean(await loadAuthJsonKey(authPath));
 }
 function firstNonEmpty(...values) {
   for (const value of values) {
@@ -1102,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);
@@ -1188,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,
@@ -1528,7 +1678,7 @@ function registerDaiProvider(pi) {
         return [];
       }
       const file = await loadFileConfig();
-      const key = file.apiKey ?? process.env.DAI_API_KEY;
+      const key = file.apiKey ?? process.env.DAI_API_KEY ?? await loadAuthJsonKey();
       const origin = normalizeOrigin(file.baseUrl ?? DEFAULT_BASE_URL2);
       const catalog = await discoverDaiCatalog({
         origin,
@@ -1565,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,
@@ -1822,7 +1973,7 @@ ${urls}` }],
         return;
       }
       const file = await loadFileConfig();
-      const ready = hasApiKey(file);
+      const ready = await hasApiKey(file);
       const models = [
         `image ${DEFAULT_IMAGE_MODEL}`,
         `video ${DEFAULT_VIDEO_MODEL}`,

package/package.json

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