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

Merge remote-tracking branch 'origin/chat_and_cache' into websocket

Browse source
This commit is contained in:
anovazzi1 2023-04-25 16:41:20 -03:00
commit 8932a959c0
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -89,7 +89,7 @@ class ChatManager:
async def send_json(self, client_id: str, message: ChatMessage): async def send_json(self, client_id: str, message: ChatMessage):
websocket = self.active_connections[client_id] websocket = self.active_connections[client_id]
await websocket.send_json(json.dumps(message.dict())) await websocket.send_json(message.dict())
async def process_message(self, client_id: str, payload: Dict): async def process_message(self, client_id: str, payload: Dict):
# Process the graph data and chat message # Process the graph data and chat message
@ -136,11 +136,16 @@ class ChatManager:
try: try:
chat_history = self.chat_history.get_history(client_id) chat_history = self.chat_history.get_history(client_id)
await websocket.send_json(json.dumps(chat_history)) # iterate and make BaseModel into dict
chat_history = [chat.dict() for chat in chat_history]
await websocket.send_json(chat_history)
while True: while True:
json_payload = await websocket.receive_json() json_payload = await websocket.receive_json()
payload = json.loads(json_payload) try:
payload = json.loads(json_payload)
except TypeError:
payload = json_payload
with self.cache_manager.set_client_id(client_id): with self.cache_manager.set_client_id(client_id):
await self.process_message(client_id, payload) await self.process_message(client_id, payload)
except Exception as e: except Exception as e: