gems 0.1.4

💎 A cli, tui, and sdk for interacting with the Gemini API (WIP)
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
89
90
91
92
93
94

use crate::responses::ModelInfo;
use crate::responses::ModelsResponse;
use crate::traits::CTrait;
use crate::Client;
use anyhow::{anyhow, Error};
use derive_builder::Builder;
use reqwest::Method;
use std::str::FromStr;

#[derive(Debug, Clone, Default, PartialEq)]
pub enum Model {
    Pro25Preview,
    #[default]
    Flash20,
    Flash20Lite,
    Flash15,
    Flash15_8B,
    Pro15,
    Embedding,
    Imagen3,
    Veo2,
    Tts,
    Flash20Live,
    FlashExpImage,
}

#[allow(clippy::to_string_trait_impl)]
impl ToString for Model {
    fn to_string(&self) -> String {
        match self {
            Model::Pro25Preview => "gemini-2.5-pro-preview-03-25",
            Model::Flash20 => "gemini-2.0-flash",
            Model::Flash20Lite => "gemini-2.0-flash-lite",
            Model::Flash15 => "gemini-1.5-flash",
            Model::Flash15_8B => "gemini-1.5-flash-8b",
            Model::Pro15 => "gemini-1.5-pro",
            Model::Embedding => "text-embedding-004",
            Model::Imagen3 => "imagen-3.0-generate-002",
            Model::Veo2 => "veo-2.0-generate-001",
            Model::Tts => "gemini-2.5-flash-preview-tts",
            Model::Flash20Live => "gemini-2.0-flash-live-001",
            Model::FlashExpImage => "gemini-2.0-flash-exp-image-generation",
        }
        .to_string()
    }
}

impl FromStr for Model {
    type Err = Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "gemini-2.5-pro-preview-03-25" => Ok(Model::Pro25Preview),
            "gemini-2.0-flash" => Ok(Model::Flash20),
            "gemini-2.0-flash-lite" => Ok(Model::Flash20Lite),
            "gemini-1.5-flash" => Ok(Model::Flash15),
            "gemini-1.5-flash-8b" => Ok(Model::Flash15_8B),
            "gemini-1.5-pro" => Ok(Model::Pro15),
            "text-embedding-004" => Ok(Model::Embedding),
            "imagen-3.0-generate-002" => Ok(Model::Imagen3),
            "veo-2.0-generate-001" => Ok(Model::Veo2),
            "gemini-2.5-flash-preview-tts" => Ok(Model::Tts),
            "gemini-2.0-flash-live-001" => Ok(Model::Flash20Live),
            "gemini-2.0-flash-exp-image-generation" => Ok(Model::Flash20Live),
            _ => Err(anyhow!("Unknown model: {}", s)),
        }
    }
}
#[derive(Clone)]
pub struct Models {
    pub client: Client,
}

#[derive(Builder, Default, Clone)]
#[builder(setter(into), default)]
pub struct Mod {
    pub model: Model,
}

impl Models {
    pub async fn list(&self) -> anyhow::Result<ModelsResponse> {
        let req = self.client.request(Method::GET, "models")?;

        let res = req.send().await?;
        Ok(res.json().await?)
    }

    pub async fn get(&self, _model: Mod) -> anyhow::Result<ModelInfo> {
        let req = self.client.request(Method::GET, "")?;

        let res = req.send().await?;
        Ok(res.json().await?)
    }
}