zhaoqingang
2025-03-13 30ff0afd5d76a3a5aa48058210ae411253574ada
app/models/v2/chat.py
@@ -23,6 +23,7 @@
class ChatDataRequest(BaseModel):
    sessionId: str
    parentId: Optional[str] = ""
    query: str
    chatMode: Optional[int] = 1  # 1= 普通对话,2=联网,3=知识库,4=深度
    isDeep: Optional[int] = 1  # 1= 普通, 2=深度
@@ -40,6 +41,7 @@
            "files": self.files,
            "isDeep": self.isDeep,
            "optimizeType": self.optimizeType,
            "parentId": self.parentId,
        }
@@ -182,11 +184,16 @@
                '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")
            }