azure_openai_client 0.0.1 → 0.0.3

data/lib/azure_openai_client/api/default_api.rb

@@ -1,16 +1,17 @@
-# frozen_string_literal: true
-
 =begin
 #Azure OpenAI Service API
 
 #Azure OpenAI APIs for completions and search
 
-OpenAPI spec version: 2023-05-15
+The version of the OpenAPI document: 2023-05-15
+
+Generated by: https://openapi-generator.tech
+OpenAPI Generator version: 6.6.0
 
-Generated by: https://github.com/swagger-api/swagger-codegen.git
-Swagger Codegen version: 3.0.42
 =end
 
+require 'cgi'
+
 module AzureOpenaiClient
   class DefaultApi
     attr_accessor :api_client
@@ -18,42 +19,41 @@ module AzureOpenaiClient
     def initialize(api_client = ApiClient.default)
       @api_client = api_client
     end
-
     # Creates a completion for the chat message
-    # @param body
-    # @param api_version
-    # @param deployment_id
+    # @param deployment_id [String] 
+    # @param api_version [String] 
+    # @param chat_completions_create_request [ChatCompletionsCreateRequest] 
     # @param [Hash] opts the optional parameters
-    # @return [InlineResponse2002]
-    def chat_completions_create(body, api_version, deployment_id, opts = {})
-      data, _status_code, _headers = chat_completions_create_with_http_info(body, api_version, deployment_id, opts)
+    # @return [ChatCompletionsCreate200Response]
+    def chat_completions_create(deployment_id, api_version, chat_completions_create_request, opts = {})
+      data, _status_code, _headers = chat_completions_create_with_http_info(deployment_id, api_version, chat_completions_create_request, opts)
       data
     end
 
     # Creates a completion for the chat message
-    # @param body
-    # @param api_version
-    # @param deployment_id
+    # @param deployment_id [String] 
+    # @param api_version [String] 
+    # @param chat_completions_create_request [ChatCompletionsCreateRequest] 
     # @param [Hash] opts the optional parameters
-    # @return [Array<(InlineResponse2002, Integer, Hash)>] InlineResponse2002 data, response status code and response headers
-    def chat_completions_create_with_http_info(body, api_version, deployment_id, opts = {})
+    # @return [Array<(ChatCompletionsCreate200Response, Integer, Hash)>] ChatCompletionsCreate200Response data, response status code and response headers
+    def chat_completions_create_with_http_info(deployment_id, api_version, chat_completions_create_request, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: DefaultApi.chat_completions_create ...'
       end
-      # verify the required parameter 'body' is set
-      if @api_client.config.client_side_validation && body.nil?
-        fail ArgumentError, "Missing the required parameter 'body' when calling DefaultApi.chat_completions_create"
+      # verify the required parameter 'deployment_id' is set
+      if @api_client.config.client_side_validation && deployment_id.nil?
+        fail ArgumentError, "Missing the required parameter 'deployment_id' when calling DefaultApi.chat_completions_create"
       end
       # verify the required parameter 'api_version' is set
       if @api_client.config.client_side_validation && api_version.nil?
         fail ArgumentError, "Missing the required parameter 'api_version' when calling DefaultApi.chat_completions_create"
       end
-      # verify the required parameter 'deployment_id' is set
-      if @api_client.config.client_side_validation && deployment_id.nil?
-        fail ArgumentError, "Missing the required parameter 'deployment_id' when calling DefaultApi.chat_completions_create"
+      # verify the required parameter 'chat_completions_create_request' is set
+      if @api_client.config.client_side_validation && chat_completions_create_request.nil?
+        fail ArgumentError, "Missing the required parameter 'chat_completions_create_request' when calling DefaultApi.chat_completions_create"
       end
       # resource path
-      local_var_path = '/deployments/{deployment-id}/chat/completions'.sub('{' + 'deployment-id' + '}', deployment_id.to_s)
+      local_var_path = '/deployments/{deployment-id}/chat/completions'.sub('{' + 'deployment-id' + '}', CGI.escape(deployment_id.to_s))
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -64,25 +64,34 @@ module AzureOpenaiClient
       # HTTP header 'Accept' (if needed)
       header_params['Accept'] = @api_client.select_header_accept(['application/json'])
       # HTTP header 'Content-Type'
-      header_params['Content-Type'] = @api_client.select_header_content_type(['application/json'])
+      content_type = @api_client.select_header_content_type(['application/json'])
+      if !content_type.nil?
+        header_params['Content-Type'] = content_type
+      end
 
       # form parameters
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:body] || @api_client.object_to_http_body(body)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(chat_completions_create_request)
 
-      return_type = opts[:return_type] || 'InlineResponse2002'
+      # return_type
+      return_type = opts[:debug_return_type] || 'ChatCompletionsCreate200Response'
 
-      auth_names = opts[:auth_names] || ['apiKey', 'bearer']
-      data, status_code, headers = @api_client.call_api(:POST, local_var_path,
-        header_params: header_params,
-        query_params: query_params,
-        form_params: form_params,
-        body: post_body,
-        auth_names: auth_names,
-        return_type: return_type)
+      # auth_names
+      auth_names = opts[:debug_auth_names] || ['apiKey', 'bearer']
 
+      new_options = opts.merge(
+        :operation => :"DefaultApi.chat_completions_create",
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type
+      )
+
+      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
       if @api_client.config.debugging
         @api_client.config.logger.debug "API called: DefaultApi#chat_completions_create\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end
@@ -90,40 +99,40 @@ module AzureOpenaiClient
     end
 
     # Creates a completion for the provided prompt, parameters and chosen model.
-    # @param body
-    # @param api_version
-    # @param deployment_id
+    # @param deployment_id [String] 
+    # @param api_version [String] 
+    # @param completions_create_request [CompletionsCreateRequest] 
     # @param [Hash] opts the optional parameters
-    # @return [InlineResponse200]
-    def completions_create(body, api_version, deployment_id, opts = {})
-      data, _status_code, _headers = completions_create_with_http_info(body, api_version, deployment_id, opts)
+    # @return [CompletionsCreate200Response]
+    def completions_create(deployment_id, api_version, completions_create_request, opts = {})
+      data, _status_code, _headers = completions_create_with_http_info(deployment_id, api_version, completions_create_request, opts)
       data
     end
 
     # Creates a completion for the provided prompt, parameters and chosen model.
-    # @param body
-    # @param api_version
-    # @param deployment_id
+    # @param deployment_id [String] 
+    # @param api_version [String] 
+    # @param completions_create_request [CompletionsCreateRequest] 
     # @param [Hash] opts the optional parameters
-    # @return [Array<(InlineResponse200, Integer, Hash)>] InlineResponse200 data, response status code and response headers
-    def completions_create_with_http_info(body, api_version, deployment_id, opts = {})
+    # @return [Array<(CompletionsCreate200Response, Integer, Hash)>] CompletionsCreate200Response data, response status code and response headers
+    def completions_create_with_http_info(deployment_id, api_version, completions_create_request, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: DefaultApi.completions_create ...'
       end
-      # verify the required parameter 'body' is set
-      if @api_client.config.client_side_validation && body.nil?
-        fail ArgumentError, "Missing the required parameter 'body' when calling DefaultApi.completions_create"
+      # verify the required parameter 'deployment_id' is set
+      if @api_client.config.client_side_validation && deployment_id.nil?
+        fail ArgumentError, "Missing the required parameter 'deployment_id' when calling DefaultApi.completions_create"
       end
       # verify the required parameter 'api_version' is set
       if @api_client.config.client_side_validation && api_version.nil?
         fail ArgumentError, "Missing the required parameter 'api_version' when calling DefaultApi.completions_create"
       end
-      # verify the required parameter 'deployment_id' is set
-      if @api_client.config.client_side_validation && deployment_id.nil?
-        fail ArgumentError, "Missing the required parameter 'deployment_id' when calling DefaultApi.completions_create"
+      # verify the required parameter 'completions_create_request' is set
+      if @api_client.config.client_side_validation && completions_create_request.nil?
+        fail ArgumentError, "Missing the required parameter 'completions_create_request' when calling DefaultApi.completions_create"
       end
       # resource path
-      local_var_path = '/deployments/{deployment-id}/completions'.sub('{' + 'deployment-id' + '}', deployment_id.to_s)
+      local_var_path = '/deployments/{deployment-id}/completions'.sub('{' + 'deployment-id' + '}', CGI.escape(deployment_id.to_s))
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -134,25 +143,34 @@ module AzureOpenaiClient
       # HTTP header 'Accept' (if needed)
       header_params['Accept'] = @api_client.select_header_accept(['application/json'])
       # HTTP header 'Content-Type'
-      header_params['Content-Type'] = @api_client.select_header_content_type(['application/json'])
+      content_type = @api_client.select_header_content_type(['application/json'])
+      if !content_type.nil?
+        header_params['Content-Type'] = content_type
+      end
 
       # form parameters
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:body] || @api_client.object_to_http_body(body)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(completions_create_request)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'CompletionsCreate200Response'
 
-      return_type = opts[:return_type] || 'InlineResponse200'
+      # auth_names
+      auth_names = opts[:debug_auth_names] || ['apiKey', 'bearer']
 
-      auth_names = opts[:auth_names] || ['apiKey', 'bearer']
-      data, status_code, headers = @api_client.call_api(:POST, local_var_path,
-        header_params: header_params,
-        query_params: query_params,
-        form_params: form_params,
-        body: post_body,
-        auth_names: auth_names,
-        return_type: return_type)
+      new_options = opts.merge(
+        :operation => :"DefaultApi.completions_create",
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type
+      )
 
+      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
       if @api_client.config.debugging
         @api_client.config.logger.debug "API called: DefaultApi#completions_create\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end
@@ -160,40 +178,40 @@ module AzureOpenaiClient
     end
 
     # Get a vector representation of a given input that can be easily consumed by machine learning models and algorithms.
-    # @param body
-    # @param api_version
-    # @param deployment_id The deployment id of the model which was deployed.
+    # @param deployment_id [String] The deployment id of the model which was deployed.
+    # @param api_version [String] 
+    # @param request_body [Hash<String, Object>] 
     # @param [Hash] opts the optional parameters
-    # @return [InlineResponse2001]
-    def embeddings_create(body, api_version, deployment_id, opts = {})
-      data, _status_code, _headers = embeddings_create_with_http_info(body, api_version, deployment_id, opts)
+    # @return [EmbeddingsCreate200Response]
+    def embeddings_create(deployment_id, api_version, request_body, opts = {})
+      data, _status_code, _headers = embeddings_create_with_http_info(deployment_id, api_version, request_body, opts)
       data
     end
 
     # Get a vector representation of a given input that can be easily consumed by machine learning models and algorithms.
-    # @param body
-    # @param api_version
-    # @param deployment_id The deployment id of the model which was deployed.
+    # @param deployment_id [String] The deployment id of the model which was deployed.
+    # @param api_version [String] 
+    # @param request_body [Hash<String, Object>] 
     # @param [Hash] opts the optional parameters
-    # @return [Array<(InlineResponse2001, Integer, Hash)>] InlineResponse2001 data, response status code and response headers
-    def embeddings_create_with_http_info(body, api_version, deployment_id, opts = {})
+    # @return [Array<(EmbeddingsCreate200Response, Integer, Hash)>] EmbeddingsCreate200Response data, response status code and response headers
+    def embeddings_create_with_http_info(deployment_id, api_version, request_body, opts = {})
       if @api_client.config.debugging
         @api_client.config.logger.debug 'Calling API: DefaultApi.embeddings_create ...'
       end
-      # verify the required parameter 'body' is set
-      if @api_client.config.client_side_validation && body.nil?
-        fail ArgumentError, "Missing the required parameter 'body' when calling DefaultApi.embeddings_create"
+      # verify the required parameter 'deployment_id' is set
+      if @api_client.config.client_side_validation && deployment_id.nil?
+        fail ArgumentError, "Missing the required parameter 'deployment_id' when calling DefaultApi.embeddings_create"
       end
       # verify the required parameter 'api_version' is set
       if @api_client.config.client_side_validation && api_version.nil?
         fail ArgumentError, "Missing the required parameter 'api_version' when calling DefaultApi.embeddings_create"
       end
-      # verify the required parameter 'deployment_id' is set
-      if @api_client.config.client_side_validation && deployment_id.nil?
-        fail ArgumentError, "Missing the required parameter 'deployment_id' when calling DefaultApi.embeddings_create"
+      # verify the required parameter 'request_body' is set
+      if @api_client.config.client_side_validation && request_body.nil?
+        fail ArgumentError, "Missing the required parameter 'request_body' when calling DefaultApi.embeddings_create"
       end
       # resource path
-      local_var_path = '/deployments/{deployment-id}/embeddings'.sub('{' + 'deployment-id' + '}', deployment_id.to_s)
+      local_var_path = '/deployments/{deployment-id}/embeddings'.sub('{' + 'deployment-id' + '}', CGI.escape(deployment_id.to_s))
 
       # query parameters
       query_params = opts[:query_params] || {}
@@ -204,25 +222,34 @@ module AzureOpenaiClient
       # HTTP header 'Accept' (if needed)
       header_params['Accept'] = @api_client.select_header_accept(['application/json'])
       # HTTP header 'Content-Type'
-      header_params['Content-Type'] = @api_client.select_header_content_type(['application/json'])
+      content_type = @api_client.select_header_content_type(['application/json'])
+      if !content_type.nil?
+        header_params['Content-Type'] = content_type
+      end
 
       # form parameters
       form_params = opts[:form_params] || {}
 
       # http body (model)
-      post_body = opts[:body] || @api_client.object_to_http_body(body)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(request_body)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'EmbeddingsCreate200Response'
 
-      return_type = opts[:return_type] || 'InlineResponse2001'
+      # auth_names
+      auth_names = opts[:debug_auth_names] || ['apiKey', 'bearer']
 
-      auth_names = opts[:auth_names] || ['apiKey', 'bearer']
-      data, status_code, headers = @api_client.call_api(:POST, local_var_path,
-        header_params: header_params,
-        query_params: query_params,
-        form_params: form_params,
-        body: post_body,
-        auth_names: auth_names,
-        return_type: return_type)
+      new_options = opts.merge(
+        :operation => :"DefaultApi.embeddings_create",
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type
+      )
 
+      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
       if @api_client.config.debugging
         @api_client.config.logger.debug "API called: DefaultApi#embeddings_create\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
       end

data/lib/azure_openai_client/api_client.rb

@@ -1,22 +1,21 @@
-# frozen_string_literal: true
-
 =begin
 #Azure OpenAI Service API
 
 #Azure OpenAI APIs for completions and search
 
-OpenAPI spec version: 2023-05-15
+The version of the OpenAPI document: 2023-05-15
+
+Generated by: https://openapi-generator.tech
+OpenAPI Generator version: 6.6.0
 
-Generated by: https://github.com/swagger-api/swagger-codegen.git
-Swagger Codegen version: 3.0.42
 =end
 
 require 'date'
 require 'json'
 require 'logger'
 require 'tempfile'
+require 'time'
 require 'typhoeus'
-require 'uri'
 
 module AzureOpenaiClient
   class ApiClient
@@ -32,7 +31,7 @@ module AzureOpenaiClient
     # @option config [Configuration] Configuration for initializing the object, default to Configuration.default
     def initialize(config = Configuration.default)
       @config = config
-      @user_agent = "Swagger-Codegen/#{VERSION}/ruby"
+      @user_agent = "OpenAPI-Generator/#{VERSION}/ruby"
       @default_headers = {
         'Content-Type' => 'application/json',
         'User-Agent' => @user_agent
@@ -60,12 +59,12 @@ module AzureOpenaiClient
           fail ApiError.new('Connection timed out')
         elsif response.code == 0
           # Errors from libcurl will be made visible here
-          fail ApiError.new(code: 0,
-                            message: response.return_message)
+          fail ApiError.new(:code => 0,
+                            :message => response.return_message)
         else
-          fail ApiError.new(code: response.code,
-                            response_headers: response.headers,
-                            response_body: response.body),
+          fail ApiError.new(:code => response.code,
+                            :response_headers => response.headers,
+                            :response_body => response.body),
                response.status_message
         end
       end
@@ -88,12 +87,13 @@ module AzureOpenaiClient
     # @option opts [Object] :body HTTP body (JSON/XML)
     # @return [Typhoeus::Request] A Typhoeus Request
     def build_request(http_method, path, opts = {})
-      url = build_request_url(path)
+      url = build_request_url(path, opts)
       http_method = http_method.to_sym.downcase
 
       header_params = @default_headers.merge(opts[:header_params] || {})
       query_params = opts[:query_params] || {}
       form_params = opts[:form_params] || {}
+      follow_location = opts[:follow_location] || true
 
       update_params_for_auth! header_params, query_params, opts[:auth_names]
 
@@ -101,16 +101,17 @@ module AzureOpenaiClient
       _verify_ssl_host = @config.verify_ssl_host ? 2 : 0
 
       req_opts = {
-        method: http_method,
-        headers: header_params,
-        params: query_params,
-        params_encoding: @config.params_encoding,
-        timeout: @config.timeout,
-        ssl_verifypeer: @config.verify_ssl,
-        ssl_verifyhost: _verify_ssl_host,
-        sslcert: @config.cert_file,
-        sslkey: @config.key_file,
-        verbose: @config.debugging
+        :method => http_method,
+        :headers => header_params,
+        :params => query_params,
+        :params_encoding => @config.params_encoding,
+        :timeout => @config.timeout,
+        :ssl_verifypeer => @config.verify_ssl,
+        :ssl_verifyhost => _verify_ssl_host,
+        :sslcert => @config.cert_file,
+        :sslkey => @config.key_file,
+        :verbose => @config.debugging,
+        :followlocation => follow_location
       }
 
       # set custom cert, if provided
@@ -118,7 +119,7 @@ module AzureOpenaiClient
 
       if [:post, :patch, :put, :delete].include?(http_method)
         req_body = build_request_body(header_params, form_params, opts[:body])
-        req_opts.update body: req_body
+        req_opts.update :body => req_body
         if @config.debugging
           @config.logger.debug "HTTP request body param ~BEGIN~\n#{req_body}\n~END~\n"
         end
@@ -157,6 +158,44 @@ module AzureOpenaiClient
       data
     end
 
+    # Save response body into a file in (the defined) temporary folder, using the filename
+    # from the "Content-Disposition" header if provided, otherwise a random filename.
+    # The response body is written to the file in chunks in order to handle files which
+    # size is larger than maximum Ruby String or even larger than the maximum memory a Ruby
+    # process can use.
+    #
+    # @see Configuration#temp_folder_path
+    def download_file(request)
+      tempfile = nil
+      encoding = nil
+      request.on_headers do |response|
+        content_disposition = response.headers['Content-Disposition']
+        if content_disposition && content_disposition =~ /filename=/i
+          filename = content_disposition[/filename=['"]?([^'"\s]+)['"]?/, 1]
+          prefix = sanitize_filename(filename)
+        else
+          prefix = 'download-'
+        end
+        prefix = prefix + '-' unless prefix.end_with?('-')
+        encoding = response.body.encoding
+        tempfile = Tempfile.open(prefix, @config.temp_folder_path, encoding: encoding)
+        @tempfile = tempfile
+      end
+      request.on_body do |chunk|
+        chunk.force_encoding(encoding)
+        tempfile.write(chunk)
+      end
+      request.on_complete do |response|
+        if tempfile
+          tempfile.close
+          @config.logger.info "Temp file written to #{tempfile.path}, please copy the file to a proper folder "\
+                              "with e.g. `FileUtils.cp(tempfile.path, '/new/file/path')` otherwise the temp file "\
+                              "will be deleted automatically with GC. It's also recommended to delete the temp file "\
+                              "explicitly with `tempfile.delete`"
+        end
+      end
+    end
+
     # Check if the given MIME is a JSON MIME.
     # JSON MIME examples:
     #   application/json
@@ -191,9 +230,9 @@ module AzureOpenaiClient
       fail "Content-Type is not supported: #{content_type}" unless json_mime?(content_type)
 
       begin
-        data = JSON.parse("[#{body}]", symbolize_names: true)[0]
+        data = JSON.parse("[#{body}]", :symbolize_names => true)[0]
       rescue JSON::ParserError => e
-        if %w(String Date DateTime).include?(return_type)
+        if %w(String Date Time).include?(return_type)
           data = body
         else
           raise e
@@ -218,9 +257,9 @@ module AzureOpenaiClient
         data.to_f
       when 'Boolean'
         data == true
-      when 'DateTime'
+      when 'Time'
         # parse date time (expecting ISO 8601 format)
-        DateTime.parse data
+        Time.parse data
       when 'Date'
         # parse date time (expecting ISO 8601 format)
         Date.parse data
@@ -238,46 +277,9 @@ module AzureOpenaiClient
           data.each { |k, v| hash[k] = convert_to_type(v, sub_type) }
         end
       else
-        # models, e.g. Pet
-        AzureOpenaiClient.const_get(return_type).build_from_hash(data)
-      end
-    end
-
-    # Save response body into a file in (the defined) temporary folder, using the filename
-    # from the "Content-Disposition" header if provided, otherwise a random filename.
-    # The response body is written to the file in chunks in order to handle files which
-    # size is larger than maximum Ruby String or even larger than the maximum memory a Ruby
-    # process can use.
-    #
-    # @see Configuration#temp_folder_path
-    def download_file(request)
-      tempfile = nil
-      encoding = nil
-      request.on_headers do |response|
-        content_disposition = response.headers['Content-Disposition']
-        if content_disposition && content_disposition =~ /filename=/i
-          filename = content_disposition[/filename=['"]?([^'"\s]+)['"]?/, 1]
-          prefix = sanitize_filename(filename)
-        else
-          prefix = 'download-'
-        end
-        prefix = prefix + '-' unless prefix.end_with?('-')
-        encoding = response.body.encoding
-        tempfile = Tempfile.open(prefix, @config.temp_folder_path, encoding: encoding)
-        @tempfile = tempfile
-      end
-      request.on_body do |chunk|
-        chunk.force_encoding(encoding)
-        tempfile.write(chunk)
-      end
-      request.on_complete do |response|
-        if tempfile
-          tempfile.close
-          @config.logger.info "Temp file written to #{tempfile.path}, please copy the file to a proper folder "\
-                              "with e.g. `FileUtils.cp(tempfile.path, '/new/file/path')` otherwise the temp file "\
-                              "will be deleted automatically with GC. It's also recommended to delete the temp file "\
-                              "explicitly with `tempfile.delete`"
-        end
+        # models (e.g. Pet) or oneOf
+        klass = AzureOpenaiClient.const_get(return_type)
+        klass.respond_to?(:openapi_one_of) ? klass.build(data) : klass.build_from_hash(data)
       end
     end
 
@@ -290,13 +292,13 @@ module AzureOpenaiClient
       filename.gsub(/.*[\/\\]/, '')
     end
 
-    def build_request_url(path)
+    def build_request_url(path, opts = {})
       # Add leading and trailing slashes to path
       path = "/#{path}".gsub(/\/+/, '/')
-      @config.base_url + path
+      @config.base_url(opts[:operation]) + path
     end
 
-    # Update hearder and query params based on authentication settings.
+    # Update header and query params based on authentication settings.
     #
     # @param [Hash] header_params Header parameters
     # @param [Hash] query_params Query parameters
@@ -308,14 +310,14 @@ module AzureOpenaiClient
         case auth_setting[:in]
         when 'header' then header_params[auth_setting[:key]] = auth_setting[:value]
         when 'query'  then query_params[auth_setting[:key]] = auth_setting[:value]
-        else fail ArgumentError, 'Authentication token must be in `query` of `header`'
+        else fail ArgumentError, 'Authentication token must be in `query` or `header`'
         end
       end
     end
 
     # Sets user agent in HTTP header
     #
-    # @param [String] user_agent User agent (e.g. swagger-codegen/ruby/1.0.0)
+    # @param [String] user_agent User agent (e.g. openapi-generator/ruby/1.0.0)
     def user_agent=(user_agent)
       @user_agent = user_agent
       @default_headers['User-Agent'] = @user_agent
@@ -335,8 +337,8 @@ module AzureOpenaiClient
     # @param [Array] content_types array for Content-Type
     # @return [String] the Content-Type header  (e.g. application/json)
     def select_header_content_type(content_types)
-      # use application/json by default
-      return 'application/json' if content_types.nil? || content_types.empty?
+      # return nil by default
+      return if content_types.nil? || content_types.empty?
       # use JSON when present, otherwise use the first one
       json_content_type = content_types.find { |s| json_mime?(s) }
       json_content_type || content_types.first

data/lib/azure_openai_client/api_error.rb

@@ -1,14 +1,13 @@
-# frozen_string_literal: true
-
 =begin
 #Azure OpenAI Service API
 
 #Azure OpenAI APIs for completions and search
 
-OpenAPI spec version: 2023-05-15
+The version of the OpenAPI document: 2023-05-15
+
+Generated by: https://openapi-generator.tech
+OpenAPI Generator version: 6.6.0
 
-Generated by: https://github.com/swagger-api/swagger-codegen.git
-Swagger Codegen version: 3.0.42
 =end
 
 module AzureOpenaiClient
@@ -33,6 +32,7 @@ module AzureOpenaiClient
         end
       else
         super arg
+        @message = arg
       end
     end