fix: prediction stream raw predict url for non dedicated endpoint

vertex-sdk-bot · copybara-github · commit b91a8aa95e89 · 2025-02-12T16:04:50.000-08:00
PiperOrigin-RevId: 726231575
diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py
@@ -2583,6 +2583,7 @@ def stream_raw_predict(
         timeout: Optional[float] = None,
     ) -> Iterator[requests.models.Response]:
         """Makes a streaming prediction request using arbitrary headers.
+        For custom model, this method is only supported for dedicated endpoint.
 
         Example usage:
             ```
@@ -2632,7 +2633,7 @@ def stream_raw_predict(
         if self.stream_raw_predict_request_url is None:
             self.stream_raw_predict_request_url = f"https://{self.location}-{constants.base.API_BASE_PATH}/v1/projects/{self.project}/locations/{self.location}/endpoints/{self.name}:streamRawPredict"
 
-        url = self.raw_predict_request_url
+        url = self.stream_raw_predict_request_url
 
         if use_dedicated_endpoint:
             self._sync_gca_resource_if_skipped()
