feat:增加查询用户历史消息功能

services/agent_service.py

@@ -2,7 +2,7 @@ from langchain_core.messages import HumanMessage, AIMessageChunk, AIMessage
 
 from agent import agent, checkpointer, title_agent
 from model import ChatRequest
-from services.db_service import exists_session, save_session
+from services.db_service import exists_session, save_session, query_session_messages
 from utils import dicts_to_messages
 
 
@@ -78,3 +78,10 @@ def get_messages(thread_id: str) -> list[dict[str, str]]:
             result.append({"role": "ai", "content": msg.content})
 
     return result
+
+
+# 查询历史会话消息
+def get_user_session_messages(username: str):
+    """查询历史会话消息"""
+    print(f"查询历史会话消息，username: {username}")
+    return query_session_messages(username)

services/db_service.py

@@ -1,5 +1,8 @@
 import sqlite3
 import os
+from typing import List
+
+from model import SessionMessageResponse
 
 
 def init_db():
@@ -20,6 +23,34 @@ def init_db():
     conn.close()
 
 
+def query_session_messages(username: str) -> List[SessionMessageResponse]:
+    conn = sqlite3.connect(os.getenv("SQLITE_DB_PATH"))
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+
+    cursor.execute(
+        """
+        SELECT username, title, thread_id
+        FROM session
+        WHERE username = ?
+        ORDER BY created_time DESC
+        """,
+        (username,)
+    )
+
+    rows = cursor.fetchall()
+    conn.close()
+
+    return [
+        SessionMessageResponse(
+            username=row["username"],
+            title=row["title"],
+            thread_id=row["thread_id"],
+        )
+        for row in rows
+    ]
+
+
 def exists_session(thread_id: str, username: str) -> bool:
     conn = sqlite3.connect(os.getenv("SQLITE_DB_PATH"))
     cursor = conn.cursor()