AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-05-31 10:35:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix graph bug about second retrieval (#1688)

Browse Source 
### What problem does this PR solve?

#1651

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-07-24 14:10:45 +08:00 committed by GitHub
parent d96348eb22
commit fdfa5d0ad4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
graph/component/base.py
View File

@ -470,7 +470,10 @@ class ComponentBase(ABC):
break
if upstream_outs:
return pd.concat(upstream_outs, ignore_index=False)
df = pd.concat(upstream_outs, ignore_index=True)
if "content" in df:
df = df.drop_duplicates(subset=['content']).reset_index(drop=True)
return df
return pd.DataFrame()
def get_stream_input(self):