feat: 添加数据库迁移、清空聊天记录按钮 - 复用SAVE_HISTORY接口

core/client.py

@@ -73,6 +73,14 @@ class BackendClient:
             body={}
         )
         return self._server.send(packet).body.get("history", [])
-
+    
+    def clear_history(self) -> Dict:
+        packet = Packet(
+            id=self._next_id(),
+            type=PacketType.SAVE_HISTORY,
+            body={"messages": [{"action": "clear"}]}
+        )
+        return self._server.send(packet).body
+    
     def shutdown(self) -> None:
         self._server.shutdown()

core/embedded_db.py

@@ -71,18 +71,20 @@ class EmbeddedGraphDB:
         cursor.execute("CREATE INDEX IF NOT EXISTS idx_relation_type ON relations(relation_type)")
         cursor.execute("CREATE INDEX IF NOT EXISTS idx_relation_status ON relations(status)")
         
-        # 创建聊天记录表
         cursor.execute("""
-            CREATE TABLE IF NOT EXISTS chat_records (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                role TEXT NOT NULL,
-                content TEXT NOT NULL,
-                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-            )
+            SELECT name FROM sqlite_master 
+            WHERE type='table' AND name='chat_records'
         """)
-        
-        # 创建聊天记录索引
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_chat_created ON chat_records(created_at)")
+        if not cursor.fetchone():
+            cursor.execute("""
+                CREATE TABLE chat_records (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    role TEXT NOT NULL,
+                    content TEXT NOT NULL,
+                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+                )
+            """)
+            cursor.execute("CREATE INDEX idx_chat_created ON chat_records(created_at)")
         
         self.conn.commit()
     
@@ -445,6 +447,13 @@ class EmbeddedGraphDB:
         """, (limit,))
         return [{"role": row[0], "content": row[1]} for row in cursor.fetchall()]
     
+    def clear_chat_records(self) -> Dict:
+        """清空聊天记录（保留图数据库）"""
+        cursor = self.conn.cursor()
+        cursor.execute("DELETE FROM chat_records")
+        self.conn.commit()
+        return {"cleared": True}
+    
     def close(self):
         """关闭数据库连接"""
         if self.conn:

core/server.py

@@ -345,9 +345,12 @@ class BackendServer:
 
     def _handle_save_history(self, body: Dict) -> Dict:
         messages = body.get("messages", [])
+        if messages and messages[0].get("action") == "clear":
+            self._graph.clear_chat_records()
+            return {"status": "history_cleared"}
         result = self._graph.save_chat_records(messages)
         return {"status": "history_saved"}
-
+    
     def _send_response(self, request_id: str, response: PacketResponse) -> None:
         with self._lock:
             q = self._response_queues.pop(request_id, None)