Add ability to pass Faraday config block to client

Author: alexrudall

README.md

@@ -108,6 +108,16 @@ OpenAI.configure do |config|
 end
 ```
 
+#### Verbose Logging
+
+You can pass Faraday connection options to the client in a block, eg. to enable verbose logging:
+
+```ruby
+  client = OpenAI::Client.new do |client|
+    client.response :logger, ::Logger.new(STDOUT), bodies: true
+  end
+```
+
 #### Azure
 
 To use the [Azure OpenAI Service](https://learn.microsoft.com/en-us/azure/cognitive-services/openai/) API, you can configure the gem like this:

lib/openai/client.rb

@@ -12,13 +12,15 @@ class Client
       extra_headers
     ].freeze
     attr_reader *CONFIG_KEYS
+    attr_reader :faraday_config
 
-    def initialize(config = {})
+    def initialize(config = {}, &faraday_config)
       CONFIG_KEYS.each do |key|
         # Set instance variables like api_type & access_token. Fall back to global config
         # if not present.
         instance_variable_set("@#{key}", config[key] || OpenAI.configuration.send(key))
       end
+      @faraday_config = faraday_config
     end
 
     def chat(parameters: {})

lib/openai/http.rb

@@ -71,12 +71,18 @@ def to_json_stream(user_proc:)
     end
 
     def conn(multipart: false)
-      Faraday.new do |f|
+      connection = Faraday.new do |f|
         f.options[:timeout] = @request_timeout
         f.request(:multipart) if multipart
         f.response :raise_error
         f.response :json
       end
+
+      if @faraday_config
+        @faraday_config.call(connection)
+      end
+
+      connection
     end
 
     def uri(path:)

spec/openai/client/chat_spec.rb

@@ -4,11 +4,7 @@
       let(:messages) { [{ role: "user", content: "Hello!" }] }
       let(:stream) { false }
       let(:response) do
-        client = OpenAI::Client.new do |client|
-          client.response :logger, ::Logger.new(STDOUT), bodies: true
-        end
-
-        client.chat(
+        OpenAI::Client.new.chat(
           parameters: {
             model: model,
             messages: messages,

spec/openai/client/client_spec.rb

@@ -119,4 +119,18 @@
       expect(client.send(:headers)["OpenAI-Beta"]).to eq "assistants=v1"
     end
   end
+
+  context "with a block" do
+    let(:client) do
+      OpenAI::Client.new do |client|
+        client.response :logger, ::Logger.new(STDOUT), bodies: true
+      end
+    end
+
+    it "sets the logger" do
+      connection = Faraday.new
+      client.faraday_config.call(connection)
+      expect(connection.builder.handlers).to include Faraday::Response::Logger
+    end
+  end
 end