Merge remote-tracking branch 'upstream/main' into improving-ci-coverage-and-configuring-.coveragerc

Author: leseb

src/ogx/providers/remote/inference/nvidia/utils.py

@@ -4,8 +4,13 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+from urllib.parse import urlparse
+
 from . import NVIDIAConfig
 
+_NVIDIA_HOSTED_HOSTNAME = "integrate.api.nvidia.com"
+
 
 def _is_nvidia_hosted(config: NVIDIAConfig) -> bool:
-    return "integrate.api.nvidia.com" in str(config.base_url)
+    hostname = urlparse(str(config.base_url)).hostname
+    return hostname == _NVIDIA_HOSTED_HOSTNAME