azure_openai_client 0.0.1

data/lib/azure_openai_client/models/one_ofdeploymentid_completions_body_stop.rb

@@ -0,0 +1,198 @@
+# frozen_string_literal: true
+
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+OpenAPI spec version: 2023-05-15
+
+Generated by: https://github.com/swagger-api/swagger-codegen.git
+Swagger Codegen version: 3.0.42
+=end
+
+require 'date'
+
+module AzureOpenaiClient
+  class OneOfdeploymentidCompletionsBodyStop
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {}
+    end
+
+    # Attribute type mapping.
+    def self.openapi_types
+      {}
+    end
+
+    # List of attributes with nullable: true
+    def self.openapi_nullable
+      Set.new([
+      ])
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      if (!attributes.is_a?(Hash))
+        fail ArgumentError, "The input argument (attributes) must be a hash in `AzureOpenaiClient::OneOfdeploymentidCompletionsBodyStop` initialize method"
+      end
+
+      # check to see if the attribute exists and convert string to symbol for hash key
+      attributes = attributes.each_with_object({}) { |(k, v), h|
+        if (!self.class.attribute_map.key?(k.to_sym))
+          fail ArgumentError, "`#{k}` is not a valid attribute in `AzureOpenaiClient::OneOfdeploymentidCompletionsBodyStop`. Please check the name to make sure it's valid. List of attributes: " + self.class.attribute_map.keys.inspect
+        end
+        h[k.to_sym] = v
+      }
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properties with the reasons
+    def list_invalid_properties
+      invalid_properties = Array.new
+      invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      true
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Integer] Hash code
+    def hash
+      [].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def self.build_from_hash(attributes)
+      new.build_from_hash(attributes)
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      self.class.openapi_types.each_pair do |key, type|
+        if type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the attribute
+          # is documented as an array but the input is not
+          if attributes[self.class.attribute_map[key]].is_a?(Array)
+            self.send("#{key}=", attributes[self.class.attribute_map[key]].map { |v| _deserialize($1, v) })
+          end
+        elsif !attributes[self.class.attribute_map[key]].nil?
+          self.send("#{key}=", _deserialize(type, attributes[self.class.attribute_map[key]]))
+        elsif attributes[self.class.attribute_map[key]].nil? && self.class.openapi_nullable.include?(key)
+          self.send("#{key}=", nil)
+        end
+      end
+
+      self
+    end
+
+    # Deserializes the data based on type
+    # @param string type Data type
+    # @param string value Value to be deserialized
+    # @return [Object] Deserialized data
+    def _deserialize(type, value)
+      case type.to_sym
+      when :DateTime
+        DateTime.parse(value)
+      when :Date
+        Date.parse(value)
+      when :String
+        value.to_s
+      when :Integer
+        value.to_i
+      when :Float
+        value.to_f
+      when :Boolean
+        if value.to_s =~ /\A(true|t|yes|y|1)\z/i
+          true
+        else
+          false
+        end
+      when :Object
+        # generic object (usually a Hash), return directly
+        value
+      when /\AArray<(?<inner_type>.+)>\z/
+        inner_type = Regexp.last_match[:inner_type]
+        value.map { |v| _deserialize(inner_type, v) }
+      when /\AHash<(?<k_type>.+?), (?<v_type>.+)>\z/
+        k_type = Regexp.last_match[:k_type]
+        v_type = Regexp.last_match[:v_type]
+        {}.tap do |hash|
+          value.each do |k, v|
+            hash[_deserialize(k_type, k)] = _deserialize(v_type, v)
+          end
+        end
+      else # model
+        AzureOpenaiClient.const_get(type).build_from_hash(value)
+      end
+    end
+
+    # Returns the string representation of the object
+    # @return [String] String presentation of the object
+    def to_s
+      to_hash.to_s
+    end
+
+    # to_body is an alias to to_hash (backward compatibility)
+    # @return [Hash] Returns the object in the form of hash
+    def to_body
+      to_hash
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        if value.nil?
+          is_nullable = self.class.openapi_nullable.include?(attr)
+          next if !is_nullable || (is_nullable && !instance_variable_defined?(:"@#{attr}"))
+        end
+
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+    # Outputs non-array value in the form of hash
+    # For object, use to_hash. Otherwise, just return the value
+    # @param [Object] value Any valid value
+    # @return [Hash] Returns the value in the form of hash
+    def _to_hash(value)
+      if value.is_a?(Array)
+        value.compact.map { |v| _to_hash(v) }
+      elsif value.is_a?(Hash)
+        {}.tap do |hash|
+          value.each { |k, v| hash[k] = _to_hash(v) }
+        end
+      elsif value.respond_to? :to_hash
+        value.to_hash
+      else
+        value
+      end
+    end
+  end
+end

data/lib/azure_openai_client/version.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+OpenAPI spec version: 2023-05-15
+
+Generated by: https://github.com/swagger-api/swagger-codegen.git
+Swagger Codegen version: 3.0.42
+=end
+
+module AzureOpenaiClient
+  VERSION = '0.0.1'
+end

data/lib/azure_openai_client.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+OpenAPI spec version: 2023-05-15
+
+Generated by: https://github.com/swagger-api/swagger-codegen.git
+Swagger Codegen version: 3.0.42
+=end
+
+# Common files
+require 'azure_openai_client/api_client'
+require 'azure_openai_client/api_error'
+require 'azure_openai_client/version'
+require 'azure_openai_client/configuration'
+
+# Models
+require 'azure_openai_client/models/chat_completions_body'
+require 'azure_openai_client/models/deploymentid_completions_body'
+require 'azure_openai_client/models/deploymentid_embeddings_body'
+require 'azure_openai_client/models/deploymentsdeploymentidchatcompletions_messages'
+require 'azure_openai_client/models/error_response'
+require 'azure_openai_client/models/error_response_error'
+require 'azure_openai_client/models/inline_response_200'
+require 'azure_openai_client/models/inline_response_200_1'
+require 'azure_openai_client/models/inline_response_200_1_data'
+require 'azure_openai_client/models/inline_response_200_1_usage'
+require 'azure_openai_client/models/inline_response_200_2'
+require 'azure_openai_client/models/inline_response_200_2_choices'
+require 'azure_openai_client/models/inline_response_200_2_message'
+require 'azure_openai_client/models/inline_response_200_2_usage'
+require 'azure_openai_client/models/inline_response_200_choices'
+require 'azure_openai_client/models/inline_response_200_logprobs'
+require 'azure_openai_client/models/inline_response_200_usage'
+require 'azure_openai_client/models/one_ofchat_completions_body_stop'
+require 'azure_openai_client/models/one_ofdeploymentid_completions_body_prompt'
+require 'azure_openai_client/models/one_ofdeploymentid_completions_body_stop'
+
+# APIs
+require 'azure_openai_client/api/default_api'
+
+module AzureOpenaiClient
+  class << self
+    # Customize default settings for the SDK using block.
+    #   AzureOpenaiClient.configure do |config|
+    #     config.username = "xxx"
+    #     config.password = "xxx"
+    #   end
+    # If no block given, return the default Configuration object.
+    def configure
+      if block_given?
+        yield(Configuration.default)
+      else
+        Configuration.default
+      end
+    end
+  end
+end

data/spec/api/default_api_spec.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+OpenAPI spec version: 2023-05-15
+
+Generated by: https://github.com/swagger-api/swagger-codegen.git
+Swagger Codegen version: 3.0.42
+=end
+
+require 'spec_helper'
+require 'json'
+
+# Unit tests for AzureOpenaiClient::DefaultApi
+# Automatically generated by swagger-codegen (github.com/swagger-api/swagger-codegen)
+# Please update as you see appropriate
+describe 'DefaultApi' do
+  before do
+    # run before each test
+    @instance = AzureOpenaiClient::DefaultApi.new
+  end
+
+  after do
+    # run after each test
+  end
+
+  describe 'test an instance of DefaultApi' do
+    it 'should create an instance of DefaultApi' do
+      expect(@instance).to be_instance_of(AzureOpenaiClient::DefaultApi)
+    end
+  end
+
+  # unit tests for chat_completions_create
+  # Creates a completion for the chat message
+  # @param body
+  # @param api_version
+  # @param deployment_id
+  # @param [Hash] opts the optional parameters
+  # @return [InlineResponse2002]
+  describe 'chat_completions_create test' do
+    it 'should work' do
+      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+    end
+  end
+
+  # unit tests for completions_create
+  # Creates a completion for the provided prompt, parameters and chosen model.
+  # @param body
+  # @param api_version
+  # @param deployment_id
+  # @param [Hash] opts the optional parameters
+  # @return [InlineResponse200]
+  describe 'completions_create test' do
+    it 'should work' do
+      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+    end
+  end
+
+  # unit tests for embeddings_create
+  # 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 [Hash] opts the optional parameters
+  # @return [InlineResponse2001]
+  describe 'embeddings_create test' do
+    it 'should work' do
+      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+    end
+  end
+
+end

data/spec/api_client_spec.rb

@@ -0,0 +1,229 @@
+# frozen_string_literal: true
+
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+OpenAPI spec version: 2023-05-15
+
+Generated by: https://github.com/swagger-api/swagger-codegen.git
+Swagger Codegen version: 3.0.42
+=end
+
+require 'spec_helper'
+
+describe AzureOpenaiClient::ApiClient do
+  context 'initialization' do
+    context 'URL stuff' do
+      context 'host' do
+        it 'removes http from host' do
+          AzureOpenaiClient.configure { |c| c.host = 'http://example.com' }
+          expect(AzureOpenaiClient::Configuration.default.host).to eq('example.com')
+        end
+
+        it 'removes https from host' do
+          AzureOpenaiClient.configure { |c| c.host = 'https://wookiee.com' }
+          expect(AzureOpenaiClient::ApiClient.default.config.host).to eq('wookiee.com')
+        end
+
+        it 'removes trailing path from host' do
+          AzureOpenaiClient.configure { |c| c.host = 'hobo.com/v4' }
+          expect(AzureOpenaiClient::Configuration.default.host).to eq('hobo.com')
+        end
+      end
+
+      context 'base_path' do
+        it "prepends a slash to base_path" do
+          AzureOpenaiClient.configure { |c| c.base_path = 'v4/dog' }
+          expect(AzureOpenaiClient::Configuration.default.base_path).to eq('/v4/dog')
+        end
+
+        it "doesn't prepend a slash if one is already there" do
+          AzureOpenaiClient.configure { |c| c.base_path = '/v4/dog' }
+          expect(AzureOpenaiClient::Configuration.default.base_path).to eq('/v4/dog')
+        end
+
+        it "ends up as a blank string if nil" do
+          AzureOpenaiClient.configure { |c| c.base_path = nil }
+          expect(AzureOpenaiClient::Configuration.default.base_path).to eq('')
+        end
+      end
+    end
+  end
+
+  describe 'params_encoding in #build_request' do
+    let(:config) { AzureOpenaiClient::Configuration.new }
+    let(:api_client) { AzureOpenaiClient::ApiClient.new(config) }
+
+    it 'defaults to nil' do
+      expect(AzureOpenaiClient::Configuration.default.params_encoding).to eq(nil)
+      expect(config.params_encoding).to eq(nil)
+
+      request = api_client.build_request(:get, '/test')
+      expect(request.options[:params_encoding]).to eq(nil)
+    end
+
+    it 'can be customized' do
+      config.params_encoding = :multi
+      request = api_client.build_request(:get, '/test')
+      expect(request.options[:params_encoding]).to eq(:multi)
+    end
+  end
+
+  describe 'timeout in #build_request' do
+    let(:config) { AzureOpenaiClient::Configuration.new }
+    let(:api_client) { AzureOpenaiClient::ApiClient.new(config) }
+
+    it 'defaults to 0' do
+      expect(AzureOpenaiClient::Configuration.default.timeout).to eq(0)
+      expect(config.timeout).to eq(0)
+
+      request = api_client.build_request(:get, '/test')
+      expect(request.options[:timeout]).to eq(0)
+    end
+
+    it 'can be customized' do
+      config.timeout = 100
+      request = api_client.build_request(:get, '/test')
+      expect(request.options[:timeout]).to eq(100)
+    end
+  end
+
+  describe '#deserialize' do
+    it "handles Array<Integer>" do
+      api_client = AzureOpenaiClient::ApiClient.new
+      headers = { 'Content-Type' => 'application/json' }
+      response = double('response', headers: headers, body: '[12, 34]')
+      data = api_client.deserialize(response, 'Array<Integer>')
+      expect(data).to be_instance_of(Array)
+      expect(data).to eq([12, 34])
+    end
+
+    it 'handles Array<Array<Integer>>' do
+      api_client = AzureOpenaiClient::ApiClient.new
+      headers = { 'Content-Type' => 'application/json' }
+      response = double('response', headers: headers, body: '[[12, 34], [56]]')
+      data = api_client.deserialize(response, 'Array<Array<Integer>>')
+      expect(data).to be_instance_of(Array)
+      expect(data).to eq([[12, 34], [56]])
+    end
+
+    it 'handles Hash<String, String>' do
+      api_client = AzureOpenaiClient::ApiClient.new
+      headers = { 'Content-Type' => 'application/json' }
+      response = double('response', headers: headers, body: '{"message": "Hello"}')
+      data = api_client.deserialize(response, 'Hash<String, String>')
+      expect(data).to be_instance_of(Hash)
+      expect(data).to eq(message: 'Hello')
+    end
+  end
+
+  describe "#object_to_hash" do
+    it 'ignores nils and includes empty arrays' do
+      # uncomment below to test object_to_hash for model
+      api_client = AzureOpenaiClient::ApiClient.new
+      _model = AzureOpenaiClient::InlineResponse200.new
+      # update the model attribute below
+      _model.id = 1
+      # update the expected value (hash) below
+      expected = { id: 1 }
+      expect(api_client.object_to_hash(_model)).to eq(expected)
+    end
+  end
+
+  describe '#build_collection_param' do
+    let(:param) { ['aa', 'bb', 'cc'] }
+    let(:api_client) { AzureOpenaiClient::ApiClient.new }
+
+    it 'works for csv' do
+      expect(api_client.build_collection_param(param, :csv)).to eq('aa,bb,cc')
+    end
+
+    it 'works for ssv' do
+      expect(api_client.build_collection_param(param, :ssv)).to eq('aa bb cc')
+    end
+
+    it 'works for tsv' do
+      expect(api_client.build_collection_param(param, :tsv)).to eq("aa\tbb\tcc")
+    end
+
+    it 'works for pipes' do
+      expect(api_client.build_collection_param(param, :pipes)).to eq('aa|bb|cc')
+    end
+
+    it 'works for multi' do
+      expect(api_client.build_collection_param(param, :multi)).to eq(['aa', 'bb', 'cc'])
+    end
+
+    it 'fails for invalid collection format' do
+      expect {
+        api_client.build_collection_param(param, :INVALID)
+      }.to raise_error(RuntimeError, 'unknown collection format: :INVALID')
+    end
+  end
+
+  describe '#json_mime?' do
+    let(:api_client) { AzureOpenaiClient::ApiClient.new }
+
+    it 'works' do
+      expect(api_client.json_mime?(nil)).to eq false
+      expect(api_client.json_mime?('')).to eq false
+
+      expect(api_client.json_mime?('application/json')).to eq true
+      expect(api_client.json_mime?('application/json; charset=UTF8')).to eq true
+      expect(api_client.json_mime?('APPLICATION/JSON')).to eq true
+
+      expect(api_client.json_mime?('application/xml')).to eq false
+      expect(api_client.json_mime?('text/plain')).to eq false
+      expect(api_client.json_mime?('application/jsonp')).to eq false
+    end
+  end
+
+  describe '#select_header_accept' do
+    let(:api_client) { AzureOpenaiClient::ApiClient.new }
+
+    it 'works' do
+      expect(api_client.select_header_accept(nil)).to be_nil
+      expect(api_client.select_header_accept([])).to be_nil
+
+      expect(api_client.select_header_accept(['application/json'])).to eq('application/json')
+      expect(api_client.select_header_accept(['application/xml', 'application/json; charset=UTF8'])).to eq('application/json; charset=UTF8')
+      expect(api_client.select_header_accept(['APPLICATION/JSON', 'text/html'])).to eq('APPLICATION/JSON')
+
+      expect(api_client.select_header_accept(['application/xml'])).to eq('application/xml')
+      expect(api_client.select_header_accept(['text/html', 'application/xml'])).to eq('text/html,application/xml')
+    end
+  end
+
+  describe '#select_header_content_type' do
+    let(:api_client) { AzureOpenaiClient::ApiClient.new }
+
+    it 'works' do
+      expect(api_client.select_header_content_type(nil)).to eq('application/json')
+      expect(api_client.select_header_content_type([])).to eq('application/json')
+
+      expect(api_client.select_header_content_type(['application/json'])).to eq('application/json')
+      expect(api_client.select_header_content_type(['application/xml', 'application/json; charset=UTF8'])).to eq('application/json; charset=UTF8')
+      expect(api_client.select_header_content_type(['APPLICATION/JSON', 'text/html'])).to eq('APPLICATION/JSON')
+      expect(api_client.select_header_content_type(['application/xml'])).to eq('application/xml')
+      expect(api_client.select_header_content_type(['text/plain', 'application/xml'])).to eq('text/plain')
+    end
+  end
+
+  describe '#sanitize_filename' do
+    let(:api_client) { AzureOpenaiClient::ApiClient.new }
+
+    it 'works' do
+      expect(api_client.sanitize_filename('sun')).to eq('sun')
+      expect(api_client.sanitize_filename('sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('../sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('/var/tmp/sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('./sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('..\sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('\var\tmp\sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('c:\var\tmp\sun.gif')).to eq('sun.gif')
+      expect(api_client.sanitize_filename('.\sun.gif')).to eq('sun.gif')
+    end
+  end
+end

data/spec/configuration_spec.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe AzureOpenaiClient::Configuration do
+  let(:config) { AzureOpenaiClient::Configuration.new }
+
+  describe '#initialize' do
+    it 'sets default values' do
+      expect(config.scheme).to eq('https')
+      expect(config.host).to eq('{endpoint}')
+      expect(config.base_path).to eq('https://{endpoint}/openai')
+      expect(config.api_key).to eq({})
+      expect(config.api_key_prefix).to eq({})
+      expect(config.timeout).to eq(0)
+      expect(config.client_side_validation).to be(true)
+      expect(config.verify_ssl).to be(true)
+      expect(config.verify_ssl_host).to be(true)
+      expect(config.params_encoding).to be_nil
+      expect(config.cert_file).to be_nil
+      expect(config.key_file).to be_nil
+      expect(config.debugging).to be(false)
+      expect(config.inject_format).to be(false)
+      expect(config.force_ending_format).to be(false)
+    end
+  end
+
+  describe '#base_url' do
+    it 'returns the base URL with the correct scheme, host, and base_path' do
+      config.scheme = 'https'
+      config.host = 'example.com'
+      config.base_path = '/openai'
+      expect(config.base_url).to eq('https://example.com/openai')
+    end
+  end
+
+  describe '#api_key_with_prefix' do
+    it 'returns the API key with prefix if set' do
+      config.api_key['api_key'] = '12345'
+      config.api_key_prefix['api_key'] = 'Token'
+      expect(config.api_key_with_prefix('api_key')).to eq('Token 12345')
+    end
+
+    it 'returns the API key without prefix if not set' do
+      config.api_key['api_key'] = '12345'
+      expect(config.api_key_with_prefix('api_key')).to eq('12345')
+    end
+  end
+
+  describe '#basic_auth_token' do
+    it 'returns the Basic Auth token string' do
+      config.username = 'user'
+      config.password = 'pass'
+      expect(config.basic_auth_token).to eq('Basic dXNlcjpwYXNz')
+    end
+  end
+
+  describe '#auth_settings' do
+    it 'returns the Auth Settings hash for the API client' do
+      config.api_key['api-key'] = '12345'
+      config.access_token = 'token_12345'
+      expect(config.auth_settings).to eq(
+        'apiKey' => {
+          type: 'api_key',
+          in: 'header',
+          key: 'api-key',
+          value: '12345'
+        },
+        'bearer' => {
+          type: 'oauth2',
+          in: 'header',
+          key: 'Authorization',
+          value: 'Bearer token_12345'
+        }
+      )
+    end
+  end
+end