Allow user to choose a conversation model

examples/conversation.rs

@@ -25,7 +25,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     tracing::info!("Starting conversation...");
 
-    let mut conversation = Dialogue::new();
+    let mut conversation = Dialogue::new("gemini-pro");
     loop {
         let message: String = Input::with_theme(&ColorfulTheme::default())
             .with_prompt("user")

src/client.rs

@@ -116,7 +116,7 @@ impl<T: TokenProvider + Clone> GeminiClient<T> {
     }
 
     /// Prompts a conversation to the model.
-    pub async fn prompt_conversation(&self, messages: &[Message]) -> Result<Message> {
+    pub async fn prompt_conversation(&self, messages: &[Message], model: &str) -> Result<Message> {
         let request = GenerateContentRequest {
             contents: messages
                 .iter()
@@ -129,7 +129,7 @@ impl<T: TokenProvider + Clone> GeminiClient<T> {
             tools: None,
         };
 
-        let response = self.generate_content(&request, "gemini-pro").await?;
+        let response = self.generate_content(&request, model).await?;
 
         // Check for errors in the response.
         let mut candidates = GeminiClient::<T>::collect_text_from_response(&response)?;

src/dialogue.rs

@@ -48,12 +48,16 @@ impl Message {
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
 pub struct Dialogue {
+    model: String,
     messages: Vec<Message>,
 }
 
 impl Dialogue {
-    pub fn new() -> Self {
+    pub fn new(model: &str) -> Self {
-        Dialogue { messages: vec![] }
+        Dialogue {
+            model: model.to_string(),
+            messages: vec![],
+        }
     }
 
     pub async fn do_turn<T: TokenProvider + Clone>(
@@ -62,7 +66,9 @@ impl Dialogue {
         message: &str,
     ) -> Result<Message> {
         self.messages.push(Message::new(Role::User, message));
-        let response = gemini.prompt_conversation(&self.messages).await?;
+        let response = gemini
+            .prompt_conversation(&self.messages, &self.model)
+            .await?;
         self.messages.push(response.clone());
         Ok(response)
     }