From 43d7759dcb21be5bff65d6982069b74042617347 Mon Sep 17 00:00:00 2001
From: zrguo <49157727+LarFii@users.noreply.github.com>
Date: Wed, 23 Oct 2024 11:50:29 +0800
Subject: [PATCH] Update base.py

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

diff --git a/lightrag/base.py b/lightrag/base.py
index 50be4f62..cecd5edd 100644
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -18,9 +18,13 @@ class QueryParam:
     mode: Literal["local", "global", "hybrid", "naive"] = "global"
     only_need_context: bool = False
     response_type: str = "Multiple Paragraphs"
+    # Number of top-k items to retrieve; corresponds to entities in "local" mode and relationships in "global" mode.
     top_k: int = 60
+    # Number of tokens for the original chunks.
     max_token_for_text_unit: int = 4000
+    # Number of tokens for the relationship descriptions
     max_token_for_global_context: int = 4000
+    # Number of tokens for the entity descriptions
     max_token_for_local_context: int = 4000