vite-plugin-opencode-assistant 1.0.44 → 1.0.48

package/lib/core/api.d.ts

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

package/lib/core/api.js

@@ -77,21 +77,6 @@ class OpenCodeAPI {
     __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 import_shared.PerformanceTimer("HTTP Request", {
@@ -205,7 +190,7 @@ class OpenCodeAPI {
       throw lastError;
     });
   }
-  getCheapestModel() {
+  getAvailableModels() {
     return __async(this, null, function* () {
       var _a, _b;
       try {
@@ -218,7 +203,6 @@ class OpenCodeAPI {
         const connectedProviders = new Set(response.connected);
         const allModels = [];
         for (const provider of response.all) {
-          if (provider.id === "opencode") continue;
           if (!connectedProviders.has(provider.id)) {
             log.debug("Skipping not connected provider", { providerID: provider.id });
             continue;
@@ -234,34 +218,14 @@ class OpenCodeAPI {
           }
         }
         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,
-          releaseDate: availableModel.releaseDate,
-          totalModels: allModels.length,
-          failedModels: this.failedFreeModels.size,
+        log.debug("Found available models for warmup", {
+          count: allModels.length,
           connectedProviders: response.connected
         });
-        return {
-          providerID: availableModel.providerID,
-          modelID: availableModel.modelID
-        };
+        return allModels;
       } catch (error) {
-        log.warn("Failed to get cheapest model", { error });
-        return null;
+        log.warn("Failed to get available models", { error });
+        return [];
       }
     });
   }
@@ -335,12 +299,10 @@ class OpenCodeAPI {
       throw lastError;
     });
   }
-  warmupChromeMcp(projectDir, viteOrigin) {
+  executeWarmupChromeMcp(projectDir, operation, viteOrigin, selectedModel) {
     return __async(this, null, function* () {
-      if (!this.warmupChromeMcpConfig) return;
-      const timer = log.timer("warmupChromeMcp", { viteOrigin });
+      const timer = log.timer(`${operation}WarmupChromeMcp`, { viteOrigin, operation });
       let warmupSessionId = null;
-      let freeModel = null;
       const chromeAvailable = yield (0, import_shared.checkChromeDevToolsAvailable)();
       if (!chromeAvailable) {
         const error = new import_shared.ChromeMcpWarmupError(
@@ -348,14 +310,14 @@ class OpenCodeAPI {
           "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", {
+        log.warn(`Chrome DevTools not available for ${operation}`, {
           port: import_shared.CHROME_DEVTOOLS_PORT,
           hint: "Enable Chrome remote debugging at chrome://inspect/#remote-debugging"
         });
-        timer.end("Chrome DevTools not available");
-        throw error;
+        timer.end(`Chrome DevTools not available for ${operation}`);
+        return { success: false, error };
       }
-      log.debug("Chrome DevTools is available, proceeding with warmup");
+      log.debug(`Chrome DevTools is available, proceeding with ${operation} warmup`);
       try {
         const warmupSession = yield this.createSession(
           projectDir,
@@ -363,14 +325,26 @@ class OpenCodeAPI {
           "__chrome_mcp_warmup__"
         );
         warmupSessionId = warmupSession.id;
-        freeModel = yield this.getCheapestModel();
-        if (freeModel) {
-          log.debug("Using cheapest model for warmup", {
-            providerID: freeModel.providerID,
-            modelID: freeModel.modelID
-          });
+        let modelToUse = selectedModel;
+        if (!modelToUse) {
+          const models = yield this.getAvailableModels();
+          if (models.length > 0) {
+            modelToUse = {
+              providerID: models[0].providerID,
+              modelID: models[0].modelID
+            };
+            log.debug(`Using cheapest model for ${operation} warmup`, {
+              providerID: modelToUse.providerID,
+              modelID: modelToUse.modelID
+            });
+          } else {
+            log.debug(`No model available for ${operation}, using default model`);
+          }
         } else {
-          log.debug("No model available, using default model");
+          log.debug(`Using selected model for ${operation} warmup`, {
+            providerID: modelToUse.providerID,
+            modelID: modelToUse.modelID
+          });
         }
         const WARMUP_TIMEOUT = 6e4;
         const data = yield this.createHttpRequest(
@@ -388,10 +362,10 @@ class OpenCodeAPI {
                 text: "Test if the chrome-devtools_list_pages tool is available. If available, reply with: ready. If not available, explain why."
               }
             ]
-          }, freeModel && {
+          }, modelToUse && {
             model: {
-              providerID: freeModel.providerID,
-              modelID: freeModel.modelID
+              providerID: modelToUse.providerID,
+              modelID: modelToUse.modelID
             }
           })),
           WARMUP_TIMEOUT
@@ -412,56 +386,43 @@ class OpenCodeAPI {
             `AI responded with: ${responseText.substring(0, 200)}`
           );
         }
-        timer.end("Chrome MCP warmed up");
+        timer.end(`Chrome MCP ${operation} warmed up successfully`);
+        return { success: true };
       } catch (e) {
         if (e instanceof import_shared.ChromeMcpWarmupError) {
           if (e.type === import_shared.ChromeMcpWarmupErrorType.SESSION_ERROR) {
             timer.end("Session creation failed");
           }
-          log.warn(`Chrome MCP warmup failed: ${e.type}`, __spreadValues({
+          log.warn(`Chrome MCP ${operation} warmup failed: ${e.type}`, {
             message: e.message,
             details: e.details
-          }, freeModel && {
-            model: `${freeModel.providerID}/${freeModel.modelID}`
-          }));
-          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)
           });
+          timer.end(`Chrome MCP ${operation} warmup failed: ${e.type}`);
+          return { success: false, error: e };
         }
         const errorMessage = e instanceof Error ? e.message : String(e);
         if (errorMessage.includes("timeout") || errorMessage.includes("Timeout")) {
           const error2 = new import_shared.ChromeMcpWarmupError(
             import_shared.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."
+            "AI did not respond within 60 seconds. Please check if the OpenCode AI model is properly configured and available."
           );
-          log.warn("Chrome MCP warmup timeout", __spreadValues({
+          log.warn(`Chrome MCP ${operation} warmup timeout`, {
             error: errorMessage
-          }, freeModel && {
-            model: `${freeModel.providerID}/${freeModel.modelID}`
-          }));
-          timer.end("Chrome MCP warmup timeout");
-          throw error2;
+          });
+          timer.end(`Chrome MCP ${operation} warmup timeout`);
+          return { success: false, error: error2 };
         }
         const error = new import_shared.ChromeMcpWarmupError(
           import_shared.ChromeMcpWarmupErrorType.UNKNOWN,
-          "Unknown error during Chrome MCP warmup",
+          `Unknown error during Chrome MCP ${operation} warmup`,
           errorMessage
         );
-        log.warn("Chrome MCP warmup failed with unknown error", __spreadValues({
+        log.warn(`Chrome MCP ${operation} warmup failed with unknown error`, {
           error: errorMessage
-        }, freeModel && {
-          model: `${freeModel.providerID}/${freeModel.modelID}`
-        }));
-        timer.end("Chrome MCP warmup failed");
-        throw error;
+        });
+        timer.end(`Chrome MCP ${operation} warmup failed`);
+        return { success: false, error };
       } finally {
         if (warmupSessionId) {
           try {
@@ -476,6 +437,27 @@ class OpenCodeAPI {
       }
     });
   }
+  warmupChromeMcp(projectDir, viteOrigin) {
+    return __async(this, null, function* () {
+      if (!this.warmupChromeMcpConfig) return;
+      const models = yield this.getAvailableModels();
+      const cheapestModel = models[0];
+      const result = yield this.executeWarmupChromeMcp(
+        projectDir,
+        "warmup",
+        viteOrigin,
+        cheapestModel ? { providerID: cheapestModel.providerID, modelID: cheapestModel.modelID } : void 0
+      );
+      if (!result.success) {
+        throw result.error;
+      }
+    });
+  }
+  retryWarmupChromeMcp(projectDir, viteOrigin, selectedModel) {
+    return __async(this, null, function* () {
+      return this.executeWarmupChromeMcp(projectDir, "retry", viteOrigin, selectedModel);
+    });
+  }
   getOrCreateSession(projectDir) {
     return __async(this, null, function* () {
       const timer = log.timer("getOrCreateSession", { projectDir });
@@ -497,148 +479,6 @@ class OpenCodeAPI {
       return url;
     });
   }
-  retryWarmupChromeMcp(projectDir, viteOrigin) {
-    return __async(this, null, function* () {
-      const timer = log.timer("retryWarmupChromeMcp", { viteOrigin });
-      let warmupSessionId = null;
-      let freeModel = null;
-      const chromeAvailable = yield (0, import_shared.checkChromeDevToolsAvailable)();
-      if (!chromeAvailable) {
-        const error = new import_shared.ChromeMcpWarmupError(
-          import_shared.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: import_shared.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(
-          projectDir,
-          import_shared.DEFAULT_RETRIES,
-          "__chrome_mcp_warmup__"
-        );
-        warmupSessionId = warmupSession.id;
-        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 WARMUP_TIMEOUT = 6e4;
-        const data = yield this.createHttpRequest(
-          {
-            hostname: this.hostname,
-            port: this.getPort(),
-            path: `/session/${warmupSessionId}/message`,
-            method: "POST",
-            headers: { "Content-Type": "application/json" }
-          },
-          JSON.stringify(__spreadValues({
-            parts: [
-              {
-                type: "text",
-                text: "Test if the chrome-devtools_list_pages tool is available. If available, reply with: ready. If not available, explain why."
-              }
-            ]
-          }, freeModel && {
-            model: {
-              providerID: freeModel.providerID,
-              modelID: freeModel.modelID
-            }
-          })),
-          WARMUP_TIMEOUT
-        );
-        log.debug("Chrome MCP warmup response:", { data });
-        const responseText = (0, import_shared.extractTextFromResponse)(data);
-        if (!responseText) {
-          throw new import_shared.ChromeMcpWarmupError(
-            import_shared.ChromeMcpWarmupErrorType.AI_RESPONSE_ERROR,
-            "AI did not respond to the warmup request",
-            "Empty response from AI"
-          );
-        }
-        const lowerResponse = responseText.toLowerCase();
-        if (!lowerResponse.includes("ready")) {
-          throw new import_shared.ChromeMcpWarmupError(
-            import_shared.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 { success: true };
-      } catch (e) {
-        if (e instanceof import_shared.ChromeMcpWarmupError) {
-          if (e.type === import_shared.ChromeMcpWarmupErrorType.SESSION_ERROR) {
-            timer.end("Session creation failed");
-          }
-          log.warn(`Chrome MCP warmup retry failed: ${e.type}`, __spreadValues({
-            message: e.message,
-            details: e.details
-          }, freeModel && {
-            model: `${freeModel.providerID}/${freeModel.modelID}`
-          }));
-          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 import_shared.ChromeMcpWarmupError(
-            import_shared.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", __spreadValues({
-            error: errorMessage
-          }, freeModel && {
-            model: `${freeModel.providerID}/${freeModel.modelID}`
-          }));
-          timer.end("Chrome MCP warmup retry timeout");
-          return { success: false, error: error2 };
-        }
-        const error = new import_shared.ChromeMcpWarmupError(
-          import_shared.ChromeMcpWarmupErrorType.UNKNOWN,
-          "Unknown error during Chrome MCP warmup retry",
-          errorMessage
-        );
-        log.warn("Chrome MCP warmup retry failed with unknown error", __spreadValues({
-          error: errorMessage
-        }, freeModel && {
-          model: `${freeModel.providerID}/${freeModel.modelID}`
-        }));
-        timer.end("Chrome MCP warmup retry failed");
-        return { success: false, error };
-      } finally {
-        if (warmupSessionId) {
-          try {
-            yield this.deleteSession(warmupSessionId, 5);
-          } catch (e) {
-            log.warn("Failed to delete warmup session after retries", {
-              error: e,
-              warmupSessionId
-            });
-          }
-        }
-      }
-    });
-  }
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/lib/core/service.d.ts

@@ -1,5 +1,6 @@
 import type { ResultPromise } from "execa";
 import type http from "http";
+import type { ModelInfo } from "@vite-plugin-opencode-assistant/shared";
 import type { OpenCodeOptions, ServiceStartupTask } from "@vite-plugin-opencode-assistant/shared";
 import { ChromeMcpWarmupErrorType } from "@vite-plugin-opencode-assistant/shared";
 import type { OpenCodeAPI } from "./api.js";
@@ -26,7 +27,11 @@ export declare class OpenCodeService {
     constructor(config: Required<OpenCodeOptions>, api: OpenCodeAPI, sseClients: Set<http.ServerResponse>, onPortAllocated: (port: number) => void, onProxyPortAllocated: (port: number) => void);
     private sendTaskUpdate;
     start(corsOrigins?: string[], contextApiUrl?: string, logsApiUrl?: string, viteOrigin?: string): Promise<void>;
-    retryWarmupChromeMcp(viteOrigin?: string): Promise<{
+    getAvailableModels(): Promise<ModelInfo[]>;
+    retryWarmupChromeMcp(viteOrigin?: string, selectedModel?: {
+        providerID: string;
+        modelID: string;
+    }): Promise<{
         success: boolean;
         errorType?: string;
         errorMessage?: string;

package/lib/core/service.js

@@ -270,9 +270,18 @@ Please install OpenCode first:
       return this.startPromise;
     });
   }
-  retryWarmupChromeMcp(viteOrigin) {
+  getAvailableModels() {
     return __async(this, null, function* () {
-      const result = yield this.api.retryWarmupChromeMcp(this.workspaceRoot, viteOrigin);
+      return this.api.getAvailableModels();
+    });
+  }
+  retryWarmupChromeMcp(viteOrigin, selectedModel) {
+    return __async(this, null, function* () {
+      const result = yield this.api.retryWarmupChromeMcp(
+        this.workspaceRoot,
+        viteOrigin,
+        selectedModel
+      );
       if (result.success) {
         this.chromeMcpWarmupFailed = false;
         this.sendTaskUpdate("ready");

package/lib/endpoints/types.d.ts

@@ -1,4 +1,4 @@
-import type { PageContext, SessionInfo, ServiceStartupTask } from "@vite-plugin-opencode-assistant/shared";
+import type { PageContext, SessionInfo, ServiceStartupTask, ModelInfo } from "@vite-plugin-opencode-assistant/shared";
 import type http from "http";
 export interface EndpointContext {
     get webUrl(): string | null;
@@ -15,7 +15,11 @@ export interface EndpointContext {
     deleteSession: (id: string) => Promise<void>;
     resolveWidgetPath: () => string;
     resolveWidgetStylePath: () => string;
-    retryWarmupChromeMcp: () => Promise<{
+    getAvailableModels: () => Promise<ModelInfo[]>;
+    retryWarmupChromeMcp: (selectedModel?: {
+        providerID: string;
+        modelID: string;
+    }) => Promise<{
         success: boolean;
         errorType?: string;
         errorMessage?: string;

package/lib/endpoints/warmup.js

@@ -2,6 +2,7 @@ var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : /* @__PURE__ */ Symbol.for("Symbol." + name);
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -35,6 +36,7 @@ var __async = (__this, __arguments, generator) => {
     step((generator = generator.apply(__this, __arguments)).next());
   });
 };
+var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")]) ? it.call(obj) : (obj = obj[__knownSymbol("iterator")](), it = {}, method = (key, fn) => (fn = obj[key]) && (it[key] = (arg) => new Promise((yes, no, done) => (arg = fn.call(obj, arg), done = arg.done, Promise.resolve(arg.value).then((value) => yes({ value, done }), no)))), method("next"), method("return"), it);
 var warmup_exports = {};
 __export(warmup_exports, {
   setupWarmupEndpoint: () => setupWarmupEndpoint
@@ -44,33 +46,78 @@ var import_shared = require("@vite-plugin-opencode-assistant/shared");
 const log = (0, import_shared.createLogger)("Endpoints:Warmup");
 function setupWarmupEndpoint(server, ctx) {
   server.middlewares.use("/__opencode_warmup__", (req, res) => __async(null, null, function* () {
+    if (req.method === "GET") {
+      try {
+        const models = yield ctx.getAvailableModels();
+        res.setHeader("Content-Type", "application/json");
+        res.writeHead(200);
+        res.end(JSON.stringify({ success: true, models }));
+      } catch (e) {
+        log.error("Failed to get available models", { error: e });
+        res.setHeader("Content-Type", "application/json");
+        res.writeHead(500);
+        res.end(JSON.stringify({ success: false, models: [] }));
+      }
+      return;
+    }
     if (req.method !== "POST") {
       res.writeHead(405);
       res.end("Method not allowed");
       return;
     }
     try {
-      const result = yield ctx.retryWarmupChromeMcp();
+      let body = "";
+      try {
+        for (var iter = __forAwait(req), more, temp, error; more = !(temp = yield iter.next()).done; more = false) {
+          const chunk = temp.value;
+          body += chunk;
+        }
+      } catch (temp) {
+        error = [temp];
+      } finally {
+        try {
+          more && (temp = iter.return) && (yield temp.call(iter));
+        } finally {
+          if (error)
+            throw error[0];
+        }
+      }
+      let selectedModel;
+      if (body) {
+        try {
+          const parsed = JSON.parse(body);
+          if (parsed.providerID && parsed.modelID) {
+            selectedModel = { providerID: parsed.providerID, modelID: parsed.modelID };
+          }
+        } catch (e) {
+          log.debug("Failed to parse request body, using default model");
+        }
+      }
+      const result = yield ctx.retryWarmupChromeMcp(selectedModel);
       res.setHeader("Content-Type", "application/json");
       res.writeHead(200);
       if (result.success) {
         res.end(JSON.stringify({ success: true }));
       } else {
-        res.end(JSON.stringify({
-          success: false,
-          errorType: result.errorType,
-          error: result.errorMessage
-        }));
+        res.end(
+          JSON.stringify({
+            success: false,
+            errorType: result.errorType,
+            error: result.errorMessage
+          })
+        );
       }
     } catch (e) {
       log.error("Failed to retry warmup", { error: e });
       res.setHeader("Content-Type", "application/json");
       res.writeHead(500);
-      res.end(JSON.stringify({
-        success: false,
-        errorType: "UNKNOWN",
-        error: String(e)
-      }));
+      res.end(
+        JSON.stringify({
+          success: false,
+          errorType: "UNKNOWN",
+          error: String(e)
+        })
+      );
     }
   }));
 }

package/lib/index.js

@@ -146,7 +146,8 @@ function createOpenCodePlugin(options = {}) {
           deleteSession: (id) => api.deleteSession(id),
           resolveWidgetPath: import_paths.resolveWidgetPath,
           resolveWidgetStylePath: import_paths.resolveWidgetStylePath,
-          retryWarmupChromeMcp: () => service.retryWarmupChromeMcp(getViteOrigin())
+          getAvailableModels: () => service.getAvailableModels(),
+          retryWarmupChromeMcp: (selectedModel) => service.retryWarmupChromeMcp(getViteOrigin(), selectedModel)
         });
         (_a2 = server.httpServer) == null ? void 0 : _a2.on("listening", () => __async(null, null, function* () {
           var _a3;