Skip to main content

vtcode_core/llm/providers/gemini/
client.rs

1use super::*;
2
3#[async_trait]
4impl LLMClient for GeminiProvider {
5    async fn generate(&mut self, prompt: &str) -> Result<LLMResponse, LLMError> {
6        let request = LLMRequest {
7            messages: vec![Message::user(prompt.to_string())],
8            model: self.model.to_string(),
9            ..Default::default()
10        };
11        Ok(LLMProvider::generate(self, request).await?)
12    }
13
14    fn model_id(&self) -> &str {
15        &self.model
16    }
17}