add bin path option

NixOS · Feb 9, 2024 · 9afa045 · 9afa045
1 parent d48979f
commit 9afa045
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkgs/tools/misc/ollama/default.nix b/pkgs/tools/misc/ollama/default.nix
@@ -2,6 +2,7 @@
 , buildGoModule
 , fetchFromGitHub
 , llama-cpp
+, llama-cpp-bin-path ? "bin/llama-cpp-server"
 }:
 
 buildGoModule rec {
@@ -26,7 +27,7 @@ buildGoModule rec {
 
   postPatch = ''
     substituteInPlace llm/llama.go \
-      --subst-var-by llamaCppServer "${llama-cpp}/bin/llama-cpp-server"
+      --subst-var-by llamaCppServer "${llama-cpp}/${llama-cpp-bin-path}"
     substituteInPlace server/routes_test.go --replace "0.0.0" "${version}"
   '';