From 4855e878767acfd5ae5ec7ba92ab7848c0f98f78 Mon Sep 17 00:00:00 2001
From: AkaraChen <akarachen@outlook.com>
Date: Tue, 31 Dec 2024 13:46:06 +0800
Subject: [PATCH] fix: editor

---
 .../nodes/_base/components/agent-strategy.tsx         | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx b/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx
index 209f3108c9..1e9b8ff827 100644
--- a/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx
+++ b/web/app/components/workflow/nodes/_base/components/agent-strategy.tsx
@@ -16,6 +16,7 @@ import type { ComponentProps } from 'react'
 import { useDefaultModel, useLanguage } from '@/app/components/header/account-setting/model-provider-page/hooks'
 import Editor from './prompt/editor'
 import { strategyParamToCredientialForm } from '../../agent/panel'
+import { useWorkflowStore } from '../../../store'
 
 export type Strategy = {
   agent_strategy_provider_name: string
@@ -52,6 +53,10 @@ export const AgentStrategy = (props: AgentStrategyProps) => {
   const { t } = useTranslation()
   const language = useLanguage()
   const defaultModel = useDefaultModel(ModelTypeEnum.textGeneration)
+  const workflowStore = useWorkflowStore()
+  const {
+    setControlPromptEditorRerenderKey,
+  } = workflowStore.getState()
   const override: ComponentProps<typeof Form<CustomField>>['override'] = [
     [FormTypeEnum.textNumber],
     (schema, props) => {
@@ -130,9 +135,14 @@ export const AgentStrategy = (props: AgentStrategyProps) => {
         const onChange = (value: string) => {
           props.onChange({ ...props.value, [schema.variable]: value })
         }
+        const handleGenerated = (value: string) => {
+          onChange(value)
+          setControlPromptEditorRerenderKey(Math.random())
+        }
         return <Editor
           value={value}
           onChange={onChange}
+          onGenerated={handleGenerated}
           title={schema.label[language]}
           headerClassName='bg-transparent px-0 text-text-secondary system-sm-semibold-uppercase !text-base'
           containerClassName='bg-transparent'
@@ -151,7 +161,6 @@ export const AgentStrategy = (props: AgentStrategyProps) => {
                 completion_params: {},
               } : undefined
           }
-          onGenerated={onChange}
           placeholderClassName='px-2 py-1'
           inputClassName='px-2 py-1 bg-components-input-bg-normal focus:bg-components-input-bg-active focus:border-components-input-border-active focus:border rounded-lg'
         />