systemprompt-ai 0.1.18

Core AI module for systemprompt.io
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::models::providers::anthropic::AnthropicThinking;

pub mod tokens {
    pub const THINKING_BUDGET: u32 = 10240;
}

pub fn build_thinking_config(model: &str) -> Option<AnthropicThinking> {
    if supports_extended_thinking(model) {
        Some(AnthropicThinking {
            thinking_type: "enabled".to_string(),
            budget_tokens: tokens::THINKING_BUDGET,
        })
    } else {
        None
    }
}

pub fn supports_extended_thinking(model: &str) -> bool {
    model.contains("claude-3-5") || model.contains("claude-3.5")
}