From 21c185403a70b0eff0abe7d47ebbe0deffbee2d5 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Sat, 29 Mar 2025 12:02:14 -0700 Subject: [PATCH] work with all types --- .../inline/agents/meta_reference/agent_instance.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index df6c9bb36..a977b0c33 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -481,7 +481,17 @@ class ChatAgent(ShieldRunnerMixin): contexts.append(raw_document_text) attached_context = "\n".join(contexts) - input_messages[-1].content += attached_context + if isinstance(input_messages[-1].content, str): + input_messages[-1].content += attached_context + elif isinstance(input_messages[-1].content, list): + input_messages[-1].content.append( + TextContentItem(text=attached_context) + ) + else: + input_messages[-1].content = [ + input_messages[-1].content, + TextContentItem(text=attached_context), + ] session_info = await self.storage.get_session_info(session_id) # if the session has a memory bank id, let the memory tool use it