diff --git a/libs/langchain/langchain/agents/openai_assistant/base.py b/libs/langchain/langchain/agents/openai_assistant/base.py index acc681f68f929..9f0ad64fe3a6f 100644 --- a/libs/langchain/langchain/agents/openai_assistant/base.py +++ b/libs/langchain/langchain/agents/openai_assistant/base.py @@ -584,6 +584,9 @@ def _get_response(self, run: Any) -> Any: answer: Any = [ msg_content for msg in new_messages for msg_content in msg.content ] + attachments = [ + attachment for msg in new_messages for attachment in msg.attachments + ] if all( ( isinstance(content, openai.types.beta.threads.TextContentBlock) @@ -601,6 +604,7 @@ def _get_response(self, run: Any) -> Any: "output": answer, "thread_id": run.thread_id, "run_id": run.id, + "attachments": attachments, }, log="", run_id=run.id,