Browse Source

Merge pull request #66 from RooVetGit/alert-autofix-1

Fix code scanning alert no. 1: Incomplete URL substring sanitization
Matt Rubens 1 year ago
parent
commit
20c7c575e2
1 changed files with 2 additions and 1 deletions
  1. 2 1
      src/api/providers/openai.ts

+ 2 - 1
src/api/providers/openai.ts

@@ -17,7 +17,8 @@ export class OpenAiHandler implements ApiHandler {
 	constructor(options: ApiHandlerOptions) {
 	constructor(options: ApiHandlerOptions) {
 		this.options = options
 		this.options = options
 		// Azure API shape slightly differs from the core API shape: https://github.com/openai/openai-node?tab=readme-ov-file#microsoft-azure-openai
 		// Azure API shape slightly differs from the core API shape: https://github.com/openai/openai-node?tab=readme-ov-file#microsoft-azure-openai
-		if (this.options.openAiBaseUrl?.toLowerCase().includes("azure.com")) {
+		const urlHost = new URL(this.options.openAiBaseUrl ?? "").host;
+		if (urlHost === "azure.com" || urlHost.endsWith(".azure.com")) {
 			this.client = new AzureOpenAI({
 			this.client = new AzureOpenAI({
 				baseURL: this.options.openAiBaseUrl,
 				baseURL: this.options.openAiBaseUrl,
 				apiKey: this.options.openAiApiKey,
 				apiKey: this.options.openAiApiKey,