From ff2ec78f96aff5b4c501e20301ffe7f88cf113e7 Mon Sep 17 00:00:00 2001
From: Roo Code <roomote@roocode.com>
Date: Mon, 3 Nov 2025 19:51:59 +0000
Subject: [PATCH] refactor: make NVIDIA API hostname detection more specific

- Changed from broad .includes("nvidia.com") to specific hostnames
- Now checks for integrate.api.nvidia.com, build.nvidia.com, or *.api.nvidia.com
- Follows review feedback for more precise endpoint detection
---
 src/api/providers/openai.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/api/providers/openai.ts b/src/api/providers/openai.ts
index a20f6ce977..75e96f0a26 100644
--- a/src/api/providers/openai.ts
+++ b/src/api/providers/openai.ts
@@ -428,8 +428,12 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl
 
 	private _isNvidiaApi(baseUrl?: string): boolean {
 		const urlHost = this._getUrlHost(baseUrl)
-		// NVIDIA API endpoints typically use integrate.api.nvidia.com or build.nvidia.com
-		return urlHost.includes("nvidia.com")
+		// NVIDIA API endpoints for AI models
+		return (
+			urlHost === "integrate.api.nvidia.com" ||
+			urlHost === "build.nvidia.com" ||
+			urlHost.endsWith(".api.nvidia.com")
+		)
 	}
 
 	/**