zhuyongxin
0d9cce75f9
- 新增诊断会话(diagnosis_session/agent_step/tool_invocation)三表 - AgentLoggingHook 持久化 agent_step，记录决策链和耗时 - LookupKnowledgeTool 写入 tool_invocation，记录L0/L1检索质量 - TokenTrackingChatModel 捕获真实token用量 - Chat接口支持意图路由：简单问题单Agent，复杂问题多Agent(Planner+Executor) - Prompt外置到 src/main/resources/prompts/ - 删除旧 diagnosis_record 表及相关文件 - 新增SessionContextHolder(ThreadLocal传递sessionId) - QuestionComplexity 复杂度判断工具 - 测试覆盖三张新表的Repository
23 lines
506 B
Java
23 lines
506 B
Java
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();
|
||
}
|
||
}
|