commit bcb04b09678682c65b5ca5c309c64e6196ac7ab8 · pyrox.dev/nixpkgs

+12

nixos/modules/services/misc/open-webui.nix

···

       60
        
               '';

     

       61
        
             };

     

       62
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       63
        
             openFirewall = lib.mkOption {

     

       64
        
               type = types.bool;

     

       65
        
               default = false;

     
···

       86
        
       

     

       87
        
             serviceConfig = {

     

       88
        
               ExecStart = "${lib.getExe cfg.package} serve --host ${cfg.host} --port ${toString cfg.port}";

     

       0
        
       
     

       89
        
               WorkingDirectory = cfg.stateDir;

     

       90
        
               StateDirectory = "open-webui";

     

       91
        
               RuntimeDirectory = "open-webui";

···

       60
        
               '';

     

       61
        
             };

     

       62
        
       

     

       63
       +
             environmentFile = lib.mkOption {

     

       64
       +
               description = ''

     

       65
       +
                 Environment file to be passed to the systemd service.

     

       66
       +
                 Useful for passing secrets to the service to prevent them from being

     

       67
       +
                 world-readable in the Nix store.

     

       68
       +
               '';

     

       69
       +
               type = lib.types.nullOr lib.types.path;

     

       70
       +
               default = null;

     

       71
       +
               example = "/var/lib/secrets/openWebuiSecrets";

     

       72
       +
             };

     

       73
       +
       

     

       74
        
             openFirewall = lib.mkOption {

     

       75
        
               type = types.bool;

     

       76
        
               default = false;

     
···

       97
        
       

     

       98
        
             serviceConfig = {

     

       99
        
               ExecStart = "${lib.getExe cfg.package} serve --host ${cfg.host} --port ${toString cfg.port}";

     

       100
       +
               EnvironmentFile = lib.optional (cfg.environmentFile != null) cfg.environmentFile;

     

       101
        
               WorkingDirectory = cfg.stateDir;

     

       102
        
               StateDirectory = "open-webui";

     

       103
        
               RuntimeDirectory = "open-webui";

+17 -1

nixos/tests/open-webui.nix

···

       1
       -
       { lib, ... }:

     

       2
        
       let

     

       3
        
         mainPort = "8080";

     

       0
        
       
     

       4
        
       in

     

       5
        
       {

     

       6
        
         name = "open-webui";

     
···

       18
        
                   # Requires network connection

     

       19
        
                   RAG_EMBEDDING_MODEL = "";

     

       20
        
                 };

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       21
        
               };

     

       22
        
             };

     

       23
        
         };

     

       24
        
       

     

       25
        
         testScript = ''

     

       0
        
       
     

       0
        
       
     

       26
        
           machine.start()

     

       27
        
       

     

       28
        
           machine.wait_for_unit("open-webui.service")

     

       29
        
           machine.wait_for_open_port(${mainPort})

     

       30
        
       

     

       31
        
           machine.succeed("curl http://127.0.0.1:${mainPort}")

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       32
        
         '';

     

       33
        
       }

···

       1
       +
       { config, lib, ... }:

     

       2
        
       let

     

       3
        
         mainPort = "8080";

     

       4
       +
         webuiName = "NixOS Test";

     

       5
        
       in

     

       6
        
       {

     

       7
        
         name = "open-webui";

     
···

       19
        
                   # Requires network connection

     

       20
        
                   RAG_EMBEDDING_MODEL = "";

     

       21
        
                 };

     

       22
       +
       

     

       23
       +
                 # Test that environment variables can be

     

       24
       +
                 # overridden through a file.

     

       25
       +
                 environmentFile = config.node.pkgs.writeText "test.env" ''

     

       26
       +
                   WEBUI_NAME="${webuiName}"

     

       27
       +
                 '';

     

       28
        
               };

     

       29
        
             };

     

       30
        
         };

     

       31
        
       

     

       32
        
         testScript = ''

     

       33
       +
           import json

     

       34
       +
       

     

       35
        
           machine.start()

     

       36
        
       

     

       37
        
           machine.wait_for_unit("open-webui.service")

     

       38
        
           machine.wait_for_open_port(${mainPort})

     

       39
        
       

     

       40
        
           machine.succeed("curl http://127.0.0.1:${mainPort}")

     

       41
       +
       

     

       42
       +
           # Load the Web UI config JSON and parse it.

     

       43
       +
           webui_config_json = machine.succeed("curl http://127.0.0.1:${mainPort}/api/config")

     

       44
       +
           webui_config = json.loads(webui_config_json)

     

       45
       +
       

     

       46
       +
           # Check that the name was overridden via the environmentFile option.

     

       47
       +
           assert webui_config["name"] == "${webuiName} (Open WebUI)"

     

       48
        
         '';

     

       49
        
       }