diff --git a/src/backend/langflow/api/chat_manager.py b/src/backend/langflow/api/chat_manager.py index ca31139ab..8e05127fe 100644 --- a/src/backend/langflow/api/chat_manager.py +++ b/src/backend/langflow/api/chat_manager.py @@ -120,10 +120,20 @@ class ChatManager: raise e # Send a response back to the frontend, if needed intermediate_steps = intermediate_steps or "" + history = self.chat_history.get_history(client_id, filter=False) + file_responses = [] + if history: + for msg in history: + if isinstance(msg, FileResponse): + file_responses.append(msg) + if msg.type == "start": + break + response = ChatResponse( message=result or "", intermediate_steps=intermediate_steps.strip(), type="end", + files=file_responses, ) self.chat_history.add_message(client_id, response) diff --git a/src/backend/langflow/api/schemas.py b/src/backend/langflow/api/schemas.py index c9b210210..e4adacf9e 100644 --- a/src/backend/langflow/api/schemas.py +++ b/src/backend/langflow/api/schemas.py @@ -16,6 +16,7 @@ class ChatResponse(ChatMessage): intermediate_steps: str type: str is_bot: bool = True + files: list = [] @validator("type") def validate_message_type(cls, v):