cb8d941adf
- Renamed directory: codex-backend-openapi-models -> llmx-backend-openapi-models
- Updated all Cargo.toml files:
- Package names: codex-* -> llmx-*
- Library names: codex_* -> llmx_*
- Workspace dependencies updated
- Renamed Rust source files:
- codex*.rs -> llmx*.rs (all modules)
- codex_conversation -> llmx_conversation
- codex_delegate -> llmx_delegate
- codex_message_processor -> llmx_message_processor
- codex_tool_* -> llmx_tool_*
- Updated all Rust imports:
- use codex_* -> use llmx_*
- mod codex* -> mod llmx*
- Updated environment variables in code:
- CODEX_HOME -> LLMX_HOME
- .codex -> .llmx paths
- Updated protocol crate lib name for proper linking
Note: Some compilation errors remain (type inference issues) but all
renaming is complete. Will fix compilation in next phase.
🤖 Generated with Claude Code
76 lines
2.2 KiB
Rust
76 lines
2.2 KiB
Rust
use async_trait::async_trait;
|
|
|
|
use crate::function_tool::FunctionCallError;
|
|
use crate::mcp_tool_call::handle_mcp_tool_call;
|
|
use crate::tools::context::ToolInvocation;
|
|
use crate::tools::context::ToolOutput;
|
|
use crate::tools::context::ToolPayload;
|
|
use crate::tools::registry::ToolHandler;
|
|
use crate::tools::registry::ToolKind;
|
|
|
|
pub struct McpHandler;
|
|
|
|
#[async_trait]
|
|
impl ToolHandler for McpHandler {
|
|
fn kind(&self) -> ToolKind {
|
|
ToolKind::Mcp
|
|
}
|
|
|
|
async fn handle(&self, invocation: ToolInvocation) -> Result<ToolOutput, FunctionCallError> {
|
|
let ToolInvocation {
|
|
session,
|
|
turn,
|
|
call_id,
|
|
payload,
|
|
..
|
|
} = invocation;
|
|
|
|
let payload = match payload {
|
|
ToolPayload::Mcp {
|
|
server,
|
|
tool,
|
|
raw_arguments,
|
|
} => (server, tool, raw_arguments),
|
|
_ => {
|
|
return Err(FunctionCallError::RespondToModel(
|
|
"mcp handler received unsupported payload".to_string(),
|
|
));
|
|
}
|
|
};
|
|
|
|
let (server, tool, raw_arguments) = payload;
|
|
let arguments_str = raw_arguments;
|
|
|
|
let response = handle_mcp_tool_call(
|
|
session.as_ref(),
|
|
turn.as_ref(),
|
|
call_id.clone(),
|
|
server,
|
|
tool,
|
|
arguments_str,
|
|
)
|
|
.await;
|
|
|
|
match response {
|
|
llmx_protocol::models::ResponseInputItem::McpToolCallOutput { result, .. } => {
|
|
Ok(ToolOutput::Mcp { result })
|
|
}
|
|
llmx_protocol::models::ResponseInputItem::FunctionCallOutput { output, .. } => {
|
|
let llmx_protocol::models::FunctionCallOutputPayload {
|
|
content,
|
|
content_items,
|
|
success,
|
|
} = output;
|
|
Ok(ToolOutput::Function {
|
|
content,
|
|
content_items,
|
|
success,
|
|
})
|
|
}
|
|
_ => Err(FunctionCallError::RespondToModel(
|
|
"mcp handler received unexpected response variant".to_string(),
|
|
)),
|
|
}
|
|
}
|
|
}
|