mirror of
https://github.com/openai/codex.git
synced 2026-04-29 02:41:12 +03:00
8a89d3aeda
Some effects of this change: - New formatting changes across many files. No functionality changes should occur from that. - Calls to `set_env` are considered unsafe, since this only happens in tests we wrap them in `unsafe` blocks
80 lines
2.4 KiB
Rust
80 lines
2.4 KiB
Rust
use std::sync::Arc;
|
|
use std::sync::atomic::AtomicU64;
|
|
|
|
use crate::Codex;
|
|
use crate::config::Config;
|
|
use crate::protocol::Event;
|
|
use crate::protocol::EventMsg;
|
|
use crate::protocol::Op;
|
|
use crate::protocol::Submission;
|
|
use crate::util::notify_on_sigint;
|
|
use tokio::sync::Notify;
|
|
|
|
/// Spawn a new [`Codex`] and initialise the session.
|
|
///
|
|
/// Returns the wrapped [`Codex`] **and** the `SessionInitialized` event that
|
|
/// is received as a response to the initial `ConfigureSession` submission so
|
|
/// that callers can surface the information to the UI.
|
|
pub async fn init_codex(config: Config) -> anyhow::Result<(CodexWrapper, Event, Arc<Notify>)> {
|
|
let ctrl_c = notify_on_sigint();
|
|
let codex = CodexWrapper::new(Codex::spawn(ctrl_c.clone())?);
|
|
let init_id = codex
|
|
.submit(Op::ConfigureSession {
|
|
model: config.model.clone(),
|
|
instructions: config.instructions.clone(),
|
|
approval_policy: config.approval_policy,
|
|
sandbox_policy: config.sandbox_policy,
|
|
disable_response_storage: config.disable_response_storage,
|
|
notify: config.notify.clone(),
|
|
cwd: config.cwd.clone(),
|
|
})
|
|
.await?;
|
|
|
|
// The first event must be `SessionInitialized`. Validate and forward it to
|
|
// the caller so that they can display it in the conversation history.
|
|
let event = codex.next_event().await?;
|
|
if event.id != init_id
|
|
|| !matches!(
|
|
&event,
|
|
Event {
|
|
id: _id,
|
|
msg: EventMsg::SessionConfigured { .. },
|
|
}
|
|
)
|
|
{
|
|
return Err(anyhow::anyhow!(
|
|
"expected SessionInitialized but got {event:?}"
|
|
));
|
|
}
|
|
|
|
Ok((codex, event, ctrl_c))
|
|
}
|
|
|
|
pub struct CodexWrapper {
|
|
next_id: AtomicU64,
|
|
codex: Codex,
|
|
}
|
|
|
|
impl CodexWrapper {
|
|
fn new(codex: Codex) -> Self {
|
|
Self {
|
|
next_id: AtomicU64::new(0),
|
|
codex,
|
|
}
|
|
}
|
|
|
|
/// Returns the id of the Submission.
|
|
pub async fn submit(&self, op: Op) -> crate::error::Result<String> {
|
|
let id = self
|
|
.next_id
|
|
.fetch_add(1, std::sync::atomic::Ordering::SeqCst)
|
|
.to_string();
|
|
self.codex.submit(Submission { id: id.clone(), op }).await?;
|
|
Ok(id)
|
|
}
|
|
|
|
pub async fn next_event(&self) -> crate::error::Result<Event> {
|
|
self.codex.next_event().await
|
|
}
|
|
}
|