rig-core 0.35.0

An opinionated library for building LLM powered applications.
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

//! Migrated from `examples/anthropic_think_tool.rs`.

use rig::client::{CompletionClient, ProviderClient};
use rig::completion::Prompt;
use rig::providers::anthropic;
use rig::tools::ThinkTool;

use crate::support::{assert_contains_any_case_insensitive, assert_nonempty_response};

#[tokio::test]
#[ignore = "requires ANTHROPIC_API_KEY"]
async fn think_tool_menu_planning() {
    let agent = anthropic::Client::from_env()
        .agent(anthropic::completion::CLAUDE_SONNET_4_6)
        .name("Anthropic Thinker")
        .preamble(
            "You are a helpful assistant that can solve complex problems. \
             Use the 'think' tool to reason through complex problems step by step.",
        )
        .tool(ThinkTool)
        .build();

    let response = agent
        .prompt(
            "I need to plan a dinner party for 8 people, including 2 vegetarians and \
             1 person with a gluten allergy. Create appetizers, mains, and desserts.",
        )
        .max_turns(10)
        .await
        .expect("think tool prompt should succeed");

    assert_nonempty_response(&response);
    assert_contains_any_case_insensitive(&response, &["appetizer", "main", "dessert"]);
}