improve qa dataset

76a1d2c2 · jyong · 3b374a46 · 76a1d2c2
Commit 76a1d2c2 authored Jul 27, 2023 by jyong
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

dataset_retriever_tool.py api/core/tool/dataset_retriever_tool.py +1 -1

No files found.
--- a/api/core/tool/dataset_retriever_tool.py
+++ b/api/core/tool/dataset_retriever_tool.py
@@ -110,7 +110,7 @@ class DatasetRetrieverTool(BaseTool):
            if segments:
                for segment in segments:
                    if segment.answer:
-                        document_context_list.append(segment.answer)
+                        document_context_list.append(f'question:{segment.content} \nanswer:{segment.answer}')
                    else:
                        document_context_list.append(segment.content)