From 377bd48ee59cc23db360d7c39c761d4a09b601e6 Mon Sep 17 00:00:00 2001
From: ooks-io <ooks@protonmail.com>
Date: Wed, 29 Jan 2025 17:26:11 +1100
Subject: [PATCH] ollama: init

---
 hosts/ooksdesk/default.nix                    |  2 ++
 modules/nixos/workstation/options.nix         |  1 +
 .../nixos/workstation/services/default.nix    |  1 +
 modules/nixos/workstation/services/ollama.nix | 21 +++++++++++++++++++
 4 files changed, 25 insertions(+)
 create mode 100644 modules/nixos/workstation/services/ollama.nix

diff --git a/hosts/ooksdesk/default.nix b/hosts/ooksdesk/default.nix
index e05050a..c7a3fdd 100644
--- a/hosts/ooksdesk/default.nix
+++ b/hosts/ooksdesk/default.nix
@@ -25,6 +25,8 @@
         browser = "firefox";
         terminal = "ghostty";
       };
+      # FIXME
+      programs.ollama.enable = false;
     };
     console = {
       profile = "standard";
diff --git a/modules/nixos/workstation/options.nix b/modules/nixos/workstation/options.nix
index 6c385a4..da3d19a 100644
--- a/modules/nixos/workstation/options.nix
+++ b/modules/nixos/workstation/options.nix
@@ -29,6 +29,7 @@ in {
       firefox.enable = mkEnableOption "";
       foot.enable = mkEnableOption "";
       ghostty.enable = mkEnableOption "";
+      ollama.enable = mkEnableOption "";
     };
   };
 }
diff --git a/modules/nixos/workstation/services/default.nix b/modules/nixos/workstation/services/default.nix
index 0dce40b..a50d475 100644
--- a/modules/nixos/workstation/services/default.nix
+++ b/modules/nixos/workstation/services/default.nix
@@ -3,5 +3,6 @@
     ./dbus.nix
     ./gnome-services.nix
     ./geoclue.nix
+    ./ollama.nix
   ];
 }
diff --git a/modules/nixos/workstation/services/ollama.nix b/modules/nixos/workstation/services/ollama.nix
new file mode 100644
index 0000000..549c6f1
--- /dev/null
+++ b/modules/nixos/workstation/services/ollama.nix
@@ -0,0 +1,21 @@
+{
+  lib,
+  config,
+  ...
+}: let
+  inherit (lib) mkIf;
+  cfg = config.ooknet.workstation.programs.ollama;
+in {
+  config = mkIf cfg.enable {
+    # FIXME:
+    # https://github.com/NixOS/nixpkgs/issues/376930
+    services.ollama = {
+      enable = true;
+      acceleration = "rocm";
+      rocmOverrideGfx = "10.1.0";
+      environmentVariables = {
+        HCC_AMDGPU_TARGET = "gfx1010";
+      };
+    };
+  };
+}