yangdx commited on
Commit
6bc0be1
·
1 Parent(s): eadf22c

Revert: get llm-binding and embedding-binding from cli

Browse files
Files changed (1) hide show
  1. lightrag/api/utils_api.py +16 -4
lightrag/api/utils_api.py CHANGED
@@ -258,6 +258,22 @@ def parse_args() -> argparse.Namespace:
258
  help="Enable automatic scanning when the program starts",
259
  )
260
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
261
  args = parser.parse_args()
262
 
263
  # convert relative path to absolute path
@@ -277,10 +293,6 @@ def parse_args() -> argparse.Namespace:
277
  args.vector_storage = get_env_value(
278
  "LIGHTRAG_VECTOR_STORAGE", DefaultRAGStorageConfig.VECTOR_STORAGE
279
  )
280
-
281
- # Inject binding configuration
282
- args.llm_binding = get_env_value("LLM_BINDING", "ollama")
283
- args.embedding_binding = get_env_value("EMBEDDING_BINDING", "ollama")
284
  args.llm_binding_host = get_env_value(
285
  "LLM_BINDING_HOST", get_default_host(args.llm_binding)
286
  )
 
258
  help="Enable automatic scanning when the program starts",
259
  )
260
 
261
+ # LLM and embedding bindings
262
+ parser.add_argument(
263
+ "--llm-binding",
264
+ type=str,
265
+ default=get_env_value("LLM_BINDING", "ollama"),
266
+ choices=["lollms", "ollama", "openai", "openai-ollama", "azure_openai"],
267
+ help="LLM binding type (default: from env or ollama)",
268
+ )
269
+ parser.add_argument(
270
+ "--embedding-binding",
271
+ type=str,
272
+ default=get_env_value("EMBEDDING_BINDING", "ollama"),
273
+ choices=["lollms", "ollama", "openai", "azure_openai"],
274
+ help="Embedding binding type (default: from env or ollama)",
275
+ )
276
+
277
  args = parser.parse_args()
278
 
279
  # convert relative path to absolute path
 
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
  )