绘图历史信息

app/api/agent.py

@@ -181,6 +181,8 @@ async def session_log(agent_id: str, conversation_id: str, db: Session = Depends
                         if "file_name" in i.get("content", {}):
                             tmp_data["files"] = [{"file_name": i.get("content", {}).get("file_name"),
                                                   "file_url": i.get("content", {}).get("file_url")}]
+                        if "images" in i.get("content", {}):
+                            tmp_data["images"] = i.get("content", {}).get("images")
                     else:
                         tmp_data["answer"] = i.get("content")
                     data.append(tmp_data)

app/api/chat.py

@@ -339,6 +339,7 @@ async def handle_client(websocket: WebSocket,
                                     else:  # 正常输出
                                         answer = data.get("answer", "")
                                         if isinstance(answer, str):
+
                                             answer_str += answer
 
                                         elif isinstance(answer, dict):
@@ -356,7 +357,7 @@ async def handle_client(websocket: WebSocket,
                                     result = {"message": answer_str, "type": "close", "message_files": message_files}
                                     try:
                                         SessionService(db).update_session(chat_id,
-                                                                          message={"role": "assistant", "content": answer_str},conversation_id=data.get("conversation_id"))
+                                                                          message={"role": "assistant", "content": {"answer":answer_str, "images":[i.get("url") for i in message_files]}},conversation_id=data.get("conversation_id"))
                                     except Exception as e:
                                         logger.error("保存dify的会话异常！")
                                         logger.error(e)