always use engine if pro modes are enabled (#449)
This commit is contained in:
@@ -11,7 +11,7 @@ import log from "electron-log";
|
|||||||
import { getLanguageModelProviders } from "../shared/language_model_helpers";
|
import { getLanguageModelProviders } from "../shared/language_model_helpers";
|
||||||
import { LanguageModelProvider } from "../ipc_types";
|
import { LanguageModelProvider } from "../ipc_types";
|
||||||
import { createDyadEngine } from "./llm_engine_provider";
|
import { createDyadEngine } from "./llm_engine_provider";
|
||||||
import { findLanguageModel } from "./findLanguageModel";
|
|
||||||
import { LM_STUDIO_BASE_URL } from "./lm_studio_utils";
|
import { LM_STUDIO_BASE_URL } from "./lm_studio_utils";
|
||||||
|
|
||||||
const dyadEngineUrl = process.env.DYAD_ENGINE_URL;
|
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,
|
// IMPORTANT: some providers like OpenAI have an empty string gateway prefix,
|
||||||
// so we do a nullish and not a truthy check here.
|
// so we do a nullish and not a truthy check here.
|
||||||
if (providerConfig.gatewayPrefix != null || dyadEngineUrl) {
|
if (providerConfig.gatewayPrefix != null || dyadEngineUrl) {
|
||||||
const languageModel = await findLanguageModel(model);
|
const isEngineEnabled =
|
||||||
const engineProMode =
|
|
||||||
settings.enableProSmartFilesContextMode ||
|
settings.enableProSmartFilesContextMode ||
|
||||||
settings.enableProLazyEditsMode;
|
settings.enableProLazyEditsMode;
|
||||||
// Currently engine is only used for turbo edits.
|
|
||||||
const isEngineEnabled = Boolean(
|
|
||||||
engineProMode && languageModel?.type === "cloud",
|
|
||||||
);
|
|
||||||
const provider = isEngineEnabled
|
const provider = isEngineEnabled
|
||||||
? createDyadEngine({
|
? createDyadEngine({
|
||||||
apiKey: dyadApiKey,
|
apiKey: dyadApiKey,
|
||||||
@@ -146,7 +141,7 @@ export async function getModelClient(
|
|||||||
const autoModelClient = {
|
const autoModelClient = {
|
||||||
model: provider(
|
model: provider(
|
||||||
`${providerConfig.gatewayPrefix || ""}${modelName}`,
|
`${providerConfig.gatewayPrefix || ""}${modelName}`,
|
||||||
engineProMode
|
isEngineEnabled
|
||||||
? {
|
? {
|
||||||
files,
|
files,
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user