unified-agent-api-codex 0.3.5

Async wrapper around the Codex CLI for programmatic prompting
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

//! Demonstrates `codex debug ...` surfaces via the wrapper.
//!
//! Usage:
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- help`
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- help app-server`
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- app-server`
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- app-server help`
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- app-server send-message-v2 "hello"`
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- models [--bundled]`
//! - `cargo run -p unified-agent-api-codex --example debug_cmd -- prompt-input [--image <FILE> ...] [PROMPT]`
//!
//! Environment:
//! - `CODEX_BINARY` (optional): path to the `codex` CLI binary (defaults to `codex` in PATH).
//! - `CODEX_HOME` (optional): Codex home directory for config/auth.

use codex::{
    DebugAppServerHelpRequest, DebugAppServerRequest, DebugAppServerSendMessageV2Request,
    DebugHelpRequest, DebugModelsRequest, DebugPromptInputRequest,
};

#[path = "support/real_cli.rs"]
mod real_cli;

fn usage() {
    eprintln!("usage: debug_cmd <help|app-server ...|models [--bundled]|prompt-input [--image <FILE> ...] [PROMPT]>");
    eprintln!("examples:");
    eprintln!("  debug_cmd help");
    eprintln!("  debug_cmd help app-server");
    eprintln!("  debug_cmd app-server");
    eprintln!("  debug_cmd app-server help");
    eprintln!("  debug_cmd app-server send-message-v2 \"hello\"");
    eprintln!("  debug_cmd models --bundled");
    eprintln!("  debug_cmd prompt-input --image ./diagram.png \"summarize this prompt payload\"");
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut args: Vec<String> = std::env::args().skip(1).collect();
    if args.is_empty() {
        usage();
        return Ok(());
    }

    let client = real_cli::default_client();

    match args[0].to_ascii_lowercase().as_str() {
        "help" => {
            let command = args.drain(1..).collect::<Vec<_>>();
            let output = client
                .debug_help(DebugHelpRequest::new().command(command))
                .await?;
            print!("{}", output.stdout);
            if !output.stderr.is_empty() {
                eprintln!("{}", output.stderr);
            }
        }
        "app-server" => {
            if args.len() == 1 {
                let output = client
                    .debug_app_server(DebugAppServerRequest::new())
                    .await?;
                print!("{}", output.stdout);
                if !output.stderr.is_empty() {
                    eprintln!("{}", output.stderr);
                }
                return Ok(());
            }

            match args[1].to_ascii_lowercase().as_str() {
                "help" => {
                    let command = args.drain(2..).collect::<Vec<_>>();
                    let output = client
                        .debug_app_server_help(DebugAppServerHelpRequest::new().command(command))
                        .await?;
                    print!("{}", output.stdout);
                    if !output.stderr.is_empty() {
                        eprintln!("{}", output.stderr);
                    }
                }
                "send-message-v2" => {
                    if args.len() < 3 {
                        usage();
                        return Ok(());
                    }
                    let message = args.drain(2..).collect::<Vec<_>>().join(" ");
                    let output = client
                        .debug_app_server_send_message_v2(DebugAppServerSendMessageV2Request::new(
                            message,
                        ))
                        .await?;
                    print!("{}", output.stdout);
                    if !output.stderr.is_empty() {
                        eprintln!("{}", output.stderr);
                    }
                }
                _ => usage(),
            }
        }
        "models" => {
            let bundled = args.iter().skip(1).any(|arg| arg == "--bundled");
            let output = client
                .debug_models(DebugModelsRequest::new().bundled(bundled))
                .await?;
            print!("{}", output.stdout);
            if !output.stderr.is_empty() {
                eprintln!("{}", output.stderr);
            }
        }
        "prompt-input" => {
            let mut request = DebugPromptInputRequest::new();
            let mut idx = 1;
            let mut prompt_parts = Vec::new();
            while idx < args.len() {
                if args[idx] == "--image" {
                    idx += 1;
                    if idx >= args.len() {
                        usage();
                        return Ok(());
                    }
                    request = request.image(&args[idx]);
                } else {
                    prompt_parts.push(args[idx].clone());
                }
                idx += 1;
            }
            if !prompt_parts.is_empty() {
                request = request.prompt(prompt_parts.join(" "));
            }
            let output = client.debug_prompt_input(request).await?;
            print!("{}", output.stdout);
            if !output.stderr.is_empty() {
                eprintln!("{}", output.stderr);
            }
        }
        _ => usage(),
    }

    Ok(())
}