package impl

import (
	"sdk-generator/internal/data/model"
	"sdk-generator/tmpl/dataTemp"
	"sdk-generator/utils"
)

type AiGenerateTaskImpl struct {
	dataTemp.DataTemp
	db *utils.Db
}

func NewAiGenerateTaskImpl(db *utils.Db) *AiGenerateTaskImpl {
	return &AiGenerateTaskImpl{
		DataTemp: *dataTemp.NewDataTemp(db, new(model.AiGenerateTask)),
		db:       db,
	}
}

func (m *AiGenerateTaskImpl) PrimaryKey() string {
	return "id"
}

func (m *AiGenerateTaskImpl) GetTemp() *dataTemp.DataTemp {
	return &m.DataTemp
}