feat: [STG-1607] Yield finished SSE event instead of silently dropping it

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 8
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase%2Fstagehand-573d364768ac1902ee5ed8b2485d3b293bda0ea8ff7898aef1a3fd6be79b594a.yml
 openapi_spec_hash: 107ec840f4330885dd2232a05a66fed7
-config_hash: 0209737a4ab2a71afececb0ff7459998
+config_hash: ab1c1122e862f9bdb6db4df36d489701

lib/stagehand/internal/stream.rb

@@ -16,16 +16,14 @@ class Stream
       # @return [Enumerable<generic<Elem>>]
       private def iterator
         # rubocop:disable Metrics/BlockLength
+        # rubocop:disable Lint/DuplicateBranch
         @iterator ||= Stagehand::Internal::Util.chain_fused(@stream) do |y|
-          consume = false
-
           @stream.each do |msg|
-            next if consume
-
             case msg
             in {data: String => data} if data.start_with?("{\"data\":{\"status\":\"finished\"")
-              consume = true
-              next
+              decoded = JSON.parse(data, symbolize_names: true)
+              unwrapped = Stagehand::Internal::Util.dig(decoded, @unwrap)
+              y << Stagehand::Internal::Type::Converter.coerce(@model, unwrapped)
             in {data: String => data} if data.start_with?("error")
               decoded = Kernel.then do
                 JSON.parse(data, symbolize_names: true)
@@ -49,6 +47,7 @@ class Stream
             end
           end
         end
+        # rubocop:enable Lint/DuplicateBranch
         # rubocop:enable Metrics/BlockLength
       end
     end