Open in playground copy tools & output (#341)

lunary-ai · Jun 6, 2024 · a6c2822 · a6c2822
1 parent 82d9915
commit a6c2822
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/packages/frontend/pages/prompts/[[...id]].tsx b/packages/frontend/pages/prompts/[[...id]].tsx
@@ -181,10 +181,17 @@ function Playground() {
         const run = await fetcher.get(`/runs/${clone}?projectId=${project?.id}`)
 
         if (run?.input) {
+          setTemplateVersion({
+            // ...templateVersion,
+            content: run.input,
+            extra: { ...run.params, model: run.name },
+          })
+
+          setTemplate({ mode: "openai" })
+
           setOutput(run.output)
-          setTemplateVersion({ ...templateVersion, content: run.input })
 
-          setTemplate({ mode: "openai", extra: run.params })
+          setOutputTokens(run.tokens?.completion)
         }
 
         setLoading(false)
@@ -375,11 +382,13 @@ function Playground() {
     setStreaming(false)
   }
 
-  // reset output when the template or template version changes
+  // reset output when the template or template version changes, but not if cloned
   useEffect(() => {
-    setOutput(null)
-    setError(null)
-    setOutputTokens(0)
+    if (!router.query.clone) {
+      setOutput(null)
+      setError(null)
+      setOutputTokens(0)
+    }
   }, [
     template?.id,
     templateVersion?.id,