| | |
| | | } |
| | | |
| | | def log_to_json(self): |
| | | query = {} |
| | | if self.query: |
| | | query = json.loads(self.query) |
| | | if self.message_type == 1: |
| | | return { |
| | | 'id': self.id, |
| | | 'role': "user", |
| | | 'content': self.content, |
| | | 'files': query.get("files", []), |
| | | } |
| | | else: |
| | | query = {} |
| | | if self.query: |
| | | query = json.loads(self.query) |
| | | |
| | | res = { |
| | | 'id': self.id, |
| | |
| | | 'chat_mode': self.chat_mode, |
| | | "parentId": query.get("parentId"), |
| | | "isDeep": query.get("isDeep", 1), |
| | | "mindmap": True if self.mindmap else False, |
| | | } |
| | | if self.chat_mode == complex_knowledge_chat: |
| | | res['reference'] = json.loads(self.node_data) if self.node_data else {} |