fix embedding for azure

asklar · Aug 24, 2023 · 1cf807a · 1cf807a
1 parent 6f9b808
commit 1cf807a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/OpenAI.WinRT.nuspec b/OpenAI.WinRT.nuspec
@@ -2,7 +2,7 @@
 <package >
   <metadata>
     <id>OpenAI.WinRT</id>
-    <version>0.0.16</version>
+    <version>0.0.17</version>
     <title>OpenAI.WinRT</title>
     <authors>Alexander Sklar</authors>
     <requireLicenseAcceptance>false</requireLicenseAcceptance>

diff --git a/OpenAIClient.cpp b/OpenAIClient.cpp
@@ -33,10 +33,15 @@ namespace winrt::OpenAI::implementation
       auto client = winrt::OpenAI::OpenAIClient();
       client.UseBearerTokenAuthorization(false);
       client.ApiKey(apiKey);
-      constexpr std::wstring_view uriTemplate = L"{}/openai/deployments/{}/chat/completions?api-version=2023-03-15-preview";
+      constexpr std::wstring_view completionUriTemplate = L"{}/openai/deployments/{}/chat/completions?api-version=2023-03-15-preview";
+
+      auto completionUri = winrt::Windows::Foundation::Uri{ std::vformat(completionUriTemplate, std::make_wformat_args(endpoint.AbsoluteUri(), deployment))};
+      client.CompletionUri(completionUri);
+
+      constexpr std::wstring_view embeddingUriTemplate = L"{}/openai/deployments/{}/embeddings?api-version=2023-05-15";
+      auto embeddingUri = winrt::Windows::Foundation::Uri{ std::vformat(embeddingUriTemplate, std::make_wformat_args(endpoint.AbsoluteUri(), deployment)) };
+      client.EmbeddingUri(embeddingUri);
 
-      auto uri = winrt::Windows::Foundation::Uri{ std::vformat(uriTemplate, std::make_wformat_args(endpoint.AbsoluteUri(), deployment))};
-      client.CompletionUri(uri);
       client.IsChatModel(true);
 
       return client;