rigg-client 1.4.3

Azure AI Search and Microsoft Foundry REST API client and authentication for rigg
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

//! Unified AI text generation via ailloy
//!
//! Uses the globally configured ailloy provider for AI requests.

use ailloy::{ChatOptions, Client, Message};

/// Generate text using the globally configured ailloy provider.
///
/// Uses the default chat node from `~/.config/ailloy/config.yaml`.
/// Run `ailloy config` to set up a provider.
pub async fn generate_text(system_prompt: &str, user_prompt: &str) -> anyhow::Result<String> {
    generate_text_with_limit(system_prompt, user_prompt, 2000).await
}

/// Generate text with a custom max_tokens limit.
pub async fn generate_text_with_limit(
    system_prompt: &str,
    user_prompt: &str,
    max_tokens: u32,
) -> anyhow::Result<String> {
    let client = Client::from_config()?;
    // No temperature override: several current models (gpt-5.x chat) only
    // accept their default temperature.
    let opts = ChatOptions::builder().max_tokens(max_tokens).build();
    let response = client
        .chat_with(
            &[Message::system(system_prompt), Message::user(user_prompt)],
            &opts,
        )
        .await?;
    Ok(response.content)
}

/// Check if ailloy is configured with a default chat node.
pub fn is_configured() -> bool {
    ailloy::config::Config::load()
        .map(|c| c.default_chat_node().is_ok())
        .unwrap_or(false)
}