vite-plugin-opencode-assistant 1.0.15 → 1.0.16

package/es/core/api.d.ts

@@ -1,15 +1,27 @@
 import type { SessionInfo } from "@vite-plugin-opencode-assistant/shared";
+import { ChromeMcpWarmupError } from "@vite-plugin-opencode-assistant/shared";
 export declare class OpenCodeAPI {
     private hostname;
     private getPort;
+    private getProxyPort;
     private warmupChromeMcpConfig;
-    constructor(hostname: string, getPort: () => number, warmupChromeMcpConfig?: boolean);
+    private failedFreeModels;
+    constructor(hostname: string, getPort: () => number, getProxyPort: () => number, warmupChromeMcpConfig?: boolean);
+    markModelAsFailed(providerID: string, modelID: string): void;
+    clearFailedModels(): void;
     private createHttpRequest;
-    getSessions(retries?: number): Promise<SessionInfo[]>;
-    createSession(retries?: number, title?: string): Promise<SessionInfo>;
+    getSessions(projectDir: string, retries?: number): Promise<SessionInfo[]>;
+    createSession(projectDir: string, retries?: number, title?: string): Promise<SessionInfo>;
+    getCheapestModel(): Promise<{
+        providerID: string;
+        modelID: string;
+    } | null>;
     deleteSession(sessionId: string, retries?: number): Promise<void>;
     getToolIds(retries?: number): Promise<string[]>;
-    warmupChromeMcp(viteOrigin?: string): Promise<void>;
-    getOrCreateSession(): Promise<string>;
-    retryWarmupChromeMcp(viteOrigin?: string): Promise<boolean>;
+    warmupChromeMcp(projectDir: string, viteOrigin?: string): Promise<void>;
+    getOrCreateSession(projectDir: string): Promise<string>;
+    retryWarmupChromeMcp(projectDir: string, viteOrigin?: string): Promise<{
+        success: boolean;
+        error?: ChromeMcpWarmupError;
+    }>;
 }

package/es/core/api.js

@@ -1,5 +1,22 @@
 var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
 var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
 var __async = (__this, __arguments, generator) => {
   return new Promise((resolve, reject) => {
@@ -26,43 +43,37 @@ import {
   PerformanceTimer,
   createLogger,
   DEFAULT_RETRIES,
-  RETRY_DELAY
+  RETRY_DELAY,
+  ChromeMcpWarmupErrorType,
+  ChromeMcpWarmupError,
+  CHROME_DEVTOOLS_PORT,
+  checkChromeDevToolsAvailable,
+  sleep,
+  base64Encode,
+  extractTextFromResponse
 } from "@vite-plugin-opencode-assistant/shared";
 const log = createLogger("API");
-function sleep(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
-function base64Encode(str) {
-  return Buffer.from(str).toString("base64");
-}
-function extractTextFromResponse(data) {
-  if (!data || typeof data !== "object") return null;
-  const obj = data;
-  if (obj.parts && Array.isArray(obj.parts)) {
-    const textParts = obj.parts.filter(
-      (p) => p && typeof p === "object" && p.type === "text"
-    ).map((p) => p.text).filter(Boolean);
-    if (textParts.length > 0) return textParts.join("");
-  }
-  if (obj.text && typeof obj.text === "string") {
-    return obj.text;
-  }
-  if (obj.content && typeof obj.content === "string") {
-    return obj.content;
-  }
-  if (obj.message && typeof obj.message === "string") {
-    return obj.message;
-  }
-  if (typeof data === "string") {
-    return data;
-  }
-  return null;
-}
 class OpenCodeAPI {
-  constructor(hostname, getPort, warmupChromeMcpConfig = false) {
+  constructor(hostname, getPort, getProxyPort, warmupChromeMcpConfig = false) {
     __publicField(this, "hostname", hostname);
     __publicField(this, "getPort", getPort);
+    __publicField(this, "getProxyPort", getProxyPort);
     __publicField(this, "warmupChromeMcpConfig", warmupChromeMcpConfig);
+    __publicField(this, "failedFreeModels", /* @__PURE__ */ new Set());
+  }
+  markModelAsFailed(providerID, modelID) {
+    const key = `${providerID}:${modelID}`;
+    this.failedFreeModels.add(key);
+    log.debug("Marked model as failed", {
+      providerID,
+      modelID,
+      key,
+      failedCount: this.failedFreeModels.size
+    });
+  }
+  clearFailedModels() {
+    this.failedFreeModels.clear();
+    log.debug("Cleared failed models cache");
   }
   createHttpRequest(options, body, timeout) {
     const timer = new PerformanceTimer("HTTP Request", {
@@ -98,20 +109,26 @@ class OpenCodeAPI {
       req.end();
     });
   }
-  getSessions() {
-    return __async(this, arguments, function* (retries = DEFAULT_RETRIES) {
-      const timer = log.timer("getSessions", { retries });
+  getSessions(_0) {
+    return __async(this, arguments, function* (projectDir, retries = DEFAULT_RETRIES) {
+      const timer = log.timer("getSessions", { retries, projectDir });
       let lastError = null;
       for (let i = 0; i < retries; i++) {
         try {
-          log.debug(`Attempt ${i + 1}/${retries}`, { operation: "getSessions" });
+          log.debug(`Attempt ${i + 1}/${retries}`, { operation: "getSessions", projectDir });
           const sessions = yield this.createHttpRequest({
             hostname: this.hostname,
             port: this.getPort(),
-            path: "/session"
+            path: "/session",
+            headers: {
+              "x-opencode-directory": encodeURIComponent(projectDir)
+            }
           });
+          const sessionsWithUrl = sessions.map((s) => __spreadProps(__spreadValues({}, s), {
+            url: `http://${this.hostname}:${this.getProxyPort()}/${base64Encode(s.directory)}/session/${s.id}`
+          }));
           timer.end(`Found ${sessions.length} sessions`);
-          return sessions;
+          return sessionsWithUrl;
         } catch (e) {
           lastError = e instanceof Error ? e : new Error(String(e));
           log.debug(`Attempt ${i + 1} failed: ${lastError.message}`, {
@@ -129,15 +146,16 @@ class OpenCodeAPI {
       throw lastError;
     });
   }
-  createSession() {
-    return __async(this, arguments, function* (retries = DEFAULT_RETRIES, title) {
-      const timer = log.timer("createSession", { retries, title });
+  createSession(_0) {
+    return __async(this, arguments, function* (projectDir, retries = DEFAULT_RETRIES, title) {
+      const timer = log.timer("createSession", { retries, title, projectDir });
       let lastError = null;
       for (let i = 0; i < retries; i++) {
         try {
           log.debug(`Attempt ${i + 1}/${retries}`, {
             operation: "createSession",
-            title
+            title,
+            projectDir
           });
           const requestBody = title ? JSON.stringify({ title }) : void 0;
           const session = yield this.createHttpRequest(
@@ -146,12 +164,17 @@ class OpenCodeAPI {
               port: this.getPort(),
               path: "/session",
               method: "POST",
-              headers: requestBody ? { "Content-Type": "application/json" } : void 0
+              headers: __spreadProps(__spreadValues({}, requestBody ? { "Content-Type": "application/json" } : {}), {
+                "x-opencode-directory": encodeURIComponent(projectDir)
+              })
             },
             requestBody
           );
+          const sessionWithUrl = __spreadProps(__spreadValues({}, session), {
+            url: `http://${this.hostname}:${this.getProxyPort()}/${base64Encode(projectDir)}/session/${session.id}`
+          });
           timer.end(`Created session: ${session.id}`);
-          return session;
+          return sessionWithUrl;
         } catch (e) {
           lastError = e instanceof Error ? e : new Error(String(e));
           log.debug(`Attempt ${i + 1} failed: ${lastError.message}`, {
@@ -169,6 +192,63 @@ class OpenCodeAPI {
       throw lastError;
     });
   }
+  getCheapestModel() {
+    return __async(this, null, function* () {
+      var _a, _b;
+      try {
+        const response = yield this.createHttpRequest({
+          hostname: this.hostname,
+          port: this.getPort(),
+          path: "/provider",
+          method: "GET"
+        });
+        const connectedProviders = new Set(response.connected);
+        const allModels = [];
+        for (const provider of response.all) {
+          if (!connectedProviders.has(provider.id)) {
+            log.debug("Skipping not connected provider", { providerID: provider.id });
+            continue;
+          }
+          for (const [modelID, model] of Object.entries(provider.models)) {
+            allModels.push({
+              providerID: provider.id,
+              modelID,
+              name: model.name,
+              inputCost: (_b = (_a = model.cost) == null ? void 0 : _a.input) != null ? _b : 0
+            });
+          }
+        }
+        allModels.sort((a, b) => a.inputCost - b.inputCost);
+        const availableModel = allModels.find(
+          (model) => !this.failedFreeModels.has(`${model.providerID}:${model.modelID}`)
+        );
+        if (!availableModel) {
+          log.debug("All models have failed", {
+            totalModels: allModels.length,
+            failedModels: this.failedFreeModels.size,
+            connectedProviders: response.connected
+          });
+          return null;
+        }
+        log.debug("Found cheapest available model for warmup", {
+          providerID: availableModel.providerID,
+          modelID: availableModel.modelID,
+          name: availableModel.name,
+          inputCost: availableModel.inputCost,
+          totalModels: allModels.length,
+          failedModels: this.failedFreeModels.size,
+          connectedProviders: response.connected
+        });
+        return {
+          providerID: availableModel.providerID,
+          modelID: availableModel.modelID
+        };
+      } catch (error) {
+        log.warn("Failed to get cheapest model", { error });
+        return null;
+      }
+    });
+  }
   deleteSession(_0) {
     return __async(this, arguments, function* (sessionId, retries = DEFAULT_RETRIES) {
       const timer = log.timer("deleteSession", { sessionId, retries });
@@ -239,13 +319,33 @@ class OpenCodeAPI {
       throw lastError;
     });
   }
-  warmupChromeMcp(viteOrigin) {
+  warmupChromeMcp(projectDir, viteOrigin) {
     return __async(this, null, function* () {
       if (!this.warmupChromeMcpConfig) return;
       const timer = log.timer("warmupChromeMcp", { viteOrigin });
       let warmupSessionId = null;
+      let freeModel = null;
+      const chromeAvailable = yield checkChromeDevToolsAvailable();
+      if (!chromeAvailable) {
+        const error = new ChromeMcpWarmupError(
+          ChromeMcpWarmupErrorType.CHROME_NOT_CONNECTED,
+          "Chrome DevTools Protocol is not available",
+          "Chrome remote debugging is not enabled or not running on port 9222. Please enable Chrome remote debugging first."
+        );
+        log.warn("Chrome DevTools not available", {
+          port: CHROME_DEVTOOLS_PORT,
+          hint: "Enable Chrome remote debugging at chrome://inspect/#remote-debugging"
+        });
+        timer.end("Chrome DevTools not available");
+        throw error;
+      }
+      log.debug("Chrome DevTools is available, proceeding with warmup");
       try {
-        const warmupSession = yield this.createSession(DEFAULT_RETRIES, "__chrome_mcp_warmup__");
+        const warmupSession = yield this.createSession(
+          projectDir,
+          DEFAULT_RETRIES,
+          "__chrome_mcp_warmup__"
+        );
         warmupSessionId = warmupSession.id;
         const prompt = [
           "Call the browser tool list_pages immediately to establish the Chrome DevTools MCP connection.",
@@ -255,7 +355,16 @@ class OpenCodeAPI {
           "After the tool call is complete, reply with exactly: ready",
           "If the tool call fails, reply with exactly: fail"
         ].join(" ");
-        const WARMUP_TIMEOUT = 3e4;
+        const WARMUP_TIMEOUT = 6e4;
+        freeModel = yield this.getCheapestModel();
+        if (freeModel) {
+          log.debug("Using cheapest model for warmup", {
+            providerID: freeModel.providerID,
+            modelID: freeModel.modelID
+          });
+        } else {
+          log.debug("No model available, using default model");
+        }
         const data = yield this.createHttpRequest(
           {
             hostname: this.hostname,
@@ -264,21 +373,80 @@ class OpenCodeAPI {
             method: "POST",
             headers: { "Content-Type": "application/json" }
           },
-          JSON.stringify({
+          JSON.stringify(__spreadValues({
             system: "You are warming up Chrome DevTools MCP during startup. You must use the available browser tools immediately before replying.",
             parts: [{ type: "text", text: prompt }]
-          }),
+          }, freeModel && {
+            model: {
+              providerID: freeModel.providerID,
+              modelID: freeModel.modelID
+            }
+          })),
           WARMUP_TIMEOUT
         );
         const responseText = extractTextFromResponse(data);
-        if (!(responseText == null ? void 0 : responseText.toLowerCase().includes("ready"))) {
-          throw new Error(`Chrome MCP warmup failed: ${responseText || "No response"}`);
+        if (!responseText) {
+          throw new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.AI_RESPONSE_ERROR,
+            "AI did not respond to the warmup request",
+            "Empty response from AI"
+          );
+        }
+        const lowerResponse = responseText.toLowerCase();
+        if (lowerResponse.includes("fail")) {
+          throw new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.CHROME_NOT_CONNECTED,
+            "Chrome DevTools MCP is not connected",
+            "AI reported that browser tools are not available. This should not happen if Chrome DevTools check passed."
+          );
+        }
+        if (!lowerResponse.includes("ready")) {
+          throw new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.AI_RESPONSE_ERROR,
+            "AI response does not indicate success",
+            `AI responded with: ${responseText.substring(0, 200)}`
+          );
         }
         timer.end("Chrome MCP warmed up");
       } catch (e) {
-        log.warn("Failed to warm up Chrome MCP", { error: e });
-        timer.end("Chrome MCP warmup skipped");
-        throw e;
+        if (e instanceof ChromeMcpWarmupError) {
+          if (e.type === ChromeMcpWarmupErrorType.SESSION_ERROR) {
+            timer.end("Session creation failed");
+          }
+          log.warn(`Chrome MCP warmup failed: ${e.type}`, {
+            message: e.message,
+            details: e.details
+          });
+          timer.end(`Chrome MCP warmup failed: ${e.type}`);
+          throw e;
+        }
+        if (freeModel) {
+          this.markModelAsFailed(freeModel.providerID, freeModel.modelID);
+          log.debug("Marked model as failed due to warmup error", {
+            providerID: freeModel.providerID,
+            modelID: freeModel.modelID,
+            error: e instanceof Error ? e.message : String(e)
+          });
+        }
+        const errorMessage = e instanceof Error ? e.message : String(e);
+        if (errorMessage.includes("timeout") || errorMessage.includes("Timeout")) {
+          const error2 = new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.AI_TIMEOUT,
+            "AI response timeout",
+            "AI did not respond within 30 seconds. Please check if the OpenCode AI model is properly configured and available."
+          );
+          log.warn("Chrome MCP warmup timeout", { error: errorMessage });
+          timer.end("Chrome MCP warmup timeout");
+          throw error2;
+        }
+        const error = new ChromeMcpWarmupError(
+          ChromeMcpWarmupErrorType.UNKNOWN,
+          "Unknown error during Chrome MCP warmup",
+          errorMessage
+        );
+        log.warn("Chrome MCP warmup failed with unknown error", { error: errorMessage });
+        timer.end("Chrome MCP warmup failed");
+        throw error;
       } finally {
         if (warmupSessionId) {
           try {
@@ -293,34 +461,53 @@ class OpenCodeAPI {
       }
     });
   }
-  getOrCreateSession() {
+  getOrCreateSession(projectDir) {
     return __async(this, null, function* () {
-      const timer = log.timer("getOrCreateSession");
-      const projectDir = process.cwd();
+      const timer = log.timer("getOrCreateSession", { projectDir });
       log.debug("Getting sessions...", { projectDir });
-      const sessions = yield this.getSessions();
+      const sessions = yield this.getSessions(projectDir);
       log.debug(`Found ${sessions.length} sessions`, {
         sessions: sessions.map((s) => ({ id: s.id, directory: s.directory }))
       });
       const matchingSession = sessions.find((s) => s.directory === projectDir);
       if (matchingSession) {
-        const url2 = `http://${this.hostname}:${this.getPort()}/${base64Encode(projectDir)}/session/${matchingSession.id}`;
+        const url2 = `http://${this.hostname}:${this.getProxyPort()}/${base64Encode(projectDir)}/session/${matchingSession.id}`;
         timer.end(`Using existing session: ${matchingSession.id}`);
         return url2;
       }
       log.debug("Creating new session...", { projectDir });
-      const newSession = yield this.createSession();
-      const url = `http://${this.hostname}:${this.getPort()}/${base64Encode(projectDir)}/session/${newSession.id}`;
+      const newSession = yield this.createSession(projectDir);
+      const url = `http://${this.hostname}:${this.getProxyPort()}/${base64Encode(projectDir)}/session/${newSession.id}`;
       timer.end(`Created new session: ${newSession.id}`);
       return url;
     });
   }
-  retryWarmupChromeMcp(viteOrigin) {
+  retryWarmupChromeMcp(projectDir, viteOrigin) {
     return __async(this, null, function* () {
       const timer = log.timer("retryWarmupChromeMcp", { viteOrigin });
       let warmupSessionId = null;
+      let freeModel = null;
+      const chromeAvailable = yield checkChromeDevToolsAvailable();
+      if (!chromeAvailable) {
+        const error = new ChromeMcpWarmupError(
+          ChromeMcpWarmupErrorType.CHROME_NOT_CONNECTED,
+          "Chrome DevTools Protocol is not available",
+          "Chrome remote debugging is not enabled or not running on port 9222. Please enable Chrome remote debugging first."
+        );
+        log.warn("Chrome DevTools not available for retry", {
+          port: CHROME_DEVTOOLS_PORT,
+          hint: "Enable Chrome remote debugging at chrome://inspect/#remote-debugging"
+        });
+        timer.end("Chrome DevTools not available for retry");
+        return { success: false, error };
+      }
+      log.debug("Chrome DevTools is available, proceeding with retry warmup");
       try {
-        const warmupSession = yield this.createSession(DEFAULT_RETRIES, "__chrome_mcp_warmup__");
+        const warmupSession = yield this.createSession(
+          projectDir,
+          DEFAULT_RETRIES,
+          "__chrome_mcp_warmup__"
+        );
         warmupSessionId = warmupSession.id;
         const prompt = [
           "Call the browser tool list_pages immediately to establish the Chrome DevTools MCP connection.",
@@ -331,6 +518,15 @@ class OpenCodeAPI {
           "If the tool call fails, reply with exactly: fail"
         ].join(" ");
         const WARMUP_TIMEOUT = 6e4;
+        freeModel = yield this.getCheapestModel();
+        if (freeModel) {
+          log.debug("Using cheapest model for retry warmup", {
+            providerID: freeModel.providerID,
+            modelID: freeModel.modelID
+          });
+        } else {
+          log.debug("No model available for retry, using default model");
+        }
         const data = yield this.createHttpRequest(
           {
             hostname: this.hostname,
@@ -339,23 +535,82 @@ class OpenCodeAPI {
             method: "POST",
             headers: { "Content-Type": "application/json" }
           },
-          JSON.stringify({
+          JSON.stringify(__spreadValues({
             system: "You are warming up Chrome DevTools MCP during startup. You must use the available browser tools immediately before replying.",
             parts: [{ type: "text", text: prompt }]
-          }),
+          }, freeModel && {
+            model: {
+              providerID: freeModel.providerID,
+              modelID: freeModel.modelID
+            }
+          })),
           WARMUP_TIMEOUT
         );
         log.debug("Chrome MCP warmup response:", { data });
         const responseText = extractTextFromResponse(data);
-        if (!(responseText == null ? void 0 : responseText.toLowerCase().includes("ready"))) {
-          throw new Error(`Chrome MCP warmup failed: ${responseText || "No response"}`);
+        if (!responseText) {
+          throw new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.AI_RESPONSE_ERROR,
+            "AI did not respond to the warmup request",
+            "Empty response from AI"
+          );
+        }
+        const lowerResponse = responseText.toLowerCase();
+        if (lowerResponse.includes("fail")) {
+          throw new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.CHROME_NOT_CONNECTED,
+            "Chrome DevTools MCP is not connected",
+            "AI reported that browser tools are not available. This should not happen if Chrome DevTools check passed."
+          );
+        }
+        if (!lowerResponse.includes("ready")) {
+          throw new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.AI_RESPONSE_ERROR,
+            "AI response does not indicate success",
+            `AI responded with: ${responseText.substring(0, 200)}`
+          );
         }
         timer.end("Chrome MCP warmed up successfully");
-        return true;
+        return { success: true };
       } catch (e) {
-        log.warn("Failed to retry warm up Chrome MCP", { error: e });
+        if (e instanceof ChromeMcpWarmupError) {
+          if (e.type === ChromeMcpWarmupErrorType.SESSION_ERROR) {
+            timer.end("Session creation failed");
+          }
+          log.warn(`Chrome MCP warmup retry failed: ${e.type}`, {
+            message: e.message,
+            details: e.details
+          });
+          timer.end(`Chrome MCP warmup retry failed: ${e.type}`);
+          return { success: false, error: e };
+        }
+        if (freeModel) {
+          this.markModelAsFailed(freeModel.providerID, freeModel.modelID);
+          log.debug("Marked model as failed due to retry warmup error", {
+            providerID: freeModel.providerID,
+            modelID: freeModel.modelID,
+            error: e instanceof Error ? e.message : String(e)
+          });
+        }
+        const errorMessage = e instanceof Error ? e.message : String(e);
+        if (errorMessage.includes("timeout") || errorMessage.includes("Timeout")) {
+          const error2 = new ChromeMcpWarmupError(
+            ChromeMcpWarmupErrorType.AI_TIMEOUT,
+            "AI response timeout",
+            "AI did not respond within 60 seconds. Please check if the OpenCode AI model is properly configured and available."
+          );
+          log.warn("Chrome MCP warmup retry timeout", { error: errorMessage });
+          timer.end("Chrome MCP warmup retry timeout");
+          return { success: false, error: error2 };
+        }
+        const error = new ChromeMcpWarmupError(
+          ChromeMcpWarmupErrorType.UNKNOWN,
+          "Unknown error during Chrome MCP warmup retry",
+          errorMessage
+        );
+        log.warn("Chrome MCP warmup retry failed with unknown error", { error: errorMessage });
         timer.end("Chrome MCP warmup retry failed");
-        return false;
+        return { success: false, error };
       } finally {
         if (warmupSessionId) {
           try {