Mirrors/kotaemon
1
0
Fork 0
mirror of https://github.com/Cinnamon/kotaemon.git synced 2026-02-23 19:49:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: update custom warning message in simple qa

Browse Source
This commit is contained in:
trducng
2024-12-02 00:37:57 +00:00
parent 7e2ad7af7a
commit 874aa388b7
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
libs/ktem/ktem/reasoning/simple.py
View File

@@ -674,7 +674,7 @@ class FullQAPipeline(BaseReasoning):
)
return with_citation, without_citation
def show_citations(self, answer, docs):
def show_citations(self, answer, docs, warning_message=None):
# show the evidence
with_citation, without_citation = self.prepare_citations(answer, docs)
if not with_citation and not without_citation:
@@ -688,14 +688,17 @@ class FullQAPipeline(BaseReasoning):
yield Document(channel="info", content=None)
# yield warning message
if max_llm_rerank_score < CONTEXT_RELEVANT_WARNING_SCORE:
yield Document(
channel="info",
content=(
"<h5>WARNING! Context relevance score is low. "
"Double check the model answer for correctness.</h5>"
),
)
if warning_message:
yield Document(channel="info", content=warning_message)
else:
if max_llm_rerank_score < CONTEXT_RELEVANT_WARNING_SCORE:
yield Document(
channel="info",
content=(
"<h5>WARNING! Context relevance score is low. "
"Double check the model answer for correctness.</h5>"
),
)
# show QA score
qa_score = (