volute 0.40.1 → 0.41.0

package/dist/{chunk-S4SXBQ7F.js → chunk-VK2YD7VJ.js}

@@ -2,15 +2,37 @@
 import {
   readGlobalConfig,
   writeGlobalConfig
-} from "./chunk-DNU6XLE2.js";
+} from "./chunk-N64Q7FMP.js";
 import {
   logger_default
 } from "./chunk-T2TP6ZC6.js";
 
 // packages/daemon/src/lib/ai-service.ts
-import { complete, getEnvApiKey, getModel, getModels, getProviders } from "@mariozechner/pi-ai";
-import { getOAuthApiKey } from "@mariozechner/pi-ai/oauth";
+import { defaultProviderAuthContext } from "@earendil-works/pi-ai";
+import { getOAuthApiKey } from "@earendil-works/pi-ai/oauth";
+import {
+  builtinModels,
+  getBuiltinModel,
+  getBuiltinModels,
+  getBuiltinProviders
+} from "@earendil-works/pi-ai/providers/all";
 var aiLog = logger_default.child("ai-service");
+var models = builtinModels();
+var authContext = defaultProviderAuthContext();
+async function resolveProviderKey(providerId, configKey) {
+  const provider = models.getProvider(providerId);
+  const model = getBuiltinModels(providerId)[0];
+  if (provider?.auth.apiKey && model) {
+    try {
+      const credential = configKey ? { type: "api_key", key: configKey } : void 0;
+      const res = await provider.auth.apiKey.resolve({ model, ctx: authContext, credential });
+      if (res?.auth.apiKey) return res.auth.apiKey;
+    } catch (err) {
+      aiLog.debug(`api key resolution failed for ${providerId}`, logger_default.errorData(err));
+    }
+  }
+  return configKey;
+}
 var providerRefreshHook;
 function setProviderRefreshHook(hook) {
   providerRefreshHook = hook;
@@ -49,7 +71,7 @@ function removeAiConfig() {
   delete config.ai;
   writeGlobalConfig(config);
 }
-function getConfiguredProviders() {
+async function getConfiguredProviders() {
   const ai = getAiConfig();
   const configured = /* @__PURE__ */ new Set();
   if (ai) {
@@ -58,8 +80,8 @@ function getConfiguredProviders() {
       if (p.apiKey || p.oauth) configured.add(id);
     }
   }
-  for (const id of getProviders()) {
-    if (!configured.has(id) && getEnvApiKey(id)) {
+  for (const id of getBuiltinProviders()) {
+    if (!configured.has(id) && await resolveProviderKey(id)) {
       configured.add(id);
     }
   }
@@ -87,14 +109,14 @@ function templateForProvider(provider) {
   if (provider === "openai-codex") return "codex";
   return "pi";
 }
-function resolveTemplate(modelId) {
+async function resolveTemplate(modelId) {
   if (!modelId) {
     const enabled = getEnabledModels();
     if (enabled.length > 0) {
       const model2 = findModel(enabled[0]);
       if (model2) return templateForProvider(model2.provider);
     }
-    const providers = getConfiguredProviders();
+    const providers = await getConfiguredProviders();
     if (providers.length === 1) return templateForProvider(providers[0]);
     if (providers.length > 0 && !providers.includes("anthropic")) {
       return templateForProvider(providers[0]);
@@ -119,17 +141,13 @@ function setEnabledModels(modelIds) {
   const config = readGlobalConfig();
   writeGlobalConfig({ ...config, ai });
 }
-function getAvailableModels() {
-  const providers = getConfiguredProviders();
-  const models = [];
+async function getAvailableModels() {
+  const providers = await getConfiguredProviders();
+  const result = [];
   for (const provider of providers) {
-    try {
-      models.push(...getModels(provider));
-    } catch (err) {
-      aiLog.warn(`failed to load models for provider ${provider}`, logger_default.errorData(err));
-    }
+    result.push(...getBuiltinModels(provider));
   }
-  return models;
+  return result;
 }
 async function resolveApiKey(providerId) {
   const ai = getAiConfig();
@@ -150,8 +168,7 @@ async function resolveApiKey(providerId) {
       aiLog.warn(`OAuth key resolution failed for ${providerId}`, logger_default.errorData(err));
     }
   }
-  if (providerConfig?.apiKey) return providerConfig.apiKey;
-  return getEnvApiKey(providerId) ?? void 0;
+  return resolveProviderKey(providerId, providerConfig?.apiKey);
 }
 function qualifyModelId(modelId) {
   if (modelId.includes(":")) return modelId;
@@ -164,22 +181,13 @@ function unqualifyModelId(modelId) {
   return idx >= 0 ? modelId.slice(idx + 1) : modelId;
 }
 function findModel(modelId) {
-  const providers = getConfiguredProviders();
-  for (const provider of providers) {
-    try {
-      const model = getModel(provider, modelId);
-      if (model) return model;
-    } catch (err) {
-      aiLog.debug(`model lookup failed for ${modelId} in ${provider}`, logger_default.errorData(err));
-    }
+  for (const provider of getBuiltinProviders()) {
+    const model = getBuiltinModel(provider, modelId);
+    if (model) return model;
   }
-  for (const provider of providers) {
-    try {
-      const found = getModels(provider).find((m) => m.id.startsWith(modelId));
-      if (found) return found;
-    } catch (err) {
-      aiLog.debug(`prefix search failed for ${modelId} in ${provider}`, logger_default.errorData(err));
-    }
+  for (const provider of getBuiltinProviders()) {
+    const found = getBuiltinModels(provider).find((m) => m.id.startsWith(modelId));
+    if (found) return found;
   }
   return void 0;
 }
@@ -201,7 +209,7 @@ async function aiComplete(systemPrompt, userMessage, modelId) {
   }
   try {
     const apiKey = await resolveApiKey(model.provider);
-    const response = await complete(
+    const response = await models.complete(
       model,
       {
         systemPrompt,