From 1ef97b2f32caecdb2eef9fe3755390f36d96eae5 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 16 Mar 2023 17:02:09 +0800
Subject: feat: support multiple models (#71)

---
 src/repl/handler.rs | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/repl/handler.rs')

diff --git a/src/repl/handler.rs b/src/repl/handler.rs
index 159127c..014a851 100644
--- a/src/repl/handler.rs
+++ b/src/repl/handler.rs
@@ -12,6 +12,7 @@ use std::cell::RefCell;
 
 pub enum ReplCmd {
     Submit(String),
+    SetModel(String),
     SetRole(String),
     UpdateConfig(String),
     Prompt(String),
@@ -65,6 +66,10 @@ impl ReplCmdHandler {
                 self.config.write().save_conversation(&input, &buffer)?;
                 *self.reply.borrow_mut() = buffer;
             }
+            ReplCmd::SetModel(name) => {
+                self.config.write().set_model(&name)?;
+                print_now!("\n");
+            }
             ReplCmd::SetRole(name) => {
                 let output = self.config.write().change_role(&name)?;
                 print_now!("{}\n\n", output.trim_end());
-- 
cgit v1.2.3