smartui 0.2.0

A terminal utility that integrates with Google Gemini API
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

use crate::error::Result;
use crate::gemini::GeminiClientTrait;
use crate::tts::TTSManager;
use async_trait::async_trait;

#[async_trait]
pub trait GenerateContent {
    async fn generate_content(&self, prompt: &str) -> Result<String>;
}

pub struct FindCommandCommand<T: GeminiClientTrait> {
    client: T,
}

impl<T: GeminiClientTrait> FindCommandCommand<T> {
    pub fn new(client: T) -> Self {
        Self { client }
    }
}

#[async_trait]
impl<T: GeminiClientTrait + Send + Sync> GenerateContent for FindCommandCommand<T> {
    async fn generate_content(&self, prompt: &str) -> Result<String> {
        let command_prompt = format!(
            "Find a command for the following task:\n\n{}",
            prompt
        );
        
        self.client.generate_content(command_prompt).await
    }
}

pub async fn execute<T: GeminiClientTrait + Send + Sync>(
    client: T,
    task: String,
    tts_manager: Option<&TTSManager>,
) -> Result<()> {
    let find_command = FindCommandCommand::new(client);
    let command = find_command.generate_content(&task).await?;
    println!("{}", command);

    if let Some(tts) = tts_manager {
        tts.speak(&command).await?;
    }

    Ok(())
}