gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
Show stats 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(
"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)
)