From 8e5d4e55b1a5158a1f35adaf044dec545159f426 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 6 Mar 2024 08:35:40 +0800 Subject: refactor: rename model's `max_tokens` to `max_input_tokens` (#339) BREAKING CHANGE: rename model's `max_tokens` to `max_input_tokens` --- src/client/mistral.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/client/mistral.rs') diff --git a/src/client/mistral.rs b/src/client/mistral.rs index 1bb4889..ba96116 100644 --- a/src/client/mistral.rs +++ b/src/client/mistral.rs @@ -11,6 +11,7 @@ use serde::Deserialize; const API_URL: &str = "https://api.mistral.ai/v1/chat/completions"; const MODELS: [(&str, usize, &str); 5] = [ + // https://docs.mistral.ai/platform/endpoints/ ("mistral-small-latest", 32000, "text"), ("mistral-medium-latest", 32000, "text"), ("mistral-larget-latest", 32000, "text"), @@ -39,10 +40,10 @@ impl MistralClient { let client_name = Self::name(local_config); MODELS .into_iter() - .map(|(name, max_tokens, capabilities)| { + .map(|(name, max_input_tokens, capabilities)| { Model::new(client_name, name) .set_capabilities(capabilities.into()) - .set_max_tokens(Some(max_tokens)) + .set_max_input_tokens(Some(max_input_tokens)) .set_tokens_count_factors(OPENAI_TOKENS_COUNT_FACTORS) }) .collect() -- cgit v1.2.3