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)