refactor: simplify impl client trait (#445)

1 files changed, 6 insertions, 3 deletions

diff --git a/src/client/openai_compatible.rs b/src/client/openai_compatible.rs
index b4cb5f4..1626b17 100644
--- a/src/client/openai_compatible.rs
+++ b/src/client/openai_compatible.rs
@@ -4,7 +4,6 @@ use super::{ExtraConfig, Model, ModelConfig, OpenAICompatibleClient, PromptType,
 use crate::utils::PromptKind;
 
 use anyhow::Result;
-use async_trait::async_trait;
 use reqwest::{Client as ReqwestClient, RequestBuilder};
 use serde::Deserialize;
 
@@ -18,8 +17,6 @@ pub struct OpenAICompatibleConfig {
     pub extra: Option<ExtraConfig>,
 }
 
-openai_compatible_client!(OpenAICompatibleClient);
-
 impl OpenAICompatibleClient {
     list_models_fn!(OpenAICompatibleConfig);
     config_get_fn!(api_key, get_api_key);
@@ -61,3 +58,9 @@ impl OpenAICompatibleClient {
         Ok(builder)
     }
 }
+
+impl_client_trait!(
+    OpenAICompatibleClient,
+    crate::client::openai::openai_send_message,
+    crate::client::openai::openai_send_message_streaming
+);