inference-gateway-sdk 0.22.1

Rust SDK for interacting with various language models through the Inference Gateway
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

//! Ergonomics for the image-generation request.
//!
//! The spec now types `CreateImageRequest::quality` and `::size`
//! (`CreateImageRequestQuality` and `ImageSize`), so the former hand-written
//! enums are gone - use the generated types directly. Only the `Default` impl
//! remains: typify cannot derive one for the enums in the schema set, so it is
//! maintained here and mirrors `chat_request.rs` so callers set just the fields
//! they care about.

use std::num::NonZeroU64;

use crate::generated::schemas::{CreateImageRequest, CreateImageRequestResponseFormat};

impl Default for CreateImageRequest {
    /// Mirrors the schema defaults: `n` = 1, `response_format` = `url`.
    fn default() -> Self {
        Self {
            model: None,
            n: NonZeroU64::MIN, // schema default: 1
            prompt: String::new(),
            quality: None,
            response_format: CreateImageRequestResponseFormat::Url,
            size: None,
        }
    }
}

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

    #[test]
    fn default_matches_schema_defaults() {
        let req = CreateImageRequest::default();
        assert_eq!(req.n, NonZeroU64::MIN);
        assert_eq!(req.response_format, CreateImageRequestResponseFormat::Url);
        assert!(req.quality.is_none());
    }
}