@assistant-ui/core 0.2.11 → 0.2.14

package/src/react/RuntimeAdapter.ts

@@ -1,20 +1,37 @@
+import { useEffect } from "react";
 import { useResource, resource } from "@assistant-ui/tap";
 import type { AssistantRuntime } from "..";
 import {
-  RuntimeAdapterResource,
   baseRuntimeAdapterTransformScopes,
+  ThreadListClient,
 } from "../store/internal";
-import { attachTransformScopes } from "@assistant-ui/store";
+import {
+  attachTransformScopes,
+  useAssistantClientRef,
+} from "@assistant-ui/store";
 import { DataRenderers } from "./client/DataRenderers";
 import { Tools } from "./client/Tools";
 
-export const RuntimeAdapter = resource(function RuntimeAdapter(
-  runtime: AssistantRuntime,
-) {
-  return useResource(RuntimeAdapterResource(runtime));
-});
+const useRuntimeAdapter = (runtime: AssistantRuntime) => {
+  const clientRef = useAssistantClientRef();
+
+  useEffect(() => {
+    return runtime.registerModelContextProvider(
+      clientRef.current!.modelContext(),
+    );
+  }, [runtime, clientRef]);
+
+  return useResource(
+    ThreadListClient({
+      runtime: runtime.threads,
+      __internal_assistantRuntime: runtime,
+    }),
+  );
+};
+
+export const RuntimeAdapter = resource(useRuntimeAdapter);
 
-attachTransformScopes(RuntimeAdapter, (scopes, parent) => {
+attachTransformScopes(useRuntimeAdapter, (scopes, parent) => {
   baseRuntimeAdapterTransformScopes(scopes, parent);
 
   if (!scopes.tools && parent.tools.source === null) {

package/src/react/client/DataRenderers.ts

@@ -11,61 +11,58 @@ import type { DataMessagePartComponent } from "../types/MessagePartComponentType
  * directly for a renderer scope, or prefer {@link useAssistantDataUI} /
  * {@link makeAssistantDataUI} when registering from React components.
  */
-export const DataRenderers = resource(
-  function DataRenderers(): ClientOutput<"dataRenderers"> {
-    const [state, setState] = useState<DataRenderersState>(() => ({
-      renderers: {},
-      fallbacks: [],
-    }));
+const useDataRenderers = (): ClientOutput<"dataRenderers"> => {
+  const [state, setState] = useState<DataRenderersState>(() => ({
+    renderers: {},
+    fallbacks: [],
+  }));
+
+  const setDataUI = useCallback(
+    (name: string, render: DataMessagePartComponent) => {
+      setState((prev) => {
+        return {
+          ...prev,
+          renderers: {
+            ...prev.renderers,
+            [name]: [...(prev.renderers[name] ?? []), render],
+          },
+        };
+      });
 
-    const setDataUI = useCallback(
-      (name: string, render: DataMessagePartComponent) => {
+      return () => {
         setState((prev) => {
           return {
             ...prev,
             renderers: {
               ...prev.renderers,
-              [name]: [...(prev.renderers[name] ?? []), render],
+              [name]: prev.renderers[name]?.filter((r) => r !== render) ?? [],
             },
           };
         });
+      };
+    },
+    [],
+  );
 
-        return () => {
-          setState((prev) => {
-            return {
-              ...prev,
-              renderers: {
-                ...prev.renderers,
-                [name]: prev.renderers[name]?.filter((r) => r !== render) ?? [],
-              },
-            };
-          });
-        };
-      },
-      [],
-    );
+  const setFallbackDataUI = useCallback((render: DataMessagePartComponent) => {
+    setState((prev) => ({
+      ...prev,
+      fallbacks: [...prev.fallbacks, render],
+    }));
 
-    const setFallbackDataUI = useCallback(
-      (render: DataMessagePartComponent) => {
-        setState((prev) => ({
-          ...prev,
-          fallbacks: [...prev.fallbacks, render],
-        }));
+    return () => {
+      setState((prev) => ({
+        ...prev,
+        fallbacks: prev.fallbacks.filter((r) => r !== render),
+      }));
+    };
+  }, []);
 
-        return () => {
-          setState((prev) => ({
-            ...prev,
-            fallbacks: prev.fallbacks.filter((r) => r !== render),
-          }));
-        };
-      },
-      [],
-    );
+  return {
+    getState: () => state,
+    setDataUI,
+    setFallbackDataUI,
+  };
+};
 
-    return {
-      getState: () => state,
-      setDataUI,
-      setFallbackDataUI,
-    };
-  },
-);
+export const DataRenderers = resource(useDataRenderers);