Open in playground copy tools & output (#341)

lunary-ai · May 24, 2024 · fc332dc · fc332dc
1 parent b974325
commit fc332dc
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git 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
@@ -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,