Add a --max-tokens option

Author: SamuelSarle

src/lib.rs

@@ -36,7 +36,7 @@ pub fn generate(input: impl AsRef<str>, options: &Options) -> Result<Vec<String>
     });
 
     if let Some(max_tokens) = options.max_tokens {
-        req["max_output_tokens"] = max_tokens.into();
+        req["max_completion_tokens"] = max_tokens.into();
     }
 
     let mut resp = ureq::Agent::config_builder()

src/prompter/main.rs

@@ -15,9 +15,14 @@ struct Options {
     #[clap(flatten)]
     flags: StandardOptions,
 
+    /// Choose which model to use.
     #[clap(long, short = 'm')]
     model: Option<String>,
 
+    /// Maximum number of tokens to generate.
+    #[clap(long)]
+    max_tokens: Option<usize>,
+
     input: Option<String>,
     output: Option<String>,
 }
@@ -99,6 +104,7 @@ pub fn main() -> Result<SysexitsError> {
         .endpoint(endpoint)
         .model(model)
         .api_key(api_key)
+        .maybe_max_tokens(options.max_tokens)
         .build();
 
     let response =