@assistant-ui/react
Version:
TypeScript/React library for AI Chat
53 lines • 2.3 kB
JavaScript
"use client";
import { useEffect, useMemo, useRef, useState } from "react";
import { LocalRuntimeCore } from "./LocalRuntimeCore.js";
import { useRuntimeAdapters } from "../adapters/RuntimeAdapterProvider.js";
import { useRemoteThreadListRuntime } from "../remote-thread-list/useRemoteThreadListRuntime.js";
import { useCloudThreadListAdapter } from "../remote-thread-list/adapter/cloud.js";
import { AssistantRuntimeImpl } from "../../../internal.js";
import { useAssistantState } from "../../../context/react/index.js";
const useLocalThreadRuntime = (adapter, { initialMessages, ...options }) => {
const { modelContext, ...threadListAdapters } = useRuntimeAdapters() ?? {};
const opt = useMemo(() => ({
...options,
adapters: {
...threadListAdapters,
...options.adapters,
chatModel: adapter,
},
}), [adapter, options, threadListAdapters]);
const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
const threadIdRef = useRef(undefined);
threadIdRef.current = useAssistantState(({ threadListItem }) => threadListItem.remoteId);
useEffect(() => {
runtime.threads
.getMainThreadRuntimeCore()
.__internal_setGetThreadId(() => threadIdRef.current);
}, [runtime]);
useEffect(() => {
return () => {
runtime.threads.getMainThreadRuntimeCore().detach();
};
}, [runtime]);
useEffect(() => {
runtime.threads.getMainThreadRuntimeCore().__internal_setOptions(opt);
runtime.threads.getMainThreadRuntimeCore().__internal_load();
}, [runtime, opt]);
useEffect(() => {
if (!modelContext)
return undefined;
return runtime.registerModelContextProvider(modelContext);
}, [modelContext, runtime]);
return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);
};
export const useLocalRuntime = (adapter, { cloud, ...options } = {}) => {
const cloudAdapter = useCloudThreadListAdapter({ cloud });
return useRemoteThreadListRuntime({
runtimeHook: function RuntimeHook() {
return useLocalThreadRuntime(adapter, options);
},
adapter: cloudAdapter,
allowNesting: true,
});
};
//# sourceMappingURL=useLocalRuntime.js.map