142 lines
3.2 KiB
Nix
142 lines
3.2 KiB
Nix
{
|
|
lib,
|
|
buildNpmPackage,
|
|
fetchFromGitHub,
|
|
python3,
|
|
nixosTests,
|
|
}:
|
|
let
|
|
pname = "open-webui";
|
|
version = "0.2.4";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "open-webui";
|
|
repo = "open-webui";
|
|
rev = "v${version}";
|
|
hash = "sha256-jWO0mo26C+QTIX5j3ucDk/no+vQnAh7Q6JwB3lLM83k=";
|
|
};
|
|
|
|
frontend = buildNpmPackage {
|
|
inherit pname version src;
|
|
|
|
npmDepsHash = "sha256-QIgYHZusuq2QD8p8MGsNVhCbz6fR+qP9UuU/kbBkadc=";
|
|
|
|
# Disabling `pyodide:fetch` as it downloads packages during `buildPhase`
|
|
# Until this is solved, running python packages from the browser will not work.
|
|
postPatch = ''
|
|
substituteInPlace package.json \
|
|
--replace-fail "npm run pyodide:fetch && vite build" "vite build" \
|
|
'';
|
|
|
|
env.CYPRESS_INSTALL_BINARY = "0"; # disallow cypress from downloading binaries in sandbox
|
|
|
|
installPhase = ''
|
|
runHook preInstall
|
|
|
|
mkdir -p $out/share
|
|
cp -a build $out/share/open-webui
|
|
|
|
runHook postInstall
|
|
'';
|
|
};
|
|
in
|
|
python3.pkgs.buildPythonApplication rec {
|
|
inherit pname version src;
|
|
pyproject = true;
|
|
|
|
# The custom hook tries to run `npm install` in `buildPhase`.
|
|
# We don't have to worry, as nodejs depedencies are managed by `frontend` drv.
|
|
postPatch = ''
|
|
substituteInPlace pyproject.toml \
|
|
--replace-fail '[tool.hatch.build.hooks.custom]' "" \
|
|
--replace-fail ', build = "open_webui/frontend"' ""
|
|
'';
|
|
|
|
pythonRelaxDeps = true;
|
|
|
|
pythonRemoveDeps = [
|
|
# using `psycopg2` instead
|
|
"psycopg2-binary"
|
|
# using `opencv4`
|
|
"opencv-python-headless"
|
|
# package request: https://github.com/NixOS/nixpkgs/issues/317065
|
|
"rapidocr-onnxruntime"
|
|
# package request: https://github.com/NixOS/nixpkgs/issues/317066
|
|
"langfuse"
|
|
# package request: https://github.com/NixOS/nixpkgs/issues/317068
|
|
"langchain-chroma"
|
|
];
|
|
|
|
dependencies = with python3.pkgs; [
|
|
beautifulsoup4
|
|
fastapi
|
|
uvicorn
|
|
python-multipart
|
|
flask
|
|
flask-cors
|
|
python-socketio
|
|
python-jose
|
|
passlib
|
|
requests
|
|
aiohttp
|
|
peewee
|
|
peewee-migrate
|
|
psycopg2
|
|
pymysql
|
|
bcrypt
|
|
litellm
|
|
boto3
|
|
argon2-cffi
|
|
apscheduler
|
|
google-generativeai
|
|
langchain
|
|
langchain-community
|
|
fake-useragent
|
|
chromadb
|
|
sentence-transformers
|
|
pypdf
|
|
docx2txt
|
|
python-pptx
|
|
unstructured
|
|
markdown
|
|
pypandoc
|
|
pandas
|
|
openpyxl
|
|
pyxlsb
|
|
xlrd
|
|
validators
|
|
opencv4
|
|
fpdf2
|
|
rank-bm25
|
|
faster-whisper
|
|
pyjwt
|
|
black
|
|
youtube-transcript-api
|
|
pytube
|
|
];
|
|
|
|
build-system = with python3.pkgs; [
|
|
hatchling
|
|
pythonRelaxDepsHook
|
|
];
|
|
|
|
pythonImportsCheck = [ "open_webui" ];
|
|
|
|
postInstall = ''
|
|
wrapProgram $out/bin/open-webui \
|
|
--set FRONTEND_BUILD_DIR "${frontend}/share/open-webui"
|
|
'';
|
|
|
|
passthru.tests = {
|
|
inherit (nixosTests) open-webui;
|
|
};
|
|
|
|
meta = {
|
|
description = "Full-stack of open-webui. open-webui is a user-friendly WebUI for LLMs (Formerly Ollama WebUI)";
|
|
homepage = "https://github.com/open-webui/open-webui";
|
|
changelog = "https://github.com/open-webui/open-webui/blob/${src.rev}/CHANGELOG.md";
|
|
license = lib.licenses.mit;
|
|
maintainers = with lib.maintainers; [ shivaraj-bh ];
|
|
mainProgram = "open-webui";
|
|
};
|
|
}
|