mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-08-28 05:27:24 +00:00
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:
parent
0878604b4e
commit
ff2ec78f96
1 changed files with 6 additions and 2 deletions
|
|
@ -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")
|
||||
)
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue