H
commited on
Commit
·
ce5e150
1
Parent(s):
fc6e68a
Fix generate param empty_response (#2010)
Browse files### What problem does this PR solve?
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
agent/component/generate.py
CHANGED
@@ -67,7 +67,8 @@ class Generate(ComponentBase):
|
|
67 |
|
68 |
def set_cite(self, retrieval_res, answer):
|
69 |
retrieval_res.dropna(subset=["vector", "content_ltks"], inplace=True)
|
70 |
-
|
|
|
71 |
answer, idx = retrievaler.insert_citations(answer, [ck["content_ltks"] for _, ck in retrieval_res.iterrows()],
|
72 |
[ck["vector"] for _, ck in retrieval_res.iterrows()],
|
73 |
LLMBundle(self._canvas.get_tenant_id(), LLMType.EMBEDDING,
|
|
|
67 |
|
68 |
def set_cite(self, retrieval_res, answer):
|
69 |
retrieval_res.dropna(subset=["vector", "content_ltks"], inplace=True)
|
70 |
+
if "empty_response" in retrieval_res.columns:
|
71 |
+
retrieval_res["empty_response"].fillna("", inplace=True)
|
72 |
answer, idx = retrievaler.insert_citations(answer, [ck["content_ltks"] for _, ck in retrieval_res.iterrows()],
|
73 |
[ck["vector"] for _, ck in retrieval_res.iterrows()],
|
74 |
LLMBundle(self._canvas.get_tenant_id(), LLMType.EMBEDDING,
|