Add custom function with separate keyword extraction for user's query and a separate prompt

2025-01-14 22:10:47 +05:30
parent 61349b0fe7
commit ef61ffe444
4 changed files with 313 additions and 1 deletions
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -31,6 +31,8 @@ class QueryParam:
    max_token_for_global_context: int = 4000
    # Number of tokens for the entity descriptions
    max_token_for_local_context: int = 4000
+    hl_keywords: list[str] = field(default_factory=list)
+    ll_keywords: list[str] = field(default_factory=list)


@dataclass