From 405e8746c9a0e9ba42590cc7e4c3145fda6228ca Mon Sep 17 00:00:00 2001
From: Alex Grozav <alex@grozav.com>
Date: Wed, 26 Jun 2024 12:17:32 +0300
Subject: [PATCH] feat(editor): Allow two-way connection for AI Nodes in new
 canvas (no-changelog) (#9864)

---
 .../src/components/canvas/elements/handles/HandleRenderer.vue | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/editor-ui/src/components/canvas/elements/handles/HandleRenderer.vue b/packages/editor-ui/src/components/canvas/elements/handles/HandleRenderer.vue
index 0f8968a31..767c52a4a 100644
--- a/packages/editor-ui/src/components/canvas/elements/handles/HandleRenderer.vue
+++ b/packages/editor-ui/src/components/canvas/elements/handles/HandleRenderer.vue
@@ -23,11 +23,11 @@ const $style = useCssModule();
 const handleType = computed(() => (props.mode === 'input' ? 'target' : 'source'));
 
 const isConnectableStart = computed(() => {
-	return props.mode === 'output';
+	return props.mode === 'output' || props.type !== NodeConnectionType.Main;
 });
 
 const isConnectableEnd = computed(() => {
-	return props.mode === 'input';
+	return props.mode === 'input' || props.type !== NodeConnectionType.Main;
 });
 
 const Render = (renderProps: { label?: string }) => {