yangdx commited on
Commit
bf28994
·
1 Parent(s): 2a71867

Fix parameters missing in app creation in case openai is selected

Browse files
Files changed (1) hide show
  1. lightrag/api/lightrag_server.py +3 -0
lightrag/api/lightrag_server.py CHANGED
@@ -715,6 +715,9 @@ def create_app(args):
715
  llm_model_func=azure_openai_model_complete
716
  if args.llm_binding == "azure_openai"
717
  else openai_alike_model_complete,
 
 
 
718
  embedding_func=embedding_func,
719
  )
720
 
 
715
  llm_model_func=azure_openai_model_complete
716
  if args.llm_binding == "azure_openai"
717
  else openai_alike_model_complete,
718
+ llm_model_name=args.llm_model,
719
+ llm_model_max_async=args.max_async,
720
+ llm_model_max_token_size=args.max_tokens,
721
  embedding_func=embedding_func,
722
  )
723