nixpkgs/nixos/tests/open-webui.nix
pinage404 83b1350904 nixos/open-webui: fix opensearch
When [adding Open WebUI as search engine in Firefox](https://docs.openwebui.com/tutorials/integrations/browser-search-engine/#for-firefox) it always links to `http://localhost:3000` instead of the configured port

It is because [the response use `WEBUI_URL` environment variable](1d225dd804/backend/open_webui/main.py (L2370)) which [is set by default to `1d225dd804/backend/open_webui/env.py (L106))
2024-09-30 17:28:11 +02:00

64 lines
1.7 KiB
Nix

{ config, lib, ... }:
let
mainPort = "8080";
webuiName = "NixOS Test";
in
{
name = "open-webui";
meta = with lib.maintainers; {
maintainers = [ shivaraj-bh ];
};
nodes = {
machine =
{ ... }:
{
services.open-webui = {
enable = true;
environment = {
# Requires network connection
RAG_EMBEDDING_MODEL = "";
};
# Test that environment variables can be
# overridden through a file.
environmentFile = config.node.pkgs.writeText "test.env" ''
WEBUI_NAME="${webuiName}"
'';
};
};
};
testScript = ''
import json
import xml.etree.ElementTree as xml
machine.start()
machine.wait_for_unit("open-webui.service")
machine.wait_for_open_port(${mainPort})
machine.succeed("curl http://127.0.0.1:${mainPort}")
# Load the Web UI config JSON and parse it.
webui_config_json = machine.succeed("curl http://127.0.0.1:${mainPort}/api/config")
webui_config = json.loads(webui_config_json)
# Check that the name was overridden via the environmentFile option.
assert webui_config["name"] == "${webuiName} (Open WebUI)"
webui_opensearch_xml = machine.succeed("curl http://127.0.0.1:${mainPort}/opensearch.xml")
webui_opensearch = xml.fromstring(webui_opensearch_xml)
webui_opensearch_url = webui_opensearch.find(
".//{http://a9.com/-/spec/opensearch/1.1/}Url"
)
assert (
webui_opensearch_url is not None
), f"no url tag found in {webui_opensearch_xml}"
assert (
webui_opensearch_url.get("template") == "http://localhost:8080/?q={searchTerms}"
), "opensearch url doesn't match the configured port"
'';
}