feat(builder) Add save Agent functionality (#7361)

Add save functionality
pull/7350/head^2
Swifty 2024-07-10 11:01:12 +02:00 committed by GitHub
parent e10c4ee4cd
commit f94e81f48b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 73 additions and 57 deletions

View File

@ -261,8 +261,7 @@ const FlowEditor: React.FC<{ flowID?: string; className?: string }> = ({
return inputData; return inputData;
}; };
const saveAgent = async () => {
const runAgent = async () => {
try { try {
console.log("All nodes before formatting:", nodes); console.log("All nodes before formatting:", nodes);
const blockIdToNodeIdMap = {}; const blockIdToNodeIdMap = {};
@ -326,6 +325,20 @@ const FlowEditor: React.FC<{ flowID?: string; className?: string }> = ({
setNodes(updatedNodes); setNodes(updatedNodes);
return newAgentId;
} catch (error) {
console.error('Error running agent:', error);
}
};
const runAgent = async () => {
try {
const newAgentId = await saveAgent();
if (!newAgentId) {
console.error('Error saving agent');
return;
}
const executeData = await api.executeFlow(newAgentId); const executeData = await api.executeFlow(newAgentId);
const runId = executeData.id; const runId = executeData.id;
@ -406,7 +419,10 @@ const updateNodesWithExecutionData = (executionData: any[]) => {
value={agentDescription} value={agentDescription}
onChange={(e) => setAgentDescription(e.target.value)} onChange={(e) => setAgentDescription(e.target.value)}
/> />
<Button onClick={runAgent}>Run Agent</Button> <div style={{ display: 'flex', flexDirection: 'column', gap: '10px' }}> {/* Added gap for spacing */}
<Button onClick={saveAgent}>Save Agent</Button>
<Button onClick={runAgent}>Save & Run Agent</Button>
</div>
</div> </div>
</ReactFlow> </ReactFlow>
</div> </div>