Warm up pyannote and FunASR runtimes

2026-05-28 13:05:27 +02:00
parent 7ff93b73b3
commit c46c587e65
14 changed files with 696 additions and 20 deletions
@@ -181,6 +181,54 @@ public sealed class PyannoteTranscriptFinalizerTests
        Assert.Equal("hf_azure", commandRunner.Environment["HF_TOKEN"]);
    }

+    [Fact]
+    public async Task WarmUpBuildsImageAndDownloadsConfiguredModelWithoutAudioInput()
+    {
+        var commandRunner = new CapturingCommandRunner("");
+        var finalizer = CreateFinalizer(commandRunner, token: "hf_test");
+        var diarization = new PyannoteDiarizationOptions
+        {
+            Enabled = true,
+            DockerCommand = "docker",
+            BaseImage = "python:3.11-slim",
+            Image = "meeting-assistant-pyannote-warmup:local",
+            ModelsFolder = Path.Combine(Path.GetTempPath(), "meeting-assistant-tests", Guid.NewGuid().ToString("N"), "models"),
+            Model = "pyannote/speaker-diarization-3.1",
+            Token = "hf_test",
+            TokenEnv = "",
+            CommandTimeout = TimeSpan.FromMinutes(1)
+        };
+
+        await finalizer.WarmUpAsync(diarization, CancellationToken.None);
+
+        Assert.Contains(commandRunner.Commands, command => command.Arguments.Contains("inspect"));
+        Assert.Contains(commandRunner.Commands, command => command.Arguments.Contains("build"));
+        var runCommand = commandRunner.Commands.Last();
+        Assert.Contains("run", runCommand.Arguments);
+        Assert.Contains("meeting-assistant-pyannote-warmup:local", runCommand.Arguments);
+        Assert.DoesNotContain(runCommand.Arguments, argument => argument.Contains("/workspace/input.wav", StringComparison.Ordinal));
+        Assert.Contains(runCommand.Arguments, argument => argument.Contains("Pipeline.from_pretrained", StringComparison.Ordinal));
+        Assert.Contains(runCommand.Arguments, argument => argument.Contains("pyannote/speaker-diarization-3.1", StringComparison.Ordinal));
+        Assert.Equal("hf_test", commandRunner.Environment["HF_TOKEN"]);
+    }
+
+    [Fact]
+    public async Task WarmUpSkipsWhenTokenIsMissing()
+    {
+        var commandRunner = new CapturingCommandRunner("");
+        var finalizer = CreateFinalizer(commandRunner, token: null);
+
+        await finalizer.WarmUpAsync(new PyannoteDiarizationOptions
+        {
+            Enabled = true,
+            Token = "",
+            TokenEnv = "",
+            ModelsFolder = Path.Combine(Path.GetTempPath(), "meeting-assistant-tests", Guid.NewGuid().ToString("N"), "models")
+        }, CancellationToken.None);
+
+        Assert.Empty(commandRunner.Commands);
+    }
+
    [Fact]
    public async Task FinalizerSkipsPyannoteWhenTokenIsMissing()
    {