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
This commit is contained in:
Roo Code 2025-11-03 19:51:59 +00:00
parent 0878604b4e
commit ff2ec78f96

View file

@ -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")
)
}
/**