From 2c6aafb0d73e23b1c42ddb5095f1afb736c38c61 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 1 Sep 2024 22:18:09 +0800 Subject: refactor: openai_compatible client config (#820) --- src/client/openai_compatible.rs | 28 ++-------------------------- 1 file changed, 2 insertions(+), 26 deletions(-) (limited to 'src/client') diff --git a/src/client/openai_compatible.rs b/src/client/openai_compatible.rs index da261b3..b2797e7 100644 --- a/src/client/openai_compatible.rs +++ b/src/client/openai_compatible.rs @@ -11,8 +11,6 @@ pub struct OpenAICompatibleConfig { pub name: Option, pub api_base: Option, pub api_key: Option, - pub chat_endpoint: Option, - pub embeddings_endpoint: Option, #[serde(default)] pub models: Vec, pub patch: Option, @@ -55,18 +53,7 @@ fn prepare_chat_completions( let api_key = self_.get_api_key().ok(); let api_base = get_api_base_ext(self_)?; - let chat_endpoint = match self_.config.chat_endpoint.clone() { - Some(v) => { - if v.starts_with('/') { - v - } else { - format!("/{}", v) - } - } - None => "/chat/completions".into(), - }; - - let url = format!("{api_base}{chat_endpoint}"); + let url = format!("{api_base}/chat/completions"); let body = openai_build_chat_completions_body(data, &self_.model); @@ -83,18 +70,7 @@ fn prepare_embeddings(self_: &OpenAICompatibleClient, data: EmbeddingsData) -> R let api_key = self_.get_api_key().ok(); let api_base = get_api_base_ext(self_)?; - let embeddings_endpoint = match self_.config.embeddings_endpoint.clone() { - Some(v) => { - if v.starts_with('/') { - v - } else { - format!("/{}", v) - } - } - None => "/embeddings".into(), - }; - - let url = format!("{api_base}{embeddings_endpoint}"); + let url = format!("{api_base}/embeddings"); let body = openai_build_embeddings_body(data, &self_.model); -- cgit v1.2.3