From c879a4e01daeb249c9a90138b26271bb621295e6 Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Thu, 19 Jun 2025 12:04:24 -0700
Subject: [PATCH] always use engine if pro modes are enabled (#449)

---
 src/ipc/utils/get_model_client.ts | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/src/ipc/utils/get_model_client.ts b/src/ipc/utils/get_model_client.ts
index 7ddf9b6..4d31847 100644
--- a/src/ipc/utils/get_model_client.ts
+++ b/src/ipc/utils/get_model_client.ts
@@ -11,7 +11,7 @@ import log from "electron-log";
 import { getLanguageModelProviders } from "../shared/language_model_helpers";
 import { LanguageModelProvider } from "../ipc_types";
 import { createDyadEngine } from "./llm_engine_provider";
-import { findLanguageModel } from "./findLanguageModel";
+
 import { LM_STUDIO_BASE_URL } from "./lm_studio_utils";
 
 const dyadEngineUrl = process.env.DYAD_ENGINE_URL;
@@ -102,14 +102,9 @@ export async function getModelClient(
     // IMPORTANT: some providers like OpenAI have an empty string gateway prefix,
     // so we do a nullish and not a truthy check here.
     if (providerConfig.gatewayPrefix != null || dyadEngineUrl) {
-      const languageModel = await findLanguageModel(model);
-      const engineProMode =
+      const isEngineEnabled =
         settings.enableProSmartFilesContextMode ||
         settings.enableProLazyEditsMode;
-      // Currently engine is only used for turbo edits.
-      const isEngineEnabled = Boolean(
-        engineProMode && languageModel?.type === "cloud",
-      );
       const provider = isEngineEnabled
         ? createDyadEngine({
             apiKey: dyadApiKey,
@@ -146,7 +141,7 @@ export async function getModelClient(
       const autoModelClient = {
         model: provider(
           `${providerConfig.gatewayPrefix || ""}${modelName}`,
-          engineProMode
+          isEngineEnabled
             ? {
                 files,
               }