rstructor 0.2.10

Rust equivalent of Python's Instructor + Pydantic: Extract structured, validated data from LLMs (OpenAI, Anthropic, Grok, Gemini) using type-safe Rust structs and enums
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
40
41
42
43

// tests/recursive_structures_tests.rs
#[cfg(test)]
mod recursive_tests {
    use rstructor::{GeminiClient, GeminiModel, Instructor, LLMClient};
    use serde::{Deserialize, Serialize};
    use std::env;

    #[derive(Instructor, Serialize, Deserialize, Debug)]
    #[llm(description = "A node in a file system tree")]
    struct FileNode {
        name: String,
        #[llm(description = "If true, this is a directory and can have children")]
        is_dir: bool,
        #[llm(description = "Children nodes if this is a directory")]
        #[allow(clippy::vec_box)]
        children: Option<Vec<Box<FileNode>>>,
        #[llm(description = "Size in bytes if this is a file")]
        size: Option<u64>,
    }

    #[tokio::test]
    async fn test_gemini_recursive_schema() {
        let api_key = env::var("GEMINI_API_KEY").expect("GEMINI_API_KEY must be set");
        let client = GeminiClient::new(api_key)
            .unwrap()
            .model(GeminiModel::Gemini31ProPreview)
            .temperature(0.0)
            .no_retries();

        let prompt = "Represent a small directory structure: A root folder 'src' containing a file 'lib.rs' (500 bytes) and a subfolder 'backend' which is empty.";
        let result: rstructor::Result<FileNode> = client.materialize(prompt).await;

        // Recursive schemas often fail if the generator doesn't handle $ref or creates infinite loops
        assert!(
            result.is_ok(),
            "Recursive FileNode failed: {:?}",
            result.err()
        );
        let root = result.unwrap();
        assert_eq!(root.name, "src");
        assert_eq!(root.children.unwrap().len(), 2);
    }
}