yangdx
commited on
Commit
·
707baaf
1
Parent(s):
9f31afd
Handle special case of CLI argument openai-ollama
Browse files
lightrag/api/utils_api.py
CHANGED
@@ -293,6 +293,12 @@ def parse_args() -> argparse.Namespace:
|
|
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 |
)
|
|
|
293 |
args.vector_storage = get_env_value(
|
294 |
"LIGHTRAG_VECTOR_STORAGE", DefaultRAGStorageConfig.VECTOR_STORAGE
|
295 |
)
|
296 |
+
|
297 |
+
# Handle openai-ollama special case
|
298 |
+
if args.llm_binding == "openai-ollama":
|
299 |
+
args.llm_binding = "openai"
|
300 |
+
args.embedding_binding = "ollama"
|
301 |
+
|
302 |
args.llm_binding_host = get_env_value(
|
303 |
"LLM_BINDING_HOST", get_default_host(args.llm_binding)
|
304 |
)
|