ai-lib-core 1.0.0

AI-Protocol execution runtime core (protocol, client, pipeline, transport)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

//! Structured output module for ai-lib-rust.
//!
//! Provides JSON mode, schema validation, and structured output capabilities:
//! - `OutputValidator`: Validate JSON against schemas
//! - `ValidationResult`: Result of validation operations
//! - `ValidationError`: Detailed validation errors
//!
//! # Examples
//!
//! ```
//! use ai_lib_core::structured::{OutputValidator, ValidationResult};
//! use serde_json::json;
//!
//! let schema = json!({
//!     "type": "object",
//!     "properties": {
//!         "name": {"type": "string"},
//!         "age": {"type": "integer"}
//!     },
//!     "required": ["name"]
//! });
//!
//! let validator = OutputValidator::lenient(schema);
//! let data = json!({"name": "Alice", "age": 30});
//! let result = validator.validate(data);
//!
//! assert!(result.is_valid());
//! ```

pub mod error;
pub mod json_mode;
pub mod schema;
pub mod validator;

// Re-export commonly used types
pub use error::{ValidationError, ValidationResult};
pub use json_mode::{JsonMode, JsonModeConfig, StructuredOutput};
pub use schema::{json_schema_from_type, schema_from_type_name, SchemaGenerator};
pub use validator::{IntoValidatorData, OutputValidator};

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_json_mode_display() {
        assert_eq!(JsonMode::Json.to_string(), "json_object");
        assert_eq!(JsonMode::JsonSchema.to_string(), "json_schema");
        assert_eq!(JsonMode::Off.to_string(), "");
    }

    #[test]
    fn test_json_mode_from_str() {
        assert_eq!("json_object".parse::<JsonMode>().unwrap(), JsonMode::Json);
        assert_eq!(
            "json_schema".parse::<JsonMode>().unwrap(),
            JsonMode::JsonSchema
        );
        assert_eq!("off".parse::<JsonMode>().unwrap(), JsonMode::Off);
        assert_eq!("".parse::<JsonMode>().unwrap(), JsonMode::Off);

        assert!("invalid".parse::<JsonMode>().is_err());
    }
}