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

fix linting

Browse Source
This commit is contained in:
yangdx
2025-01-25 16:57:47 +08:00
parent 66dcc551ef
commit cf2f83377c
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lightrag/api/lightrag_server.py
View File

@@ -725,10 +725,7 @@ def create_app(args):
from lightrag.llm.ollama import ollama_model_complete, ollama_embed from lightrag.llm.ollama import ollama_model_complete, ollama_embed
if args.llm_binding == "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 from lightrag.llm.openai import openai_complete_if_cache, openai_embed
if ( if args.llm_binding == "azure_openai" or args.embedding_binding == "azure_openai":
args.llm_binding == "azure_openai"
or args.embedding_binding == "azure_openai"
):
from lightrag.llm.azure_openai import ( from lightrag.llm.azure_openai import (
azure_openai_complete_if_cache, azure_openai_complete_if_cache,
azure_openai_embed, azure_openai_embed,

4
lightrag/operate.py
View File

@@ -1543,7 +1543,9 @@ async def naive_query(
sys_prompt_temp = PROMPTS["naive_rag_response"] sys_prompt_temp = PROMPTS["naive_rag_response"]
sys_prompt = sys_prompt_temp.format( sys_prompt = sys_prompt_temp.format(
content_data=section, response_type=query_param.response_type, history=history_context content_data=section,
response_type=query_param.response_type,
history=history_context,
) )
if query_param.only_need_prompt: if query_param.only_need_prompt: