<?xml version="1.0" encoding="UTF-8"?>
|
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
<mapper namespace="com.zy.ai.mapper.AiTokenUsageMapper">
|
|
<resultMap id="BaseResultMap" type="com.zy.ai.entity.AiTokenUsage">
|
<id column="id" property="id" />
|
<result column="prompt_tokens" property="promptTokens" />
|
<result column="completion_tokens" property="completionTokens" />
|
<result column="total_tokens" property="totalTokens" />
|
<result column="llm_call_count" property="llmCallCount" />
|
<result column="update_time" property="updateTime" />
|
</resultMap>
|
|
<update id="incrementTokens">
|
UPDATE sys_ai_token_usage
|
SET prompt_tokens = prompt_tokens + #{promptTokens},
|
completion_tokens = completion_tokens + #{completionTokens},
|
total_tokens = total_tokens + #{totalTokens},
|
llm_call_count = llm_call_count + #{callCount},
|
update_time = NOW()
|
WHERE id = 1
|
</update>
|
|
</mapper>
|
