fix: prediction stream raw predict url for non dedicated endpoint

PiperOrigin-RevId: 725777307
googleapis · Feb 12, 2025 · d9577d5 · d9577d5
1 parent 0d73929
commit d9577d5
Showing 1 changed file with 2 additions and 1 deletion.
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()