gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimize RAG initialization for openai-ollama

Browse Source
This commit is contained in:
yangdx
2025-02-23 11:54:36 +08:00
parent d84b90bcd4
commit 637d6756b8
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/api/lightrag_server.py
View File

@@ -314,7 +314,7 @@ def create_app(args):
) )
# Initialize RAG # Initialize RAG
if args.llm_binding in ["lollms", "ollama", "openai-ollama"]: if args.llm_binding in ["lollms", "ollama", "openai"]:
rag = LightRAG( rag = LightRAG(
working_dir=args.working_dir, working_dir=args.working_dir,
llm_model_func=lollms_model_complete llm_model_func=lollms_model_complete
@@ -353,12 +353,10 @@ def create_app(args):
namespace_prefix=args.namespace_prefix, namespace_prefix=args.namespace_prefix,
auto_manage_storages_states=False, auto_manage_storages_states=False,
) )
else: else: # azure_openai
rag = LightRAG( rag = LightRAG(
working_dir=args.working_dir, working_dir=args.working_dir,
llm_model_func=azure_openai_model_complete llm_model_func=azure_openai_model_complete,
if args.llm_binding == "azure_openai"
else openai_alike_model_complete,
chunk_token_size=int(args.chunk_size), chunk_token_size=int(args.chunk_size),
chunk_overlap_token_size=int(args.chunk_overlap_size), chunk_overlap_token_size=int(args.chunk_overlap_size),
llm_model_kwargs={ llm_model_kwargs={