@assistant-ui/react-ag-ui 0.0.29 → 0.0.31

package/src/useAgUiRuntime.ts

@@ -15,12 +15,23 @@ import type {
 } from "@assistant-ui/core";
 import type { ReadonlyJSONValue } from "assistant-stream/utils";
 import { makeLogger } from "./runtime/logger";
-import type { UseAgUiRuntimeOptions } from "./runtime/types";
+import type {
+  AgUiInterrupt,
+  AgUiResumeEntry,
+  UseAgUiRuntimeOptions,
+} from "./runtime/types";
 import { AgUiThreadRuntimeCore } from "./runtime/AgUiThreadRuntimeCore";
 
+export type AgUiAssistantRuntime = AssistantRuntime & {
+  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];
+  unstable_submitInterruptResponses: (
+    responses: readonly AgUiResumeEntry[],
+  ) => Promise<void>;
+};
+
 export function useAgUiRuntime(
   options: UseAgUiRuntimeOptions,
-): AssistantRuntime {
+): AgUiAssistantRuntime {
   const logger = useMemo(() => makeLogger(options.logger), [options.logger]);
   const [_version, setVersion] = useState(0);
   const notifyUpdate = useCallback(() => setVersion((v) => v + 1), []);
@@ -174,7 +185,16 @@ export function useAgUiRuntime(
     [adapterAdapters, core, _version, hasExecutingTools],
   );
 
-  const runtime = useExternalStoreRuntime(store);
+  const baseRuntime = useExternalStoreRuntime(store);
+
+  const runtime = useMemo<AgUiAssistantRuntime>(() => {
+    const wrapper = Object.create(baseRuntime) as AgUiAssistantRuntime;
+    wrapper.unstable_getPendingInterrupts = () =>
+      core.getPendingInterrupts()?.interrupts ?? [];
+    wrapper.unstable_submitInterruptResponses = (responses) =>
+      core.submitInterruptResponses(responses);
+    return wrapper;
+  }, [baseRuntime, core]);
 
   useEffect(() => {
     core.attachRuntime(runtime);