Add Label Assistant Reply Task

website/src/hooks/tasks/labeling/useLabelAssistantReply.ts

@@ -0,0 +1,22 @@
+import { TaskResponse } from "../useGenericTaskAPI";
+import { LabelingTaskType, useLabelingTask } from "./useLabelingTask";
+
+export interface LabelAssistantReplyTask {
+  id: string;
+  type: "label_prompter_reply";
+  message_id: string;
+  valid_labels: string[];
+  reply: string;
+  conversation: {
+    messages: Array<{
+      text: string;
+      is_assistant: boolean;
+      message_id: string;
+    }>;
+  };
+}
+
+export type LabelAssistantReplyTaskResponse = TaskResponse<LabelAssistantReplyTask>;
+
+export const useLabelAssistantReplyTask = () =>
+  useLabelingTask<LabelAssistantReplyTask>(LabelingTaskType.label_assistant_reply);

website/src/hooks/tasks/labeling/useLabelInitialPrompt.tsx

@@ -0,0 +1,15 @@
+import { TaskResponse } from "../useGenericTaskAPI";
+import { LabelingTaskType, useLabelingTask } from "./useLabelingTask";
+
+export interface LabelInitialPromptTask {
+  id: string;
+  type: "label_initial_prompt";
+  message_id: string;
+  valid_labels: string[];
+  prompt: string;
+}
+
+export type LabelInitialPromptTaskResponse = TaskResponse<LabelInitialPromptTask>;
+
+export const useLabelInitialPromptTask = () =>
+  useLabelingTask<LabelInitialPromptTask>(LabelingTaskType.label_initial_prompt);

website/src/hooks/tasks/labeling/useLabelPrompterReply.ts

@@ -0,0 +1,22 @@
+import { TaskResponse } from "../useGenericTaskAPI";
+import { LabelingTaskType, useLabelingTask } from "./useLabelingTask";
+
+export interface LabelPrompterReplyTask {
+  id: string;
+  type: "label_prompter_reply";
+  message_id: string;
+  valid_labels: string[];
+  reply: string;
+  conversation: {
+    messages: Array<{
+      text: string;
+      is_assistant: boolean;
+      message_id: string;
+    }>;
+  };
+}
+
+export type LabelPrompterReplyTaskResponse = TaskResponse<LabelPrompterReplyTask>;
+
+export const useLabelPrompterReplyTask = () =>
+  useLabelingTask<LabelPrompterReplyTask>(LabelingTaskType.label_prompter_reply);

website/src/hooks/tasks/labeling/useLabelingTask.ts

@@ -1,17 +1,13 @@
-import { TaskResponse, useGenericTaskAPI } from "./useGenericTaskAPI";
+import { useGenericTaskAPI } from "../useGenericTaskAPI";
 
-export interface LabelInitialPromptTask {
-  id: string;
-  type: "label_initial_prompt";
-  message_id: string;
-  valid_labels: string[];
-  prompt: string;
+export const enum LabelingTaskType {
+  label_initial_prompt = "label_initial_prompt",
+  label_prompter_reply = "label_prompter_reply",
+  label_assistant_reply = "label_assistant_reply",
 }
 
-export type LabelInitialPromptTaskResponse = TaskResponse<LabelInitialPromptTask>;
-
-export const useLabelInitialPromptTask = () => {
-  const { tasks, isLoading, trigger, reset, error } = useGenericTaskAPI<LabelInitialPromptTask>("label_initial_prompt");
+export const useLabelingTask = <TaskType>(endpoint: LabelingTaskType) => {
+  const { tasks, isLoading, trigger, reset, error } = useGenericTaskAPI<TaskType>(endpoint);
 
   const submit = (id: string, message_id: string, text: string, validLabels: string[], labelWeights: number[]) => {
     console.assert(validLabels.length === labelWeights.length);

website/src/hooks/tasks/useLabelPrompterReply.ts

@@ -1,31 +0,0 @@
-import { TaskResponse, useGenericTaskAPI } from "./useGenericTaskAPI";
-
-export interface LabelPrompterReplyTask {
-  id: string;
-  type: "label_prompter_reply";
-  message_id: string;
-  valid_labels: string[];
-  reply: string;
-  conversation: {
-    messages: Array<{
-      text: string;
-      is_assistant: boolean;
-      message_id: string;
-    }>;
-  };
-}
-
-export type LabelPrompterReplyTaskResponse = TaskResponse<LabelPrompterReplyTask>;
-
-export const useLabelPrompterReplyTask = () => {
-  const { tasks, isLoading, trigger, reset, error } = useGenericTaskAPI<LabelPrompterReplyTask>("label_prompter_reply");
-
-  const submit = (id: string, message_id: string, text: string, validLabels: string[], labelWeights: number[]) => {
-    console.assert(validLabels.length === labelWeights.length);
-    const labels = Object.fromEntries(validLabels.map((label, i) => [label, labelWeights[i]]));
-
-    return trigger({ id, update_type: "text_labels", content: { labels, text, message_id } });
-  };
-
-  return { tasks, isLoading, submit, reset, error };
-};