gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle special case of CLI argument openai-ollama

Browse Source
This commit is contained in:
yangdx
2025-02-22 15:27:19 +08:00
parent e935fed50e
commit d84b90bcd4
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/api/utils_api.py
View File

@@ -293,6 +293,12 @@ def parse_args() -> argparse.Namespace:
args.vector_storage = get_env_value( args.vector_storage = get_env_value(
"LIGHTRAG_VECTOR_STORAGE", DefaultRAGStorageConfig.VECTOR_STORAGE "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( args.llm_binding_host = get_env_value(
"LLM_BINDING_HOST", get_default_host(args.llm_binding) "LLM_BINDING_HOST", get_default_host(args.llm_binding)
) )