feat: runtime prop in ThreadConfig (#390)

Yonom · Jul 4, 2024 · 19bf8bf · 19bf8bf
1 parent bfd91ea
commit 19bf8bf
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/packages/react-ui/src/components/thread-config.tsx b/packages/react-ui/src/components/thread-config.tsx
@@ -87,6 +87,8 @@ export type StringsConfig = {
 const ThreadConfigContext = createContext<ThreadConfig>({});
 
 export type ThreadConfig = {
+  runtime?: AssistantRuntime;
+
   assistantAvatar?: AvatarProps;
 
   welcome?: ThreadWelcomeConfig;
@@ -96,7 +98,7 @@ export type ThreadConfig = {
   strings?: StringsConfig;
 };
 
-export const useThreadConfig = () => {
+export const useThreadConfig = (): Omit<ThreadConfig, "runtime"> => {
   return useContext(ThreadConfigContext);
 };
 
@@ -108,11 +110,17 @@ export const ThreadConfigProvider: FC<ThreadConfigProviderProps> = ({
   children,
   config,
 }) => {
-  return (
+  const configProvider = (
     <ThreadConfigContext.Provider value={config}>
       {children}
     </ThreadConfigContext.Provider>
   );
+  if (!config.runtime) return configProvider;
+  return (
+    <AssistantRuntimeProvider runtime={config.runtime}>
+      {configProvider}
+    </AssistantRuntimeProvider>
+  );
 };
 
 ThreadConfigProvider.displayName = "ThreadConfigProvider";