pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: limit there to be only one line between two nodes #918 (#1310)

Browse Source 
### What problem does this PR solve?

feat: limit there to be only one line between two nodes #918

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2024-06-28 12:01:06 +08:00
committed by GitHub
parent 0acf4194ca
commit 6a7c2112f7
3 changed files with 30 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
web/src/pages/flow/hooks.ts
View File

@ -13,7 +13,7 @@ import React, {
useEffect,
useState,
} from 'react';
import { Node, Position, ReactFlowInstance } from 'reactflow';
import { Connection, Node, Position, ReactFlowInstance } from 'reactflow';
// import { shallow } from 'zustand/shallow';
import { variableEnabledFieldMap } from '@/constants/chat';
import {
@ -25,9 +25,9 @@ import { useDebounceEffect } from 'ahooks';
import { FormInstance } from 'antd';
import { humanId } from 'human-id';
import { useParams } from 'umi';
import { NodeMap, Operator } from './constant';
import { NodeMap, Operator, RestrictedUpstreamMap } from './constant';
import useGraphStore, { RFState } from './store';
import { buildDslComponentsByGraph } from './utils';
import { buildDslComponentsByGraph, getOperatorTypeFromId } from './utils';
const selector = (state: RFState) => ({
nodes: state.nodes,
@ -247,3 +247,26 @@ export const useSetLlmSetting = (form?: FormInstance) => {
form?.setFieldsValue({ ...switchBoxValues, ...otherValues });
}, [form, initialLlmSetting]);
};
export const useValidateConnection = () => {
const edges = useGraphStore((state) => state.edges);
// restricted lines cannot be connected successfully.
const isValidConnection = useCallback(
(connection: Connection) => {
// limit there to be only one line between two nodes
const hasLine = edges.some(
(x) => x.source === connection.source && x.target === connection.target,
);
const ret =
!hasLine &&
RestrictedUpstreamMap[
getOperatorTypeFromId(connection.source) as Operator
]?.every((x) => x !== getOperatorTypeFromId(connection.target));
return ret;
},
[edges],
);
return isValidConnection;
};