objectiveai-sdk 2.0.6

ObjectiveAI SDK, definitions, and utilities
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
44
45
46
47
48
49
50
51
52

use rand::Rng;
use rand::seq::SliceRandom;

use crate::agent::completions::message::{
    ImageUrl, ImageUrlDetail, RichContentPart,
};
use crate::functions::expression::{ImageInputSchema, InputValue};

pub const fn permutations(_schema: &ImageInputSchema) -> usize {
    4usize
}

pub fn generate<R: Rng>(
    _schema: &ImageInputSchema,
    mut rng: R,
) -> Generator<R> {
    let mut indices: Vec<usize> = (0..4).collect();
    indices.shuffle(&mut rng);
    Generator {
        indices,
        pos: 0,
        rng,
    }
}

pub struct Generator<R: Rng> {
    indices: Vec<usize>,
    pos: usize,
    rng: R,
}

impl<R: Rng> Iterator for Generator<R> {
    type Item = InputValue;
    fn next(&mut self) -> Option<InputValue> {
        if self.pos >= self.indices.len() {
            self.indices.shuffle(&mut self.rng);
            self.pos = 0;
        }
        let index = self.indices[self.pos];
        self.pos += 1;
        let url = super::string::random_string(&mut self.rng);
        let detail = match index {
            0 => None,
            1 => Some(ImageUrlDetail::Auto),
            2 => Some(ImageUrlDetail::Low),
            _ => Some(ImageUrlDetail::High),
        };
        Some(InputValue::RichContentPart(RichContentPart::ImageUrl {
            image_url: ImageUrl { url, detail },
        }))
    }
}