首页会话-模型修改

zhaoqingang

2025-04-08 0650b889a36d9b9fd42415b9b9819676f839ae9b

 app/models/v2/chat.py

@@ -46,6 +46,13 @@



class SetModelRequest(BaseModel):
    chatType: int
    modelType: int
    modelName: str
    modelProvider: str



class ComplexChatModel(Base):
    __tablename__ = 'complex_chat'
@@ -64,6 +71,9 @@
    mode = Column(String(36))
    parameters = Column(Text)
    chat_mode = Column(Integer) #1= 普通对话，2=联网，3=知识库,4=深度
    chat_model = Column(String(255)) # 模型
    chat_model_ds = Column(String(255)) # 模型
    chat_provider = Column(String(255)) # 模型提供商

    def to_json(self):
        return {
@@ -99,24 +109,20 @@
        session = self.db.query(ComplexChatModel).filter_by(id=chat_id).first()
        return session

    async def update_complex_chat_by_id(self, chat_id: str, session, message: dict, conversation_id=None) -> ComplexChatModel | None:
        if not session:
            session = await self.get_complex_chat_by_id(chat_id)
        if session:
    async def update_complex_chat_by_id(self, chat_id: str, kwargs:dict) -> None:

            try:
                # TODO
                session.update_date = current_time()
            self.db.query(ComplexChatModel).filter_by(id=chat_id).update(kwargs)
                self.db.commit()
                self.db.refresh(session)
            except Exception as e:
                # logger.error(e)
                self.db.rollback()
        return session


    async def update_or_insert_by_id(self, chat_id: str, **kwargs) -> ComplexChatModel:
        existing_session = await self.get_complex_chat_by_id(chat_id)
        if existing_session:
            return await self.update_complex_chat_by_id(chat_id, existing_session, kwargs.get("message"))
            return await self.update_complex_chat_by_id(chat_id, kwargs)

        existing_session = await self.create_complex_chat(chat_id, **kwargs)
        return existing_session
@@ -142,6 +148,10 @@
        return session


    async def aget_complex_chat(self) -> List:
        return self.db.query(ComplexChatModel).filter(ComplexChatModel.status!=Dialog_STATSU_DELETE).all()



class ComplexChatSessionModel(Base):
    __tablename__ = "complex_chat_sessions"