mirror of
https://github.com/pchuan98/codex.git
synced 2026-07-01 00:31:56 +08:00
jif-oai
d996f5366f
Part 1 of guardian as extension. This bind all the logic to spawn another agent from an extension and it adds `ThreadId` in the start thread collaborator
39 lines
1.1 KiB
Rust
39 lines
1.1 KiB
Rust
use std::future::Future;
|
|
use std::pin::Pin;
|
|
|
|
use codex_protocol::ThreadId;
|
|
|
|
/// Future returned by one injected subagent-spawn helper.
|
|
pub type AgentSpawnFuture<'a, T, E> = Pin<Box<dyn Future<Output = Result<T, E>> + Send + 'a>>;
|
|
|
|
/// Constructor-injected host helper for extensions that need to spawn subagents.
|
|
///
|
|
/// The extension owns the request shape and resulting handle types. The host
|
|
/// provides the implementation when it constructs the extension.
|
|
pub trait AgentSpawner<R>: Send + Sync {
|
|
type Spawned;
|
|
type Error;
|
|
|
|
fn spawn_subagent<'a>(
|
|
&'a self,
|
|
forked_from_thread_id: ThreadId,
|
|
request: R,
|
|
) -> AgentSpawnFuture<'a, Self::Spawned, Self::Error>;
|
|
}
|
|
|
|
impl<R, S, E, F> AgentSpawner<R> for F
|
|
where
|
|
F: Fn(ThreadId, R) -> AgentSpawnFuture<'static, S, E> + Send + Sync,
|
|
{
|
|
type Spawned = S;
|
|
type Error = E;
|
|
|
|
fn spawn_subagent<'a>(
|
|
&'a self,
|
|
forked_from_thread_id: ThreadId,
|
|
request: R,
|
|
) -> AgentSpawnFuture<'a, Self::Spawned, Self::Error> {
|
|
self(forked_from_thread_id, request)
|
|
}
|
|
}
|