Просмотр исходного кода

fix(fixed-layout-demo): agent tool add from flow-operation-service (#754)

xiamidaxia 4 месяцев назад
Родитель
Сommit
341822eb56
1 измененных файлов с 1 добавлено и 5 удалено
  1. 1 5
      apps/demo-fixed-layout/src/components/agent-adder/index.tsx

+ 1 - 5
apps/demo-fixed-layout/src/components/agent-adder/index.tsx

@@ -6,8 +6,6 @@
 import {
   type FlowNodeEntity,
   FlowNodeRenderData,
-  FlowDocument,
-  useService,
   useClientContext,
 } from '@flowgram.ai/fixed-layout-editor';
 import { IconPlus } from '@douyinfe/semi-icons';
@@ -22,12 +20,10 @@ export function AgentAdder(props: PropsType) {
   const { node } = props;
 
   const nodeData = node.firstChild?.getData<FlowNodeRenderData>(FlowNodeRenderData);
-  const document = useService(FlowDocument) as FlowDocument;
   const ctx = useClientContext();
 
   async function addPort() {
-    document.addNode({
-      ...ToolNodeRegistry.onAdd!(ctx, node),
+    ctx.operation.addNode(ToolNodeRegistry.onAdd!(ctx, node), {
       parent: node,
     });
   }