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

Update lightrag_server.py

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-01-28 15:03:26 +01:00
committed by GitHub
parent 80451af839
commit 9fdea743e4
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/api/lightrag_server.py
View File

@@ -484,8 +484,17 @@ def parse_args() -> argparse.Namespace:
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)",
) )
parser.add_argument(
"--simulated-model-name",
type=int,
default=get_env_value("SIMULATED_MODEL_NAME", LIGHTRAG_MODEL, str),
help="Number of conversation history turns to include (default: from env or 3)",
)
args = parser.parse_args() args = parser.parse_args()
LIGHTRAG_MODEL = args.simulated_model_name
return args return args