From cb963bba4c94e50f2f617005d04f461458872f7a Mon Sep 17 00:00:00 2001
From: Colin <colin@uninsane.org>
Date: Sun, 13 Oct 2024 02:03:55 +0000
Subject: [PATCH] ollama: ship mistral

---
 modules/services/ollama.nix | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/modules/services/ollama.nix b/modules/services/ollama.nix
index 0c285ad1d..e5d936640 100644
--- a/modules/services/ollama.nix
+++ b/modules/services/ollama.nix
@@ -15,7 +15,14 @@
 { config, lib, pkgs, ... }:
 let
   cfg = config.sane.services.ollama;
-  models = "${pkgs.ollamaPackages.llama3_2_3b}/share/ollama/models";
+  modelSources = pkgs.symlinkJoin {
+    name = "ollama-models";
+    paths = with pkgs.ollamaPackages; [
+      llama3_2_3b
+      mistral
+    ];
+  };
+  models = "${modelSources}/share/ollama/models";
 in
 {
   options.sane.services.ollama = with lib; {