yangdx commited on
Commit
707baaf
·
1 Parent(s): 9f31afd

Handle special case of CLI argument openai-ollama

Browse files
Files changed (1) hide show
  1. lightrag/api/utils_api.py +6 -0
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
  )