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/main.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index ecbfc60..589e865 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -35,6 +35,12 @@ fn main() -> Result<()> {
             .for_each(|v| println!("{}", v.name));
         exit(0);
     }
+    if cli.list_models {
+        config::MODELS
+            .iter()
+            .for_each(|(name, _)| println!("{}", name));
+        exit(0);
+    }
     let role = match &cli.role {
         Some(name) => Some(
             config
@@ -44,6 +50,9 @@ fn main() -> Result<()> {
         ),
         None => None,
     };
+    if let Some(model) = &cli.model {
+        config.write().set_model(model)?;
+    }
     config.write().role = role;
     if cli.no_highlight {
         config.write().highlight = false;
-- 
cgit v1.2.3