feat: support multiple models (#71)

1 files changed, 6 insertions, 0 deletions

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>,