gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

linting

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-01-28 18:20:45 +01:00
parent 4ab1deaf25
commit b5d09725f5
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/api/lightrag_server.py
View File

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