From 0fd54a5494788a54df3f18f1d6c52839f3d8328c Mon Sep 17 00:00:00 2001
From: Leyla Becker <git@jan-leila.com>
Date: Sat, 26 Jul 2025 17:53:37 -0500
Subject: [PATCH] added more options to aiCode

---
 .../home-manager/leyla/packages/vscode/default.nix |  1 +
 .../programs/vscode/aiCode.nix                     | 14 ++++++++++++++
 2 files changed, 15 insertions(+)

diff --git a/configurations/home-manager/leyla/packages/vscode/default.nix b/configurations/home-manager/leyla/packages/vscode/default.nix
index 009d7f0..9ad87bf 100644
--- a/configurations/home-manager/leyla/packages/vscode/default.nix
+++ b/configurations/home-manager/leyla/packages/vscode/default.nix
@@ -60,6 +60,7 @@ in {
             aiCode = {
               enable = ai-tooling-enabled;
               ollamaHost = "http://defiant:11434";
+              inlineCompletion.model = "qwen3:4b";
             };
           };
 
diff --git a/modules/home-manager-modules/programs/vscode/aiCode.nix b/modules/home-manager-modules/programs/vscode/aiCode.nix
index cd23f92..838a439 100644
--- a/modules/home-manager-modules/programs/vscode/aiCode.nix
+++ b/modules/home-manager-modules/programs/vscode/aiCode.nix
@@ -16,6 +16,18 @@ in {
             description = "what host should be used for ollama";
             default = null;
           };
+          inlineCompletion = {
+            enable = lib.mkOption {
+              type = lib.types.bool;
+              description = "should inline completion be enabled";
+              default = true;
+            };
+            model = lib.mkOption {
+              type = lib.types.nullOr lib.types.str;
+              description = "what model should be used for ollama";
+              default = null;
+            };
+          };
         };
       };
       config = lib.mkIf config.extraExtensions.aiCode.enable {
@@ -24,6 +36,8 @@ in {
         ];
         userSettings = {
           "aiCode.ollamaHost" = lib.mkIf (config.extraExtensions.aiCode.ollamaHost != null) config.extraExtensions.aiCode.ollamaHost;
+          "aiCode.inlineCompletion.enable" = config.extraExtensions.aiCode.inlineCompletion.enable;
+          "aiCode.inlineCompletion.model" = lib.mkIf (config.extraExtensions.aiCode.inlineCompletion.model != null) config.extraExtensions.aiCode.inlineCompletion.model;
         };
       };
     }));