From c80f89f545bed87608660d2de98d96515ad33e40 Mon Sep 17 00:00:00 2001 From: Kostia Date: Wed, 4 Jan 2023 01:59:15 +0200 Subject: [PATCH] Added context messages to fix #339 --- website/src/components/ContextMessages.tsx | 19 +++++++++++++++++++ .../pages/evaluate/rank_assistant_replies.tsx | 6 +++++- .../src/pages/evaluate/rank_user_replies.tsx | 4 ++++ 3 files changed, 28 insertions(+), 1 deletion(-) create mode 100644 website/src/components/ContextMessages.tsx diff --git a/website/src/components/ContextMessages.tsx b/website/src/components/ContextMessages.tsx new file mode 100644 index 00000000..2b491ac7 --- /dev/null +++ b/website/src/components/ContextMessages.tsx @@ -0,0 +1,19 @@ +import { Box } from "@chakra-ui/react"; +import { Message } from "./Messages"; + +export const ContextMessages = ({ messages } : { messages: Message[]}) => { + console.log(messages) + return ( + + {messages.map((message, i) => { + return ( + + {message.is_assistant ? "Assistant: " : "User: "} + {message.text} + + ); + })} + + ); +}; + diff --git a/website/src/pages/evaluate/rank_assistant_replies.tsx b/website/src/pages/evaluate/rank_assistant_replies.tsx index e18d228a..1c5208d3 100644 --- a/website/src/pages/evaluate/rank_assistant_replies.tsx +++ b/website/src/pages/evaluate/rank_assistant_replies.tsx @@ -1,7 +1,9 @@ import { useColorMode } from "@chakra-ui/react"; import Head from "next/head"; import { useState } from "react"; +import { ContextMessages } from "src/components/ContextMessages"; import { LoadingScreen } from "src/components/Loading/LoadingScreen"; +import { Message } from "src/components/Messages"; import { Sortable } from "src/components/Sortable/Sortable"; import { SurveyCard } from "src/components/Survey/SurveyCard"; import { TaskControls } from "src/components/Survey/TaskControls"; @@ -64,7 +66,8 @@ const RankAssistantReplies = () => { } const replies = tasks[0].task.replies as string[]; - + const messages = tasks[0].task.conversation.messages as Message[]; + return ( <> @@ -77,6 +80,7 @@ const RankAssistantReplies = () => {

Given the following replies, sort them from best to worst, best being first, worst being last.

+ diff --git a/website/src/pages/evaluate/rank_user_replies.tsx b/website/src/pages/evaluate/rank_user_replies.tsx index ea57f9a0..7c603c3a 100644 --- a/website/src/pages/evaluate/rank_user_replies.tsx +++ b/website/src/pages/evaluate/rank_user_replies.tsx @@ -9,6 +9,8 @@ import fetcher from "src/lib/fetcher"; import poster from "src/lib/poster"; import useSWRImmutable from "swr/immutable"; import useSWRMutation from "swr/mutation"; +import {Message} from "src/components/Messages" +import { ContextMessages } from "src/components/ContextMessages"; const RankUserReplies = () => { const [tasks, setTasks] = useState([]); @@ -63,6 +65,7 @@ const RankUserReplies = () => { ); } const replies = tasks[0].task.replies as string[]; + const messages = tasks[0].task.conversation.messages as Message[]; return ( <> @@ -76,6 +79,7 @@ const RankUserReplies = () => {

Given the following replies, sort them from best to worst, best being first, worst being last.

+