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

(limited to 'src/cli.rs')

diff --git a/src/cli.rs b/src/cli.rs
index cfc1476..9211878 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -3,6 +3,9 @@ use clap::Parser;
 #[derive(Parser, Debug)]
 #[command(author, version, about, long_about = None)]
 pub struct Cli {
+    /// Choose a model
+    #[clap(short, long)]
+    pub model: Option<String>,
     /// Add a GPT prompt
     #[clap(short, long)]
     pub prompt: Option<String>,
@@ -15,6 +18,9 @@ pub struct Cli {
     /// List all roles
     #[clap(long)]
     pub list_roles: bool,
+    /// List all models
+    #[clap(long)]
+    pub list_models: bool,
     /// Select a role
     #[clap(short, long)]
     pub role: Option<String>,
-- 
cgit v1.2.3