'commit'

Model/HaiBaoModel.py

@@ -0,0 +1,24 @@
+import logging
+from DbKit.Db import Db
+
+logger = logging.getLogger("HaiBaoModel")
+
+class HaiBaoModel:
+    def __init__(self):
+        self.db = Db()
+
+    async def insert_record(self, id, prompt, image_url, scheme_content, created_at):
+        """插入生成记录"""
+        params = {
+            "id": id,
+            "prompt": prompt,
+            "image_url": image_url,
+            "scheme_content": scheme_content,
+            "created_at": created_at
+        }
+        return await self.db.execute_update("HaiBao.insertHistory", params)
+
+    async def get_history(self, limit=50):
+        """获取历史记录"""
+        params = {"limit": limit}
+        return await self.db.find("HaiBao.getHistory", params)