Skip to content

Commit ff2ec78

Browse files
committed
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
1 parent 0878604 commit ff2ec78

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/api/providers/openai.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -428,8 +428,12 @@ export class OpenAiHandler extends BaseProvider implements SingleCompletionHandl
428428

429429
private _isNvidiaApi(baseUrl?: string): boolean {
430430
const urlHost = this._getUrlHost(baseUrl)
431-
// NVIDIA API endpoints typically use integrate.api.nvidia.com or build.nvidia.com
432-
return urlHost.includes("nvidia.com")
431+
// NVIDIA API endpoints for AI models
432+
return (
433+
urlHost === "integrate.api.nvidia.com" ||
434+
urlHost === "build.nvidia.com" ||
435+
urlHost.endsWith(".api.nvidia.com")
436+
)
433437
}
434438

435439
/**

0 commit comments

Comments
 (0)