Fix dictation readiness after app refresh

2026-04-20 17:23:17 +07:00
parent ac45f23213
commit 93095aa1d2
5 changed files with 187 additions and 12 deletions
@@ -0,0 +1,44 @@
+import { useCallback } from "react";
+
+import type { HostRuntimeAgentDirectoryStatus } from "@/runtime/host-runtime";
+import { useSessionStore } from "@/stores/session-store";
+import { getVoiceReadinessState } from "@/utils/server-info-capabilities";
+
+function isLegacyDictationReady(agentDirectoryStatus: HostRuntimeAgentDirectoryStatus): boolean {
+  return (
+    agentDirectoryStatus === "ready" ||
+    agentDirectoryStatus === "revalidating" ||
+    agentDirectoryStatus === "error_after_ready"
+  );
+}
+
+export function useIsDictationReady({
+  serverId,
+  isConnected,
+  agentDirectoryStatus,
+}: {
+  serverId: string;
+  isConnected: boolean;
+  agentDirectoryStatus: HostRuntimeAgentDirectoryStatus;
+}): boolean {
+  const dictationCapabilityEnabled = useSessionStore(
+    useCallback(
+      (state) => {
+        const serverInfo = state.sessions[serverId]?.serverInfo ?? null;
+        return (
+          getVoiceReadinessState({
+            serverInfo,
+            mode: "dictation",
+          })?.enabled ?? null
+        );
+      },
+      [serverId],
+    ),
+  );
+
+  if (!isConnected) {
+    return false;
+  }
+
+  return dictationCapabilityEnabled ?? isLegacyDictationReady(agentDirectoryStatus);
+}