Use hooks for "evaluate" tasks

This commit is contained in:
AbdBarho
2023-01-08 19:50:49 +01:00
parent f33787b4cc
commit d0042cebb3
5 changed files with 52 additions and 102 deletions
@@ -0,0 +1,9 @@
import { useGenericTaskAPI } from "../useGenericTaskAPI";
interface RankInitialPromptsTask {
id: string;
type: "rank_initial_prompts";
prompts: string[];
}
export const useRankInitialPromptsTask = () => useGenericTaskAPI<RankInitialPromptsTask>("rank_initial_prompts");
@@ -0,0 +1,25 @@
import { useGenericTaskAPI } from "../useGenericTaskAPI";
interface BaseRankRepliesTask {
id: string;
replies: string[];
conversation: {
messages: Array<{
text: string;
is_assistant: boolean;
message_id: string;
}>;
};
}
interface RankAssistantRepliesTask extends BaseRankRepliesTask {
type: "rank_assistant_replies";
}
interface RankPrompterRepliesTask extends BaseRankRepliesTask {
type: "rank_prompter_replies";
}
export const useRankAssistantRepliesTask = () => useGenericTaskAPI<RankAssistantRepliesTask>("rank_assistant_replies");
export const useRankPrompterRepliesTask = () => useGenericTaskAPI<RankPrompterRepliesTask>("rank_prompter_replies");