feat: getRuntime() context API (#465)

assistant-ui · Jul 11, 2024 · b036399 · b036399
1 parent 85abdfa
commit b036399
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/packages/react/src/context/stores/AssistantActions.tsx b/packages/react/src/context/stores/AssistantActions.tsx
@@ -4,6 +4,7 @@ import { MutableRefObject } from "react";
 
 export type AssistantActionsState = Readonly<{
   switchToThread: (threadId: string | null) => void;
+  getRuntime: () => AssistantRuntime;
 }>;
 
 export const makeAssistantActionsStore = (
@@ -12,5 +13,6 @@ export const makeAssistantActionsStore = (
   create<AssistantActionsState>(() =>
     Object.freeze({
       switchToThread: () => runtimeRef.current.switchToThread(null),
+      getRuntime: () => runtimeRef.current,
     }),
   );
diff --git a/packages/react/src/context/stores/ThreadActions.ts b/packages/react/src/context/stores/ThreadActions.ts
@@ -24,6 +24,7 @@ export type ThreadActionsState = Readonly<{
   cancelRun: () => void;
 
   addToolResult: (options: AddToolResultOptions) => void;
+  getRuntime: () => ThreadRuntime;
 }>;
 
 export const makeThreadActionStore = (
@@ -40,6 +41,7 @@ export const makeThreadActionStore = (
       append: (message) => runtimeRef.current.append(message),
       cancelRun: () => runtimeRef.current.cancelRun(),
       addToolResult: (options) => runtimeRef.current.addToolResult(options),
+      getRuntime: () => runtimeRef.current,
     }),
   );
 };