From d961b5e9290edef3bee5cd6adc4a636af209c1e7 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 09 一月 2025 18:52:21 +0800
Subject: [PATCH] tmp
---
app/models/v2/session_model.py | 22 +++++++++++++++-------
1 files changed, 15 insertions(+), 7 deletions(-)
diff --git a/app/models/v2/session_model.py b/app/models/v2/session_model.py
index aa85f19..fad5bc1 100644
--- a/app/models/v2/session_model.py
+++ b/app/models/v2/session_model.py
@@ -7,7 +7,7 @@
from pydantic import BaseModel
from sqlalchemy import Column, String, Integer, DateTime, JSON, TEXT, Index
-from Log import logger
+# from Log import logger
from app.models.agent_model import AgentType
from app.models.base_model import Base
@@ -35,14 +35,17 @@
reference = Column(TEXT)
conversation_id = Column(String(36), index=True)
event_type = Column(String(16))
+ session_type = Column(String(16))
# to_dict 鏂规硶
def to_dict(self):
return {
- 'id': self.id,
+ 'session_id': self.id,
'name': self.name,
'agent_type': self.agent_type,
'agent_id': self.agent_id,
+ 'event_type': self.event_type,
+ 'session_type': self.session_type,
'create_date': self.create_date.strftime("%Y-%m-%d %H:%M:%S"),
'update_date': self.update_date.strftime("%Y-%m-%d %H:%M:%S"),
}
@@ -71,10 +74,12 @@
-class ChatDialogData(BaseModel):
+class ChatData(BaseModel):
sessionId: Optional[str] = ""
- question: str
- chatId: str
+
+ class Config:
+ extra = 'allow' # 鍏佽鍏朵粬鍔ㄦ�佸瓧娈�
+
class ChatSessionDao:
@@ -98,17 +103,20 @@
session = self.db.query(ChatSessionModel).filter_by(id=session_id).first()
return session
- async def update_session_by_id(self, session_id: str, session, message: dict) -> ChatSessionModel | None:
+ async def update_session_by_id(self, session_id: str, session, message: dict, conversation_id=None) -> ChatSessionModel | None:
+ print(message)
if not session:
session = await self.get_session_by_id(session_id)
if session:
try:
+ if conversation_id:
+ session.conversation_id=conversation_id
session.add_message(message)
session.update_date = current_time()
self.db.commit()
self.db.refresh(session)
except Exception as e:
- logger.error(e)
+ # logger.error(e)
self.db.rollback()
return session
--
Gitblit v1.8.0