tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Add input and output token tracking to SessionMetadata (#1979)

Browse Source
This commit is contained in:
Young Kim
2025-04-01 11:08:28 -07:00
committed by GitHub
parent 45e0f7f717
commit 2fbf601446
5 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/agents/reference.rs
+2
View File
@@ -169,6 +169,8 @@ impl Agent for ReferenceAgent {
let mut metadata = session::read_metadata(&session_file)?;
metadata.working_dir = session.working_dir;
metadata.total_tokens = usage.usage.total_tokens;
metadata.input_tokens = usage.usage.input_tokens;
metadata.output_tokens = usage.usage.output_tokens;
// The message count is the number of messages in the session + 1 for the response
// The message count does not include the tool response till next iteration
metadata.message_count = messages.len() + 1;
crates/goose/src/agents/summarize.rs
+2
View File
@@ -266,6 +266,8 @@ impl Agent for SummarizeAgent {
let mut metadata = session::read_metadata(&session_file)?;
metadata.working_dir = session.working_dir;
metadata.total_tokens = usage.usage.total_tokens;
metadata.input_tokens = usage.usage.input_tokens;
metadata.output_tokens = usage.usage.output_tokens;
// The message count is the number of messages in the session + 1 for the response
// The message count does not include the tool response till next iteration
metadata.message_count = messages.len() + 1;
crates/goose/src/agents/truncate.rs
+2
View File
@@ -280,6 +280,8 @@ impl Agent for TruncateAgent {
let mut metadata = session::read_metadata(&session_file)?;
metadata.working_dir = session.working_dir;
metadata.total_tokens = usage.usage.total_tokens;
metadata.input_tokens = usage.usage.input_tokens;
metadata.output_tokens = usage.usage.output_tokens;
// The message count is the number of messages in the session + 1 for the response
// The message count does not include the tool response till next iteration
metadata.message_count = messages.len() + 1;