Add support request timeout

Author: naoa

bin/console

@@ -12,7 +12,7 @@ require "cohere"
 # Pry.start
 
 client = Cohere::Client.new(
-  api_key: ENV['COHERE_API_KEY']
+  ENV['COHERE_API_KEY']
 )
 
 require "irb"

lib/cohere/client.rb

@@ -8,8 +8,9 @@ class Client
 
     ENDPOINT_URL = "https://api.cohere.ai/v1"
 
-    def initialize(api_key)
+    def initialize(api_key, timeout: nil)
       @api_key = api_key
+      @timeout = timeout
     end
 
     def chat(
@@ -181,7 +182,7 @@ def summarize(
 
     # standard:disable Lint/DuplicateMethods
     def connection
-      @connection ||= Faraday.new(url: ENDPOINT_URL) do |faraday|
+      @connection ||= Faraday.new(url: ENDPOINT_URL, request: { timeout: @timeout }) do |faraday|
         if api_key
           faraday.request :authorization, :Bearer, api_key
         end