azure_openai_client 0.0.1 → 0.0.2

data/lib/azure_openai_client/models/embeddings_create_request.rb

@@ -0,0 +1,252 @@
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+The version of the OpenAPI document: 2023-05-15
+
+Generated by: https://openapi-generator.tech
+OpenAPI Generator version: 6.6.0
+
+=end
+
+require 'date'
+require 'time'
+
+module AzureOpenaiClient
+  class EmbeddingsCreateRequest
+    attr_accessor :input
+
+    # A unique identifier representing your end-user, which can help monitoring and detecting abuse.
+    attr_accessor :user
+
+    # input type of embedding search to use
+    attr_accessor :input_type
+
+    # ID of the model to use. You can use the Models_List operation to see all of your available models, or see our Models_Get overview for descriptions of them.
+    attr_accessor :model
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'input' => :'input',
+        :'user' => :'user',
+        :'input_type' => :'input_type',
+        :'model' => :'model'
+      }
+    end
+
+    # Returns all the JSON keys this model knows about
+    def self.acceptable_attributes
+      attribute_map.values
+    end
+
+    # Attribute type mapping.
+    def self.openapi_types
+      {
+        :'input' => :'EmbeddingsCreateRequestInput',
+        :'user' => :'String',
+        :'input_type' => :'String',
+        :'model' => :'String'
+      }
+    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::EmbeddingsCreateRequest` 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::EmbeddingsCreateRequest`. 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
+      }
+
+      if attributes.key?(:'input')
+        self.input = attributes[:'input']
+      end
+
+      if attributes.key?(:'user')
+        self.user = attributes[:'user']
+      end
+
+      if attributes.key?(:'input_type')
+        self.input_type = attributes[:'input_type']
+      end
+
+      if attributes.key?(:'model')
+        self.model = attributes[:'model']
+      end
+    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
+      if @input.nil?
+        invalid_properties.push('invalid value for "input", input cannot be nil.')
+      end
+
+      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?
+      return false if @input.nil?
+      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 &&
+          input == o.input &&
+          user == o.user &&
+          input_type == o.input_type &&
+          model == o.model
+    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
+      [input, user, input_type, model].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)
+      attributes = attributes.transform_keys(&:to_sym)
+      self.class.openapi_types.each_pair do |key, type|
+        if attributes[self.class.attribute_map[key]].nil? && self.class.openapi_nullable.include?(key)
+          self.send("#{key}=", nil)
+        elsif 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]]))
+        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 :Time
+        Time.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
+        # models (e.g. Pet) or oneOf
+        klass = AzureOpenaiClient.const_get(type)
+        klass.respond_to?(:openapi_one_of) ? klass.build(value) : klass.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/models/embeddings_create_request_input.rb

@@ -0,0 +1,105 @@
+=begin
+#Azure OpenAI Service API
+
+#Azure OpenAI APIs for completions and search
+
+The version of the OpenAPI document: 2023-05-15
+
+Generated by: https://openapi-generator.tech
+OpenAPI Generator version: 6.6.0
+
+=end
+
+require 'date'
+require 'time'
+
+module AzureOpenaiClient
+  # Input text to get embeddings for, encoded as a string. To get embeddings for multiple inputs in a single request, pass an array of strings. Each input must not exceed 2048 tokens in length. Unless you are embedding code, we suggest replacing newlines (\\n) in your input with a single space, as we have observed inferior results when newlines are present.
+  module EmbeddingsCreateRequestInput
+    class << self
+      # List of class defined in oneOf (OpenAPI v3)
+      def openapi_one_of
+        [
+          :'Array<String>',
+          :'String'
+        ]
+      end
+
+      # Builds the object
+      # @param [Mixed] Data to be matched against the list of oneOf items
+      # @return [Object] Returns the model or the data itself
+      def build(data)
+        # Go through the list of oneOf items and attempt to identify the appropriate one.
+        # Note:
+        # - We do not attempt to check whether exactly one item matches.
+        # - No advanced validation of types in some cases (e.g. "x: { type: string }" will happily match { x: 123 })
+        #   due to the way the deserialization is made in the base_object template (it just casts without verifying).
+        # - TODO: scalar values are de facto behaving as if they were nullable.
+        # - TODO: logging when debugging is set.
+        openapi_one_of.each do |klass|
+          begin
+            next if klass == :AnyType # "nullable: true"
+            typed_data = find_and_cast_into_type(klass, data)
+            return typed_data if typed_data
+          rescue # rescue all errors so we keep iterating even if the current item lookup raises
+          end
+        end
+
+        openapi_one_of.include?(:AnyType) ? data : nil
+      end
+
+      private
+
+        SchemaMismatchError = Class.new(StandardError)
+
+        # Note: 'File' is missing here because in the regular case we get the data _after_ a call to JSON.parse.
+        def find_and_cast_into_type(klass, data)
+          return if data.nil?
+
+          case klass.to_s
+          when 'Boolean'
+            return data if data.instance_of?(TrueClass) || data.instance_of?(FalseClass)
+          when 'Float'
+            return data if data.instance_of?(Float)
+          when 'Integer'
+            return data if data.instance_of?(Integer)
+          when 'Time'
+            return Time.parse(data)
+          when 'Date'
+            return Date.parse(data)
+          when 'String'
+            return data if data.instance_of?(String)
+          when 'Object' # "type: object"
+            return data if data.instance_of?(Hash)
+          when /\AArray<(?<sub_type>.+)>\z/ # "type: array"
+            if data.instance_of?(Array)
+              sub_type = Regexp.last_match[:sub_type]
+              return data.map { |item| find_and_cast_into_type(sub_type, item) }
+            end
+          when /\AHash<String, (?<sub_type>.+)>\z/ # "type: object" with "additionalProperties: { ... }"
+            if data.instance_of?(Hash) && data.keys.all? { |k| k.instance_of?(Symbol) || k.instance_of?(String) }
+              sub_type = Regexp.last_match[:sub_type]
+              return data.each_with_object({}) { |(k, v), hsh| hsh[k] = find_and_cast_into_type(sub_type, v) }
+            end
+          else # model
+            const = AzureOpenaiClient.const_get(klass)
+            if const
+              if const.respond_to?(:openapi_one_of) # nested oneOf model
+                model = const.build(data)
+                return model if model
+              else
+                # raise if data contains keys that are not known to the model
+                raise unless (data.keys - const.acceptable_attributes).empty?
+                model = const.build_from_hash(data)
+                return model if model && model.valid?
+              end
+            end
+          end
+
+          raise # if no match by now, raise
+        rescue
+          raise SchemaMismatchError, "#{data} doesn't match the #{klass} type"
+        end
+    end
+  end
+end

data/lib/azure_openai_client/models/error_response.rb

@@ -1,17 +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 'date'
+require 'time'
 
 module AzureOpenaiClient
   class ErrorResponse
@@ -20,14 +20,19 @@ module AzureOpenaiClient
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
-        'error': :'error'
+        :'error' => :'error'
       }
     end
 
+    # Returns all the JSON keys this model knows about
+    def self.acceptable_attributes
+      attribute_map.values
+    end
+
     # Attribute type mapping.
     def self.openapi_types
       {
-        'error': :'Object'
+        :'error' => :'ErrorResponseError'
       }
     end
 
@@ -102,8 +107,11 @@ module AzureOpenaiClient
     # @return [Object] Returns the model itself
     def build_from_hash(attributes)
       return nil unless attributes.is_a?(Hash)
+      attributes = attributes.transform_keys(&:to_sym)
       self.class.openapi_types.each_pair do |key, type|
-        if type =~ /\AArray<(.*)>/i
+        if attributes[self.class.attribute_map[key]].nil? && self.class.openapi_nullable.include?(key)
+          self.send("#{key}=", nil)
+        elsif 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)
@@ -111,8 +119,6 @@ module AzureOpenaiClient
           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
 
@@ -125,8 +131,8 @@ module AzureOpenaiClient
     # @return [Object] Deserialized data
     def _deserialize(type, value)
       case type.to_sym
-      when :DateTime
-        DateTime.parse(value)
+      when :Time
+        Time.parse(value)
       when :Date
         Date.parse(value)
       when :String
@@ -156,7 +162,9 @@ module AzureOpenaiClient
           end
         end
       else # model
-        AzureOpenaiClient.const_get(type).build_from_hash(value)
+        # models (e.g. Pet) or oneOf
+        klass = AzureOpenaiClient.const_get(type)
+        klass.respond_to?(:openapi_one_of) ? klass.build(value) : klass.build_from_hash(value)
       end
     end
 

data/lib/azure_openai_client/models/error_response_error.rb

@@ -1,17 +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 'date'
+require 'time'
 
 module AzureOpenaiClient
   class ErrorResponseError
@@ -26,20 +26,25 @@ module AzureOpenaiClient
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
-        'code': :'code',
-        'message': :'message',
-        'param': :'param',
-        'type': :'type'
+        :'code' => :'code',
+        :'message' => :'message',
+        :'param' => :'param',
+        :'type' => :'type'
       }
     end
 
+    # Returns all the JSON keys this model knows about
+    def self.acceptable_attributes
+      attribute_map.values
+    end
+
     # Attribute type mapping.
     def self.openapi_types
       {
-        'code': :'Object',
-        'message': :'Object',
-        'param': :'Object',
-        'type': :'Object'
+        :'code' => :'String',
+        :'message' => :'String',
+        :'param' => :'String',
+        :'type' => :'String'
       }
     end
 
@@ -129,8 +134,11 @@ module AzureOpenaiClient
     # @return [Object] Returns the model itself
     def build_from_hash(attributes)
       return nil unless attributes.is_a?(Hash)
+      attributes = attributes.transform_keys(&:to_sym)
       self.class.openapi_types.each_pair do |key, type|
-        if type =~ /\AArray<(.*)>/i
+        if attributes[self.class.attribute_map[key]].nil? && self.class.openapi_nullable.include?(key)
+          self.send("#{key}=", nil)
+        elsif 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)
@@ -138,8 +146,6 @@ module AzureOpenaiClient
           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
 
@@ -152,8 +158,8 @@ module AzureOpenaiClient
     # @return [Object] Deserialized data
     def _deserialize(type, value)
       case type.to_sym
-      when :DateTime
-        DateTime.parse(value)
+      when :Time
+        Time.parse(value)
       when :Date
         Date.parse(value)
       when :String
@@ -183,7 +189,9 @@ module AzureOpenaiClient
           end
         end
       else # model
-        AzureOpenaiClient.const_get(type).build_from_hash(value)
+        # models (e.g. Pet) or oneOf
+        klass = AzureOpenaiClient.const_get(type)
+        klass.respond_to?(:openapi_one_of) ? klass.build(value) : klass.build_from_hash(value)
       end
     end
 

data/lib/azure_openai_client/version.rb

@@ -1,16 +1,15 @@
-# 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
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

data/lib/azure_openai_client.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
 
 # Common files
@@ -18,26 +17,27 @@ 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/chat_completions_create200_response'
+require 'azure_openai_client/models/chat_completions_create200_response_choices_inner'
+require 'azure_openai_client/models/chat_completions_create200_response_choices_inner_message'
+require 'azure_openai_client/models/chat_completions_create200_response_usage'
+require 'azure_openai_client/models/chat_completions_create_request'
+require 'azure_openai_client/models/chat_completions_create_request_messages_inner'
+require 'azure_openai_client/models/chat_completions_create_request_stop'
+require 'azure_openai_client/models/completions_create200_response'
+require 'azure_openai_client/models/completions_create200_response_choices_inner'
+require 'azure_openai_client/models/completions_create200_response_choices_inner_logprobs'
+require 'azure_openai_client/models/completions_create200_response_usage'
+require 'azure_openai_client/models/completions_create_request'
+require 'azure_openai_client/models/completions_create_request_prompt'
+require 'azure_openai_client/models/completions_create_request_stop'
+require 'azure_openai_client/models/embeddings_create200_response'
+require 'azure_openai_client/models/embeddings_create200_response_data_inner'
+require 'azure_openai_client/models/embeddings_create200_response_usage'
+require 'azure_openai_client/models/embeddings_create_request'
+require 'azure_openai_client/models/embeddings_create_request_input'
 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'

data/openapi-codegen.sh

@@ -0,0 +1,12 @@
+#!/bin/sh
+
+gem_version=$1
+release_notes=$2
+
+openapi-generator generate -c openapi_config.yaml \
+  --additional-properties gemVersion="$gem_version",\
+  --release-note "$release_notes"
+
+
+echo "Running Linter"
+rubocop -A

data/openapi_config.yaml

@@ -0,0 +1,14 @@
+inputSpec: inference.json
+generatorName: ruby
+outputDir: .
+additionalProperties:
+  gemName: azure_openai_client
+  gemLicense: MIT
+  gemHomepage: "https://github.com/etsenake/azure-openai-client"
+  gemAuthor: "Josh Etsenake"
+  gemAuthorEmail: "josh.etsenake@fullscript.com"
+  hideGenerationTimestamp: true
+gitUserId: etsenake
+gitRepoId: azure-openai-client
+globalProperties:
+  resolveFully: true