tmp

zhaoqingang

2025-01-09 d961b5e9290edef3bee5cd6adc4a636af209c1e7

 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