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)
This commit is contained in:
parent
73a03287a5
commit
8d2f8ed561
1 changed files with 2 additions and 1 deletions
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue