yangdx commited on
Commit
60fdd58
Β·
1 Parent(s): 74ae495

Add environment variable for Ollama model tag

Browse files
Files changed (1) hide show
  1. lightrag/api/lightrag_server.py +4 -2
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 = "lightrag:latest"
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="")