diff --git a/app/chain/__init__.py b/app/chain/__init__.py index d0be3c4c..1774e757 100644 --- a/app/chain/__init__.py +++ b/app/chain/__init__.py @@ -486,7 +486,7 @@ class ChainBase(metaclass=ABCMeta): # 发送事件 self.eventmanager.send_event(etype=EventType.NoticeMessage, data=message.dict()) # 保存消息 - self.messagehelper.put(message, role="user") + self.messagehelper.put(message, role="user", title=message.title) self.messageoper.add(**message.dict()) # 发送 self.run_module("post_message", message=message) @@ -499,7 +499,7 @@ class ChainBase(metaclass=ABCMeta): :return: 成功或失败 """ note_list = [media.to_dict() for media in medias] - self.messagehelper.put(message, role="user", note=note_list) + self.messagehelper.put(message, role="user", note=note_list, title=message.title) self.messageoper.add(**message.dict(), note=note_list) return self.run_module("post_medias_message", message=message, medias=medias) @@ -511,7 +511,7 @@ class ChainBase(metaclass=ABCMeta): :return: 成功或失败 """ note_list = [torrent.torrent_info.to_dict() for torrent in torrents] - self.messagehelper.put(message, role="user", note=note_list) + self.messagehelper.put(message, role="user", note=note_list, title=message.title) self.messageoper.add(**message.dict(), note=note_list) return self.run_module("post_torrents_message", message=message, torrents=torrents) diff --git a/app/db/__init__.py b/app/db/__init__.py index e90642fa..2ecd065c 100644 --- a/app/db/__init__.py +++ b/app/db/__init__.py @@ -147,6 +147,10 @@ class Base: @db_update def create(self, db: Session): + # 如果是对象,转换为json字符串 + for key, value in self.__dict__.items(): + if ObjectUtils.is_obj(value): + setattr(self, key, json.dumps(value)) db.add(self) @classmethod