Codex MCP installation (#345)

This commit is contained in:
Solomon
2025-08-08 18:24:04 +01:00
committed by GitHub
parent 1999986304
commit 0e7d4ddbdf
9 changed files with 343 additions and 135 deletions

View File

@@ -123,7 +123,7 @@ impl BaseCodingAgent {
Self::Amp => Some(vec!["amp", "mcpServers"]), // Nested path for Amp
Self::Gemini => Some(vec!["mcpServers"]),
//ExecutorConfig::Aider => None, // Aider doesn't support MCP. https://github.com/Aider-AI/aider/issues/3314
Self::Codex => None, // Codex uses TOML config, frontend doesn't handle TOML yet
Self::Codex => Some(vec!["mcp_servers"]), // Codex uses TOML with mcp_servers
}
}

View File

@@ -351,20 +351,20 @@ impl LocalContainerService {
}
// Fire event when CodingAgent execution has finished
if let Some(true) = config.read().await.analytics_enabled {
if matches!(
if let Some(true) = config.read().await.analytics_enabled
&& matches!(
&ctx.execution_process.run_reason,
ExecutionProcessRunReason::CodingAgent
) && let Some(analytics) = &analytics
{
analytics.analytics_service.track_event(&analytics.user_id, "task_attempt_finished", Some(json!({
)
&& let Some(analytics) = &analytics
{
analytics.analytics_service.track_event(&analytics.user_id, "task_attempt_finished", Some(json!({
"task_id": ctx.task.id.to_string(),
"project_id": ctx.task.project_id.to_string(),
"attempt_id": ctx.task_attempt.id.to_string(),
"execution_success": matches!(ctx.execution_process.status, ExecutionProcessStatus::Completed),
"exit_code": ctx.execution_process.exit_code,
})));
}
}
}

View File

@@ -33,6 +33,7 @@ openssl-sys = { workspace = true }
rmcp = { version = "0.2.1", features = ["server", "transport-io"] }
schemars = "0.8"
regex = "1.11.1"
toml = "0.8"
sentry = { version = "0.41.0", features = ["anyhow", "backtrace", "panic", "debug-images"] }
sentry-tracing = { version = "0.41.0", features = ["backtrace"] }
reqwest = { version = "0.12", features = ["json"] }

View File

@@ -0,0 +1,56 @@
//! Utilities for reading and writing external agent config files (not the server's own config).
//!
//! These helpers abstract over JSON vs TOML formats used by different agents.
use executors::executors::BaseCodingAgent;
use serde_json::Value;
use tokio::fs;
/// Determine if the agent's config file is TOML-based.
fn is_toml_config(agent: &BaseCodingAgent) -> bool {
matches!(agent, BaseCodingAgent::Codex)
}
/// Read an agent's external config file (JSON or TOML) and normalize it to serde_json::Value.
pub async fn read_agent_config(
config_path: &std::path::Path,
agent: &BaseCodingAgent,
) -> Result<Value, Box<dyn std::error::Error + Send + Sync>> {
let file_content = fs::read_to_string(config_path).await.unwrap_or_else(|_| {
if is_toml_config(agent) {
"".to_string()
} else {
"{}".to_string()
}
});
if is_toml_config(agent) {
// Parse TOML then convert to JSON Value
if file_content.trim().is_empty() {
return Ok(serde_json::json!({}));
}
let toml_val: toml::Value = toml::from_str(&file_content)?;
let json_string = serde_json::to_string(&toml_val)?;
Ok(serde_json::from_str(&json_string)?)
} else {
Ok(serde_json::from_str(&file_content)?)
}
}
/// Write an agent's external config (as serde_json::Value) back to disk in the agent's format (JSON or TOML).
pub async fn write_agent_config(
config_path: &std::path::Path,
agent: &BaseCodingAgent,
config: &Value,
) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
if is_toml_config(agent) {
// Convert JSON Value back to TOML
let toml_value: toml::Value = serde_json::from_str(&serde_json::to_string(config)?)?;
let toml_content = toml::to_string_pretty(&toml_value)?;
fs::write(config_path, toml_content).await?;
} else {
let json_content = serde_json::to_string_pretty(config)?;
fs::write(config_path, json_content).await?;
}
Ok(())
}

View File

@@ -1 +1,2 @@
pub mod agent_config;
pub mod task_server;

View File

@@ -17,7 +17,11 @@ use tokio::fs;
use ts_rs::TS;
use utils::{assets::config_path, response::ApiResponse};
use crate::{error::ApiError, DeploymentImpl};
use crate::{
error::ApiError,
mcp::agent_config::{read_agent_config, write_agent_config},
DeploymentImpl,
};
pub fn router() -> Router<DeploymentImpl> {
Router::new()
@@ -210,11 +214,8 @@ async fn update_mcp_servers_in_config(
fs::create_dir_all(parent).await?;
}
// Read existing config file or create empty object if it doesn't exist
let file_content = fs::read_to_string(config_path)
.await
.unwrap_or_else(|_| "{}".to_string());
let mut config: Value = serde_json::from_str(&file_content)?;
// Read existing config (JSON or TOML depending on agent)
let mut config = read_agent_config(config_path, agent).await?;
let mcp_path = agent.mcp_attribute_path().unwrap();
@@ -224,9 +225,8 @@ async fn update_mcp_servers_in_config(
// Set the MCP servers using the correct attribute path
set_mcp_servers_in_config_path(agent, &mut config, &mcp_path, &new_servers)?;
// Write the updated config back to file
let updated_content = serde_json::to_string_pretty(&config)?;
fs::write(config_path, updated_content).await?;
// Write the updated config back to file (JSON or TOML depending on agent)
write_agent_config(config_path, agent, &config).await?;
let new_count = new_servers.len();
let message = match (old_servers, new_count) {
@@ -246,10 +246,8 @@ async fn read_mcp_servers_from_config(
config_path: &std::path::Path,
agent: &BaseCodingAgent,
) -> Result<HashMap<String, Value>, Box<dyn std::error::Error + Send + Sync>> {
let file_content = fs::read_to_string(config_path)
.await
.unwrap_or_else(|_| "{}".to_string());
let raw_config: Value = serde_json::from_str(&file_content)?;
// Read config in appropriate format (JSON or TOML) and normalize to serde_json::Value
let raw_config = read_agent_config(config_path, agent).await?;
let mcp_path = agent.mcp_attribute_path().unwrap();
let servers = get_mcp_servers_from_config_path(agent, &raw_config, &mcp_path);
Ok(servers)