43 lines
1.6 KiB
TypeScript
43 lines
1.6 KiB
TypeScript
|
import { createAgentUIStreamResponse, type UIMessage } from "ai";
|
||
|
|
import { createTeacherAgent, type TeacherAgent } from "../agent/teacher-agent.ts";
|
||
|
|
import { getChat, saveMessages } from "../store/chats.ts";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* One chat turn: run the teacher agent over the client's message list and
|
||
|
|
* stream the UI message stream back. Framework-free (plain Request/Response)
|
||
|
|
* so tests can drive it directly with a scripted mock model — the Next route
|
||
|
|
* handler in app/api/chat is a one-line wrapper.
|
||
|
|
*
|
||
|
|
* The client owns the message list between turns (useChat); the server
|
||
|
|
* persists the canonical state (client messages + agent response) onFinish.
|
||
|
|
*/
|
||
|
|
export async function handleChatRequest(req: Request, agent?: TeacherAgent): Promise<Response> {
|
||
|
|
let body: { chatId?: unknown; messages?: unknown };
|
||
|
|
try {
|
||
|
|
body = await req.json();
|
||
|
|
} catch {
|
||
|
|
return Response.json({ error: "Invalid JSON body" }, { status: 400 });
|
||
|
|
}
|
||
|
|
const chatId = typeof body.chatId === "string" ? body.chatId : "";
|
||
|
|
if (!chatId || !getChat(chatId)) {
|
||
|
|
return Response.json({ error: "Unknown chat" }, { status: 404 });
|
||
|
|
}
|
||
|
|
if (!Array.isArray(body.messages)) {
|
||
|
|
return Response.json({ error: "messages must be an array" }, { status: 400 });
|
||
|
|
}
|
||
|
|
|
||
|
|
return createAgentUIStreamResponse({
|
||
|
|
agent: agent ?? createTeacherAgent(),
|
||
|
|
uiMessages: body.messages as UIMessage[],
|
||
|
|
onFinish: ({ messages, isAborted }) => {
|
||
|
|
// Persist even when aborted: the partial transcript (including any
|
||
|
|
// pending question part) is what makes resume work.
|
||
|
|
try {
|
||
|
|
saveMessages(chatId, messages as UIMessage[]);
|
||
|
|
} catch (err) {
|
||
|
|
console.error(`[chat ${chatId}] persist failed:`, err, { isAborted });
|
||
|
|
}
|
||
|
|
},
|
||
|
|
});
|
||
|
|
}
|