package com.superbiz.agent.hook;

/**
 * Token 用量持有者（基于 ThreadLocal）
 * ChatModel 调用后写入实际 token 数，AgentLoggingHook 读取
 */
public class TokenUsageHolder {

    private static final ThreadLocal<Integer> TOKEN_COUNT = new ThreadLocal<>();

    public static void set(Integer count) {
        TOKEN_COUNT.set(count);
    }

    public static Integer get() {
        return TOKEN_COUNT.get();
    }

    public static void clear() {
        TOKEN_COUNT.remove();
    }
}