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);
|
}
|
