Update lightrag_server.py

2025-01-28 15:03:26 +01:00
parent 80451af839
commit 9fdea743e4
1 changed files with 9 additions and 0 deletions
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -484,8 +484,17 @@ def parse_args() -> argparse.Namespace:
        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()

+    LIGHTRAG_MODEL = args.simulated_model_name
+
    return args