From 8d2f8ed5612406c34466c4f863569b027f608e72 Mon Sep 17 00:00:00 2001 From: H <43509927+guoyuhao2330@users.noreply.github.com> Date: Tue, 20 Aug 2024 09:04:23 +0800 Subject: [PATCH] Fix generate param empty_response (#2010) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- agent/component/generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/agent/component/generate.py b/agent/component/generate.py index dab99632a..f7937dbb9 100644 --- a/agent/component/generate.py +++ b/agent/component/generate.py @@ -67,7 +67,8 @@ class Generate(ComponentBase): def set_cite(self, retrieval_res, answer): retrieval_res.dropna(subset=["vector", "content_ltks"], inplace=True) - retrieval_res["empty_response"].fillna("", inplace=True) + if "empty_response" in retrieval_res.columns: + retrieval_res["empty_response"].fillna("", inplace=True) answer, idx = retrievaler.insert_citations(answer, [ck["content_ltks"] for _, ck in retrieval_res.iterrows()], [ck["vector"] for _, ck in retrieval_res.iterrows()], LLMBundle(self._canvas.get_tenant_id(), LLMType.EMBEDDING,