diff --git a/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx b/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx
index 52e9975e..b84be017 100644
--- a/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx
@@ -122,14 +122,14 @@ const AIFlowChart = () => {
source: 'apiService',
target: model.id,
label: `apis(${model.api_count})`,
- style: { stroke: '#ddd', cursor: 'pointer' }
+ markerEnd: { type: 'arrow' },
+ data: { id: model.id }
})),
...modelData.map((model) => ({
id: `${model.id}-keys-edge`,
source: model.id,
target: `${model.id}-keys`,
- animated: true,
- style: { stroke: '#ddd' }
+ animated: true
}))
]
@@ -221,10 +221,10 @@ const AIFlowChart = () => {
panOnScroll={true}
panOnScrollMode={PanOnScrollMode.Vertical}
defaultEdgeOptions={{
- type: 'custom'
+ type: 'custom',
+ style: { strokeWidth: 2 }
}}
connectionMode={ConnectionMode.Loose}
- fitView
/>
)
diff --git a/frontend/packages/core/src/pages/aiSetting/components/CustomEdge.tsx b/frontend/packages/core/src/pages/aiSetting/components/CustomEdge.tsx
index 7df93001..2fcf5824 100644
--- a/frontend/packages/core/src/pages/aiSetting/components/CustomEdge.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/components/CustomEdge.tsx
@@ -1,4 +1,4 @@
-import { BaseEdge, EdgeLabelRenderer, EdgeProps, getBezierPath } from '@xyflow/react'
+import { BaseEdge, EdgeLabelRenderer, EdgeProps, getSmoothStepPath } from '@xyflow/react'
export default function CustomEdge({
id,
@@ -13,36 +13,41 @@ export default function CustomEdge({
label,
data
}: EdgeProps) {
- const [edgePath, labelX, labelY] = getBezierPath({
+ const [edgePath, labelX, labelY] = getSmoothStepPath({
sourceX,
sourceY,
sourcePosition,
targetX,
targetY,
- targetPosition
+ targetPosition,
+ borderRadius: 16
})
+ const modelId = data?.id
+
return (
<>
-
+
{label && (
-
{label}
-
+
)}
>