about summary refs log tree commit diff
path: root/nixpkgs/nixos/tests/ollama.nix
diff options
context:
space:
mode:
Diffstat (limited to 'nixpkgs/nixos/tests/ollama.nix')
-rw-r--r--nixpkgs/nixos/tests/ollama.nix8
1 files changed, 4 insertions, 4 deletions
diff --git a/nixpkgs/nixos/tests/ollama.nix b/nixpkgs/nixos/tests/ollama.nix
index 4b21f445cdbd..30e475553eb1 100644
--- a/nixpkgs/nixos/tests/ollama.nix
+++ b/nixpkgs/nixos/tests/ollama.nix
@@ -1,10 +1,10 @@
 import ./make-test-python.nix ({ pkgs, lib, ... }:
 let
-  mainPort = "11434";
-  altPort = "11435";
+  mainPort = 11434;
+  altPort = 11435;
 
   curlRequest = port: request:
-    "curl http://127.0.0.1:${port}/api/generate -d '${builtins.toJSON request}'";
+    "curl http://127.0.0.1:${toString port}/api/generate -d '${builtins.toJSON request}'";
 
   prompt = {
     model = "tinydolphin";
@@ -38,7 +38,7 @@ in
 
     altAddress = { ... }: {
       services.ollama.enable = true;
-      services.ollama.listenAddress = "127.0.0.1:${altPort}";
+      services.ollama.port = altPort;
     };
   };