jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: annotation management frontend (#1764)

Browse source
This commit is contained in:
Joel 2023-12-18 15:41:24 +08:00 • committed by GitHub
commit 65fd4b39ce
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
122 changed files with 4718 additions and 214 deletions
Download patch file Download diff file Expand all files Collapse all files

26
web/context/provider-context.tsx
View file

@ -22,6 +22,7 @@ const ProviderContext = createContext<{
textGenerationDefaultModel?: BackendModel
mutateTextGenerationDefaultModel: () => void
embeddingsDefaultModel?: BackendModel
isEmbeddingsDefaultModelValid: boolean
mutateEmbeddingsDefaultModel: () => void
speech2textDefaultModel?: BackendModel
mutateSpeech2textDefaultModel: () => void
@ -42,16 +43,17 @@ const ProviderContext = createContext<{
speech2textModelList: [],
rerankModelList: [],
agentThoughtModelList: [],
updateModelList: () => {},
updateModelList: () => { },
textGenerationDefaultModel: undefined,
mutateTextGenerationDefaultModel: () => {},
mutateTextGenerationDefaultModel: () => { },
speech2textDefaultModel: undefined,
mutateSpeech2textDefaultModel: () => {},
mutateSpeech2textDefaultModel: () => { },
embeddingsDefaultModel: undefined,
mutateEmbeddingsDefaultModel: () => {},
isEmbeddingsDefaultModelValid: false,
mutateEmbeddingsDefaultModel: () => { },
rerankDefaultModel: undefined,
isRerankDefaultModelVaild: false,
mutateRerankDefaultModel: () => {},
mutateRerankDefaultModel: () => { },
supportRetrievalMethods: [],
plan: {
type: Plan.sandbox,
@ -59,11 +61,13 @@ const ProviderContext = createContext<{
vectorSpace: 32,
buildApps: 12,
teamMembers: 1,
annotatedResponse: 1,
},
total: {
vectorSpace: 200,
buildApps: 50,
teamMembers: 1,
annotatedResponse: 10,
},
},
isFetchedPlan: false,
@ -97,6 +101,10 @@ export const ProviderContextProvider = ({
item => item.model_name === rerankDefaultModel?.model_name && item.model_provider.provider_name === rerankDefaultModel?.model_provider.provider_name,
)
const isEmbeddingsDefaultModelValid = !!embeddingsModelList?.find(
item => item.model_name === embeddingsDefaultModel?.model_name && item.model_provider.provider_name === embeddingsDefaultModel?.model_provider.provider_name,
)
const updateModelList = (type: ModelType) => {
if (type === ModelType.textGeneration)
mutateTextGenerationModelList()
@ -118,6 +126,13 @@ export const ProviderContextProvider = ({
setEnableBilling(enabled)
if (enabled) {
setPlan(parseCurrentPlan(data))
// setPlan(parseCurrentPlan({
// ...data,
// annotation_quota_limit: {
// ...data.annotation_quota_limit,
// limit: 10,
// },
// }))
setIsFetchedPlan(true)
}
})()
@ -139,6 +154,7 @@ export const ProviderContextProvider = ({
mutateSpeech2textDefaultModel,
rerankDefaultModel,
isRerankDefaultModelVaild,
isEmbeddingsDefaultModelValid,
mutateRerankDefaultModel,
supportRetrievalMethods: supportRetrievalMethods?.retrieval_method || [],
plan,