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,