fix: avoid duplicate requests in pooled requester

Author: vicentereig

Gemfile

@@ -14,4 +14,5 @@ end
 group :test do
   gem "aruba", "~> 2.3"
   gem "webmock", "~> 3.25"
+  gem "webrick", "~> 1.8"
 end

exa-ai-ruby.gemspec

@@ -37,4 +37,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "rubocop", "~> 1.64"
   spec.add_development_dependency "aruba", "~> 2.2"
   spec.add_development_dependency "webmock", "~> 3.23"
+  spec.add_development_dependency "webrick", "~> 1.8"
 end

lib/exa/internal/transport/pooled_net_requester.rb

@@ -47,7 +47,16 @@ def execute(request)
             end
 
             _, response = enum.next
-            body = Exa::Internal::Util.fused_enum(enum)
+            body_stream = Enumerator.new do |y|
+              loop do
+                begin
+                  y << enum.next
+                rescue StopIteration
+                  break
+                end
+              end
+            end
+            body = Exa::Internal::Util.fused_enum(body_stream)
             [Integer(response.code), response, body]
           end
         end

lib/exa/responses/search_response.rb

@@ -8,7 +8,7 @@ class SearchResponse < T::Struct
       const :search_type, T.nilable(String)
       const :results, T::Array[ResultWithContent]
       const :context, T.nilable(String)
-      const :cost_dollars, T.nilable(Float)
+      const :cost_dollars, T.nilable(T.any(Float, T::Hash[Symbol, T.untyped]))
 
       def self.from_hash(hash)
         sym = Helpers.symbolize_keys(hash)
@@ -18,9 +18,22 @@ def self.from_hash(hash)
           search_type: sym[:searchType],
           results: Array(sym[:results]).map { ResultWithContent.from_hash(_1) },
           context: sym[:context],
-          cost_dollars: sym[:costDollars]&.to_f
+          cost_dollars: normalize_cost(sym[:costDollars])
         )
       end
+
+      def self.normalize_cost(value)
+        case value
+        when nil
+          nil
+        when Numeric
+          value.to_f
+        when Hash
+          Exa::Responses::Helpers.symbolize_keys(value)
+        else
+          value
+        end
+      end
     end
 
     class FindSimilarResponse < T::Struct

test/transport/pooled_net_requester_test.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "test_helper"
+require "webrick"
+
+class PooledNetRequesterTest < Minitest::Test
+  def setup
+    @server = nil
+    @server_thread = nil
+  end
+
+  def teardown
+    @server&.shutdown
+    @server_thread&.join
+  end
+
+  def test_single_http_request_per_execute_call
+    requests = Queue.new
+    port = free_port
+    @server = WEBrick::HTTPServer.new(
+      Port: port,
+      BindAddress: "127.0.0.1",
+      Logger: WEBrick::Log.new(File::NULL),
+      AccessLog: []
+    )
+    @server.mount_proc "/echo" do |req, res|
+      requests << req.body
+      res.status = 200
+      res["Content-Type"] = "application/json"
+      res.body = {ok: true}.to_json
+    end
+    @server_thread = Thread.new { @server.start }
+
+    requester = Exa::Internal::Transport::PooledNetRequester.new(size: 1)
+    client = Exa::Client.new(
+      api_key: "test-key",
+      base_url: "http://127.0.0.1:#{port}",
+      requester: requester
+    )
+
+    response = client.request(method: :post, path: "echo", body: {hello: "world"})
+    assert_equal({ok: true}, response)
+
+    assert_equal 1, requests.length, "expected a single HTTP request but multiple were made"
+  end
+
+  private
+
+  def free_port
+    server = TCPServer.new("127.0.0.1", 0)
+    port = server.addr[1]
+    server.close
+    port
+  end
+end