lib/voicebase/client.rb in voicebase-client-ruby-1.2.4 vs lib/voicebase/client.rb in voicebase-client-ruby-1.3.0

@@ -1,6 +1,7 @@
 module VoiceBase
+  class UnknownApiVersionError < StandardError; end;
   class Client
     include HTTParty
 
     attr_accessor :args
     attr_accessor :api_host
@@ -28,14 +29,19 @@
 
       if ENV['VOICEBASE_BEARER_TOKEN']
         @token = VoiceBase::Client::Token.new(ENV['VOICEBASE_BEARER_TOKEN'])
       end
 
-      if @api_version.to_f < 2.0
+      if @api_version.to_i < 2
         self.extend(VoiceBase::V1::Client)
-      else
+      elsif  @api_version.to_i == 2
         self.extend(VoiceBase::V2::Client)
+      elsif @api_version.to_i == 3
+        self.extend(VoiceBase::V3::Client)
+      else
+        raise UnknownApiVersionError
       end
+
     end
 
     def uri
       @api_host + @api_endpoint
     end