Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

agent: Remove unneeded tracking of request usage (#29894)

Browse source 
This PR removes some unneeded tracking of the model request usage in the
`ActiveThread` and `ThreadEvent::UsageUpdated` events.

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2025-05-04 21:16:53 -04:00 • committed by GitHub
parent fe177f5d69
commit 3db4744e18
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 4 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

8
crates/agent/src/thread.rs
View file

@ -1545,7 +1545,6 @@ impl Thread {
let usage = RequestUsage { limit, amount: amount as i32 };
thread.last_usage = Some(usage);
cx.emit(ThreadEvent::UsageUpdated(usage));
}
CompletionRequestStatus::ToolUseLimitReached => {
thread.tool_use_limit_reached = true;
@ -1714,11 +1713,11 @@ impl Thread {
LanguageModelCompletionEvent::StatusUpdate(
CompletionRequestStatus::UsageUpdated { amount, limit },
) => {
this.update(cx, |_, cx| {
cx.emit(ThreadEvent::UsageUpdated(RequestUsage {
this.update(cx, |thread, _cx| {
thread.last_usage = Some(RequestUsage {
limit,
amount: amount as i32,
}));
});
})?;
continue;
}
@ -2565,7 +2564,6 @@ pub enum ThreadError {
#[derive(Debug, Clone)]
pub enum ThreadEvent {
ShowError(ThreadError),
UsageUpdated(RequestUsage),
StreamedCompletion,
ReceivedTextChunk,
NewRequest,