From d6df1e84a7d4fece6c9107de867a1c4d7d2fee56 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 28 Apr 2024 10:14:12 +0800 Subject: refactor: extract prelude models to models.yaml (#451) --- src/client/ernie.rs | 24 ++---------------------- 1 file changed, 2 insertions(+), 22 deletions(-) (limited to 'src/client/ernie.rs') diff --git a/src/client/ernie.rs b/src/client/ernie.rs index 5cc546c..7695061 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -31,19 +31,6 @@ pub struct ErnieConfig { } impl ErnieClient { - list_models_fn!( - ErnieConfig, - [ - // https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t - ("ernie-4.0-8k", "text", 5120, 2048), - ("ernie-3.5-8k", "text", 5120, 2048), - ("ernie-3.5-4k", "text", 2048, 2048), - ("ernie-speed-8k", "text", 7168, 2048), - ("ernie-speed-128k", "text", 124000, 4096), - ("ernie-lite-8k", "text", 7168, 2048), - ("ernie-tiny-8k", "text", 7168, 2048), - ] - ); pub const PROMPTS: [PromptType<'static>; 2] = [ ("api_key", "API Key:", true, PromptKind::String), @@ -53,16 +40,9 @@ impl ErnieClient { fn request_builder(&self, client: &ReqwestClient, data: SendData) -> Result { let body = build_body(data, &self.model); - let endpoint = match self.model.name.as_str() { - "ernie-4.0-8k" => "completions_pro", - "ernie-3.5-8k" => "ernie-3.5-8k-0205", - "ernie-3.5-4k" => "ernie-3.5-4k-0205", - "ernie-speed-8k" => "ernie_speed", - _ => &self.model.name, - }; - let url = format!( - "{API_BASE}/wenxinworkshop/chat/{endpoint}?access_token={}", + "{API_BASE}/wenxinworkshop/chat/{}?access_token={}", + &self.model.name, unsafe { &ACCESS_TOKEN.0 } ); -- cgit v1.2.3