From d30c56a8c1c9b9aeea6c76389c43e71f362bf0b9 Mon Sep 17 00:00:00 2001
From: Georgi Stefanov <98421770+georgi-stefanov@users.noreply.github.com>
Date: Fri, 25 Apr 2025 04:39:11 +0300
Subject: [PATCH] langchain: return attachments in _get_response (#30853)

This is a PR to return the message attachments in _get_response, as when
files are generated these attachments are not returned thus generated
files cannot be retrieved

Fixes issue: https://github.com/langchain-ai/langchain/issues/30851
---
 libs/langchain/langchain/agents/openai_assistant/base.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libs/langchain/langchain/agents/openai_assistant/base.py b/libs/langchain/langchain/agents/openai_assistant/base.py
index 363a6da86e0..5ca00c3c684 100644
--- a/libs/langchain/langchain/agents/openai_assistant/base.py
+++ b/libs/langchain/langchain/agents/openai_assistant/base.py
@@ -584,6 +584,9 @@ class OpenAIAssistantRunnable(RunnableSerializable[dict, OutputType]):
             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 @@ class OpenAIAssistantRunnable(RunnableSerializable[dict, OutputType]):
                     "output": answer,
                     "thread_id": run.thread_id,
                     "run_id": run.id,
+                    "attachments": attachments,
                 },
                 log="",
                 run_id=run.id,