This commit is contained in:
Saifeddine ALOUI
2025-01-28 18:20:45 +01:00
parent 4ab1deaf25
commit b5d09725f5

View File

@@ -47,6 +47,7 @@ def estimate_tokens(text: str) -> int:
return int(tokens)
class OllamaServerInfos:
# Constants for emulated Ollama model information
LIGHTRAG_NAME = "lightrag"
@@ -61,6 +62,7 @@ class OllamaServerInfos:
GRAPH_STORAGE = "NetworkXStorage"
VECTOR_STORAGE = "NanoVectorDBStorage"
# Add infos
ollama_server_infos = OllamaServerInfos()
@@ -490,7 +492,9 @@ def parse_args() -> argparse.Namespace:
parser.add_argument(
"--simulated-model-name",
type=str,
default=get_env_value("SIMULATED_MODEL_NAME", ollama_server_infos.LIGHTRAG_MODEL),
default=get_env_value(
"SIMULATED_MODEL_NAME", ollama_server_infos.LIGHTRAG_MODEL
),
help="Number of conversation history turns to include (default: from env or 3)",
)