25 lines
702 B
Nix
25 lines
702 B
Nix
|
{ pkgs, lib, ... }: {
|
||
|
|
environment.systemPackages = with pkgs; [ ollama ];
|
||
|
|
systemd.services.ollama = {
|
||
|
|
description = "Ollama LLM server.";
|
||
|
|
wantedBy = [ "multi-user.target" ];
|
||
|
|
wants = [ "NetworkManager-wait-online.service" ];
|
||
|
|
after = [ "NetworkManager-wait-online.service" ];
|
||
|
|
serviceConfig.Type = "simple";
|
||
|
|
script = ''
|
||
|
|
HOME=/root ${lib.getExe pkgs.ollama} serve
|
||
|
|
'';
|
||
|
|
};
|
||
|
|
systemd.services.ollamamodel = {
|
||
|
|
description = "Ollama LLM model.";
|
||
|
|
wantedBy = [ "multi-user.target" ];
|
||
|
|
wants = [ "ollama.service" ];
|
||
|
|
after = [ "ollama.service" ];
|
||
|
|
serviceConfig.Type = "simple";
|
||
|
|
script = ''
|
||
|
|
sleep 5
|
||
|
|
${lib.getExe pkgs.ollama} pull mistral
|
||
|
|
'';
|
||
|
|
};
|
||
|
|
}
|