stabilityai 0.1.0

stability.ai library based on OpenAPI spec
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

//! This test is primarily to make sure that macros_rules for From traits are correct.

use stabilityai::types::TextPrompts;
fn text_prompts<T>(input: T) -> TextPrompts
where
    TextPrompts: From<T>,
{
    input.into()
}

#[test]
fn create_text_prompts() {
    let prompt = "This is &str prompt";
    let _ = text_prompts(prompt);

    let prompt = "This is String".to_string();
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = vec!["This is first", "This is second"];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = vec!["First string".to_string(), "Second string".to_string()];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let first = "First".to_string();
    let second = "Second".to_string();
    let prompt = vec![&first, &second];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = ["first", "second"];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = ["first".to_string(), "second".to_string()];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let first = "First".to_string();
    let second = "Second".to_string();
    let prompt = [&first, &second];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);
}

#[test]
fn create_text_prompts_with_weights() {
    let prompt = ("This is &str prompt", 1.0);
    let _ = text_prompts(prompt);

    let prompt = ("This is String".to_string(), 1.0);
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = vec![("This is first", 0.2), ("This is second", 0.3)];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = vec![
        ("First string".to_string(), 0.1),
        ("Second string".to_string(), 0.2),
    ];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let first = "First".to_string();
    let second = "Second".to_string();
    let prompt = vec![(&first, 0.1), (&second, 0.22)];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = [("first", 0.11), ("second", 0.22)];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let prompt = [("first".to_string(), 0.12), ("second".to_string(), 0.12)];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);

    let first = "First".to_string();
    let second = "Second".to_string();
    let prompt = [(&first, 0.111), (&second, 0.222)];
    let _ = text_prompts(&prompt);
    let _ = text_prompts(prompt);
}