From b5a2711c05993609920a1c5befe970b9961c63c0 Mon Sep 17 00:00:00 2001
From: wwwlll <wingjson@gmail.com>
Date: Fri, 6 Sep 2024 15:02:41 +0800
Subject: [PATCH] Fix agent retrieval nothing (#2283)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

### What problem does this PR solve?

Fix agent retrieval nothing

### Type of change

- [✓] Bug Fix (non-breaking change which fixes an issue)
---
 agent/component/retrieval.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/agent/component/retrieval.py b/agent/component/retrieval.py
index d49ecaa8b..dbd087ee8 100644
--- a/agent/component/retrieval.py
+++ b/agent/component/retrieval.py
@@ -54,8 +54,8 @@ class Retrieval(ComponentBase, ABC):
         for role, cnt in history[::-1][:self._param.message_history_window_size]:
             if role != "user":continue
             query.append(cnt)
-        query = "\n".join(query)
-
+        # query = "\n".join(query)
+        query = query[0]
         kbs = KnowledgebaseService.get_by_ids(self._param.kb_ids)
         if not kbs:
             raise ValueError("Can't find knowledgebases by {}".format(self._param.kb_ids))