From 4302c65a04935eeaedaa2f842d0b3458975100d4 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Wed, 29 Jan 2025 21:01:35 +0800
Subject: [PATCH] Avoid unnecessary LLM response generation

---
 lightrag/operate.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lightrag/operate.py b/lightrag/operate.py
index 66e4533c..cf096993 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -1025,6 +1025,10 @@ async def _build_query_context(
             [hl_relations_context, ll_relations_context],
             [hl_text_units_context, ll_text_units_context],
         )
+    # not necessary to use LLM to generate a response
+    if not entities_context.strip() and not relations_context.strip():
+        return None
+        
     return f"""
 -----Entities-----
 ```csv