From 96ac6477affdb75f5bf223a46b222b13475d4903 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 22 十一月 2024 11:06:04 +0800
Subject: [PATCH] 会话保存问题优化

---
 app/service/session.py |    5 +++--
 1 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/app/service/session.py b/app/service/session.py
index 7e43223..e43d8aa 100644
--- a/app/service/session.py
+++ b/app/service/session.py
@@ -1,3 +1,4 @@
+import json
 from typing import Type
 
 from sqlalchemy.orm import Session
@@ -39,7 +40,7 @@
             agent_id=agent_id,
             agent_type=agent_type,
             tenant_id=user_id,
-            message=[{"role": "user", "content": name}]
+            message=json.dumps([{"role": "user", "content": name}])
         )
         self.db.add(new_session)
         self.db.commit()
@@ -57,7 +58,7 @@
             SessionModel: 鏌ユ壘鍒扮殑浼氳瘽妯″瀷瀹炰緥锛屽鏋滄湭鎵惧埌鍒欒繑鍥濶one銆�
         """
         session = self.db.query(SessionModel).filter_by(id=session_id).first()
-        if session.message is None:
+        if  session and session.message is None:
             session.message = '[]'
         return session
 

--
Gitblit v1.8.0