@d-ai/pi 0.2.1 → 0.2.2

package/dist/index.js

@@ -1042,6 +1042,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 +1099,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) {
@@ -1528,7 +1563,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,
@@ -1822,7 +1857,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.2",
   "description": "Pi extension for D.AI image, video, and music generation",
   "license": "MIT",
   "type": "module",