From 9c6c9f10a27d0993636b453f39d8934c95c5c2b2 Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 23 Apr 2024 18:14:47 +0800 Subject: feat: builtin models can be overwrited by models config (#429) --- src/client/azure_openai.rs | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) (limited to 'src/client/azure_openai.rs') diff --git a/src/client/azure_openai.rs b/src/client/azure_openai.rs index 1726bbe..9e96692 100644 --- a/src/client/azure_openai.rs +++ b/src/client/azure_openai.rs @@ -1,5 +1,5 @@ use super::openai::openai_build_body; -use super::{convert_models, AzureOpenAIClient, ExtraConfig, Model, ModelConfig, PromptType, SendData}; +use super::{AzureOpenAIClient, ExtraConfig, Model, ModelConfig, PromptType, SendData}; use crate::utils::PromptKind; @@ -20,6 +20,7 @@ pub struct AzureOpenAIConfig { openai_compatible_client!(AzureOpenAIClient); impl AzureOpenAIClient { + list_models_fn!(AzureOpenAIConfig); config_get_fn!(api_base, get_api_base); config_get_fn!(api_key, get_api_key); @@ -35,11 +36,6 @@ impl AzureOpenAIClient { ), ]; - pub fn list_models(local_config: &AzureOpenAIConfig) -> Vec { - let client_name = Self::name(local_config); - convert_models(client_name, &local_config.models) - } - fn request_builder(&self, client: &ReqwestClient, data: SendData) -> Result { let api_base = self.get_api_base()?; let api_key = self.get_api_key()?; -- cgit v1.2.3