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

assistant_tool: Pass an Entity<Project> to Tool::run (#26312)

Browse source 
This PR updates the `Tool::run` method to take an `Entity<Project>`
instead of a `WeakEntity<Project>`.

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2025-03-07 18:30:56 -05:00 • committed by GitHub
parent 921c24e274
commit e70d0edfac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 20 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -5,7 +5,7 @@ use assistant_tool::ToolWorkingSet;
use chrono::{DateTime, Utc};
use collections::{BTreeMap, HashMap, HashSet};
use futures::StreamExt as _;
use gpui::{App, Context, Entity, EventEmitter, SharedString, Task, WeakEntity};
use gpui::{App, Context, Entity, EventEmitter, SharedString, Task};
use language_model::{
LanguageModel, LanguageModelCompletionEvent, LanguageModelRegistry, LanguageModelRequest,
LanguageModelRequestMessage, LanguageModelRequestTool, LanguageModelToolResult,
@ -72,7 +72,7 @@ pub struct Thread {
context_by_message: HashMap<MessageId, Vec<ContextId>>,
completion_count: usize,
pending_completions: Vec<PendingCompletion>,
project: WeakEntity<Project>,
project: Entity<Project>,
tools: Arc<ToolWorkingSet>,
tool_use: ToolUseState,
}
@ -94,7 +94,7 @@ impl Thread {
context_by_message: HashMap::default(),
completion_count: 0,
pending_completions: Vec::new(),
project: project.downgrade(),
project,
tools,
tool_use: ToolUseState::new(),
}
@ -135,7 +135,7 @@ impl Thread {
context_by_message: HashMap::default(),
completion_count: 0,
pending_completions: Vec::new(),
project: project.downgrade(),
project,
tools,
tool_use,
}