feat: classify data panel node sync

This commit is contained in:
Joel 2024-03-05 11:06:35 +08:00
parent 2e649c3329
commit 86656de971
2 changed files with 3 additions and 15 deletions

View File

@ -32,7 +32,7 @@ const allMockData = {
[BlockEnum.End]: EndNodeMock,
}
const nodes = [
BlockEnum.Start/* 1 */, BlockEnum.LLM/* 3 */, BlockEnum.DirectAnswer/* 2 */, BlockEnum.KnowledgeRetrieval/* 4 */, BlockEnum.QuestionClassifier/* 5 */,
BlockEnum.QuestionClassifier/* 5 */, BlockEnum.Start/* 1 */, BlockEnum.DirectAnswer/* 2 */, BlockEnum.LLM/* 3 */, BlockEnum.KnowledgeRetrieval/* 4 */,
BlockEnum.IfElse/* 6 */, BlockEnum.Code/* 7 */, BlockEnum.TemplateTransform/* 8 */, BlockEnum.HttpRequest/* 9 */, BlockEnum.Tool/* 10 */,
BlockEnum.VariableAssigner/* 11 */, BlockEnum.End/* 12 */,
].map((item, i) => {
@ -42,7 +42,7 @@ const nodes = [
type: 'custom',
position: { x: 330, y: 30 + i * 300 },
data: {
selected: i === 0, // for test: always select the first node
_selected: i === 0, // for test: always select the first node
name: item,
...payload,
},

View File

@ -1,5 +1,4 @@
import type { FC } from 'react'
import { useState } from 'react'
import type { NodeProps } from 'reactflow'
import InfoPanel from '../_base/components/info-panel'
import { NodeSourceHandle } from '../_base/components/node-handle'
@ -14,7 +13,7 @@ const Node: FC<NodeProps<QuestionClassifierNodeType>> = (props) => {
const { data } = props
const { provider, name: modelId } = data.model
// const tempTopics = data.topics
const [topics, setTopics] = useState(data.topics)
const topics = data.topics
const {
textGenerationModelList,
} = useTextGenerationCurrentProviderAndModelAndModelList()
@ -42,17 +41,6 @@ const Node: FC<NodeProps<QuestionClassifierNodeType>> = (props) => {
/>
</div>
))}
{/* For test */}
<div
className='mt-1 flex items-center h-6 justify-center bg-gray-100 rounded-md px-1 space-x-1 text-xs font-normal text-gray-700'
onClick={() => {
setTopics([...topics, {
id: `${Date.now()}`,
name: `Topic${topics.length}`,
topic: `Topic${topics.length}`,
}])
}}
>Add</div>
</div>
</div>
)