Handle special case of CLI argument openai-ollama

2025-02-22 15:27:19 +08:00
parent e935fed50e
commit d84b90bcd4
1 changed files with 6 additions and 0 deletions
--- 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)
    )