Timothy Jaeryang Baek 3 miesięcy temu
rodzic
commit
ad93341b74
3 zmienionych plików z 22 dodań i 0 usunięć
  1. 19 0
      backend/open_webui/__init__.py
  2. 2 0
      backend/requirements.txt
  3. 1 0
      pyproject.toml

+ 19 - 0
backend/open_webui/__init__.py

@@ -5,12 +5,31 @@ from pathlib import Path
 
 import typer
 import uvicorn
+from typing import Optional
+from typing_extensions import Annotated
 
 app = typer.Typer()
 
 KEY_FILE = Path.cwd() / ".webui_secret_key"
 
 
+def version_callback(value: bool):
+    if value:
+        from open_webui.env import VERSION
+
+        typer.echo(f"Open WebUI version: {VERSION}")
+        raise typer.Exit()
+
+
+@app.command()
+def main(
+    version: Annotated[
+        Optional[bool], typer.Option("--version", callback=version_callback)
+    ] = None,
+):
+    pass
+
+
 @app.command()
 def serve(
     host: str = "0.0.0.0",

+ 2 - 0
backend/requirements.txt

@@ -47,6 +47,8 @@ pymilvus==2.5.0
 qdrant-client~=1.12.0
 opensearch-py==2.7.1
 
+
+transformers
 sentence-transformers==3.3.1
 colbert-ai==0.2.21
 einops==0.8.0

+ 1 - 0
pyproject.toml

@@ -54,6 +54,7 @@ dependencies = [
     "qdrant-client~=1.12.0",
     "opensearch-py==2.7.1",
 
+    "transformers",
     "sentence-transformers==3.3.1",
     "colbert-ai==0.2.21",
     "einops==0.8.0",