feat: Implement new commands for recording control, configuration management, latency testing, and status display, updating program entry and project references.

2026-02-27 01:12:23 +01:00
parent d910fe1441
commit 482fe84eb1
8 changed files with 494 additions and 368 deletions
--- a/Commands/LatencyTestCommand.cs
+++ b/Commands/LatencyTestCommand.cs
@@ -0,0 +1,95 @@
+using System;
+using System.Diagnostics;
+using System.IO;
+using System.Threading.Tasks;
+using Spectre.Console;
+using Toak.Api;
+using Toak.Configuration;
+using Toak.Core;
+
+namespace Toak.Commands;
+
+public static class LatencyTestCommand
+{
+    public static async Task ExecuteAsync(bool verbose)
+    {
+        Logger.Verbose = verbose;
+        var config = ConfigManager.LoadConfig();
+        if (string.IsNullOrWhiteSpace(config.GroqApiKey))
+        {
+            AnsiConsole.MarkupLine("[red]Groq API Key is not configured.[/] Run 'toak onboard'.");
+            return;
+        }
+
+        AnsiConsole.MarkupLine("Generating 1-second silent audio file for testing...");
+        var testWavPath = Path.Combine(Path.GetTempPath(), "toak_latency_test.wav");
+        
+        var pInfo = new ProcessStartInfo
+        {
+            FileName = "ffmpeg",
+            Arguments = $"-f lavfi -i anullsrc=r=44100:cl=mono -t 1 -y {testWavPath}",
+            UseShellExecute = false,
+            CreateNoWindow = true,
+            RedirectStandardError = true,
+            RedirectStandardOutput = true
+        };
+        var proc = Process.Start(pInfo);
+        proc?.WaitForExit();
+
+        if (!File.Exists(testWavPath))
+        {
+            AnsiConsole.MarkupLine("[red]Failed to generate test audio file using ffmpeg.[/]");
+            return;
+        }
+
+        var groq = new GroqApiClient(config.GroqApiKey);
+        
+        try
+        {
+            await AnsiConsole.Status()
+                .StartAsync("Running latency test...", async ctx => 
+                {
+                    ctx.Status("Testing STT (Whisper)...");
+                    var sttWatch = Stopwatch.StartNew();
+                    var transcript = await groq.TranscribeAsync(testWavPath, config.WhisperLanguage, config.WhisperModel);
+                    sttWatch.Stop();
+                    
+                    ctx.Status("Testing LLM (Llama)...");
+                    var systemPrompt = PromptBuilder.BuildPrompt(config);
+                    var llmWatch = Stopwatch.StartNew();
+                    var refinedText = await groq.RefineTextAsync("Hello world, this is a latency test.", systemPrompt, config.LlmModel);
+                    llmWatch.Stop();
+
+                    var total = sttWatch.ElapsedMilliseconds + llmWatch.ElapsedMilliseconds;
+
+                    AnsiConsole.WriteLine();
+                    var table = new Table();
+                    table.AddColumn("Operation");
+                    table.AddColumn("Latency (ms)");
+                    
+                    table.AddRow("STT", sttWatch.ElapsedMilliseconds.ToString());
+                    table.AddRow("LLM", llmWatch.ElapsedMilliseconds.ToString());
+                    table.AddRow("[bold]Total[/]", $"[bold]{total}ms[/]");
+
+                    AnsiConsole.Write(table);
+
+                    if (total < 1500)
+                    {
+                        AnsiConsole.MarkupLine($"[green]Status: OK (under 1.5s target). Total time: {(total / 1000.0):0.0}s.[/]");
+                    }
+                    else
+                    {
+                        AnsiConsole.MarkupLine($"[yellow]Status: SLOW (over 1.5s target). Total time: {(total / 1000.0):0.0}s.[/]");
+                    }
+                });
+        }
+        catch (Exception ex)
+        {
+            AnsiConsole.MarkupLine($"[red]Error during test: {ex.Message}[/]");
+        }
+        finally
+        {
+            if (File.Exists(testWavPath)) File.Delete(testWavPath);
+        }
+    }
+}