From 540db884d1e8f2d35c69fe585960450407e67d64 Mon Sep 17 00:00:00 2001
From: Antoine Viallon <antoine@lesviallon.fr>
Date: Wed, 11 Dec 2024 09:58:46 +0100
Subject: [PATCH] [Desktop/Developer] enable ollama with yi-coder support by
 default

Also remove tabnine from systemPackages, since it is obsolete.
---
 desktop/developer.nix | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/desktop/developer.nix b/desktop/developer.nix
index 1ce5f3f..a8708ce 100644
--- a/desktop/developer.nix
+++ b/desktop/developer.nix
@@ -28,7 +28,7 @@ in {
     '';
   
     environment.systemPackages = with pkgs; [
-      tabnine
+      #tabnine
       numactl
       schedtool
       stress
@@ -126,6 +126,20 @@ in {
       "doc" "info" "dev" "debug" "static"
     ];
 
+    services.ollama = {
+      enable = mkDefault true;
+      loadModels = [ "yi-coder:1.5b" ];
+      group = "ollama";
+      user = "ollama";
+      package =
+        if config.aviallon.hardware.amd.enable
+          then pkgs.unstable.ollama-rocm
+        else if (config.aviallon.hardware.nvidia.enable && config.aviallon.hardware.nvidia.variant != "nouveau")
+          then pkgs.unstable.ollama-cuda
+        else pkgs.unstable.ollama
+      ;
+    };
+
     aviallon.services.journald.extraConfig = {
       Storage = mkForce "persistent";
     };