Ver código fonte

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 meses atrás
pai
commit
7eb417b24f
Nenhuma conta vinculada ao e-mail do autor do commit
2 arquivos alterados com 2 adições e 2 exclusões
  1. 1
    1
      agent/canvas.py
  2. 1
    1
      agent/component/retrieval.py

+ 1
- 1
agent/canvas.py Ver arquivo

pid = cpn["parent_id"] pid = cpn["parent_id"]
_, o = cpn["obj"].output(allow_partial=False) _, o = cpn["obj"].output(allow_partial=False)
_, oo = self.components[pid]["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] downstream = [pid]


for m in prepare2run(downstream): for m in prepare2run(downstream):

+ 1
- 1
agent/component/retrieval.py Ver arquivo

rank_feature=label_question(query, kbs)) rank_feature=label_question(query, kbs))
else: else:
kbinfos = {"chunks": [], "doc_aggs": []} kbinfos = {"chunks": [], "doc_aggs": []}
if self._param.use_kg and kbs: if self._param.use_kg and kbs:
ck = settings.kg_retrievaler.retrieval(query, ck = settings.kg_retrievaler.retrieval(query,
[kbs[0].tenant_id], [kbs[0].tenant_id],

Carregando…
Cancelar
Salvar