feat: add `.set max_output_tokens` (#468)

1 files changed, 7 insertions, 2 deletions

diff --git a/src/client/common.rs b/src/client/common.rs
index 5ddaf87..ee190d5 100644
--- a/src/client/common.rs
+++ b/src/client/common.rs
@@ -115,8 +115,7 @@ macro_rules! register_client {
             None
             $(.or_else(|| $client::init(config)))+
             .ok_or_else(|| {
-                let model = config.read().model.clone();
-                anyhow::anyhow!("Unknown client '{}'", &model.client_name)
+                anyhow::anyhow!("Unknown client '{}'", &config.read().model.client_name)
             })
         }
 
@@ -254,6 +253,10 @@ macro_rules! client_common_fns {
             &self.model
         }
 
+        fn model_mut(&mut self) -> &mut Model {
+            &mut self.model
+        }
+
         fn set_model(&mut self, model: Model) {
             self.model = model;
         }
@@ -323,6 +326,8 @@ pub trait Client: Sync + Send {
 
     fn model(&self) -> &Model;
 
+    fn model_mut(&mut self) -> &mut Model;
+
     fn set_model(&mut self, model: Model);
 
     fn build_client(&self) -> Result<ReqwestClient> {