From fc332dcdab47bf3415b06a68d6d7fae6b5f862f4 Mon Sep 17 00:00:00 2001
From: Vincelwt <vincelwt@users.noreply.github.com>
Date: Fri, 24 May 2024 17:28:47 +0100
Subject: [PATCH] Open in playground copy tools & output (#341)

---
 .../components/blocks/RunInputOutput.tsx      |  1 -
 packages/frontend/pages/prompts/[[...id]].tsx | 21 +++++++++++++------
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/packages/frontend/components/blocks/RunInputOutput.tsx b/packages/frontend/components/blocks/RunInputOutput.tsx
index 5b1aaca3..e1e4198c 100644
--- a/packages/frontend/components/blocks/RunInputOutput.tsx
+++ b/packages/frontend/components/blocks/RunInputOutput.tsx
@@ -162,7 +162,6 @@ export default function RunInputOutput({
     ? true
     : org?.plan === "team" || org?.plan === "custom"
 
-  console.log(run)
   return (
     <ErrorBoundary>
       <Stack>
diff --git a/packages/frontend/pages/prompts/[[...id]].tsx b/packages/frontend/pages/prompts/[[...id]].tsx
index 3c991cb7..f50c5fe9 100644
--- 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,