summaryrefslogtreecommitdiffstats
path: root/src/client/openai_compatible.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/openai_compatible.rs')
-rw-r--r--src/client/openai_compatible.rs8
1 files changed, 2 insertions, 6 deletions
diff --git a/src/client/openai_compatible.rs b/src/client/openai_compatible.rs
index f25d931..b4cb5f4 100644
--- a/src/client/openai_compatible.rs
+++ b/src/client/openai_compatible.rs
@@ -1,5 +1,5 @@
use super::openai::openai_build_body;
-use super::{convert_models, ExtraConfig, Model, ModelConfig, OpenAICompatibleClient, PromptType, SendData};
+use super::{ExtraConfig, Model, ModelConfig, OpenAICompatibleClient, PromptType, SendData};
use crate::utils::PromptKind;
@@ -21,6 +21,7 @@ pub struct OpenAICompatibleConfig {
openai_compatible_client!(OpenAICompatibleClient);
impl OpenAICompatibleClient {
+ list_models_fn!(OpenAICompatibleConfig);
config_get_fn!(api_key, get_api_key);
pub const PROMPTS: [PromptType<'static>; 5] = [
@@ -36,11 +37,6 @@ impl OpenAICompatibleClient {
),
];
- pub fn list_models(local_config: &OpenAICompatibleConfig) -> Vec<Model> {
- let client_name = Self::name(local_config);
- convert_models(client_name, &local_config.models)
- }
-
fn request_builder(&self, client: &ReqwestClient, data: SendData) -> Result<RequestBuilder> {
let api_key = self.get_api_key().ok();