From 2e1dc1f7d58f037e83a184f4c9e662696d1425be Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 26 Jun 2024 13:48:55 +0800 Subject: refactor: improve azure-openai (#652) --- src/client/azure_openai.rs | 12 +++++++++--- 1 file changed, 9 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 058a53e..e3e9f16 100644 --- a/src/client/azure_openai.rs +++ b/src/client/azure_openai.rs @@ -49,7 +49,7 @@ impl AzureOpenAIClient { self.model.name() ); - debug!("AzureOpenAI Request: {url} {body}"); + debug!("AzureOpenAI Chat Completions Request: {url} {body}"); let builder = client.post(url).header("api-key", api_key).json(&body); @@ -66,9 +66,15 @@ impl AzureOpenAIClient { let body = openai_build_embeddings_body(data, &self.model); - let url = format!("{api_base}/embeddings"); + let url = format!( + "{}/openai/deployments/{}/embeddings?api-version=2024-02-01", + &api_base, + self.model.name() + ); + + debug!("AzureOpenAI Embeddings Request: {url} {body}"); - let builder = client.post(url).bearer_auth(api_key).json(&body); + let builder = client.post(url).header("api-key", api_key).json(&body); Ok(builder) } -- cgit v1.2.3