Kayos
1ed4d8211f
Three examples: - basic.rs: query() one-shot prompt + cost reporting. - with_options.rs: system_prompt, model selection, effort, permission mode. - interactive.rs: Client multi-turn session with two back-to-back sends. Integration tests in tests/transport_end_to_end.rs spawn a tiny POSIX shell script (tests/fake_cli/fake-claude.sh) as a stand-in for the real claude CLI. The fake answers -v with a version, reads one user-message frame on stdin, and emits a fixed assistant + result pair on stdout, then blocks on stdin until close. This proves the spawn / write / read / disconnect lifecycle works without an authenticated claude install. Coverage: - query() round-trip: stream yields Assistant then Result. - Client round-trip: connect + send + drain to Result + disconnect. - CLI-not-found surfaces as a typed Error::CliNotFound. Cargo.lock is committed since this is, in practice, both a library and a binary (the examples link the crate). Locking dev-deps avoids surprise churn in CI.
50 lines
1.4 KiB
Rust
50 lines
1.4 KiB
Rust
//! Minimal `query()` example.
|
|
//!
|
|
//! Run:
|
|
//!
|
|
//! ```sh
|
|
//! cargo run --example basic
|
|
//! ```
|
|
//!
|
|
//! Requires the `claude` CLI on `PATH` and authenticated. Output prints each
|
|
//! assistant text block, then the total cost from the terminal `result`
|
|
//! message.
|
|
|
|
use claude_agent_sdk::{ClaudeAgentOptions, ContentBlock, Message, query};
|
|
use tokio_stream::StreamExt;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
tracing_subscriber::fmt()
|
|
.with_env_filter(
|
|
tracing_subscriber::EnvFilter::try_from_default_env()
|
|
.unwrap_or_else(|_| "claude_agent_sdk=info".into()),
|
|
)
|
|
.init();
|
|
|
|
let opts = ClaudeAgentOptions::new().with_max_turns(1);
|
|
|
|
let mut stream = query("What is 2 + 2? Reply with just the digit.", opts).await?;
|
|
|
|
while let Some(item) = stream.next().await {
|
|
match item? {
|
|
Message::Assistant(a) => {
|
|
for block in &a.message.content {
|
|
if let ContentBlock::Text(t) = block {
|
|
println!("Claude: {}", t.text);
|
|
}
|
|
}
|
|
}
|
|
Message::Result(r) => {
|
|
println!(
|
|
"\n[result] subtype={} turns={} cost_usd={:?}",
|
|
r.subtype, r.num_turns, r.total_cost_usd
|
|
);
|
|
break;
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|