From 5442392003252130da5d0db320ffc2e08b574bc6 Mon Sep 17 00:00:00 2001 From: Weixuan Fu Date: Sat, 29 Jun 2024 15:06:47 +0800 Subject: [PATCH] fix api prompt not added in create version --- ui/src/components/CreateVersionDialog.tsx | 8 +++++++- ui/src/components/copySharelink.tsx | 8 +++++++- 2 files changed, 14 insertions(+), 2 deletions(-) diff --git a/ui/src/components/CreateVersionDialog.tsx b/ui/src/components/CreateVersionDialog.tsx index 47fbf01..ff9d295 100644 --- a/ui/src/components/CreateVersionDialog.tsx +++ b/ui/src/components/CreateVersionDialog.tsx @@ -45,6 +45,12 @@ export default function CreateVersionDialog({ onClose }: Props) { throw new Error("No workflow selected"); } setSubmitting(true); + const prompt = await app.graphToPrompt(); + const graph = app.graph.serialize(); + if (!graph.extra) { + graph.extra = {}; + } + graph.extra.apiPrompt = prompt.output ?? null; const input: ShareWorkflowData = { workflow: { name: workflowsTable?.curWorkflow?.name, @@ -52,7 +58,7 @@ export default function CreateVersionDialog({ onClose }: Props) { }, version: { name: trimName, - json: JSON.stringify(app.graph.serialize()), + json: JSON.stringify(graph), }, nodeDefs: getNodeDefs(), privacy: EWorkflowPrivacy.PRIVATE, diff --git a/ui/src/components/copySharelink.tsx b/ui/src/components/copySharelink.tsx index 6a3ecbb..2878802 100644 --- a/ui/src/components/copySharelink.tsx +++ b/ui/src/components/copySharelink.tsx @@ -20,6 +20,12 @@ export async function copySharelink( link, }; } + const prompt = await app.graphToPrompt(); + const graph = app.graph.serialize(); + if (!graph.extra) { + graph.extra = {}; + } + graph.extra.apiPrompt = prompt.output ?? null; const input: ShareWorkflowData = { workflow: { name: curWorkflow?.name ?? "Untitled", @@ -27,7 +33,7 @@ export async function copySharelink( }, version: { name: "v1", - json: JSON.stringify(app.graph.serialize()), + json: JSON.stringify(graph), }, nodeDefs: getNodeDefs(), privacy: EWorkflowPrivacy.PRIVATE,