diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift index 0a0ad23edfb..9191ae3e6ec 100644 --- a/FirebaseVertexAI/Sources/GenerativeModel.swift +++ b/FirebaseVertexAI/Sources/GenerativeModel.swift @@ -185,10 +185,9 @@ public final class GenerativeModel: Sendable { isStreaming: true, options: requestOptions) + var responseIterator = generativeAIService.loadRequestStream(request: generateContentRequest) + .makeAsyncIterator() return AsyncThrowingStream { - var responseIterator = self.generativeAIService.loadRequestStream( - request: generateContentRequest - ).makeAsyncIterator() let response: GenerateContentResponse? do { response = try await responseIterator.next()