From d84b90bcd446c02e21fc2897bd026478d7b32862 Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 22 Feb 2025 15:27:19 +0800 Subject: [PATCH] Handle special case of CLI argument openai-ollama --- lightrag/api/utils_api.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/lightrag/api/utils_api.py b/lightrag/api/utils_api.py index bbb8303c..e32cdc41 100644 --- a/lightrag/api/utils_api.py +++ b/lightrag/api/utils_api.py @@ -293,6 +293,12 @@ def parse_args() -> argparse.Namespace: args.vector_storage = get_env_value( "LIGHTRAG_VECTOR_STORAGE", DefaultRAGStorageConfig.VECTOR_STORAGE ) + + # Handle openai-ollama special case + if args.llm_binding == "openai-ollama": + args.llm_binding = "openai" + args.embedding_binding = "ollama" + args.llm_binding_host = get_env_value( "LLM_BINDING_HOST", get_default_host(args.llm_binding) )