yangdx
commited on
Commit
·
6bc0be1
1
Parent(s):
eadf22c
Revert: get llm-binding and embedding-binding from cli
Browse files- lightrag/api/utils_api.py +16 -4
lightrag/api/utils_api.py
CHANGED
|
@@ -258,6 +258,22 @@ def parse_args() -> argparse.Namespace:
|
|
| 258 |
help="Enable automatic scanning when the program starts",
|
| 259 |
)
|
| 260 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 261 |
args = parser.parse_args()
|
| 262 |
|
| 263 |
# convert relative path to absolute path
|
|
@@ -277,10 +293,6 @@ def parse_args() -> argparse.Namespace:
|
|
| 277 |
args.vector_storage = get_env_value(
|
| 278 |
"LIGHTRAG_VECTOR_STORAGE", DefaultRAGStorageConfig.VECTOR_STORAGE
|
| 279 |
)
|
| 280 |
-
|
| 281 |
-
# Inject binding configuration
|
| 282 |
-
args.llm_binding = get_env_value("LLM_BINDING", "ollama")
|
| 283 |
-
args.embedding_binding = get_env_value("EMBEDDING_BINDING", "ollama")
|
| 284 |
args.llm_binding_host = get_env_value(
|
| 285 |
"LLM_BINDING_HOST", get_default_host(args.llm_binding)
|
| 286 |
)
|
|
|
|
| 258 |
help="Enable automatic scanning when the program starts",
|
| 259 |
)
|
| 260 |
|
| 261 |
+
# LLM and embedding bindings
|
| 262 |
+
parser.add_argument(
|
| 263 |
+
"--llm-binding",
|
| 264 |
+
type=str,
|
| 265 |
+
default=get_env_value("LLM_BINDING", "ollama"),
|
| 266 |
+
choices=["lollms", "ollama", "openai", "openai-ollama", "azure_openai"],
|
| 267 |
+
help="LLM binding type (default: from env or ollama)",
|
| 268 |
+
)
|
| 269 |
+
parser.add_argument(
|
| 270 |
+
"--embedding-binding",
|
| 271 |
+
type=str,
|
| 272 |
+
default=get_env_value("EMBEDDING_BINDING", "ollama"),
|
| 273 |
+
choices=["lollms", "ollama", "openai", "azure_openai"],
|
| 274 |
+
help="Embedding binding type (default: from env or ollama)",
|
| 275 |
+
)
|
| 276 |
+
|
| 277 |
args = parser.parse_args()
|
| 278 |
|
| 279 |
# convert relative path to absolute path
|
|
|
|
| 293 |
args.vector_storage = get_env_value(
|
| 294 |
"LIGHTRAG_VECTOR_STORAGE", DefaultRAGStorageConfig.VECTOR_STORAGE
|
| 295 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 296 |
args.llm_binding_host = get_env_value(
|
| 297 |
"LLM_BINDING_HOST", get_default_host(args.llm_binding)
|
| 298 |
)
|