feat: useAssistantRuntime (#704)

packages/react/src/context/providers/AssistantProvider.tsx

@@ -8,6 +8,11 @@ import { makeAssistantModelConfigStore } from "../stores/AssistantModelConfig";
 import { makeAssistantToolUIsStore } from "../stores/AssistantToolUIs";
 import { ThreadProvider } from "./ThreadProvider";
 import { makeAssistantActionsStore } from "../stores/AssistantActions";
+import {
+  AssistantRuntimeStore,
+  makeAssistantRuntimeStore,
+} from "../stores/AssistantRuntime";
+import { StoreApi } from "zustand";
 
 type AssistantProviderProps = {
   runtime: AssistantRuntime;
@@ -22,18 +27,32 @@ export const AssistantProvider: FC<
   });
 
   const [context] = useState(() => {
+    const useAssistantRuntime = makeAssistantRuntimeStore(runtime);
     const useModelConfig = makeAssistantModelConfigStore();
     const useToolUIs = makeAssistantToolUIsStore();
     const useAssistantActions = makeAssistantActionsStore(runtimeRef);
 
-    return { useModelConfig, useToolUIs, useAssistantActions };
+    return {
+      useModelConfig,
+      useToolUIs,
+      useAssistantRuntime,
+      useAssistantActions,
+    };
   });
 
   const getModelConfig = context.useModelConfig();
   useEffect(() => {
     return runtime.registerModelConfigProvider(getModelConfig);
   }, [runtime, getModelConfig]);
 
+  useEffect(
+    () =>
+      (
+        context.useAssistantRuntime as unknown as StoreApi<AssistantRuntimeStore>
+      ).setState(runtime, true),
+    [runtime, context],
+  );
+
   return (
     <AssistantContext.Provider value={context}>
       <ThreadProvider provider={runtime}>{children}</ThreadProvider>

packages/react/src/context/react/AssistantContext.ts

@@ -5,10 +5,12 @@ import type { AssistantModelConfigState } from "../stores/AssistantModelConfig";
 import type { AssistantToolUIsState } from "../stores/AssistantToolUIs";
 import { ReadonlyStore } from "../ReadonlyStore";
 import { AssistantActionsState } from "../stores/AssistantActions";
+import { AssistantRuntime } from "../../runtimes";
 
 export type AssistantContextValue = {
   useModelConfig: ReadonlyStore<AssistantModelConfigState>;
   useToolUIs: ReadonlyStore<AssistantToolUIsState>;
+  useAssistantRuntime: ReadonlyStore<AssistantRuntime>;
   useAssistantActions: ReadonlyStore<AssistantActionsState>;
 };
 

packages/react/src/context/stores/AssistantRuntime.tsx

@@ -0,0 +1,8 @@
+import { create } from "zustand";
+import { AssistantRuntime } from "../../runtimes";
+
+export type AssistantRuntimeStore = AssistantRuntime;
+
+export const makeAssistantRuntimeStore = (runtime: AssistantRuntime) => {
+  return create<AssistantRuntimeStore>(() => runtime);
+};