add current to text qa prompt

pingcap · Oct 9, 2024 · d02e132 · d02e132
1 parent c4bd854
commit d02e132
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/backend/app/rag/chat.py b/backend/app/rag/chat.py
@@ -367,6 +367,7 @@ def _get_llamaindex_callback_manager():
         callback_manager = _get_llamaindex_callback_manager()
         text_qa_template = get_prompt_by_jinja2_template(
             self.chat_engine_config.llm.text_qa_prompt,
+            current_date=datetime.now().strftime("%Y-%m-%d"),
             graph_knowledges=graph_knowledges_context,
             original_question=self.user_question,
         )

diff --git a/backend/app/rag/default_prompt.py b/backend/app/rag/default_prompt.py
@@ -121,6 +121,7 @@
 
 
 DEFAULT_TEXT_QA_PROMPT = """\
+Current Date: {{current_date}}
 ---------------------
 Knowledge graph information is below
 ---------------------