mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-28 16:20:34 +08:00
70 lines
2.2 KiB
TypeScript
70 lines
2.2 KiB
TypeScript
import { Prisma } from "@prisma/client";
|
|
import { withoutRole } from "src/lib/auth";
|
|
import { oasstApiClient } from "src/lib/oasst_api_client";
|
|
import prisma from "src/lib/prismadb";
|
|
import { getBackendUserCore } from "src/lib/users";
|
|
import type { BackendUserCore } from "src/types/Users";
|
|
|
|
/**
|
|
* Stores the task interaction with the Task Backend and then returns the next task generated.
|
|
*
|
|
* This implicity does a few things:
|
|
* 1) Records the users answer in our local database.
|
|
* 2) Accepts the task.
|
|
* 3) Sends the users answer to the Task Backend.
|
|
* 4) Records the new task in our local database.
|
|
* 5) Returns the newly created task to the client.
|
|
*/
|
|
const handler = withoutRole("banned", async (req, res, token) => {
|
|
// Parse out the local task ID and the interaction contents.
|
|
const { id: frontendId, content, update_type } = req.body;
|
|
|
|
// Record that the user has done meaningful work and is no longer new.
|
|
await prisma.user.update({ where: { id: token.sub }, data: { isNew: false } });
|
|
|
|
// Accept the task so that we can complete it, this will probably go away soon.
|
|
const registeredTask = await prisma.registeredTask.findUniqueOrThrow({ where: { id: frontendId } });
|
|
const task = registeredTask.task as Prisma.JsonObject;
|
|
const taskId = task.id as string;
|
|
await oasstApiClient.ackTask(taskId, registeredTask.id);
|
|
|
|
// Log the interaction locally to create our user_post_id needed by the Task
|
|
// Backend.
|
|
const interaction = await prisma.taskInteraction.create({
|
|
data: {
|
|
content,
|
|
task: {
|
|
connect: {
|
|
id: frontendId,
|
|
},
|
|
},
|
|
},
|
|
});
|
|
|
|
const user = await getBackendUserCore(token.sub);
|
|
let newTask;
|
|
try {
|
|
newTask = await oasstApiClient.interactTask(update_type, taskId, frontendId, interaction.id, content, user);
|
|
} catch (err) {
|
|
console.error(JSON.stringify(err));
|
|
return res.status(500).json(err);
|
|
}
|
|
|
|
// Stores the new task with our database.
|
|
const newRegisteredTask = await prisma.registeredTask.create({
|
|
data: {
|
|
task: newTask,
|
|
user: {
|
|
connect: {
|
|
id: token.sub,
|
|
},
|
|
},
|
|
},
|
|
});
|
|
|
|
// Send the next task in the sequence to the client.
|
|
res.status(200).json(newRegisteredTask);
|
|
});
|
|
|
|
export default handler;
|