diff --git a/agent/component/keyword.py b/agent/component/keyword.py index d7e668ae6..4a0b54f28 100644 --- a/agent/component/keyword.py +++ b/agent/component/keyword.py @@ -51,7 +51,11 @@ class KeywordExtract(Generate, ABC): def _run(self, history, **kwargs): query = self.get_input() - query = str(query["content"][0]) if "content" in query else "" + if hasattr(query, "to_dict") and "content" in query: + query = ", ".join(map(str, query["content"].dropna())) + else: + query = str(query) + chat_mdl = LLMBundle(self._canvas.get_tenant_id(), LLMType.CHAT, self._param.llm_id) ans = chat_mdl.chat(self._param.get_prompt(), [{"role": "user", "content": query}],