modules/ollama: Create by factoring common code from bean/*, tomatoHEADmaster

1 files changed, 26 insertions, 0 deletions

diff --git a/modules/ollama.nix b/modules/ollama.nix
new file mode 100644
index 0000000..7bbcfd1
--- /dev/null
+++ b/modules/ollama.nix
@@ -0,0 +1,26 @@
+{ lib, pkgs, config, inputs, ... }:
+
+let
+  nixpkgs-unstable = inputs.nixpkgs-unstable.
+    legacyPackages.x86_64-linux;
+  cfg = config.this.ollama;
+in
+{
+  options = {
+    this.ollama = {
+      enable = lib.mkEnableOption "ollama";
+      unstable = lib.mkEnableOption "unstable ollama package"
+                 // { default = true; };
+    };
+  };
+
+  config = lib.mkIf cfg.enable {
+    services.ollama = {
+      enable = true;
+      package = lib.mkIf cfg.unstable nixpkgs-unstable.ollama;
+      environmentVariables = {
+        OLLAMA_CONTEXT_LENGTH = "8192";
+      };
+    };
+  };
+}