feat(web): template download

This commit is contained in:
JzoNg
2026-04-10 17:05:14 +08:00
parent 670ab16ea1
commit ba951b01de
4 changed files with 14 additions and 64 deletions

View File

@@ -11,15 +11,6 @@ const mockUseEvaluationConfig = vi.hoisted(() => vi.fn())
const mockUseEvaluationNodeInfoMutation = vi.hoisted(() => vi.fn())
const mockUseSaveEvaluationConfigMutation = vi.hoisted(() => vi.fn())
const mockUseStartEvaluationRunMutation = vi.hoisted(() => vi.fn())
const mockUsePublishedPipelineInfo = vi.hoisted(() => vi.fn())
vi.mock('@/context/dataset-detail', () => ({
useDatasetDetailContextWithSelector: (selector: (state: { dataset?: { pipeline_id?: string } }) => unknown) => selector({
dataset: {
pipeline_id: 'pipeline-1',
},
}),
}))
vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({
useModelList: () => ({
@@ -64,10 +55,6 @@ vi.mock('@/service/use-evaluation', () => ({
useStartEvaluationRunMutation: (...args: unknown[]) => mockUseStartEvaluationRunMutation(...args),
}))
vi.mock('@/service/use-pipeline', () => ({
usePublishedPipelineInfo: (...args: unknown[]) => mockUsePublishedPipelineInfo(...args),
}))
vi.mock('@/service/use-workflow', () => ({
useAppWorkflow: () => ({
data: {
@@ -165,24 +152,6 @@ describe('Evaluation', () => {
isPending: false,
mutate: vi.fn(),
})
mockUsePublishedPipelineInfo.mockReturnValue({
data: {
rag_pipeline_variables: [{
belong_to_node_id: 'shared',
type: 'text-input',
label: 'Question',
variable: 'question',
required: true,
}, {
belong_to_node_id: 'shared',
type: 'number',
label: 'Top K',
variable: 'top_k',
required: false,
}],
},
isLoading: false,
})
mockUpload.mockResolvedValue({
id: 'uploaded-file-id',
name: 'evaluation.csv',
@@ -469,15 +438,15 @@ describe('Evaluation', () => {
fireEvent.click(screen.getByRole('button', { name: /Context Precision/i }))
fireEvent.click(screen.getByRole('button', { name: 'evaluation.pipeline.uploadAndRun' }))
expect(screen.getAllByText('question').length).toBeGreaterThan(0)
expect(screen.getAllByText('top_k').length).toBeGreaterThan(0)
expect(screen.getAllByText('query').length).toBeGreaterThan(0)
expect(screen.getAllByText('Expect Results').length).toBeGreaterThan(0)
const fileInput = document.querySelector<HTMLInputElement>('input[type="file"][accept=".csv,.xlsx"]')
expect(fileInput).toBeInTheDocument()
fireEvent.change(fileInput!, {
target: {
files: [new File(['case_id,input,expected'], 'pipeline-evaluation.csv', { type: 'text/csv' })],
files: [new File(['query,Expect Results'], 'pipeline-evaluation.csv', { type: 'text/csv' })],
},
})

View File

@@ -1,8 +1,6 @@
import type { StartNodeType } from '@/app/components/workflow/nodes/start/types'
import type { InputVar, Node } from '@/app/components/workflow/types'
import type { RAGPipelineVariables } from '@/models/pipeline'
import { inputVarTypeToVarType } from '@/app/components/workflow/nodes/_base/components/variable/utils'
import { inputVarTypeToVarType as pipelineInputVarTypeToVarType } from '@/app/components/workflow/nodes/data-source/utils'
import { BlockEnum, InputVarType } from '@/app/components/workflow/types'
export type InputField = {
@@ -29,18 +27,6 @@ export const getStartNodeInputFields = (nodes?: Node[]): InputField[] => {
}))
}
export const getRagPipelineInputFields = (variables?: RAGPipelineVariables): InputField[] => {
if (!Array.isArray(variables))
return []
return variables
.filter(variable => typeof variable.variable === 'string' && !!variable.variable)
.map(variable => ({
name: variable.variable,
type: pipelineInputVarTypeToVarType(variable.type),
}))
}
const escapeCsvCell = (value: string) => {
if (!/[",\n\r]/.test(value))
return value

View File

@@ -1,10 +1,8 @@
import type { EvaluationResourceType } from '../../../types'
import { useMemo } from 'react'
import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
import { usePublishedPipelineInfo } from '@/service/use-pipeline'
import { useSnippetPublishedWorkflow } from '@/service/use-snippet-workflows'
import { useAppWorkflow } from '@/service/use-workflow'
import { getGraphNodes, getRagPipelineInputFields, getStartNodeInputFields } from './input-fields-utils'
import { getGraphNodes, getStartNodeInputFields } from './input-fields-utils'
export const usePublishedInputFields = (
resourceType: EvaluationResourceType,
@@ -12,8 +10,6 @@ export const usePublishedInputFields = (
) => {
const { data: currentAppWorkflow, isLoading: isAppWorkflowLoading } = useAppWorkflow(resourceType === 'apps' ? resourceId : '')
const { data: currentSnippetWorkflow, isLoading: isSnippetWorkflowLoading } = useSnippetPublishedWorkflow(resourceType === 'snippets' ? resourceId : '')
const pipelineId = useDatasetDetailContextWithSelector(state => state.dataset?.pipeline_id)
const { data: currentPipelineWorkflow, isLoading: isPipelineWorkflowLoading } = usePublishedPipelineInfo(resourceType === 'datasets' ? (pipelineId ?? '') : '')
const inputFields = useMemo(() => {
if (resourceType === 'apps')
@@ -22,16 +18,12 @@ export const usePublishedInputFields = (
if (resourceType === 'snippets')
return getStartNodeInputFields(getGraphNodes(currentSnippetWorkflow?.graph))
if (resourceType === 'datasets')
return getRagPipelineInputFields(currentPipelineWorkflow?.rag_pipeline_variables)
return []
}, [currentAppWorkflow?.graph.nodes, currentPipelineWorkflow?.rag_pipeline_variables, currentSnippetWorkflow?.graph, resourceType])
}, [currentAppWorkflow?.graph.nodes, currentSnippetWorkflow?.graph, resourceType])
return {
inputFields,
isInputFieldsLoading: (resourceType === 'apps' && isAppWorkflowLoading)
|| (resourceType === 'snippets' && isSnippetWorkflowLoading)
|| (resourceType === 'datasets' && isPipelineWorkflowLoading),
|| (resourceType === 'snippets' && isSnippetWorkflowLoading),
}
}

View File

@@ -10,13 +10,17 @@ import { getEvaluationMockConfig } from '../../mock'
import { isEvaluationRunnable, useEvaluationResource, useEvaluationStore } from '../../store'
import UploadRunPopover from '../batch-test-panel/input-fields/upload-run-popover'
import { useInputFieldsActions } from '../batch-test-panel/input-fields/use-input-fields-actions'
import { usePublishedInputFields } from '../batch-test-panel/input-fields/use-published-input-fields'
import JudgeModelSelector from '../judge-model-selector'
import PipelineHistoryTable from '../pipeline/pipeline-history-table'
import PipelineMetricItem from '../pipeline/pipeline-metric-item'
import PipelineResultsPanel from '../pipeline/pipeline-results-panel'
import SectionHeader, { InlineSectionHeader } from '../section-header'
const PIPELINE_INPUT_FIELDS = [
{ name: 'query', type: 'string' },
{ name: 'Expect Results', type: 'string' },
]
const PipelineEvaluation = ({
resourceType,
resourceId,
@@ -44,12 +48,11 @@ const PipelineEvaluation = ({
}, [availableMetricIds, builtinMetricMap, config.builtinMetrics])
const isConfigReady = !!resource.judgeModelId && builtinMetricMap.size > 0
const isRunnable = isEvaluationRunnable(resource)
const { inputFields, isInputFieldsLoading } = usePublishedInputFields(resourceType, resourceId)
const actions = useInputFieldsActions({
resourceType,
resourceId,
inputFields,
isInputFieldsLoading,
inputFields: PIPELINE_INPUT_FIELDS,
isInputFieldsLoading: false,
isPanelReady: isConfigReady,
isRunnable,
templateFileName: config.templateFileName,
@@ -142,7 +145,7 @@ const PipelineEvaluation = ({
onOpenChange={actions.setIsUploadPopoverOpen}
triggerDisabled={actions.uploadButtonDisabled}
triggerLabel={t('pipeline.uploadAndRun')}
inputFields={inputFields}
inputFields={PIPELINE_INPUT_FIELDS}
currentFileName={actions.currentFileName}
currentFileExtension={actions.currentFileExtension}
currentFileSize={actions.currentFileSize}