assemblyai 1.0.0.pre.beta → 1.0.0.pre.beta.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bc0caf7b84497b13453af130bf9ed562274853b8a40c27c84dbe0e485271f099
-  data.tar.gz: fa25a137746aeb43fd13427e7e8c7de382b0a6684744fa7c805f32b2c9ffcc45
+  metadata.gz: fd39472d57199089cb681e33ce94ea14d140afc6018ee578281188e35de6ee58
+  data.tar.gz: 6d2e5f6222b4ec927f296434c3a263573bdcbca1d91aeebd41186b216c956d52
 SHA512:
-  metadata.gz: 3e6c4c159194b38ed20d04228f253892ec24a4a4550a6ad1f29601a790689d833d5b608a5d8ba19a0fa8cf6585a8af1c5039a62f32d3b3c8664f2aceba68c2cf
-  data.tar.gz: 5bffb7383c0b3e0e7ff9c9128c3022db428746beed340222157c49cf6b993f95d1a7cd60e93843cbfd3cd17ff6c4e9e3029b10fb4746a2d5adbe099077eda58f
+  metadata.gz: b59ba30872b39a055dbc337545776e57d6fcbd89731fe7eceb63eac274030660b9fae00f118926c83ef2fb51a457f5e7d1cb52197ac1ce9d62c94cbde96b4645
+  data.tar.gz: 92413fba5d2e366018bc971fa95bea0cd1829e2f1c9b59079c103168b35569b266c60fdd95f5825ebd1d8ad81efd5120f5c5a604f2c783d00b3c615a17843c8b

data/lib/assemblyai/lemur/client.rb

@@ -28,8 +28,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -66,8 +66,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -104,8 +104,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -146,8 +146,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -211,8 +211,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -251,8 +251,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -291,8 +291,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.
@@ -335,8 +335,8 @@ module AssemblyAI
     #   Use either transcript_ids or input_text as input into LeMUR.
     # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
     #   Use either transcript_ids or input_text as input into LeMUR.
-    # @param context [Hash] Context to provide the model. This can be a string or a free-form JSON value.Request of type Lemur::LemurBaseParamsContext, as a Hash
-    # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+    # @param context [String, Hash{String => String}] Context to provide the model. This can be a string or a free-form JSON value.
+    # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
     #   Defaults to "default".
     # @param max_output_size [Integer] Max output size in tokens, up to 4000
     # @param temperature [Float] The temperature to use for the model.

data/lib/assemblyai/lemur/types/lemur_base_params.rb

@@ -15,7 +15,7 @@ module AssemblyAI
       # @param input_text [String] Custom formatted transcript data. Maximum size is the context limit of the selected model, which defaults to 100000.
       #   Use either transcript_ids or input_text as input into LeMUR.
       # @param context [Lemur::LemurBaseParamsContext] Context to provide the model. This can be a string or a free-form JSON value.
-      # @param final_model [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+      # @param final_model [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
       #   Defaults to "default".
       # @param max_output_size [Integer] Max output size in tokens, up to 4000
       # @param temperature [Float] The temperature to use for the model.
@@ -33,7 +33,7 @@ module AssemblyAI
         @input_text = input_text
         # @type [Lemur::LemurBaseParamsContext] Context to provide the model. This can be a string or a free-form JSON value.
         @context = context
-        # @type [LEMUR_MODEL] The model that is used for the final prompt after compression is performed.
+        # @type [Lemur::LemurModel] The model that is used for the final prompt after compression is performed.
         #   Defaults to "default".
         @final_model = final_model
         # @type [Integer] Max output size in tokens, up to 4000
@@ -61,7 +61,7 @@ module AssemblyAI
           context = parsed_json["context"].to_json
           context = Lemur::LemurBaseParamsContext.from_json(json_object: context)
         end
-        final_model = Lemur::LEMUR_MODEL.key(parsed_json["final_model"]) || parsed_json["final_model"]
+        final_model = struct.final_model
         max_output_size = struct.max_output_size
         temperature = struct.temperature
         new(transcript_ids: transcript_ids, input_text: input_text, context: context, final_model: final_model,
@@ -76,7 +76,7 @@ module AssemblyAI
           "transcript_ids": @transcript_ids,
           "input_text": @input_text,
           "context": @context,
-          "final_model": Lemur::LEMUR_MODEL[@final_model] || @final_model,
+          "final_model": @final_model,
           "max_output_size": @max_output_size,
           "temperature": @temperature
         }.to_json
@@ -90,7 +90,7 @@ module AssemblyAI
         obj.transcript_ids&.is_a?(Array) != false || raise("Passed value for field obj.transcript_ids is not the expected type, validation failed.")
         obj.input_text&.is_a?(String) != false || raise("Passed value for field obj.input_text is not the expected type, validation failed.")
         obj.context.nil? || Lemur::LemurBaseParamsContext.validate_raw(obj: obj.context)
-        obj.final_model&.is_a?(Lemur::LEMUR_MODEL) != false || raise("Passed value for field obj.final_model is not the expected type, validation failed.")
+        obj.final_model&.is_a?(Lemur::LemurModel) != false || raise("Passed value for field obj.final_model is not the expected type, validation failed.")
         obj.max_output_size&.is_a?(Integer) != false || raise("Passed value for field obj.max_output_size is not the expected type, validation failed.")
         obj.temperature&.is_a?(Float) != false || raise("Passed value for field obj.temperature is not the expected type, validation failed.")
       end

data/lib/assemblyai/lemur/types/lemur_base_params_context.rb

@@ -6,15 +6,6 @@ module AssemblyAI
   class Lemur
     # Context to provide the model. This can be a string or a free-form JSON value.
     class LemurBaseParamsContext
-      attr_reader :member
-      alias kind_of? is_a?
-      # @param member [Object]
-      # @return [Lemur::LemurBaseParamsContext]
-      def initialize(member:)
-        # @type [Object]
-        @member = member
-      end
-
       # Deserialize a JSON object to an instance of LemurBaseParamsContext
       #
       # @param json_object [JSON]
@@ -23,26 +14,17 @@ module AssemblyAI
         struct = JSON.parse(json_object, object_class: OpenStruct)
         begin
           struct.is_a?(String) != false || raise("Passed value for field struct is not the expected type, validation failed.")
-          member = json_object
-          return new(member: member)
+          return json_object
         rescue StandardError
           # noop
         end
         begin
           struct.is_a?(Hash) != false || raise("Passed value for field struct is not the expected type, validation failed.")
-          member = json_object
-          return new(member: member)
+          return json_object
         rescue StandardError
           # noop
         end
-        new(member: struct)
-      end
-
-      # For Union Types, to_json functionality is delegated to the wrapped member.
-      #
-      # @return [JSON]
-      def to_json(*_args)
-        @member.to_json
+        struct
       end
 
       # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
@@ -62,14 +44,6 @@ module AssemblyAI
         end
         raise("Passed value matched no type within the union, validation failed.")
       end
-
-      # For Union Types, is_a? functionality is delegated to the wrapped member.
-      #
-      # @param obj [Object]
-      # @return [Boolean]
-      def is_a?(obj)
-        @member.is_a?(obj)
-      end
     end
   end
 end

data/lib/assemblyai/lemur/types/lemur_model.rb

@@ -2,12 +2,12 @@
 
 module AssemblyAI
   class Lemur
-    # @type [LEMUR_MODEL]
-    LEMUR_MODEL = {
-      default: "default",
-      basic: "basic",
-      assemblyai_mistral7b: "assemblyai/mistral-7b",
-      anthropic_claude21: "anthropic/claude-2-1"
-    }.freeze
+    # The model that is used for the final prompt after compression is performed.
+    class LemurModel
+      DEFAULT = "default"
+      BASIC = "basic"
+      ASSEMBLYAI_MISTRAL7B = "assemblyai/mistral-7b"
+      ANTHROPIC_CLAUDE2_1 = "anthropic/claude-2-1"
+    end
   end
 end

data/lib/assemblyai/lemur/types/lemur_question_context.rb

@@ -6,15 +6,6 @@ module AssemblyAI
   class Lemur
     # Any context about the transcripts you wish to provide. This can be a string or any object.
     class LemurQuestionContext
-      attr_reader :member
-      alias kind_of? is_a?
-      # @param member [Object]
-      # @return [Lemur::LemurQuestionContext]
-      def initialize(member:)
-        # @type [Object]
-        @member = member
-      end
-
       # Deserialize a JSON object to an instance of LemurQuestionContext
       #
       # @param json_object [JSON]
@@ -23,26 +14,17 @@ module AssemblyAI
         struct = JSON.parse(json_object, object_class: OpenStruct)
         begin
           struct.is_a?(String) != false || raise("Passed value for field struct is not the expected type, validation failed.")
-          member = json_object
-          return new(member: member)
+          return json_object
         rescue StandardError
           # noop
         end
         begin
           struct.is_a?(Hash) != false || raise("Passed value for field struct is not the expected type, validation failed.")
-          member = json_object
-          return new(member: member)
+          return json_object
         rescue StandardError
           # noop
         end
-        new(member: struct)
-      end
-
-      # For Union Types, to_json functionality is delegated to the wrapped member.
-      #
-      # @return [JSON]
-      def to_json(*_args)
-        @member.to_json
+        struct
       end
 
       # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
@@ -62,14 +44,6 @@ module AssemblyAI
         end
         raise("Passed value matched no type within the union, validation failed.")
       end
-
-      # For Union Types, is_a? functionality is delegated to the wrapped member.
-      #
-      # @param obj [Object]
-      # @return [Boolean]
-      def is_a?(obj)
-        @member.is_a?(obj)
-      end
     end
   end
 end

data/lib/assemblyai/realtime/types/audio_encoding.rb

@@ -2,7 +2,10 @@
 
 module AssemblyAI
   class Realtime
-    # @type [AUDIO_ENCODING]
-    AUDIO_ENCODING = { pcm_s16le: "pcm_s16le", pcm_mulaw: "pcm_mulaw" }.freeze
+    # The encoding of the audio data
+    class AudioEncoding
+      PCM_S16LE = "pcm_s16le"
+      PCM_MULAW = "pcm_mulaw"
+    end
   end
 end

data/lib/assemblyai/realtime/types/configure_end_utterance_silence_threshold.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require "json"
+
+module AssemblyAI
+  class Realtime
+    # Configure the threshold for how long to wait before ending an utterance. Default is 700ms.
+    class ConfigureEndUtteranceSilenceThreshold
+      attr_reader :end_utterance_silence_threshold, :additional_properties
+
+      # @param end_utterance_silence_threshold [Integer] The duration threshold in milliseconds
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Realtime::ConfigureEndUtteranceSilenceThreshold]
+      def initialize(end_utterance_silence_threshold:, additional_properties: nil)
+        # @type [Integer] The duration threshold in milliseconds
+        @end_utterance_silence_threshold = end_utterance_silence_threshold
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of ConfigureEndUtteranceSilenceThreshold
+      #
+      # @param json_object [JSON]
+      # @return [Realtime::ConfigureEndUtteranceSilenceThreshold]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        JSON.parse(json_object)
+        end_utterance_silence_threshold = struct.end_utterance_silence_threshold
+        new(end_utterance_silence_threshold: end_utterance_silence_threshold, additional_properties: struct)
+      end
+
+      # Serialize an instance of ConfigureEndUtteranceSilenceThreshold to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        { "end_utterance_silence_threshold": @end_utterance_silence_threshold }.to_json
+      end
+
+      # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+      #
+      # @param obj [Object]
+      # @return [Void]
+      def self.validate_raw(obj:)
+        obj.end_utterance_silence_threshold.is_a?(Integer) != false || raise("Passed value for field obj.end_utterance_silence_threshold is not the expected type, validation failed.")
+      end
+    end
+  end
+end

data/lib/assemblyai/realtime/types/force_end_utterance.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require "json"
+
+module AssemblyAI
+  class Realtime
+    # Manually end an utterance
+    class ForceEndUtterance
+      attr_reader :force_end_utterance, :additional_properties
+
+      # @param force_end_utterance [Boolean] A boolean value to communicate that you wish to force the end of the utterance
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Realtime::ForceEndUtterance]
+      def initialize(force_end_utterance:, additional_properties: nil)
+        # @type [Boolean] A boolean value to communicate that you wish to force the end of the utterance
+        @force_end_utterance = force_end_utterance
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of ForceEndUtterance
+      #
+      # @param json_object [JSON]
+      # @return [Realtime::ForceEndUtterance]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        JSON.parse(json_object)
+        force_end_utterance = struct.force_end_utterance
+        new(force_end_utterance: force_end_utterance, additional_properties: struct)
+      end
+
+      # Serialize an instance of ForceEndUtterance to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        { "force_end_utterance": @force_end_utterance }.to_json
+      end
+
+      # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+      #
+      # @param obj [Object]
+      # @return [Void]
+      def self.validate_raw(obj:)
+        obj.force_end_utterance.is_a?(Boolean) != false || raise("Passed value for field obj.force_end_utterance is not the expected type, validation failed.")
+      end
+    end
+  end
+end

data/lib/assemblyai/realtime/types/message_type.rb

@@ -2,12 +2,11 @@
 
 module AssemblyAI
   class Realtime
-    # @type [MESSAGE_TYPE]
-    MESSAGE_TYPE = {
-      session_begins: "SessionBegins",
-      partial_transcript: "PartialTranscript",
-      final_transcript: "FinalTranscript",
-      session_terminated: "SessionTerminated"
-    }.freeze
+    class MessageType
+      SESSION_BEGINS = "SessionBegins"
+      PARTIAL_TRANSCRIPT = "PartialTranscript"
+      FINAL_TRANSCRIPT = "FinalTranscript"
+      SESSION_TERMINATED = "SessionTerminated"
+    end
   end
 end

data/lib/assemblyai/realtime/types/realtime_base_message.rb

@@ -8,11 +8,11 @@ module AssemblyAI
     class RealtimeBaseMessage
       attr_reader :message_type, :additional_properties
 
-      # @param message_type [MESSAGE_TYPE] Describes the type of the message
+      # @param message_type [Realtime::MessageType] Describes the type of the message
       # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
       # @return [Realtime::RealtimeBaseMessage]
       def initialize(message_type:, additional_properties: nil)
-        # @type [MESSAGE_TYPE] Describes the type of the message
+        # @type [Realtime::MessageType] Describes the type of the message
         @message_type = message_type
         # @type [OpenStruct] Additional properties unmapped to the current class definition
         @additional_properties = additional_properties
@@ -24,8 +24,8 @@ module AssemblyAI
       # @return [Realtime::RealtimeBaseMessage]
       def self.from_json(json_object:)
         struct = JSON.parse(json_object, object_class: OpenStruct)
-        parsed_json = JSON.parse(json_object)
-        message_type = Realtime::MESSAGE_TYPE.key(parsed_json["message_type"]) || parsed_json["message_type"]
+        JSON.parse(json_object)
+        message_type = struct.message_type
         new(message_type: message_type, additional_properties: struct)
       end
 
@@ -33,7 +33,7 @@ module AssemblyAI
       #
       # @return [JSON]
       def to_json(*_args)
-        { "message_type": Realtime::MESSAGE_TYPE[@message_type] || @message_type }.to_json
+        { "message_type": @message_type }.to_json
       end
 
       # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
@@ -41,7 +41,7 @@ module AssemblyAI
       # @param obj [Object]
       # @return [Void]
       def self.validate_raw(obj:)
-        obj.message_type.is_a?(Realtime::MESSAGE_TYPE) != false || raise("Passed value for field obj.message_type is not the expected type, validation failed.")
+        obj.message_type.is_a?(Realtime::MessageType) != false || raise("Passed value for field obj.message_type is not the expected type, validation failed.")
       end
     end
   end

data/lib/assemblyai/realtime/types/realtime_message.rb

@@ -10,15 +10,6 @@ require_relative "realtime_error"
 module AssemblyAI
   class Realtime
     class RealtimeMessage
-      attr_reader :member
-      alias kind_of? is_a?
-      # @param member [Object]
-      # @return [Realtime::RealtimeMessage]
-      def initialize(member:)
-        # @type [Object]
-        @member = member
-      end
-
       # Deserialize a JSON object to an instance of RealtimeMessage
       #
       # @param json_object [JSON]
@@ -27,47 +18,35 @@ module AssemblyAI
         struct = JSON.parse(json_object, object_class: OpenStruct)
         begin
           Realtime::SessionBegins.validate_raw(obj: struct)
-          member = Realtime::SessionBegins.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::SessionBegins.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
         begin
           Realtime::PartialTranscript.validate_raw(obj: struct)
-          member = Realtime::PartialTranscript.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::PartialTranscript.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
         begin
           Realtime::FinalTranscript.validate_raw(obj: struct)
-          member = Realtime::FinalTranscript.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::FinalTranscript.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
         begin
           Realtime::SessionTerminated.validate_raw(obj: struct)
-          member = Realtime::SessionTerminated.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::SessionTerminated.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
         begin
           Realtime::RealtimeError.validate_raw(obj: struct)
-          member = Realtime::RealtimeError.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::RealtimeError.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
-        new(member: struct)
-      end
-
-      # For Union Types, to_json functionality is delegated to the wrapped member.
-      #
-      # @return [JSON]
-      def to_json(*_args)
-        @member.to_json
+        struct
       end
 
       # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
@@ -102,14 +81,6 @@ module AssemblyAI
         end
         raise("Passed value matched no type within the union, validation failed.")
       end
-
-      # For Union Types, is_a? functionality is delegated to the wrapped member.
-      #
-      # @param obj [Object]
-      # @return [Boolean]
-      def is_a?(obj)
-        @member.is_a?(obj)
-      end
     end
   end
 end

data/lib/assemblyai/realtime/types/realtime_transcript.rb

@@ -7,15 +7,6 @@ require_relative "final_transcript"
 module AssemblyAI
   class Realtime
     class RealtimeTranscript
-      attr_reader :member
-      alias kind_of? is_a?
-      # @param member [Object]
-      # @return [Realtime::RealtimeTranscript]
-      def initialize(member:)
-        # @type [Object]
-        @member = member
-      end
-
       # Deserialize a JSON object to an instance of RealtimeTranscript
       #
       # @param json_object [JSON]
@@ -24,26 +15,17 @@ module AssemblyAI
         struct = JSON.parse(json_object, object_class: OpenStruct)
         begin
           Realtime::PartialTranscript.validate_raw(obj: struct)
-          member = Realtime::PartialTranscript.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::PartialTranscript.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
         begin
           Realtime::FinalTranscript.validate_raw(obj: struct)
-          member = Realtime::FinalTranscript.from_json(json_object: json_object)
-          return new(member: member)
+          return Realtime::FinalTranscript.from_json(json_object: json_object)
         rescue StandardError
           # noop
         end
-        new(member: struct)
-      end
-
-      # For Union Types, to_json functionality is delegated to the wrapped member.
-      #
-      # @return [JSON]
-      def to_json(*_args)
-        @member.to_json
+        struct
       end
 
       # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
@@ -63,14 +45,6 @@ module AssemblyAI
         end
         raise("Passed value matched no type within the union, validation failed.")
       end
-
-      # For Union Types, is_a? functionality is delegated to the wrapped member.
-      #
-      # @param obj [Object]
-      # @return [Boolean]
-      def is_a?(obj)
-        @member.is_a?(obj)
-      end
     end
   end
 end

data/lib/assemblyai/realtime/types/realtime_transcript_type.rb

@@ -2,7 +2,9 @@
 
 module AssemblyAI
   class Realtime
-    # @type [REALTIME_TRANSCRIPT_TYPE]
-    REALTIME_TRANSCRIPT_TYPE = { partial_transcript: "PartialTranscript", final_transcript: "FinalTranscript" }.freeze
+    class RealtimeTranscriptType
+      PARTIAL_TRANSCRIPT = "PartialTranscript"
+      FINAL_TRANSCRIPT = "FinalTranscript"
+    end
   end
 end