assemblyai 1.0.0.pre.beta

data/lib/assemblyai/transcripts/polling_client.rb

@@ -0,0 +1,173 @@
+# frozen_string_literal: true
+
+require_relative "../../requests"
+require_relative "types/transcript_status"
+require_relative "types/transcript_list"
+require_relative "types/transcript_language_code"
+require_relative "types/transcript_boost_param"
+require_relative "types/redact_pii_audio_quality"
+require_relative "types/pii_policy"
+require_relative "types/substitution_policy"
+require_relative "types/transcript_custom_spelling"
+require_relative "types/summary_model"
+require_relative "types/summary_type"
+require_relative "types/transcript"
+require_relative "types/subtitle_format"
+require_relative "types/sentences_response"
+require_relative "types/paragraphs_response"
+require_relative "types/word_search_response"
+require_relative "types/redacted_audio_response"
+require_relative "types/polling_options"
+require "async"
+
+module AssemblyAI
+  # :nodoc:
+  class TranscriptsClient
+    # Create a transcript from an audio or video file that is accessible via a URL.
+    # .transcribe polls for completion of the transcription, while the .submit function does not.
+    #
+    # @param language_code [TRANSCRIPT_LANGUAGE_CODE]
+    # @param punctuate [Boolean] Enable Automatic Punctuation, can be true or false
+    # @param format_text [Boolean] Enable Text Formatting, can be true or false
+    # @param dual_channel [Boolean] Enable [Dual Channel](https://www.assemblyai.com/docs/models/speech-recognition#dual-channel-transcription) transcription, can be true or false.
+    # @param webhook_url [String] The URL to which AssemblyAI send webhooks upon trancription completion
+    # @param webhook_auth_header_name [String] The header name which should be sent back with webhook calls
+    # @param webhook_auth_header_value [String] Specify a header name and value to send back with a webhook call for added security
+    # @param auto_highlights [Boolean] Whether Key Phrases is enabled, either true or false
+    # @param audio_start_from [Integer] The point in time, in milliseconds, to begin transcribing in your media file
+    # @param audio_end_at [Integer] The point in time, in milliseconds, to stop transcribing in your media file
+    # @param word_boost [Array<String>] The list of custom vocabulary to boost transcription probability for
+    # @param boost_param [TRANSCRIPT_BOOST_PARAM] The word boost parameter value
+    # @param filter_profanity [Boolean] Filter profanity from the transcribed text, can be true or false
+    # @param redact_pii [Boolean] Redact PII from the transcribed text using the Redact PII model, can be true or false
+    # @param redact_pii_audio [Boolean] Generate a copy of the original media file with spoken PII "beeped" out, can be true or false. See [PII redaction](https://www.assemblyai.com/docs/models/pii-redaction) for more details.
+    # @param redact_pii_audio_quality [REDACT_PII_AUDIO_QUALITY] Controls the filetype of the audio created by redact_pii_audio. Currently supports mp3 (default) and wav. See [PII redaction](https://www.assemblyai.com/docs/models/pii-redaction) for more details.
+    # @param redact_pii_policies [Array<Transcripts::PII_POLICY>] The list of PII Redaction policies to enable. See [PII redaction](https://www.assemblyai.com/docs/models/pii-redaction) for more details.
+    # @param redact_pii_sub [SUBSTITUTION_POLICY]
+    # @param speaker_labels [Boolean] Enable [Speaker diarization](https://www.assemblyai.com/docs/models/speaker-diarization), can be true or false
+    # @param speakers_expected [Integer] Tells the speaker label model how many speakers it should attempt to identify, up to 10. See [Speaker diarization](https://www.assemblyai.com/docs/models/speaker-diarization) for more details.
+    # @param content_safety [Boolean] Enable [Content Moderation](https://www.assemblyai.com/docs/models/content-moderation), can be true or false
+    # @param content_safety_confidence [Integer] The confidence threshold for content moderation. Values must be between 25 and 100.
+    # @param iab_categories [Boolean] Enable [Topic Detection](https://www.assemblyai.com/docs/models/topic-detection), can be true or false
+    # @param language_detection [Boolean] Whether [Automatic language detection](https://www.assemblyai.com/docs/models/speech-recognition#automatic-language-detection) was enabled in the transcription request, either true or false.
+    # @param custom_spelling [Array<Hash>] Customize how words are spelled and formatted using to and from valuesRequest of type Array<Transcripts::TranscriptCustomSpelling>, as a Hash
+    #   * :from (Array<String>)
+    #   * :to (String)
+    # @param disfluencies [Boolean] Transcribe Filler Words, like "umm", in your media file; can be true or false
+    # @param sentiment_analysis [Boolean] Enable [Sentiment Analysis](https://www.assemblyai.com/docs/models/sentiment-analysis), can be true or false
+    # @param auto_chapters [Boolean] Enable [Auto Chapters](https://www.assemblyai.com/docs/models/auto-chapters), can be true or false
+    # @param entity_detection [Boolean] Enable [Entity Detection](https://www.assemblyai.com/docs/models/entity-detection), can be true or false
+    # @param speech_threshold [Float] Reject audio files that contain less than this fraction of speech.
+    #   Valid values are in the range [0, 1] inclusive.
+    # @param summarization [Boolean] Enable [Summarization](https://www.assemblyai.com/docs/models/summarization), can be true or false
+    # @param summary_model [SUMMARY_MODEL] The model to summarize the transcript
+    # @param summary_type [SUMMARY_TYPE] The type of summary
+    # @param custom_topics [Boolean] Whether custom topics is enabled, either true or false
+    # @param topics [Array<String>] The list of custom topics provided, if custom topics is enabled
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @param audio_url [String] The URL of the audio or video file to transcribe.
+    # @param request_options [RequestOptions]
+    # @param polling_options [Transcripts::PollingOptions] Configuration options for polling requests.
+    # @return [Transcripts::Transcript]
+    def transcribe(audio_url:, language_code: nil, punctuate: nil, format_text: nil, dual_channel: nil, webhook_url: nil,
+                   webhook_auth_header_name: nil, webhook_auth_header_value: nil, auto_highlights: nil, audio_start_from: nil, audio_end_at: nil, word_boost: nil, boost_param: nil, filter_profanity: nil, redact_pii: nil, redact_pii_audio: nil, redact_pii_audio_quality: nil, redact_pii_policies: nil, redact_pii_sub: nil, speaker_labels: nil, speakers_expected: nil, content_safety: nil, content_safety_confidence: nil, iab_categories: nil, language_detection: nil, custom_spelling: nil, disfluencies: nil, sentiment_analysis: nil, auto_chapters: nil, entity_detection: nil, speech_threshold: nil, summarization: nil, summary_model: nil, summary_type: nil, custom_topics: nil, topics: nil, additional_properties: nil, request_options: nil, polling_options: Transcripts::PollingOptions.new)
+      transcript = submit(audio_url: audio_url, language_code: language_code, punctuate: punctuate, format_text: format_text, dual_channel: dual_channel, webhook_url: webhook_url,
+                          webhook_auth_header_name: webhook_auth_header_name, webhook_auth_header_value: webhook_auth_header_value, auto_highlights: auto_highlights, audio_start_from: audio_start_from, audio_end_at: audio_end_at, word_boost: word_boost, boost_param: boost_param, filter_profanity: filter_profanity, redact_pii: redact_pii, redact_pii_audio: redact_pii_audio, redact_pii_audio_quality: redact_pii_audio_quality, redact_pii_policies: redact_pii_policies, redact_pii_sub: redact_pii_sub, speaker_labels: speaker_labels, speakers_expected: speakers_expected, content_safety: content_safety, content_safety_confidence: content_safety_confidence, iab_categories: iab_categories, language_detection: language_detection, custom_spelling: custom_spelling, disfluencies: disfluencies, sentiment_analysis: sentiment_analysis, auto_chapters: auto_chapters, entity_detection: entity_detection, speech_threshold: speech_threshold, summarization: summarization, summary_model: summary_model, summary_type: summary_type, custom_topics: custom_topics, topics: topics, additional_properties: additional_properties, request_options: request_options)
+      poll_transcript(transcript_id: transcript.id, polling_options: polling_options)
+    end
+
+    def poll_transcript(transcript_id:, polling_options:)
+      start_time = Time.now
+      timeout_in_seconds = polling_options.timeout / 1000 if polling_options.timeout.positive?
+      loop do
+        transcript = get(transcript_id: transcript_id)
+        if transcript.status == :completed || transcript.status == :error
+          return transcript
+        elsif polling_options.timeout.positive? && Time.now - start_time > timeout_in_seconds
+          raise StandardError, "Polling timeout"
+        end
+
+        sleep polling_options.interval / 1000
+      end
+    end
+
+    private :poll_transcript
+  end
+
+  # :nodoc:
+  class AsyncTranscriptsClient
+    # Create a transcript from an audio or video file that is accessible via a URL.
+    # .transcribe polls for completion of the transcription, while the .submit function does not.
+    #
+    # @param language_code [TRANSCRIPT_LANGUAGE_CODE]
+    # @param punctuate [Boolean] Enable Automatic Punctuation, can be true or false
+    # @param format_text [Boolean] Enable Text Formatting, can be true or false
+    # @param dual_channel [Boolean] Enable [Dual Channel](https://www.assemblyai.com/docs/models/speech-recognition#dual-channel-transcription) transcription, can be true or false.
+    # @param webhook_url [String] The URL to which AssemblyAI send webhooks upon trancription completion
+    # @param webhook_auth_header_name [String] The header name which should be sent back with webhook calls
+    # @param webhook_auth_header_value [String] Specify a header name and value to send back with a webhook call for added security
+    # @param auto_highlights [Boolean] Whether Key Phrases is enabled, either true or false
+    # @param audio_start_from [Integer] The point in time, in milliseconds, to begin transcribing in your media file
+    # @param audio_end_at [Integer] The point in time, in milliseconds, to stop transcribing in your media file
+    # @param word_boost [Array<String>] The list of custom vocabulary to boost transcription probability for
+    # @param boost_param [TRANSCRIPT_BOOST_PARAM] The word boost parameter value
+    # @param filter_profanity [Boolean] Filter profanity from the transcribed text, can be true or false
+    # @param redact_pii [Boolean] Redact PII from the transcribed text using the Redact PII model, can be true or false
+    # @param redact_pii_audio [Boolean] Generate a copy of the original media file with spoken PII "beeped" out, can be true or false. See [PII redaction](https://www.assemblyai.com/docs/models/pii-redaction) for more details.
+    # @param redact_pii_audio_quality [REDACT_PII_AUDIO_QUALITY] Controls the filetype of the audio created by redact_pii_audio. Currently supports mp3 (default) and wav. See [PII redaction](https://www.assemblyai.com/docs/models/pii-redaction) for more details.
+    # @param redact_pii_policies [Array<Transcripts::PII_POLICY>] The list of PII Redaction policies to enable. See [PII redaction](https://www.assemblyai.com/docs/models/pii-redaction) for more details.
+    # @param redact_pii_sub [SUBSTITUTION_POLICY]
+    # @param speaker_labels [Boolean] Enable [Speaker diarization](https://www.assemblyai.com/docs/models/speaker-diarization), can be true or false
+    # @param speakers_expected [Integer] Tells the speaker label model how many speakers it should attempt to identify, up to 10. See [Speaker diarization](https://www.assemblyai.com/docs/models/speaker-diarization) for more details.
+    # @param content_safety [Boolean] Enable [Content Moderation](https://www.assemblyai.com/docs/models/content-moderation), can be true or false
+    # @param content_safety_confidence [Integer] The confidence threshold for content moderation. Values must be between 25 and 100.
+    # @param iab_categories [Boolean] Enable [Topic Detection](https://www.assemblyai.com/docs/models/topic-detection), can be true or false
+    # @param language_detection [Boolean] Whether [Automatic language detection](https://www.assemblyai.com/docs/models/speech-recognition#automatic-language-detection) was enabled in the transcription request, either true or false.
+    # @param custom_spelling [Array<Hash>] Customize how words are spelled and formatted using to and from valuesRequest of type Array<Transcripts::TranscriptCustomSpelling>, as a Hash
+    #   * :from (Array<String>)
+    #   * :to (String)
+    # @param disfluencies [Boolean] Transcribe Filler Words, like "umm", in your media file; can be true or false
+    # @param sentiment_analysis [Boolean] Enable [Sentiment Analysis](https://www.assemblyai.com/docs/models/sentiment-analysis), can be true or false
+    # @param auto_chapters [Boolean] Enable [Auto Chapters](https://www.assemblyai.com/docs/models/auto-chapters), can be true or false
+    # @param entity_detection [Boolean] Enable [Entity Detection](https://www.assemblyai.com/docs/models/entity-detection), can be true or false
+    # @param speech_threshold [Float] Reject audio files that contain less than this fraction of speech.
+    #   Valid values are in the range [0, 1] inclusive.
+    # @param summarization [Boolean] Enable [Summarization](https://www.assemblyai.com/docs/models/summarization), can be true or false
+    # @param summary_model [SUMMARY_MODEL] The model to summarize the transcript
+    # @param summary_type [SUMMARY_TYPE] The type of summary
+    # @param custom_topics [Boolean] Whether custom topics is enabled, either true or false
+    # @param topics [Array<String>] The list of custom topics provided, if custom topics is enabled
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @param audio_url [String] The URL of the audio or video file to transcribe.
+    # @param request_options [RequestOptions]
+    # @param polling_options [PollingOptions] Configuration options for polling requests.
+    # @return [Transcripts::Transcript]
+    def transcribe(audio_url:, language_code: nil, punctuate: nil, format_text: nil, dual_channel: nil, webhook_url: nil,
+                   webhook_auth_header_name: nil, webhook_auth_header_value: nil, auto_highlights: nil, audio_start_from: nil, audio_end_at: nil, word_boost: nil, boost_param: nil, filter_profanity: nil, redact_pii: nil, redact_pii_audio: nil, redact_pii_audio_quality: nil, redact_pii_policies: nil, redact_pii_sub: nil, speaker_labels: nil, speakers_expected: nil, content_safety: nil, content_safety_confidence: nil, iab_categories: nil, language_detection: nil, custom_spelling: nil, disfluencies: nil, sentiment_analysis: nil, auto_chapters: nil, entity_detection: nil, speech_threshold: nil, summarization: nil, summary_model: nil, summary_type: nil, custom_topics: nil, topics: nil, additional_properties: nil, request_options: nil, polling_options: Transcripts::PollingOptions.new)
+      Async do
+        transcript = submit(audio_url: audio_url, language_code: language_code, punctuate: punctuate, format_text: format_text, dual_channel: dual_channel, webhook_url: webhook_url,
+                            webhook_auth_header_name: webhook_auth_header_name, webhook_auth_header_value: webhook_auth_header_value, auto_highlights: auto_highlights, audio_start_from: audio_start_from, audio_end_at: audio_end_at, word_boost: word_boost, boost_param: boost_param, filter_profanity: filter_profanity, redact_pii: redact_pii, redact_pii_audio: redact_pii_audio, redact_pii_audio_quality: redact_pii_audio_quality, redact_pii_policies: redact_pii_policies, redact_pii_sub: redact_pii_sub, speaker_labels: speaker_labels, speakers_expected: speakers_expected, content_safety: content_safety, content_safety_confidence: content_safety_confidence, iab_categories: iab_categories, language_detection: language_detection, custom_spelling: custom_spelling, disfluencies: disfluencies, sentiment_analysis: sentiment_analysis, auto_chapters: auto_chapters, entity_detection: entity_detection, speech_threshold: speech_threshold, summarization: summarization, summary_model: summary_model, summary_type: summary_type, custom_topics: custom_topics, topics: topics, additional_properties: additional_properties, request_options: request_options).wait
+        poll_transcript(transcript_id: transcript.id, polling_options: polling_options).wait
+      end
+    end
+
+    def poll_transcript(transcript_id:, polling_options:)
+      Async do
+        start_time = Time.now
+        timeout_in_seconds = polling_options.timeout / 1000 if polling_options.timeout.positive?
+        loop do
+          transcript = get(transcript_id: transcript_id).wait
+          if transcript.status == :completed || transcript.status == :error
+            break transcript
+          elsif polling_options.timeout.positive? && Time.now - start_time > timeout_in_seconds
+            raise StandardError, "Polling timeout"
+          end
+
+          sleep polling_options.interval / 1000
+        end
+      end
+    end
+
+    private :poll_transcript
+  end
+end

data/lib/assemblyai/transcripts/types/audio_intelligence_model_status.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module AssemblyAI
+  class Transcripts
+    # @type [AUDIO_INTELLIGENCE_MODEL_STATUS]
+    AUDIO_INTELLIGENCE_MODEL_STATUS = { success: "success", unavailable: "unavailable" }.freeze
+  end
+end

data/lib/assemblyai/transcripts/types/auto_highlight_result.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require_relative "timestamp"
+require "json"
+
+module AssemblyAI
+  class Transcripts
+    class AutoHighlightResult
+      attr_reader :count, :rank, :text, :timestamps, :additional_properties
+
+      # @param count [Integer] The total number of times the key phrase appears in the audio file
+      # @param rank [Float] The total relevancy to the overall audio file of this key phrase - a greater number means more relevant
+      # @param text [String] The text itself of the key phrase
+      # @param timestamps [Array<Transcripts::Timestamp>] The timestamp of the of the key phrase
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Transcripts::AutoHighlightResult]
+      def initialize(count:, rank:, text:, timestamps:, additional_properties: nil)
+        # @type [Integer] The total number of times the key phrase appears in the audio file
+        @count = count
+        # @type [Float] The total relevancy to the overall audio file of this key phrase - a greater number means more relevant
+        @rank = rank
+        # @type [String] The text itself of the key phrase
+        @text = text
+        # @type [Array<Transcripts::Timestamp>] The timestamp of the of the key phrase
+        @timestamps = timestamps
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of AutoHighlightResult
+      #
+      # @param json_object [JSON]
+      # @return [Transcripts::AutoHighlightResult]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        parsed_json = JSON.parse(json_object)
+        count = struct.count
+        rank = struct.rank
+        text = struct.text
+        timestamps = parsed_json["timestamps"]&.map do |v|
+          v = v.to_json
+          Transcripts::Timestamp.from_json(json_object: v)
+        end
+        new(count: count, rank: rank, text: text, timestamps: timestamps, additional_properties: struct)
+      end
+
+      # Serialize an instance of AutoHighlightResult to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        { "count": @count, "rank": @rank, "text": @text, "timestamps": @timestamps }.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.count.is_a?(Integer) != false || raise("Passed value for field obj.count is not the expected type, validation failed.")
+        obj.rank.is_a?(Float) != false || raise("Passed value for field obj.rank is not the expected type, validation failed.")
+        obj.text.is_a?(String) != false || raise("Passed value for field obj.text is not the expected type, validation failed.")
+        obj.timestamps.is_a?(Array) != false || raise("Passed value for field obj.timestamps is not the expected type, validation failed.")
+      end
+    end
+  end
+end

data/lib/assemblyai/transcripts/types/auto_highlights_result.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+require_relative "auto_highlight_result"
+require "json"
+
+module AssemblyAI
+  class Transcripts
+    # An array of results for the Key Phrases model, if it is enabled.
+    # See [Key phrases](https://www.assemblyai.com/docs/models/key-phrases) for more information.
+    class AutoHighlightsResult
+      attr_reader :results, :additional_properties
+
+      # @param results [Array<Transcripts::AutoHighlightResult>] A temporally-sequential array of Key Phrases
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Transcripts::AutoHighlightsResult]
+      def initialize(results:, additional_properties: nil)
+        # @type [Array<Transcripts::AutoHighlightResult>] A temporally-sequential array of Key Phrases
+        @results = results
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of AutoHighlightsResult
+      #
+      # @param json_object [JSON]
+      # @return [Transcripts::AutoHighlightsResult]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        parsed_json = JSON.parse(json_object)
+        results = parsed_json["results"]&.map do |v|
+          v = v.to_json
+          Transcripts::AutoHighlightResult.from_json(json_object: v)
+        end
+        new(results: results, additional_properties: struct)
+      end
+
+      # Serialize an instance of AutoHighlightsResult to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        { "results": @results }.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.results.is_a?(Array) != false || raise("Passed value for field obj.results is not the expected type, validation failed.")
+      end
+    end
+  end
+end

data/lib/assemblyai/transcripts/types/chapter.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require "json"
+
+module AssemblyAI
+  class Transcripts
+    # Chapter of the audio file
+    class Chapter
+      attr_reader :gist, :headline, :summary, :start, :end_, :additional_properties
+
+      # @param gist [String] An ultra-short summary (just a few words) of the content spoken in the chapter
+      # @param headline [String] A single sentence summary of the content spoken during the chapter
+      # @param summary [String] A one paragraph summary of the content spoken during the chapter
+      # @param start [Integer] The starting time, in milliseconds, for the chapter
+      # @param end_ [Integer] The starting time, in milliseconds, for the chapter
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Transcripts::Chapter]
+      def initialize(gist:, headline:, summary:, start:, end_:, additional_properties: nil)
+        # @type [String] An ultra-short summary (just a few words) of the content spoken in the chapter
+        @gist = gist
+        # @type [String] A single sentence summary of the content spoken during the chapter
+        @headline = headline
+        # @type [String] A one paragraph summary of the content spoken during the chapter
+        @summary = summary
+        # @type [Integer] The starting time, in milliseconds, for the chapter
+        @start = start
+        # @type [Integer] The starting time, in milliseconds, for the chapter
+        @end_ = end_
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of Chapter
+      #
+      # @param json_object [JSON]
+      # @return [Transcripts::Chapter]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        JSON.parse(json_object)
+        gist = struct.gist
+        headline = struct.headline
+        summary = struct.summary
+        start = struct.start
+        end_ = struct.end
+        new(gist: gist, headline: headline, summary: summary, start: start, end_: end_, additional_properties: struct)
+      end
+
+      # Serialize an instance of Chapter to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        { "gist": @gist, "headline": @headline, "summary": @summary, "start": @start, "end": @end_ }.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.gist.is_a?(String) != false || raise("Passed value for field obj.gist is not the expected type, validation failed.")
+        obj.headline.is_a?(String) != false || raise("Passed value for field obj.headline is not the expected type, validation failed.")
+        obj.summary.is_a?(String) != false || raise("Passed value for field obj.summary is not the expected type, validation failed.")
+        obj.start.is_a?(Integer) != false || raise("Passed value for field obj.start is not the expected type, validation failed.")
+        obj.end_.is_a?(Integer) != false || raise("Passed value for field obj.end_ is not the expected type, validation failed.")
+      end
+    end
+  end
+end

data/lib/assemblyai/transcripts/types/content_safety_label.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "json"
+
+module AssemblyAI
+  class Transcripts
+    class ContentSafetyLabel
+      attr_reader :label, :confidence, :severity, :additional_properties
+
+      # @param label [String] The label of the sensitive topic
+      # @param confidence [Float] The confidence score for the topic being discussed, from 0 to 1
+      # @param severity [Float] How severely the topic is discussed in the section, from 0 to 1
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Transcripts::ContentSafetyLabel]
+      def initialize(label:, confidence:, severity:, additional_properties: nil)
+        # @type [String] The label of the sensitive topic
+        @label = label
+        # @type [Float] The confidence score for the topic being discussed, from 0 to 1
+        @confidence = confidence
+        # @type [Float] How severely the topic is discussed in the section, from 0 to 1
+        @severity = severity
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of ContentSafetyLabel
+      #
+      # @param json_object [JSON]
+      # @return [Transcripts::ContentSafetyLabel]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        JSON.parse(json_object)
+        label = struct.label
+        confidence = struct.confidence
+        severity = struct.severity
+        new(label: label, confidence: confidence, severity: severity, additional_properties: struct)
+      end
+
+      # Serialize an instance of ContentSafetyLabel to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        { "label": @label, "confidence": @confidence, "severity": @severity }.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.label.is_a?(String) != false || raise("Passed value for field obj.label is not the expected type, validation failed.")
+        obj.confidence.is_a?(Float) != false || raise("Passed value for field obj.confidence is not the expected type, validation failed.")
+        obj.severity.is_a?(Float) != false || raise("Passed value for field obj.severity is not the expected type, validation failed.")
+      end
+    end
+  end
+end

data/lib/assemblyai/transcripts/types/content_safety_label_result.rb

@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+
+require_relative "content_safety_label"
+require_relative "timestamp"
+require "json"
+
+module AssemblyAI
+  class Transcripts
+    class ContentSafetyLabelResult
+      attr_reader :text, :labels, :sentences_idx_start, :sentences_idx_end, :timestamp, :additional_properties
+
+      # @param text [String] The transcript of the section flagged by the Content Moderation model
+      # @param labels [Array<Transcripts::ContentSafetyLabel>] An array of safety labels, one per sensitive topic that was detected in the section
+      # @param sentences_idx_start [Integer] The sentence index at which the section begins
+      # @param sentences_idx_end [Integer] The sentence index at which the section ends
+      # @param timestamp [Transcripts::Timestamp] Timestamp information for the section
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Transcripts::ContentSafetyLabelResult]
+      def initialize(text:, labels:, sentences_idx_start:, sentences_idx_end:, timestamp:, additional_properties: nil)
+        # @type [String] The transcript of the section flagged by the Content Moderation model
+        @text = text
+        # @type [Array<Transcripts::ContentSafetyLabel>] An array of safety labels, one per sensitive topic that was detected in the section
+        @labels = labels
+        # @type [Integer] The sentence index at which the section begins
+        @sentences_idx_start = sentences_idx_start
+        # @type [Integer] The sentence index at which the section ends
+        @sentences_idx_end = sentences_idx_end
+        # @type [Transcripts::Timestamp] Timestamp information for the section
+        @timestamp = timestamp
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of ContentSafetyLabelResult
+      #
+      # @param json_object [JSON]
+      # @return [Transcripts::ContentSafetyLabelResult]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        parsed_json = JSON.parse(json_object)
+        text = struct.text
+        labels = parsed_json["labels"]&.map do |v|
+          v = v.to_json
+          Transcripts::ContentSafetyLabel.from_json(json_object: v)
+        end
+        sentences_idx_start = struct.sentences_idx_start
+        sentences_idx_end = struct.sentences_idx_end
+        if parsed_json["timestamp"].nil?
+          timestamp = nil
+        else
+          timestamp = parsed_json["timestamp"].to_json
+          timestamp = Transcripts::Timestamp.from_json(json_object: timestamp)
+        end
+        new(text: text, labels: labels, sentences_idx_start: sentences_idx_start, sentences_idx_end: sentences_idx_end,
+            timestamp: timestamp, additional_properties: struct)
+      end
+
+      # Serialize an instance of ContentSafetyLabelResult to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        {
+          "text": @text,
+          "labels": @labels,
+          "sentences_idx_start": @sentences_idx_start,
+          "sentences_idx_end": @sentences_idx_end,
+          "timestamp": @timestamp
+        }.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.text.is_a?(String) != false || raise("Passed value for field obj.text is not the expected type, validation failed.")
+        obj.labels.is_a?(Array) != false || raise("Passed value for field obj.labels is not the expected type, validation failed.")
+        obj.sentences_idx_start.is_a?(Integer) != false || raise("Passed value for field obj.sentences_idx_start is not the expected type, validation failed.")
+        obj.sentences_idx_end.is_a?(Integer) != false || raise("Passed value for field obj.sentences_idx_end is not the expected type, validation failed.")
+        Transcripts::Timestamp.validate_raw(obj: obj.timestamp)
+      end
+    end
+  end
+end

data/lib/assemblyai/transcripts/types/content_safety_labels_result.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+require_relative "audio_intelligence_model_status"
+require_relative "content_safety_label_result"
+require "json"
+
+module AssemblyAI
+  class Transcripts
+    # An array of results for the Content Moderation model, if it is enabled.
+    # See [Content moderation](https://www.assemblyai.com/docs/models/content-moderation) for more information.
+    class ContentSafetyLabelsResult
+      attr_reader :status, :results, :summary, :severity_score_summary, :additional_properties
+
+      # @param status [AUDIO_INTELLIGENCE_MODEL_STATUS] The status of the Content Moderation model. Either success, or unavailable in the rare case that the model failed.
+      # @param results [Array<Transcripts::ContentSafetyLabelResult>]
+      # @param summary [Hash{String => String}] A summary of the Content Moderation confidence results for the entire audio file
+      # @param severity_score_summary [Hash{String => String}] A summary of the Content Moderation severity results for the entire audio file
+      # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+      # @return [Transcripts::ContentSafetyLabelsResult]
+      def initialize(status:, results:, summary:, severity_score_summary:, additional_properties: nil)
+        # @type [AUDIO_INTELLIGENCE_MODEL_STATUS] The status of the Content Moderation model. Either success, or unavailable in the rare case that the model failed.
+        @status = status
+        # @type [Array<Transcripts::ContentSafetyLabelResult>]
+        @results = results
+        # @type [Hash{String => String}] A summary of the Content Moderation confidence results for the entire audio file
+        @summary = summary
+        # @type [Hash{String => String}] A summary of the Content Moderation severity results for the entire audio file
+        @severity_score_summary = severity_score_summary
+        # @type [OpenStruct] Additional properties unmapped to the current class definition
+        @additional_properties = additional_properties
+      end
+
+      # Deserialize a JSON object to an instance of ContentSafetyLabelsResult
+      #
+      # @param json_object [JSON]
+      # @return [Transcripts::ContentSafetyLabelsResult]
+      def self.from_json(json_object:)
+        struct = JSON.parse(json_object, object_class: OpenStruct)
+        parsed_json = JSON.parse(json_object)
+        status = Transcripts::AUDIO_INTELLIGENCE_MODEL_STATUS.key(parsed_json["status"]) || parsed_json["status"]
+        results = parsed_json["results"]&.map do |v|
+          v = v.to_json
+          Transcripts::ContentSafetyLabelResult.from_json(json_object: v)
+        end
+        summary = struct.summary
+        severity_score_summary = struct.severity_score_summary
+        new(status: status, results: results, summary: summary, severity_score_summary: severity_score_summary,
+            additional_properties: struct)
+      end
+
+      # Serialize an instance of ContentSafetyLabelsResult to a JSON object
+      #
+      # @return [JSON]
+      def to_json(*_args)
+        {
+          "status": Transcripts::AUDIO_INTELLIGENCE_MODEL_STATUS[@status] || @status,
+          "results": @results,
+          "summary": @summary,
+          "severity_score_summary": @severity_score_summary
+        }.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.status.is_a?(Transcripts::AUDIO_INTELLIGENCE_MODEL_STATUS) != false || raise("Passed value for field obj.status is not the expected type, validation failed.")
+        obj.results.is_a?(Array) != false || raise("Passed value for field obj.results is not the expected type, validation failed.")
+        obj.summary.is_a?(Hash) != false || raise("Passed value for field obj.summary is not the expected type, validation failed.")
+        obj.severity_score_summary.is_a?(Hash) != false || raise("Passed value for field obj.severity_score_summary is not the expected type, validation failed.")
+      end
+    end
+  end
+end