package model

import (
	"context"

	"gorm.io/gorm"
)

// AIChatMessageInsert 插入AI聊天消息
func AIChatMessageInsert(ctx context.Context, db *gorm.DB, message *AIChatMessage) (int64, error) {
	result := db.WithContext(ctx).Create(message)
	if result.Error != nil {
		return 0, result.Error
	}
	return message.Id, nil
}

// AIChatMessageFindByConversation 根据对话ID查询所有消息（按创建时间升序，不分页）
func AIChatMessageFindByConversation(ctx context.Context, db *gorm.DB, conversationId int64) ([]AIChatMessage, error) {
	var messages []AIChatMessage
	err := db.WithContext(ctx).Where("conversation_id = ?", conversationId).
		Order("created_at ASC").Find(&messages).Error
	if err != nil {
		return nil, err
	}
	return messages, nil
}