From 10dbbe4ebf0062c53866c06bf0d1ea1ad23dba56 Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 8 May 2025 04:29:43 +0800 Subject: [PATCH] Fix linting --- lightrag/base.py | 2 +- lightrag/operate.py | 12 ++++++++++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/lightrag/base.py b/lightrag/base.py index e8994bba..c6035f70 100644 --- a/lightrag/base.py +++ b/lightrag/base.py @@ -93,7 +93,7 @@ class QueryParam: This allows using different models for different query modes. """ - user_prompt: str | None= None + user_prompt: str | None = None """User-provided prompt for the query. If proivded, this will be use instead of the default vaulue from prompt template. """ diff --git a/lightrag/operate.py b/lightrag/operate.py index 9ffd6111..3ffca35a 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -926,7 +926,11 @@ async def kg_query( ) # Build system prompt - user_prompt = query_param.user_prompt if query_param.user_prompt else PROMPTS["DEFAULT_USER_PROMPT"] + user_prompt = ( + query_param.user_prompt + if query_param.user_prompt + else PROMPTS["DEFAULT_USER_PROMPT"] + ) sys_prompt_temp = system_prompt if system_prompt else PROMPTS["rag_response"] sys_prompt = sys_prompt_temp.format( context_data=context, @@ -1911,7 +1915,11 @@ async def naive_query( ) # Build system prompt - user_prompt = query_param.user_prompt if query_param.user_prompt else PROMPTS["DEFAULT_USER_PROMPT"] + user_prompt = ( + query_param.user_prompt + if query_param.user_prompt + else PROMPTS["DEFAULT_USER_PROMPT"] + ) sys_prompt_temp = system_prompt if system_prompt else PROMPTS["naive_rag_response"] sys_prompt = sys_prompt_temp.format( content_data=text_units_str,