Merge branch 'feat/rag-2' into feat/workflow-draft-var-optimize

This commit is contained in:
QuantumGhost
2025-08-31 15:17:07 +08:00
1456 changed files with 86020 additions and 17662 deletions

View File

@ -1,17 +1,18 @@
import { createContext, useContext, useContextSelector } from 'use-context-selector'
import type { DataSet } from '@/models/datasets'
import type { IndexingType } from '@/app/components/datasets/create/step-two'
import type { QueryObserverResult, RefetchOptions } from '@tanstack/react-query'
type DatasetDetailContextValue = {
indexingTechnique?: IndexingType
dataset?: DataSet
mutateDatasetRes?: () => void
mutateDatasetRes?: (options?: RefetchOptions | undefined) => Promise<QueryObserverResult<DataSet, Error>>
}
const DatasetDetailContext = createContext<DatasetDetailContextValue>({})
export const useDatasetDetailContext = () => useContext(DatasetDetailContext)
export const useDatasetDetailContextWithSelector = (selector: (value: DatasetDetailContextValue) => any) => {
export const useDatasetDetailContextWithSelector = <T>(selector: (value: DatasetDetailContextValue) => T): T => {
return useContextSelector(DatasetDetailContext, selector)
}
export default DatasetDetailContext

View File

@ -61,6 +61,7 @@ type ProviderContextState = {
},
refreshLicenseLimit: () => void
isAllowTransferWorkspace: boolean
isAllowPublishAsKnowledgePipeline: boolean
}
const ProviderContext = createContext<ProviderContextState>({
modelProviders: [],
@ -107,6 +108,7 @@ const ProviderContext = createContext<ProviderContextState>({
},
refreshLicenseLimit: noop,
isAllowTransferWorkspace: false,
isAllowPublishAsKnowledgePipeline: false,
})
export const useProviderContext = () => useContext(ProviderContext)
@ -145,6 +147,7 @@ export const ProviderContextProvider = ({
const [isEducationWorkspace, setIsEducationWorkspace] = useState(false)
const { data: educationAccountInfo, isLoading: isLoadingEducationAccountInfo, isFetching: isFetchingEducationAccountInfo } = useEducationStatus(!enableEducationPlan)
const [isAllowTransferWorkspace, setIsAllowTransferWorkspace] = useState(false)
const [isAllowPublishAsKnowledgePipeline, setIsAllowPublishAsKnowledgePipeline] = useState(false)
const fetchPlan = async () => {
try {
@ -175,6 +178,8 @@ export const ProviderContextProvider = ({
setLicenseLimit({ workspace_members: data.workspace_members })
if (data.is_allow_transfer_workspace)
setIsAllowTransferWorkspace(data.is_allow_transfer_workspace)
if (data.knowledge_pipeline?.publish_enabled)
setIsAllowPublishAsKnowledgePipeline(data.knowledge_pipeline?.publish_enabled)
}
catch (error) {
console.error('Failed to fetch plan info:', error)
@ -240,6 +245,7 @@ export const ProviderContextProvider = ({
licenseLimit,
refreshLicenseLimit: fetchPlan,
isAllowTransferWorkspace,
isAllowPublishAsKnowledgePipeline,
}}>
{children}
</ProviderContext.Provider>