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-02-13 14:07:36 +08:00
parent e5adb2e0f3
commit cdd52809b0
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lightrag/api/lightrag_server.py
View File

@@ -672,6 +672,7 @@ def get_api_key_dependency(api_key: Optional[str]):
# Global configuration # Global configuration
global_top_k = 60 # default value global_top_k = 60 # default value
def create_app(args): def create_app(args):
global global_top_k global global_top_k
global_top_k = args.top_k # save top_k from args global_top_k = args.top_k # save top_k from args

8
lightrag/utils.py
View File

@@ -416,7 +416,13 @@ async def get_best_cached_response(
if best_similarity > similarity_threshold: if best_similarity > similarity_threshold:
# If LLM check is enabled and all required parameters are provided # If LLM check is enabled and all required parameters are provided
if use_llm_check and llm_func and original_prompt and best_prompt and best_response is not None: if (
use_llm_check
and llm_func
and original_prompt
and best_prompt
and best_response is not None
):
compare_prompt = PROMPTS["similarity_check"].format( compare_prompt = PROMPTS["similarity_check"].format(
original_prompt=original_prompt, cached_prompt=best_prompt original_prompt=original_prompt, cached_prompt=best_prompt
) )