From 0650b889a36d9b9fd42415b9b9819676f839ae9b Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 08 四月 2025 09:54:09 +0800
Subject: [PATCH] 首页会话-模型修改
---
app/models/v2/chat.py | 38 ++++++++++++++++++++++++--------------
1 files changed, 24 insertions(+), 14 deletions(-)
diff --git a/app/models/v2/chat.py b/app/models/v2/chat.py
index 1ee30d2..2f3a484 100644
--- a/app/models/v2/chat.py
+++ b/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:
- try:
- # TODO
- session.update_date = current_time()
- self.db.commit()
- self.db.refresh(session)
- except Exception as e:
- # logger.error(e)
- self.db.rollback()
- return session
+ async def update_complex_chat_by_id(self, chat_id: str, kwargs:dict) -> None:
+
+ try:
+ self.db.query(ComplexChatModel).filter_by(id=chat_id).update(kwargs)
+ self.db.commit()
+ except Exception as e:
+ # logger.error(e)
+ self.db.rollback()
+
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"
--
Gitblit v1.8.0