dify

dify/web/app/components/base/features/hooks.ts

@@ -0,0 +1,16 @@
+import { useContext } from 'react'
+import { useStore } from 'zustand'
+import { FeaturesContext } from './context'
+import type { FeatureStoreState } from './store'
+
+export function useFeatures<T>(selector: (state: FeatureStoreState) => T): T {
+  const store = useContext(FeaturesContext)
+  if (!store)
+    throw new Error('Missing FeaturesContext.Provider in the tree')
+
+  return useStore(store, selector)
+}
+
+export function useFeaturesStore() {
+  return useContext(FeaturesContext)
+}