Fix: retrieval debug mode issue. (#6150)

### What problem does this PR solve?

#6139

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-03-17 13:07:13 +08:00
committed by GitHub
parent 37f3486483
commit bfa8d342b3
4 changed files with 4 additions and 4 deletions

View File

@ -57,8 +57,7 @@ class Retrieval(ComponentBase, ABC):
query = self.get_input()
query = str(query["content"][0]) if "content" in query else ""
lines = query.split('\n')
user_queries = [line.split("USER:", 1)[1] for line in lines if line.startswith("USER:")]
query = user_queries[-1] if user_queries else ""
query = lines[-1] if lines else ""
kbs = KnowledgebaseService.get_by_ids(self._param.kb_ids)
if not kbs:
return Retrieval.be_output("")