refactor: improve azure-openai (#652)

1 files changed, 9 insertions, 3 deletions

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)
     }