yangdx
commited on
Commit
Β·
60fdd58
1
Parent(s):
74ae495
Add environment variable for Ollama model tag
Browse files
lightrag/api/lightrag_server.py
CHANGED
@@ -51,8 +51,8 @@ def estimate_tokens(text: str) -> int:
|
|
51 |
|
52 |
# Constants for emulated Ollama model information
|
53 |
LIGHTRAG_NAME = "lightrag"
|
54 |
-
LIGHTRAG_TAG = "latest"
|
55 |
-
LIGHTRAG_MODEL = "
|
56 |
LIGHTRAG_SIZE = 7365960935 # it's a dummy value
|
57 |
LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z"
|
58 |
LIGHTRAG_DIGEST = "sha256:lightrag"
|
@@ -161,6 +161,8 @@ def display_splash_screen(args: argparse.Namespace) -> None:
|
|
161 |
|
162 |
# System Configuration
|
163 |
ASCIIColors.magenta("\nπ οΈ System Configuration:")
|
|
|
|
|
164 |
ASCIIColors.white(" ββ Log Level: ", end="")
|
165 |
ASCIIColors.yellow(f"{args.log_level}")
|
166 |
ASCIIColors.white(" ββ Timeout: ", end="")
|
|
|
51 |
|
52 |
# Constants for emulated Ollama model information
|
53 |
LIGHTRAG_NAME = "lightrag"
|
54 |
+
LIGHTRAG_TAG = os.getenv("OLLAMA_EMULATING_MODEL_TAG", "latest")
|
55 |
+
LIGHTRAG_MODEL = f"{LIGHTRAG_NAME}:{LIGHTRAG_TAG}"
|
56 |
LIGHTRAG_SIZE = 7365960935 # it's a dummy value
|
57 |
LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z"
|
58 |
LIGHTRAG_DIGEST = "sha256:lightrag"
|
|
|
161 |
|
162 |
# System Configuration
|
163 |
ASCIIColors.magenta("\nπ οΈ System Configuration:")
|
164 |
+
ASCIIColors.white(" ββ Ollama Emulating Model: ", end="")
|
165 |
+
ASCIIColors.yellow(f"{LIGHTRAG_MODEL}")
|
166 |
ASCIIColors.white(" ββ Log Level: ", end="")
|
167 |
ASCIIColors.yellow(f"{args.log_level}")
|
168 |
ASCIIColors.white(" ββ Timeout: ", end="")
|