From 8dacea4deb0fe67c0cdbe15bd47e6e50c1dd12a1 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 24 Apr 2024 21:17:01 +0800 Subject: refactor: update mistral models (#435) --- src/client/mistral.rs | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/client/mistral.rs b/src/client/mistral.rs index e294e6a..9dbd831 100644 --- a/src/client/mistral.rs +++ b/src/client/mistral.rs @@ -10,12 +10,10 @@ use serde::Deserialize; const API_URL: &str = "https://api.mistral.ai/v1/chat/completions"; -const MODELS: [(&str, usize, &str); 5] = [ +const MODELS: [(&str, usize, &str); 3] = [ // https://docs.mistral.ai/platform/endpoints/ - ("open-mistral-7b", 32000, "text"), - ("open-mixtral-8x7b", 32000, "text"), + ("open-mixtral-8x22b", 64000, "text"), ("mistral-small-latest", 32000, "text"), - ("mistral-medium-latest", 32000, "text"), ("mistral-large-latest", 32000, "text"), ]; -- cgit v1.2.3