c4b58ed195
### What problem does this PR solve? Feat: Filter the query variable drop-down box options by type #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
77 lines
2.3 KiB
TypeScript
77 lines
2.3 KiB
TypeScript
import { IAgentNode } from '@/interfaces/database/flow';
|
|
import { Handle, NodeProps, Position } from '@xyflow/react';
|
|
import { memo, useMemo } from 'react';
|
|
import { NodeHandleId } from '../../constant';
|
|
import useGraphStore from '../../store';
|
|
import { isBottomSubAgent } from '../../utils';
|
|
import { CommonHandle } from './handle';
|
|
import { LeftHandleStyle, RightHandleStyle } from './handle-icon';
|
|
import styles from './index.less';
|
|
import NodeHeader from './node-header';
|
|
import { NodeWrapper } from './node-wrapper';
|
|
import { ToolBar } from './toolbar';
|
|
|
|
function InnerAgentNode({
|
|
id,
|
|
data,
|
|
isConnectable = true,
|
|
selected,
|
|
}: NodeProps<IAgentNode>) {
|
|
const edges = useGraphStore((state) => state.edges);
|
|
|
|
const isHeadAgent = useMemo(() => {
|
|
return !isBottomSubAgent(edges, id);
|
|
}, [edges, id]);
|
|
|
|
return (
|
|
<ToolBar selected={selected} id={id} label={data.label}>
|
|
<NodeWrapper>
|
|
{isHeadAgent && (
|
|
<>
|
|
<CommonHandle
|
|
type="target"
|
|
position={Position.Left}
|
|
isConnectable={isConnectable}
|
|
style={LeftHandleStyle}
|
|
nodeId={id}
|
|
id={NodeHandleId.End}
|
|
></CommonHandle>
|
|
<CommonHandle
|
|
type="source"
|
|
position={Position.Right}
|
|
isConnectable={isConnectable}
|
|
className={styles.handle}
|
|
style={RightHandleStyle}
|
|
nodeId={id}
|
|
id={NodeHandleId.Start}
|
|
isConnectableEnd={false}
|
|
></CommonHandle>
|
|
</>
|
|
)}
|
|
<Handle
|
|
type="target"
|
|
position={Position.Top}
|
|
isConnectable={false}
|
|
id={NodeHandleId.AgentTop}
|
|
></Handle>
|
|
<Handle
|
|
type="source"
|
|
position={Position.Bottom}
|
|
isConnectable={false}
|
|
id={NodeHandleId.AgentBottom}
|
|
style={{ left: 180 }}
|
|
></Handle>
|
|
<Handle
|
|
type="source"
|
|
position={Position.Bottom}
|
|
isConnectable={false}
|
|
id={NodeHandleId.Tool}
|
|
style={{ left: 20 }}
|
|
></Handle>
|
|
<NodeHeader id={id} name={data.name} label={data.label}></NodeHeader>
|
|
</NodeWrapper>
|
|
</ToolBar>
|
|
);
|
|
}
|
|
|
|
export const AgentNode = memo(InnerAgentNode);
|