message_cli/
message_cli.rs

1//! # Message CLI Example
2//!
3//! This example demonstrates a simple command-line interface for interacting with Claude.
4//! It creates a conversation loop where the user can chat with the AI assistant.
5//!
6//! ## Features
7//!
8//! - Initializes with a system-like message to set the assistant's behavior
9//! - Maintains conversation history for context
10//! - Simple command-line interface for user input
11//!
12//! ## Usage
13//!
14//! Run this example with:
15//!
16//! ```bash
17//! cargo run --example message_cli
18//! ```
19//!
20//! Make sure you have set the `ANTHROPIC_API_KEY` environment variable.
21
22use anthropic_api::{messages::*, Credentials};
23use std::io::{stdin, stdout, Write};
24
25#[tokio::main]
26async fn main() {
27    // Load .env file containing ANTHROPIC_API_KEY
28    let credentials = Credentials::from_env();
29
30    let mut messages = vec![Message {
31        role: MessageRole::User,
32        content: MessageContent::Text(
33            "You are a helpful AI assistant. Please introduce yourself briefly.".to_string(),
34        ),
35    }];
36
37    // Create initial message request
38    let response = MessagesBuilder::builder("claude-3-7-sonnet-20250219", messages.clone(), 2000)
39        .credentials(credentials.clone())
40        .thinking(Thinking {
41            thinking_type: ThinkingType::Enabled,
42            budget_tokens: 1000,
43        })
44        .create()
45        .await
46        .unwrap();
47
48    // Print assistant's response
49    if let Some(content) = response.content.first() {
50        match content {
51            ResponseContentBlock::Text { text } => {
52                println!("Assistant: {}", text.trim());
53                messages.push(Message {
54                    role: MessageRole::Assistant,
55                    content: MessageContent::Text(text.clone()),
56                });
57            }
58            _ => {}
59        }
60    }
61
62    // Start conversation loop
63    loop {
64        print!("User: ");
65        stdout().flush().unwrap();
66
67        let mut user_input = String::new();
68        stdin().read_line(&mut user_input).unwrap();
69
70        // Add user message
71        messages.push(Message {
72            role: MessageRole::User,
73            content: MessageContent::Text(user_input),
74        });
75
76        // Send message request
77        let response =
78            MessagesResponse::builder("claude-3-7-sonnet-20250219", messages.clone(), 1024)
79                .credentials(credentials.clone())
80                .create()
81                .await
82                .unwrap();
83
84        // Print assistant's response
85        if let Some(content) = response.content.first() {
86            match content {
87                ResponseContentBlock::Text { text } => {
88                    println!("Assistant: {}", text.trim());
89                    messages.push(Message {
90                        role: MessageRole::Assistant,
91                        content: MessageContent::Text(text.clone()),
92                    });
93                }
94                _ => {}
95            }
96        }
97    }
98}