From 9c632dc526dec3c9366ed749e11bf2fc17101ced Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 19 十一月 2024 17:05:12 +0800
Subject: [PATCH] 增加对话表保存basic的agent会话，会话列表支持basic会话查询
---
app/models/session_model.py | 28 ++++++++++++++++++++++++++++
app/api/agent.py | 7 +++++++
2 files changed, 35 insertions(+), 0 deletions(-)
diff --git a/app/api/agent.py b/app/api/agent.py
index 3178144..4e410d4 100644
--- a/app/api/agent.py
+++ b/app/api/agent.py
@@ -10,6 +10,7 @@
from app.config.config import settings
from app.models.agent_model import AgentType, AgentModel
from app.models.base_model import get_db
+from app.models.session_model import SessionModel
from app.models.user_model import UserModel
from app.service.bisheng import BishengService
from app.service.dialog import get_session_history
@@ -57,6 +58,12 @@
raise HTTPException(status_code=500, detail=str(e))
return ResponseList(code=200, msg="", data=result)
+ elif agent.agent_type == AgentType.BASIC:
+ offset = (page - 1) * limit
+ records = db.query(SessionModel).filter(SessionModel.agent_id == agent_id).offset(offset).limit(limit).all()
+ result = [item.to_dict() for item in records]
+ return ResponseList(code=200, msg="", data=result)
+
else:
return ResponseList(code=200, msg="Unsupported agent type")
diff --git a/app/models/session_model.py b/app/models/session_model.py
new file mode 100644
index 0000000..21bfb7e
--- /dev/null
+++ b/app/models/session_model.py
@@ -0,0 +1,28 @@
+import json
+from datetime import datetime
+from enum import IntEnum
+from sqlalchemy import Column, String, Enum as SQLAlchemyEnum, Integer, DateTime
+
+from app.models import AgentType
+from app.models.base_model import Base
+
+
+class SessionModel(Base):
+ __tablename__ = "sessions"
+ id = Column(String(255), primary_key=True)
+ name = Column(String(255))
+ agent_id = Column(String(255))
+ agent_type = Column(SQLAlchemyEnum(AgentType), nullable=False) # 鐩墠鍙瓨basic鐨勶紝ragflow鍜宐isheng鐨勮皟鎺ュ彛鑾峰彇
+ create_date = Column(DateTime) # 鍒涘缓鏃堕棿
+ update_date = Column(DateTime) # 鏇存柊鏃堕棿
+
+ # to_dict 鏂规硶
+ def to_dict(self):
+ return {
+ 'id': self.id,
+ 'name': self.name,
+ 'agent_type': self.agent_type,
+ 'agent_id': self.agent_id,
+ 'create_date': self.create_date,
+ 'update_date': self.update_date,
+ }
--
Gitblit v1.8.0