fix(kernel): message_count 始终为 0 的 bug
- AgentRegistry 新增 message_counts: DashMap<AgentId, u64> 跟踪字段 - 添加 increment_message_count() 方法 - Kernel.send_message() 和 send_message_stream() 中递增计数 - get_info() 返回实际计数值而非硬编码 0
This commit is contained in:
@@ -534,6 +534,9 @@ impl Kernel {
|
||||
// Run the loop
|
||||
let result = loop_runner.run(session_id, message).await?;
|
||||
|
||||
// Track message count
|
||||
self.registry.increment_message_count(agent_id);
|
||||
|
||||
Ok(MessageResponse {
|
||||
content: result.response,
|
||||
input_tokens: result.input_tokens,
|
||||
@@ -624,6 +627,7 @@ impl Kernel {
|
||||
let loop_runner = loop_runner.with_system_prompt(&system_prompt);
|
||||
|
||||
// Run with streaming
|
||||
self.registry.increment_message_count(agent_id);
|
||||
loop_runner.run_streaming(session_id, message).await
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user