소스 검색

Fix: Nan issue. (#6242)

### What problem does this PR solve?

#6065

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.18.0
Kevin Hu 7 달 전
부모
커밋
7eb417b24f
No account linked to committer's email address
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 1
    1
      agent/canvas.py
  2. 1
    1
      agent/component/retrieval.py

+ 1
- 1
agent/canvas.py 파일 보기

@@ -265,7 +265,7 @@ class Canvas:
pid = cpn["parent_id"]
_, o = cpn["obj"].output(allow_partial=False)
_, oo = self.components[pid]["obj"].output(allow_partial=False)
self.components[pid]["obj"].set_output(pd.concat([oo.dropna(axis=1), o.dropna(axis=1)], ignore_index=True))
self.components[pid]["obj"].set_output(pd.concat([oo.dropna(axis=1), o.dropna(axis=1)], ignore_index=True).dropna())
downstream = [pid]

for m in prepare2run(downstream):

+ 1
- 1
agent/component/retrieval.py 파일 보기

@@ -82,7 +82,7 @@ class Retrieval(ComponentBase, ABC):
rank_feature=label_question(query, kbs))
else:
kbinfos = {"chunks": [], "doc_aggs": []}
if self._param.use_kg and kbs:
ck = settings.kg_retrievaler.retrieval(query,
[kbs[0].tenant_id],

Loading…
취소
저장