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: minor fix

Browse Source
This commit is contained in:
trducng
2024-12-03 02:42:57 +00:00
parent a5d87f9e58
commit 455211f163
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -717,12 +717,15 @@ class FullQAPipeline(BaseReasoning):
def find_final_answer(self, docs):
final_answer = None
normal_docs = []
for doc in docs:
if doc.metadata.get("type") in FINAL_ANSWER_CONTENT_TYPES:
final_answer = doc
break
else:
normal_docs.append(doc)
return final_answer
return final_answer, normal_docs
async def ainvoke( # type: ignore
self, message: str, conv_id: str, history: list, **kwargs # type: ignore
@@ -782,7 +785,7 @@ class FullQAPipeline(BaseReasoning):
yield from infos
# check if evidences contain final answer
final_answer = self.find_final_answer(docs)
final_answer, docs = self.find_final_answer(docs)
if final_answer:
yield Document(
channel="chat",