Merge branch 'main' into issue-1143/-enhance-error-message-clarify-in…

…-playground
Agenta-AI · Jan 3, 2024 · 3bbc6b9 · 3bbc6b9
2 parents 0fd1ffc + e329fcf
commit 3bbc6b9
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/agenta-web/src/components/Playground/AddToTestSetDrawer/AddToTestSetDrawer.tsx b/agenta-web/src/components/Playground/AddToTestSetDrawer/AddToTestSetDrawer.tsx
@@ -14,6 +14,7 @@ import {
     Space,
     Switch,
     Typography,
+    Tooltip,
     message,
 } from "antd"
 import {useRouter} from "next/router"
@@ -247,7 +248,12 @@ const AddToTestSetDrawer: React.FC<Props> = ({params, isChatVariant, ...props})
                 <div className={classes.footer}>
                     {isChatVariant && (
                         <Space align="center">
-                            <Typography.Text>Turn by Turn:</Typography.Text>
+                            <Tooltip
+                                placement="topLeft"
+                                title="Add each exchange in the conversation as an individual data point"
+                            >
+                                <Typography.Text>Add all conversation turns:</Typography.Text>
+                            </Tooltip>
                             <Switch
                                 checked={Array.isArray(turnModeChat)}
                                 onChange={(checked) => {

diff --git a/agenta-web/src/components/Playground/Views/TestView.tsx b/agenta-web/src/components/Playground/Views/TestView.tsx
@@ -133,9 +133,9 @@ const BoxComponent: React.FC<BoxComponentProps> = ({
         })
         params.correct_answer = result
         if (isChatVariant) {
-            const messages = testData.chat.filter((item: ChatMessage) => !!item.content)
-            params.chat = messages.slice(0, -1)
-            params.correct_answer = messages.at(-1)
+            const messages = testData?.chat?.filter((item: ChatMessage) => !!item.content)
+            params.chat = messages?.slice(0, -1)
+            params.correct_answer = messages?.at(-1)
         }
 
         onAddToTestset(params)

diff --git a/agenta-web/src/lib/helpers/utils.ts b/agenta-web/src/lib/helpers/utils.ts
@@ -34,7 +34,11 @@ export const getAllLlmProviderKeysAsEnvVariable = () => {
 }
 
 export const renameVariables = (name: string) => {
-    return name.charAt(0).toUpperCase() + name.slice(1).replace(/_/g, " ")
+    if (name === "inputs") {
+        return "Prompt Variables"
+    } else {
+        return name.charAt(0).toUpperCase() + name.slice(1).replace(/_/g, " ")
+    }
 }
 
 export const renameVariablesCapitalizeAll = (name: string) => {