jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

feat(chat_manager.py): add base64 encoding for image data in FileResponse objects

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-25 22:15:36 -03:00
commit 70bfc77a82
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/api/chat_manager.py
View file

@ -23,6 +23,7 @@ class ChatHistory(Subject):
"""Add a message to the chat history.""" """Add a message to the chat history."""
self.history[client_id].append(message) self.history[client_id].append(message)
if not isinstance(message, FileResponse): if not isinstance(message, FileResponse):
self.notify() self.notify()
@ -127,6 +128,9 @@ class ChatManager:
# Iterate backwards through the history # Iterate backwards through the history
for msg in reversed(history): for msg in reversed(history):
if isinstance(msg, FileResponse): if isinstance(msg, FileResponse):
if msg.data_type == "image":
# Base64 encode the image
msg.data = pil_to_base64(msg.data)
file_responses.append(msg) file_responses.append(msg)
if msg.type == "start": if msg.type == "start":
break break