d1971e988a
### What problem does this PR solve? Feat: The Begin and IterationStart operators cannot be deleted using shortcut keys #4287 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
58 lines
1.5 KiB
TypeScript
58 lines
1.5 KiB
TypeScript
import LLMLabel from '@/components/llm-select/llm-label';
|
|
import { useTheme } from '@/components/theme-provider';
|
|
import { IRewriteNode } from '@/interfaces/database/flow';
|
|
import { Handle, NodeProps, Position } from '@xyflow/react';
|
|
import classNames from 'classnames';
|
|
import { get } from 'lodash';
|
|
import { LeftHandleStyle, RightHandleStyle } from './handle-icon';
|
|
import styles from './index.less';
|
|
import NodeHeader from './node-header';
|
|
|
|
export function RewriteNode({
|
|
id,
|
|
data,
|
|
isConnectable = true,
|
|
selected,
|
|
}: NodeProps<IRewriteNode>) {
|
|
const { theme } = useTheme();
|
|
return (
|
|
<section
|
|
className={classNames(
|
|
styles.logicNode,
|
|
theme === 'dark' ? styles.dark : '',
|
|
{
|
|
[styles.selectedNode]: selected,
|
|
},
|
|
)}
|
|
>
|
|
<Handle
|
|
id="c"
|
|
type="source"
|
|
position={Position.Left}
|
|
isConnectable={isConnectable}
|
|
className={styles.handle}
|
|
style={LeftHandleStyle}
|
|
></Handle>
|
|
<Handle
|
|
type="source"
|
|
position={Position.Right}
|
|
isConnectable={isConnectable}
|
|
className={styles.handle}
|
|
style={RightHandleStyle}
|
|
id="b"
|
|
></Handle>
|
|
|
|
<NodeHeader
|
|
id={id}
|
|
name={data.name}
|
|
label={data.label}
|
|
className={styles.nodeHeader}
|
|
></NodeHeader>
|
|
|
|
<div className={styles.nodeText}>
|
|
<LLMLabel value={get(data, 'form.llm_id')}></LLMLabel>
|
|
</div>
|
|
</section>
|
|
);
|
|
}
|