From 23c463677015fba7b4b563e5ffc1549b3b411964 Mon Sep 17 00:00:00 2001 From: "junjie.jiang" Date: Wed, 31 May 2023 20:12:34 +0800 Subject: [PATCH] Update to latest prompt Signed-off-by: junjie.jiang --- eqa_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eqa_search.py b/eqa_search.py index f2de589..fc43e2e 100644 --- a/eqa_search.py +++ b/eqa_search.py @@ -121,7 +121,7 @@ def enhanced_qa_search_pipe(config): p = p.map('result', 'result', sim_eval_op) p = ( - p.map('result', 'docs', lambda x:[i[2] for i in x]) + p.map('result', 'docs', lambda x: '\n'.join([i[2] for i in x]) .map(('question', 'docs', 'history'), 'prompt', ops.prompt.question_answer(llm_name=config.llm_src)) .map('prompt', 'answer', llm_op) )