绘图历史信息

zhaoqingang

2024-11-26 6ae19ce01dbdfdb1080802d854b13adda3982172

绘图历史信息

 app/api/agent.py

@@ -181,6 +181,8 @@
                        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 @@
                                    else:  # 正常输出
                                        answer = data.get("answer", "")
                                        if isinstance(answer, str):

                                            answer_str += answer

                                        elif isinstance(answer, dict):
@@ -356,7 +357,7 @@
                                    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)