chore: import template configuration as well (#203)

latitude-dev · Sep 17, 2024 · 1cc824b · 1cc824b
1 parent 0ccb20f
commit 1cc824b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 6 deletions.
diff --git a/apps/web/src/app/(private)/evaluations/_components/CreateEvaluationModal/index.tsx b/apps/web/src/app/(private)/evaluations/_components/CreateEvaluationModal/index.tsx
@@ -20,6 +20,7 @@ export type CreateEvaluationData = {
   title: string
   description: string
   prompt: string
+  configuration: EvaluationResultConfiguration
 }
 
 export default function CreateEvaluationModal({
@@ -37,7 +38,7 @@ export default function CreateEvaluationModal({
     handleTypeChange,
     handleRangeFromChange,
     handleRangeToChange,
-  } = useEvaluationConfiguration()
+  } = useEvaluationConfiguration(initialData?.configuration)
 
   const router = useRouter()
 
@@ -149,11 +150,15 @@ export default function CreateEvaluationModal({
   )
 }
 
-export function useEvaluationConfiguration() {
+export function useEvaluationConfiguration(
+  init?: EvaluationResultConfiguration,
+) {
   const [configuration, setConfiguration] =
-    useState<EvaluationResultConfiguration>({
-      type: EvaluationResultableType.Text,
-    })
+    useState<EvaluationResultConfiguration>(
+      init || {
+        type: EvaluationResultableType.Text,
+      },
+    )
 
   const handleTypeChange = useCallback((value: EvaluationResultableType) => {
     if (value === EvaluationResultableType.Number) {
@@ -228,6 +233,12 @@ export function useEvaluationConfiguration() {
     [],
   )
 
+  useEffect(() => {
+    if (!init) return
+
+    setConfiguration(init)
+  }, [init])
+
   return {
     configuration,
     handleTypeChange,

diff --git a/apps/web/src/app/(private)/evaluations/_components/Evaluations.tsx b/apps/web/src/app/(private)/evaluations/_components/Evaluations.tsx
@@ -2,7 +2,10 @@
 
 import { useState } from 'react'
 
-import { EvaluationTemplateWithCategory } from '@latitude-data/core/browser'
+import {
+  EvaluationResultableType,
+  EvaluationTemplateWithCategory,
+} from '@latitude-data/core/browser'
 import { Container } from '@latitude-data/web-ui'
 
 import ActiveEvaluations from './ActiveEvaluations'
@@ -27,6 +30,7 @@ export default function Evaluations({
             title: 'New Evaluation',
             description: '',
             prompt: '',
+            configuration: { type: EvaluationResultableType.Text },
           })
         }
       />
@@ -38,6 +42,7 @@ export default function Evaluations({
               title: template.name,
               description: template.description,
               prompt: template.prompt,
+              configuration: template.configuration,
             })
           }
         />