diff options
Diffstat (limited to 'src/client/cohere.rs')
| -rw-r--r-- | src/client/cohere.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/client/cohere.rs b/src/client/cohere.rs index e0ef6f0..b5d6647 100644 --- a/src/client/cohere.rs +++ b/src/client/cohere.rs @@ -135,7 +135,7 @@ fn build_body(data: SendData, model: &Model) -> Result<Value> { body["chat_history"] = messages.into(); } - if let Some(v) = model.max_output_tokens { + if let Some(v) = model.max_tokens_param() { body["max_tokens"] = v.into(); } if let Some(v) = temperature { |