From 3fefa79819608a7b269de2eba46440d2f4836a4e Mon Sep 17 00:00:00 2001
From: Paul Beusterien <paulbeusterien@google.com>
Date: Wed, 26 Feb 2025 09:14:50 -0800
Subject: [PATCH] fix

---
 FirebaseVertexAI/Sources/GenerativeModel.swift | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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()