package com.zy.ai.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.zy.ai.entity.AiTokenUsage;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.springframework.stereotype.Repository;

@Mapper
@Repository
public interface AiTokenUsageMapper extends BaseMapper<AiTokenUsage> {

    int incrementTokens(@Param("promptTokens") long promptTokens,
                        @Param("completionTokens") long completionTokens,
                        @Param("totalTokens") long totalTokens,
                        @Param("callCount") long callCount);
}