diff --git a/modules/services/llm/default.nix b/modules/services/llm/default.nix index ffef152..ad433f6 100644 --- a/modules/services/llm/default.nix +++ b/modules/services/llm/default.nix @@ -35,8 +35,7 @@ in frontend = { enable = mkEnableOption "Enable LLM frontend service (Open WebUI)"; # Defaults to false - uiPackage = mkPackageOption pkgs "open-webui" { }; - tikaPackage = mkPackageOption pkgs "tika" { }; # Tika for document processing with Open WebUI + package = mkPackageOption pkgs "open-webui" { }; domain = mkOption { type = types.nullOr types.str; default = null; @@ -50,7 +49,7 @@ in { environment.systemPackages = (if cfg.enable then [ cfg.package ] else [ ]) ++ - (if cfg.enable && cfg.frontend.enable then [ cfg.frontend.uiPackage cfg.frontend.tikaPackage ] else [ ]); + (if cfg.enable && cfg.frontend.enable then [ cfg.frontend.package ] else [ ]); } # Backend Ollama Service Configuration @@ -85,12 +84,12 @@ in (mkIf (cfg.enable && cfg.frontend.enable) { services.open-webui = { enable = true; - package = cfg.frontend.uiPackage; + package = cfg.frontend.package; }; - services.tika = { + services.docling-serve = { enable = true; - package = cfg.frontend.tikaPackage; + package = pkgs.unstablePkgs.docling-serve; }; pepe.core.vhost.hosts = optionalAttrs (cfg.frontend.domain != null) {