small Update

PowerShell · Jun 22, 2023 · a9c97c7 · a9c97c7
1 parent a4b0e51
commit a9c97c7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/code/OpenAI.cs b/src/code/OpenAI.cs
@@ -46,14 +46,14 @@ public OpenAI()
             }
 
 
-            if (endpoint.EndsWith(".azure-api.net", StringComparison.Ordinal))
+            if (endpoint.EndsWith(".azure-api.net", StringComparison.Ordinal) || endpoint.EndsWith(".azure-api.net/", StringComparison.Ordinal))
             {
                 AzureKeyCredentialPolicy policy = new AzureKeyCredentialPolicy(new AzureKeyCredential(apiKey), "Ocp-Apim-Subscription-Key");
                 options.AddPolicy(policy, Azure.Core.HttpPipelinePosition.PerRetry);
 
                 client = new OpenAIClient(new Uri(endpoint), new AzureKeyCredential("placeholder"), options);
             }
-            else if (endpoint.EndsWith(".openai.azure.com", StringComparison.Ordinal))
+            else if (endpoint.EndsWith(".openai.azure.com", StringComparison.Ordinal) || endpoint.EndsWith(".openai.azure.com/", StringComparison.Ordinal))
             {
                 client = new OpenAIClient(new Uri(endpoint), new AzureKeyCredential(apiKey));
             }
@@ -205,7 +205,7 @@ internal string GetCompletion(string prompt, bool debug, CancellationToken cance
                 return output;
 
             }
-            catch (Azure.RequestFailedException e)
+            catch (RequestFailedException e)
             {
                 return $"{PSStyle.Instance.Foreground.BrightRed}HTTP EXCEPTION: {e.Message}";
             }