Remove beta client duplication

The current implementation is breaking any attempt of connection pool usage.

Author: madwork

lib/middleware/beta.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module OpenAI
+  class BetaMiddleware < Faraday::Middleware
+    BETA_REGEX = %r{^/#{OpenAI.configuration.api_version}/(assistants|batches|threads|vector_stores)}i
+
+    def on_request(env)
+      if env[:url].path.match?(BETA_REGEX)
+        env[:request_headers].merge!(
+          {
+            "OpenAI-Beta" => "assistants=v2"
+          }
+        )
+      end
+    end
+  end
+end

lib/openai.rb

@@ -90,3 +90,5 @@ def self.rough_token_count(content = "")
     [1, estimate].max
   end
 end
+
+require_relative "middleware/beta"

lib/openai/assistants.rb

@@ -1,9 +1,7 @@
 module OpenAI
   class Assistants
-    BETA_VERSION = "v2".freeze
-
     def initialize(client:)
-      @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
+      @client = client
     end
 
     def list

lib/openai/batches.rb

@@ -1,7 +1,7 @@
 module OpenAI
   class Batches
     def initialize(client:)
-      @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
+      @client = client
     end
 
     def list(parameters: {})

lib/openai/client.rb

@@ -107,12 +107,6 @@ def azure?
       @api_type&.to_sym == :azure
     end
 
-    def beta(apis)
-      dup.tap do |client|
-        client.add_headers("OpenAI-Beta": apis.map { |k, v| "#{k}=#{v}" }.join(";"))
-      end
-    end
-
     private
 
     attr_reader :connection, :multipart_connection
@@ -122,6 +116,7 @@ def build_connection(multipart: false)
         faraday.options[:timeout] = @request_timeout
         faraday.request(:multipart) if multipart
         faraday.use MiddlewareErrors if @log_errors
+        faraday.use BetaMiddleware
         faraday.response :raise_error
         faraday.response :json
       end

lib/openai/images.rb

@@ -1,6 +1,6 @@
 module OpenAI
   class Images
-    def initialize(client: nil)
+    def initialize(client:)
       @client = client
     end
 

lib/openai/messages.rb

@@ -1,7 +1,7 @@
 module OpenAI
   class Messages
     def initialize(client:)
-      @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
+      @client = client
     end
 
     def list(thread_id:, parameters: {})

lib/openai/run_steps.rb

@@ -1,7 +1,7 @@
 module OpenAI
   class RunSteps
     def initialize(client:)
-      @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
+      @client = client
     end
 
     def list(thread_id:, run_id:, parameters: {})

lib/openai/runs.rb

@@ -1,7 +1,7 @@
 module OpenAI
   class Runs
     def initialize(client:)
-      @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
+      @client = client
     end
 
     def list(thread_id:, parameters: {})

lib/openai/threads.rb

@@ -1,7 +1,7 @@
 module OpenAI
   class Threads
     def initialize(client:)
-      @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
+      @client = client
     end
 
     def retrieve(id:)