From 4ab1deaf25e12b21b09d14e93bbf92c349a9e958 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Tue, 28 Jan 2025 15:32:41 +0100
Subject: [PATCH] Fixed --simulated-model-name argument

---
 lightrag/api/lightrag_server.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index cf556f04..56156eb7 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -489,8 +489,8 @@ def parse_args() -> argparse.Namespace:
 
     parser.add_argument(
         "--simulated-model-name",
-        type=int,
-        default=get_env_value("SIMULATED_MODEL_NAME", ollama_server_infos.LIGHTRAG_MODEL, str),
+        type=str,
+        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)",
     )