From 740ca2413a1992d151c0a9ea62c12e69321c0973 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 26 Apr 2024 07:43:35 +0800
Subject: refactor: simplify impl client trait (#445)

---
 src/client/azure_openai.rs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'src/client/azure_openai.rs')

diff --git a/src/client/azure_openai.rs b/src/client/azure_openai.rs
index 9e96692..ab8c995 100644
--- a/src/client/azure_openai.rs
+++ b/src/client/azure_openai.rs
@@ -4,7 +4,6 @@ use super::{AzureOpenAIClient, ExtraConfig, Model, ModelConfig, PromptType, Send
 use crate::utils::PromptKind;
 
 use anyhow::Result;
-use async_trait::async_trait;
 use reqwest::{Client as ReqwestClient, RequestBuilder};
 use serde::Deserialize;
 
@@ -17,8 +16,6 @@ pub struct AzureOpenAIConfig {
     pub extra: Option<ExtraConfig>,
 }
 
-openai_compatible_client!(AzureOpenAIClient);
-
 impl AzureOpenAIClient {
     list_models_fn!(AzureOpenAIConfig);
     config_get_fn!(api_base, get_api_base);
@@ -55,3 +52,9 @@ impl AzureOpenAIClient {
         Ok(builder)
     }
 }
+
+impl_client_trait!(
+    AzureOpenAIClient,
+    crate::client::openai::openai_send_message,
+    crate::client::openai::openai_send_message_streaming
+);
-- 
cgit v1.2.3