From 66dcc551ef3d390ec272a6896b85618f8fac5ec7 Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 25 Jan 2025 16:29:59 +0800 Subject: [PATCH] Fix LLM binding variable name in create_app function Corrected variable name from llm_binding_host Updated conditional checks for LLM bindings --- lightrag/api/lightrag_server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 87a72a98..11c333ec 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -719,14 +719,14 @@ def create_app(args): # Create working directory if it doesn't exist Path(args.working_dir).mkdir(parents=True, exist_ok=True) - if args.llm_binding_host == "lollms" or args.embedding_binding == "lollms": + if args.llm_binding == "lollms" or args.embedding_binding == "lollms": from lightrag.llm.lollms import lollms_model_complete, lollms_embed - if args.llm_binding_host == "ollama" or args.embedding_binding == "ollama": + if args.llm_binding == "ollama" or args.embedding_binding == "ollama": from lightrag.llm.ollama import ollama_model_complete, ollama_embed - if args.llm_binding_host == "openai" or args.embedding_binding == "openai": + if args.llm_binding == "openai" or args.embedding_binding == "openai": from lightrag.llm.openai import openai_complete_if_cache, openai_embed if ( - args.llm_binding_host == "azure_openai" + args.llm_binding == "azure_openai" or args.embedding_binding == "azure_openai" ): from lightrag.llm.azure_openai import (