| import type { FC } from 'react' |
| import React from 'react' |
| import type { ParameterExtractorNodeType } from './types' |
| import { |
| useTextGenerationCurrentProviderAndModelAndModelList, |
| } from '@/app/components/header/account-setting/model-provider-page/hooks' |
| import ModelSelector from '@/app/components/header/account-setting/model-provider-page/model-selector' |
| import type { NodeProps } from '@/app/components/workflow/types' |
|
|
| const Node: FC<NodeProps<ParameterExtractorNodeType>> = ({ |
| data, |
| }) => { |
| const { provider, name: modelId } = data.model || {} |
| const { |
| textGenerationModelList, |
| } = useTextGenerationCurrentProviderAndModelAndModelList() |
| const hasSetModel = provider && modelId |
| return ( |
| <div className='mb-1 px-3 py-1'> |
| {hasSetModel && ( |
| <ModelSelector |
| defaultModel={{ provider, model: modelId }} |
| modelList={textGenerationModelList} |
| readonly |
| /> |
| )} |
| </div> |
| ) |
| } |
|
|
| export default React.memo(Node) |
|
|