From 8a05ff459f48070c5f35f8dff1f7ee2843edc300 Mon Sep 17 00:00:00 2001 From: yangdx Date: Tue, 21 Jan 2025 03:13:13 +0800 Subject: [PATCH] Add environment variable for Ollama model tag --- lightrag/api/lightrag_server.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index fd656be0..30423d0e 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -51,8 +51,8 @@ def estimate_tokens(text: str) -> int: # Constants for emulated Ollama model information LIGHTRAG_NAME = "lightrag" -LIGHTRAG_TAG = "latest" -LIGHTRAG_MODEL = "lightrag:latest" +LIGHTRAG_TAG = os.getenv("OLLAMA_EMULATING_MODEL_TAG", "latest") +LIGHTRAG_MODEL = f"{LIGHTRAG_NAME}:{LIGHTRAG_TAG}" LIGHTRAG_SIZE = 7365960935 # it's a dummy value LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z" LIGHTRAG_DIGEST = "sha256:lightrag" @@ -161,6 +161,8 @@ def display_splash_screen(args: argparse.Namespace) -> None: # System Configuration ASCIIColors.magenta("\nšŸ› ļø System Configuration:") + ASCIIColors.white(" ā”œā”€ Ollama Emulating Model: ", end="") + ASCIIColors.yellow(f"{LIGHTRAG_MODEL}") ASCIIColors.white(" ā”œā”€ Log Level: ", end="") ASCIIColors.yellow(f"{args.log_level}") ASCIIColors.white(" ā”œā”€ Timeout: ", end="")