use log::{debug, info};
#[allow(dead_code)]
pub struct GeminiClient {
api_key: String,
model: String,
}
impl GeminiClient {
pub fn setup(api_key: &str) -> Self {
info!("Setting up GeminiClient with API key");
GeminiClient {
api_key: api_key.to_string(),
model: String::new(),
}
}
pub fn model(mut self, model: &str) -> Self {
info!("Setting model to {}", model);
self.model = model.to_string();
self
}
pub fn generate_content(&self, prompt: &str) -> String {
info!("Generating content for prompt: '{}'", prompt);
let response = format!(
"Generated content for prompt: '{}', using model: '{}'",
prompt, self.model
);
debug!("Generated response: {}", response);
response
}
}