nixos/hosts/architect/llm.nix
2024-04-28 01:29:48 +01:00

81 lines
1.6 KiB
Nix

{ pkgs, ... }:
let
frontendDomain = "pino.giugl.io";
backendDomain = "ollama.giugl.io";
frontendPort = 3030;
ollamaPort = 11434;
listenAddress = "127.0.0.1:${toString ollamaPort}";
ollamaPkg = pkgs.unstablePkgs.ollama;
in
{
environment = {
systemPackages = [ ollamaPkg pkgs.aichat ];
variables = {
OLLAMA_ORIGINS = "*";
};
};
services.ollama = {
inherit listenAddress;
enable = true;
acceleration = "cuda";
package = ollamaPkg;
};
architect.vhost.${frontendDomain} = {
dnsInterfaces = [ "tailscale" ];
locations."/" = {
host = "127.0.0.1";
port = frontendPort;
allowLan = true;
allowWAN = true;
extraConfig = ''
proxy_read_timeout 600s;
'';
};
};
architect.vhost.${backendDomain} = {
dnsInterfaces = [ "tailscale" ];
locations."/" = {
host = "127.0.0.1";
port = 11434;
allowLan = true;
allowWAN = true;
extraConfig = ''
proxy_buffering off;
proxy_read_timeout 600s;
proxy_set_header Host localhost:11434;
'';
};
};
virtualisation.oci-containers = {
containers = {
ollama-webui = {
image = "ghcr.io/open-webui/open-webui:main";
autoStart = true;
ports = [
"127.0.0.1:${toString frontendPort}:8080"
];
environment = {
OLLAMA_BASE_URL = "https://${backendDomain}";
};
extraOptions = [
"--pull=always"
];
volumes = [
"/var/lib/ollama-webui:/app/backend/data"
];
};
};
};
}