Merge pull request #56 from AssemblyAI/niels/content-safety

Fix content safety model parsing

Author: Swimburger

src/AssemblyAI.IntegrationTests/AudioIntelligenceTests.cs

@@ -0,0 +1,47 @@
+using AssemblyAI.Transcripts;
+
+namespace AssemblyAI.IntegrationTests;
+
+[TestFixture]
+public class AudioIntelligenceTests
+{
+    private const string RemoteAudioUrl = "https://assembly.ai/espn.m4a";
+
+    [Test]
+    public async Task ShouldReturnContentSafety()
+    {
+        var client = Helpers.CreateClient();
+
+        var transcript = await client.Transcripts.TranscribeAsync(new TranscriptParams
+            {
+                AudioUrl = RemoteAudioUrl,
+                ContentSafety = true,
+                ContentSafetyConfidence = 50,
+            }
+        ).ConfigureAwait(false);
+
+        Assert.That(transcript, Is.Not.Null);
+        Assert.Multiple(() =>
+        {
+            Assert.That(transcript.Id, Is.Not.Null);
+            Assert.That(transcript.Text, Is.Not.Empty);
+            Assert.That(transcript.Status, Is.EqualTo(TranscriptStatus.Completed));
+            Assert.That(transcript.ContentSafetyLabels, Is.Not.Null);
+            Assert.Multiple(() =>
+                {
+                    Assert.That(
+                        transcript.ContentSafetyLabels!.Status,
+                        Is.EqualTo(AudioIntelligenceModelStatus.Success)
+                    );
+                    Assert.That(
+                        transcript.ContentSafetyLabels!.Results,
+                        Is.Not.Empty
+                    );
+                    var result = transcript.ContentSafetyLabels!.Results.First();
+                    Assert.That(result.Text, Is.Not.Empty);
+                    Assert.That(result.Labels, Is.Not.Empty);
+                }
+            );
+        });
+    }
+}

src/AssemblyAI/Transcripts/Types/ContentSafetyLabel.cs

@@ -23,7 +23,7 @@ public record ContentSafetyLabel
     /// How severely the topic is discussed in the section, from 0 to 1
     /// </summary>
     [JsonPropertyName("severity")]
-    public required double Severity { get; set; }
+    public required double? Severity { get; set; }
 
     public override string ToString()
     {