package db

import (
	"gitlab.fjmaimaimai.com/allied-creation/su-micro/cmd/ep/chat/internal/pkg/db/models"
	"gorm.io/gorm"
)

func Migrate(db *gorm.DB) {
	db.AutoMigrate(
		&models.ChatSession{},
		&models.ChatSessionRecord{},
		&models.ChatDocument{},
		&models.ChatDatasetDocumentMapping{},
		&models.ChatDataset{},
	)
}