| | |
| | | |
| | | class ChatDataRequest(BaseModel): |
| | | sessionId: str |
| | | parentId: Optional[str] = "" |
| | | query: str |
| | | chatMode: Optional[int] = 1 # 1= 普通对话,2=联网,3=知识库,4=深度 |
| | | isDeep: Optional[int] = 1 # 1= 普通, 2=深度 |
| | |
| | | "files": self.files, |
| | | "isDeep": self.isDeep, |
| | | "optimizeType": self.optimizeType, |
| | | "parentId": self.parentId, |
| | | } |
| | | |
| | | |
| | |
| | | 'content': self.content, |
| | | } |
| | | else: |
| | | query = {} |
| | | if self.query: |
| | | query = json.loads(self.query) |
| | | return { |
| | | 'id': self.id, |
| | | 'role': "assistant", |
| | | 'answer': self.content, |
| | | 'chat_mode': self.chat_mode, |
| | | 'node_list': json.loads(self.node_data) if self.node_data else [], |
| | | "parentId": query.get("parentId") |
| | | } |
| | | |
| | | |