nano-ai 0.0.2

✨ Gemini Nano Adapter for WASM Frameworks.
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

#![allow(unused)]
#![allow(deprecated)]

use crate::{adapter::*, error::NanoAIError, types::*};
use js_sys::{Array, Object};
use serde_wasm_bindgen::from_value;
use wasm_bindgen_futures::JsFuture;

pub struct NanoAI {
    session: Option<Session>,
}

impl Default for NanoAI {
    fn default() -> Self {
        Self::new()
    }
}

impl NanoAI {
    pub fn new() -> Self {
        NanoAI { session: None }
    }

    pub async fn get_capabilities(&self) -> Result<Capabilities, NanoAIError> {
        if !AI.is_undefined() {
            let promise = AI.language_model().capabilities();
            let result = JsFuture::from(promise).await;
            match result {
                Ok(value) => Ok(from_value(value)?),
                Err(err) => Err(NanoAIError::UnexpectedJsError(err)),
            }
        } else {
            Err(NanoAIError::BrowserNotSupported)
        }
    }

    pub async fn create_session(&mut self, _options: Option<Object>) -> Result<(), NanoAIError> {
        if !AI.is_undefined() {
            let session_promise = AI.language_model().create();
            let result = JsFuture::from(session_promise).await;

            match result {
                Ok(value) => {
                    self.session = Some(value.into());
                    Ok(())
                }
                Err(_err) => Err(NanoAIError::SessionCreationFailed("".to_string())),
            }
        } else {
            Err(NanoAIError::BrowserNotSupported)
        }
    }

    pub async fn send_prompt(&self, input: &str) -> Result<String, NanoAIError> {
        if !AI.is_undefined() {
            if let Some(session) = &self.session {
                let promise = session.prompt(input);
                let result = JsFuture::from(promise).await;

                match result {
                    Ok(value) => Ok(from_value(value)?),
                    Err(err) => Err(NanoAIError::PromptFailed(format!("{:?}", err))),
                }
            } else {
                Err(NanoAIError::SessionCreationFailed(
                    "Session Creation Failed!".to_string(),
                ))
            }
        } else {
            Err(NanoAIError::BrowserNotSupported)
        }
    }

    pub async fn stream_prompt(&self, input: &str) -> Result<Vec<StreamingChunk>, NanoAIError> {
        if !AI.is_undefined() {
            if let Some(session) = &self.session {
                let promise = session.prompt_streaming(input);
                let stream = JsFuture::from(promise).await;

                match stream {
                    Ok(value) => {
                        let array: Array = value.into();
                        let chunks = array
                            .iter()
                            .map(from_value::<StreamingChunk>)
                            .collect::<Result<Vec<_>, _>>()?;
                        Ok(chunks)
                    }
                    Err(err) => Err(NanoAIError::PromptFailed(format!("{:?}", err))),
                }
            } else {
                Err(NanoAIError::SessionCreationFailed(
                    "Session Creation Failed!".to_string(),
                ))
            }
        } else {
            Err(NanoAIError::BrowserNotSupported)
        }
    }

    pub async fn destroy_session(&mut self) -> Result<(), NanoAIError> {
        if !AI.is_undefined() {
            if let Some(session) = &self.session {
                let promise = session.destroy();
                JsFuture::from(promise)
                    .await
                    .map_err(NanoAIError::UnexpectedJsError)?;
                self.session = None;
                Ok(())
            } else {
                Err(NanoAIError::SessionCreationFailed(
                    "Session Creation Failed!".to_string(),
                ))
            }
        } else {
            Err(NanoAIError::BrowserNotSupported)
        }
    }
}