Smart files context (#184)

src/ipc/utils/get_model_client.ts

@@ -50,6 +50,7 @@ export async function getModelClient(
 ): Promise<{
   modelClient: ModelClient;
   backupModelClients: ModelClient[];
+  isEngineEnabled?: boolean;
 }> {
   const allProviders = await getLanguageModelProviders();
 
@@ -103,9 +104,12 @@ export async function getModelClient(
     // so we do a nullish and not a truthy check here.
     if (providerConfig.gatewayPrefix != null || dyadLocalEngine) {
       const languageModel = await findLanguageModel(model);
+      const engineProMode =
+        settings.enableProSmartFilesContextMode ||
+        settings.enableProLazyEditsMode;
       // Currently engine is only used for turbo edits.
       const isEngineEnabled = Boolean(
-        settings.enableProLazyEditsMode &&
+        engineProMode &&
           languageModel?.type === "cloud" &&
           languageModel?.supportsTurboEdits,
       );
@@ -113,6 +117,10 @@ export async function getModelClient(
         ? createDyadEngine({
             apiKey: dyadApiKey,
             baseURL: dyadLocalEngine ?? "https://engine.dyad.sh/v1",
+            dyadOptions: {
+              enableLazyEdits: settings.enableProLazyEditsMode,
+              enableSmartFilesContext: settings.enableProSmartFilesContextMode,
+            },
           })
         : createOpenAICompatible({
             name: "dyad-gateway",
@@ -126,7 +134,7 @@ export async function getModelClient(
       const autoModelClient = {
         model: provider(
           `${providerConfig.gatewayPrefix || ""}${modelName}`,
-          settings.enableProLazyEditsMode
+          engineProMode
             ? {
                 files,
               }
@@ -161,6 +169,7 @@ export async function getModelClient(
             providerConfig,
           ).modelClient,
           backupModelClients: [autoModelClient],
+          isEngineEnabled,
         };
       } else {
         return {

src/ipc/utils/llm_engine_provider.ts

@@ -41,6 +41,11 @@ Custom fetch implementation. You can use it as a middleware to intercept request
 or to provide a custom fetch implementation for e.g. testing.
 */
   fetch?: FetchFunction;
+
+  dyadOptions: {
+    enableLazyEdits?: boolean;
+    enableSmartFilesContext?: boolean;
+  };
 }
 
 export interface DyadEngineProvider {
@@ -62,7 +67,7 @@ Creates a chat model for text generation.
 }
 
 export function createDyadEngine(
-  options: ExampleProviderSettings = {},
+  options: ExampleProviderSettings,
 ): DyadEngineProvider {
   const baseURL = withoutTrailingSlash(
     options.baseURL ?? "https://api.example.com/v1",
@@ -124,7 +129,9 @@ export function createDyadEngine(
               if (files?.length) {
                 parsedBody.dyad_options = {
                   files,
-                  enable_lazy_edits: true,
+                  enable_lazy_edits: options.dyadOptions.enableLazyEdits,
+                  enable_smart_files_context:
+                    options.dyadOptions.enableSmartFilesContext,
                 };
               }