Add user_prompt the QueryParam

2025-05-08 03:38:47 +08:00
parent 08e532eaf3
commit ae1c9f8d10
3 changed files with 22 additions and 2 deletions
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -93,6 +93,11 @@ class QueryParam:
    This allows using different models for different query modes.
    """

+    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.
+    """
+

@dataclass
 class StorageNameSpace(ABC):