@studious-lms/server 1.15.0 → 1.16.1

package/src/pipelines/labChatPrompt.ts

@@ -1,23 +1,10 @@
 import type { ModelMessage } from "ai";
+import {
+  buildChatModelMessages,
+  type ChatHistoryMessage,
+} from "../lib/chatAttachmentPrompt.js";
 
-export type ConversationMessage = {
-  id?: string;
-  content: string;
-  senderId: string;
-  createdAt?: Date | string | number;
-  sender?: {
-    username?: string | null;
-    profile?: {
-      displayName?: string | null;
-    } | null;
-  } | null;
-  attachments?: {
-    id: string;
-    name: string;
-    type: string;
-    url?: string;
-  }[];
-};
+export type ConversationMessage = ChatHistoryMessage;
 
 const LAB_CHAT_BASE_INSTRUCTIONS = [
   "IMPORTANT INSTRUCTIONS:",
@@ -47,20 +34,27 @@ const LAB_CHAT_REFERENCE_RULES = [
 
 const LAB_CHAT_DOC_RULES = [
   "WHEN CREATING FILES (generate_document tool):",
-  "- Call generate_document with title, content (markdown/plain), and format.",
+  "- Call generate_document with title, content, and format.",
   "- You can generate MANY file types — not just PDF. Always pass format explicitly.",
   "- Match format to what the teacher asked for:",
-  "  * slides / presentation / deck → pptx",
-  "  * spreadsheet / grade table / data → xlsx or csv",
+  "  * slides / presentation / deck → pptx (separate slides with --- or Slide N: headers in content)",
+  "  * spreadsheet / grade table / data → xlsx or csv (include markdown tables or row data in content)",
   "  * editable handout / letter → docx",
   "  * printable worksheet / reading → pdf",
-  "  * diagram / chart image → png",
+  "  * diagram / chart / flowchart / figure / graph / visual → png (describe the visual in detail: labels, arrows, shapes, data — NOT pdf)",
   "  * notes / readme → md or txt",
+  "  * source code → py/js/ts only when the teacher explicitly wants code files",
   "- Supported: pdf, docx, xlsx, pptx, csv, md, html, txt, json, xml, tex, png, jpeg, gif, zip, py, js, ts, css, sh.",
-  "- Use markdown structure in content: # headings, paragraphs, - bullets, 1. numbered lists.",
-  "- Write complete classroom-ready content — no placeholders.",
+  "- Write complete classroom-ready content — no placeholders like [diagram here].",
   "- One file per tool call; split long packs into logical separate documents if needed.",
-  "- In your spoken reply, name the actual format (e.g. \"PowerPoint\", \"spreadsheet\") — never say you only make PDFs.",
+  "- In your spoken reply, name the actual format (e.g. \"PowerPoint\", \"spreadsheet\", \"diagram image\") — never say you only make PDFs.",
+  "- Never include download links or URLs; the file appears as an attachment on this message automatically after you call generate_document.",
+];
+
+const LAB_CHAT_ATTACHMENT_RULES = [
+  "WHEN THE TEACHER ATTACHES FILES:",
+  "- Read attached PDFs, Word docs, PowerPoints, spreadsheets, images, and text files — content is provided inline.",
+  "- Reference specifics from the attachment when answering or creating materials.",
 ];
 
 const LAB_CHAT_WORKSHEET_RULES = [
@@ -100,33 +94,6 @@ FILE GENERATION: You can attach pdf, docx, xlsx, pptx, csv, images, and more via
 
 const buildPrompt = (sections: string[]): string => sections.filter(Boolean).join("\n\n");
 
-const sortMessagesChronologically = (
-  recentMessages: ConversationMessage[],
-): ConversationMessage[] =>
-  recentMessages
-    .map((message, index) => ({ message, index }))
-    .sort((left, right) => {
-      const leftTime = left.message.createdAt
-        ? new Date(left.message.createdAt).getTime()
-        : null;
-      const rightTime = right.message.createdAt
-        ? new Date(right.message.createdAt).getTime()
-        : null;
-
-      if (leftTime != null && rightTime != null) {
-        return leftTime - rightTime;
-      }
-      if (leftTime != null) {
-        return -1;
-      }
-      if (rightTime != null) {
-        return 1;
-      }
-
-      return left.index - right.index;
-    })
-    .map(({ message }) => message);
-
 const LAB_CHAT_TOOL_RULES = [
   "HOW TO CREATE CONTENT (TOOLS):",
   "- You have tools to search class materials and to PROPOSE content for the teacher to review and accept.",
@@ -147,6 +114,7 @@ export const buildLabChatAgentSystemPrompt = (context: string): string =>
     LAB_CHAT_REFERENCE_RULES.join("\n"),
     LAB_CHAT_TOOL_RULES.join("\n"),
     LAB_CHAT_DOC_RULES.join("\n"),
+    LAB_CHAT_ATTACHMENT_RULES.join("\n"),
     LAB_CHAT_WORKSHEET_RULES.join("\n"),
     LAB_CHAT_SECTION_RULES.join("\n"),
     LAB_CHAT_ASSIGNMENT_RULES.join("\n"),
@@ -158,7 +126,7 @@ export const buildLabChatAgentSystemPrompt = (context: string): string =>
  * Returns a `system` string and AI-SDK `ModelMessage[]` history. Per-request
  * class context (IDs) is folded into the system string.
  */
-export const buildLabChatAgentMessages = ({
+export const buildLabChatAgentMessages = async ({
   context,
   classContext,
   recentMessages,
@@ -168,54 +136,14 @@ export const buildLabChatAgentMessages = ({
   classContext: string;
   recentMessages: ConversationMessage[];
   isAIUser: (senderId: string) => boolean;
-}): { system: string; messages: ModelMessage[] } => {
+}): Promise<{ system: string; messages: ModelMessage[] }> => {
   const system = `${buildLabChatAgentSystemPrompt(context)}\n\nCLASS CONTEXT (use these IDs when proposing assignments, worksheets, or attaching files):\n${classContext}`;
 
-  const messages: ModelMessage[] = sortMessagesChronologically(recentMessages).map(
-    (message) => {
-      const role = isAIUser(message.senderId) ? "assistant" : "user";
-      const senderName =
-        message.sender?.profile?.displayName || message.sender?.username || "Teacher";
-
-      if (role === "assistant") {
-        return { role, content: message.content };
-      }
-
-      // Build user message with attachments
-      const textContent = `${senderName}: ${message.content}`;
-      const attachments = message.attachments ?? [];
-
-      if (attachments.length === 0) {
-        return { role, content: textContent };
-      }
-
-      // Separate images (multimodal) from other files (text description)
-      const imageAttachments = attachments.filter((a) => a.type.startsWith('image/'));
-      const otherAttachments = attachments.filter((a) => !a.type.startsWith('image/'));
-
-      const parts: Array<{ type: string; text?: string; image?: string }> = [];
-
-      // Add text content with non-image file descriptions
-      let fullText = textContent;
-      if (otherAttachments.length > 0) {
-        const fileList = otherAttachments.map((a) => `${a.name} (${a.type})`).join(', ');
-        fullText += `\n[Attached files: ${fileList}]`;
-      }
-      parts.push({ type: 'text', text: fullText });
-
-      // Add image attachments as image parts for multimodal
-      for (const img of imageAttachments) {
-        if (img.url) {
-          parts.push({ type: 'image', image: img.url });
-        } else {
-          // Fallback: mention the image name in text if no URL available
-          parts.push({ type: 'text', text: `[Image attached: ${img.name}]` });
-        }
-      }
-
-      return { role, content: parts } as ModelMessage;
-    },
-  );
+  const messages = await buildChatModelMessages({
+    messages: recentMessages,
+    isAIUser,
+    defaultSenderLabel: 'Teacher',
+  });
 
   return { system, messages };
 };

package/src/routers/worksheet.ts

@@ -30,6 +30,7 @@ import {
   answerWorksheetQuestion,
   cancelGrading,
   regradeQuestion,
+  regradeAll,
   gradeAnswer,
   addCommentToResponse,
 } from "../services/worksheet.js";
@@ -370,6 +371,16 @@ export const worksheetRouter = createTRPCRouter({
       regradeQuestion(input.worksheetResponseId, input.progressId, ctx.user!.id)
     ),
 
+  regradeAll: protectedProcedure
+    .input(
+      z.object({
+        worksheetResponseId: z.string(),
+      })
+    )
+    .mutation(({ ctx, input }) =>
+      regradeAll(input.worksheetResponseId, ctx.user!.id)
+    ),
+
   gradeAnswer: protectedProcedure
     .input(
       z.object({

package/src/services/aiProposalTools.ts

@@ -237,7 +237,7 @@ export function labProposalTools(proposals: LabProposals): ToolSet {
 
     generate_document: {
       description:
-        'Generate a downloadable file and attach it to this message. You are NOT limited to PDF — pick the best format for the request (pptx for slides/presentations, xlsx/csv for data, docx for editable handouts, pdf for print-ready docs, png for diagrams, etc.). Always set format explicitly.',
+        'Generate a downloadable file and attach it to this message. Pick the best format for the request — pptx for slides, xlsx/csv for data, docx for editable handouts, pdf for print-ready docs, png/jpeg for charts/diagrams/figures (NOT pdf for visuals). Use py/js/ts only when the teacher explicitly wants source code. Always set format explicitly.',
       inputSchema: jsonSchema({
         type: 'object',
         properties: {
@@ -245,12 +245,12 @@ export function labProposalTools(proposals: LabProposals): ToolSet {
           content: {
             type: 'string',
             description:
-              'Full body — markdown or plain text with headings, lists, tables, slide notes, etc.',
+              'Full body. Structure by format: pptx — separate slides with "---" or "Slide N:" headers; xlsx/csv — markdown tables or row data; png/jpeg — describe the visual in detail (shapes, labels, arrows, data points, colors); pdf/docx — headings, lists, tables; code formats — complete runnable source.',
           },
           format: {
             type: 'string',
             description:
-              'Output file type. REQUIRED. pptx=slides/presentation, xlsx/csv=spreadsheet/data, docx=Word handout, pdf=printable doc, md/txt/html=text, png/jpeg/gif=image, py/js/ts=code.',
+              'Output file type. REQUIRED. pptx=slides, xlsx/csv=spreadsheet, docx=Word, pdf=printable doc, png/jpeg/gif=chart/diagram/figure image, md/txt/html=text. py/js/ts/css/sh=source code only when requested.',
             enum: [
               'pdf', 'docx', 'doc', 'xlsx', 'pptx', 'csv', 'md', 'html', 'txt',
               'json', 'xml', 'tex', 'png', 'jpeg', 'jpg', 'gif', 'zip', 'tar',

package/src/services/newtonChat.ts

@@ -159,13 +159,6 @@ export async function postToNewtonChat(
   ensureMessageHasContentOrAttachments(content, fileIds);
   const normalizedContent = content.trim();
 
-  if (!normalizedContent) {
-    throw new TRPCError({
-      code: "BAD_REQUEST",
-      message: "Newton chat messages must include text content",
-    });
-  }
-
   const result = await prisma.$transaction(async (tx) => {
     const message = await tx.message.create({
       data: {

package/src/services/openaiDocumentGen.ts

@@ -6,7 +6,7 @@ import OpenAI from 'openai';
 import type { Response } from 'openai/resources/responses/responses.js';
 import { logger } from '../utils/logger.js';
 
-const DOCUMENT_GEN_MODEL = 'gpt-4o-mini';
+const DOCUMENT_GEN_MODEL = 'gpt-5.5';
 
 /**
  * Extensions OpenAI Code Interpreter can read/write in its sandbox.
@@ -67,11 +67,11 @@ const FORMAT_MIME: Record<DocumentFormat, string> = {
 };
 
 const FORMAT_HINTS: Record<DocumentFormat, string> = {
-  pdf: 'fpdf2 or reportlab — structured document with headings and body text',
-  docx: 'python-docx — Word document with headings and paragraphs',
+  pdf: 'fpdf2 or reportlab — polished document with headings, body text, and tables',
+  docx: 'python-docx — Word document with heading styles, paragraphs, and lists',
   doc: 'python-docx saving as .doc if supported, otherwise .docx',
-  xlsx: 'openpyxl or pandas — spreadsheet with labelled columns/rows',
-  pptx: 'python-pptx — slides with title + bullet content per slide',
+  xlsx: 'openpyxl or pandas — spreadsheet with labelled columns, formulas where useful',
+  pptx: 'python-pptx — full slide deck: title slide, section slides, bullets, diagrams on slides',
   csv: 'pandas or csv module — tabular data with headers',
   md: 'write the markdown file directly',
   html: 'write a complete, styled HTML page',
@@ -79,19 +79,58 @@ const FORMAT_HINTS: Record<DocumentFormat, string> = {
   json: 'json.dumps with readable indentation',
   xml: 'well-formed XML document',
   tex: 'LaTeX source that compiles to the requested content',
-  png: 'matplotlib or Pillow — chart, diagram, or visual',
-  jpeg: 'matplotlib or Pillow — save as JPEG',
-  jpg: 'matplotlib or Pillow — save as JPEG',
+  png: 'matplotlib, networkx, or Pillow — render the described chart/diagram as a finished image',
+  jpeg: 'matplotlib or Pillow — high-quality visual saved as JPEG',
+  jpg: 'matplotlib or Pillow — high-quality visual saved as JPEG',
   gif: 'matplotlib or Pillow — save as GIF',
   zip: 'zipfile — bundle multiple related files if needed',
   tar: 'tarfile — archive if multiple files are required',
-  py: 'valid Python source code file',
-  js: 'valid JavaScript source file',
-  ts: 'valid TypeScript source file',
+  py: 'valid, runnable Python source with comments',
+  js: 'valid JavaScript source with comments',
+  ts: 'valid TypeScript source with comments',
   css: 'stylesheet file',
   sh: 'shell script file',
 };
 
+/** Extra quality rules appended for formats that need more than a one-line library hint. */
+const FORMAT_QUALITY_RULES: Partial<Record<DocumentFormat, string[]>> = {
+  png: [
+    'Interpret the source content as a visual spec — draw it, do not dump text into the image.',
+    'Charts: labelled axes, legend, title, readable font sizes, 150+ DPI.',
+    'Diagrams/flowcharts: use matplotlib patches, networkx, or Pillow drawing — boxes, arrows, labels.',
+    'Geometry/maps: accurate relative positions; label every important part.',
+    'Use color meaningfully; avoid blank or placeholder images.',
+  ],
+  jpeg: [
+    'Same visual quality rules as PNG — polished chart or diagram, not a screenshot of text.',
+  ],
+  jpg: [
+    'Same visual quality rules as PNG — polished chart or diagram, not a screenshot of text.',
+  ],
+  gif: [
+    'Render a clear educational visual; static GIF is fine unless animation is described.',
+  ],
+  pptx: [
+    'Split slides on "---" lines or "Slide N:" / "# " headings in the source content.',
+    'Title slide first, then content slides with concise bullets (not walls of text).',
+    'If the content describes a diagram, draw it on a slide (shapes/arrows) or embed a generated chart.',
+    'Use distinct slide layouts where appropriate (title, title+content).',
+  ],
+  xlsx: [
+    'Parse markdown tables or tabular lists in the source into real rows/columns.',
+    'Bold header row; auto-fit column widths where possible.',
+  ],
+  pdf: [
+    'If the source describes a figure/chart, generate and embed it — do not leave a "[diagram]" placeholder.',
+  ],
+  docx: [
+    'Use Heading 1/2 styles for section titles; bullets and numbered lists for enumerations.',
+  ],
+  html: [
+    'Include embedded CSS for readable typography; use tables or simple layout for structured content.',
+  ],
+};
+
 export interface GenerateDocumentInput {
   title: string;
   /** Markdown or plain-text source material for the output file. */
@@ -218,6 +257,7 @@ function safeBasename(title: string): string {
 }
 
 function buildDocumentGenInstructions(format: DocumentFormat): string {
+  const qualityRules = FORMAT_QUALITY_RULES[format] ?? [];
   return [
     'You are a document generator for K-12 and university teachers.',
     'When given a title, output format, and source content:',
@@ -225,14 +265,17 @@ function buildDocumentGenInstructions(format: DocumentFormat): string {
     `2. For this format use: ${FORMAT_HINTS[format]}.`,
     `3. You MUST output a ${format.toUpperCase()} file — not any other format.`,
     `4. The saved filename MUST end with .${extensionForFormat(format)}.`,
-    '5. Structure content clearly — headings, sections, lists, tables/charts as appropriate.',
+    '5. Faithfully adapt ALL source content to the output medium — do not summarize away detail.',
     '6. Include the document title prominently where the format allows.',
     '7. Put ONLY the educational content in the file — no meta commentary.',
     '8. You MUST save the file to disk in the working directory before you finish.',
     ...(format !== 'pdf'
       ? [`9. Do NOT create a PDF when ${format.toUpperCase()} was requested.`]
       : []),
-  ].join(' ');
+    ...(qualityRules.length > 0
+      ? ['', 'Format-specific quality:', ...qualityRules.map((r, i) => `${i + 1}. ${r}`)]
+      : []),
+  ].join('\n');
 }
 
 /** Generate a file via OpenAI Code Interpreter. */
@@ -260,8 +303,7 @@ export async function generateDocumentWithOpenAI(
         `Output filename (exact): ${filename}`,
         `Output format: ${ext}`,
         `Document title: ${input.title}`,
-        'Source content (markdown or plain text — adapt to the output format):',
-        input.content,
+        ['Source content — adapt fully to the output medium (do not paste raw markdown into slides/images):', input.content].join('\n'),
       ].join('\n\n'),
     });
 

package/src/services/worksheet.ts

@@ -6,7 +6,7 @@ import { TRPCError } from "@trpc/server";
 import { JobStatus, Prisma, type WorksheetQuestionType } from "@prisma/client";
 import { prisma } from "../lib/prisma.js";
 import { getFileUrl } from "../lib/fileUpload.js";
-import { cancelGradePipeline, regradeWorksheetPipeline } from "../pipelines/gradeWorksheet.js";
+import { cancelGradePipeline, regradeWorksheetPipeline, regradeAllPipeline } from "../pipelines/gradeWorksheet.js";
 import { enqueueRunJob } from "../workers/queues.js";
 import { logger } from "../utils/logger.js";
 import {
@@ -629,6 +629,25 @@ export async function regradeQuestion(
   return regradeWorksheetPipeline(worksheetResponseId, progressId);
 }
 
+export async function regradeAll(
+  worksheetResponseId: string,
+  userId: string
+) {
+  const worksheetResponse = await findWorksheetResponseByIdForTeacher(
+    worksheetResponseId,
+    userId
+  );
+
+  if (!worksheetResponse) {
+    throw new TRPCError({
+      code: "FORBIDDEN",
+      message: "Only class teachers can regrade all worksheet answers",
+    });
+  }
+
+  await regradeAllPipeline(worksheetResponseId);
+}
+
 export async function gradeAnswer(
   questionId: string,
   studentWorksheetResponseId: string,

package/src/workers/processors/aiNewtonChat.ts

@@ -64,7 +64,6 @@ export const aiNewtonChatWorker = createRunJobWorker({
 
     await generateAndSendNewtonResponse(
       newtonChat.id,
-      triggeringMessage.content,
       conversationId,
       newtonChat.submission,
       aiMessageId,

package/tests/integration/routers/newtonChat.test.ts

@@ -50,7 +50,7 @@ describe('Newton Chat Router', () => {
     conversationId = tutorConversation.conversationId;
   });
 
-  test('rejects attachment-only Newton chat posts', async () => {
+  test('allows attachment-only Newton chat posts', async () => {
     const [uploadTarget] = await user2Caller.message.getUploadUrls({
       conversationId,
       files: [
@@ -70,12 +70,13 @@ describe('Newton Chat Router', () => {
       },
     });
 
-    await expect(
-      user2Caller.newtonChat.postToNewtonChat({
-        newtonChatId,
-        content: '   ',
-        fileIds: [uploadTarget!.id],
-      }),
-    ).rejects.toThrow();
+    const message = await user2Caller.newtonChat.postToNewtonChat({
+      newtonChatId,
+      content: '   ',
+      fileIds: [uploadTarget!.id],
+    });
+
+    expect(message.attachments).toHaveLength(1);
+    expect(message.attachments[0]?.name).toBe('newton-attachment-only.pdf');
   });
 });

package/tests/unit/lib/chatAttachmentPrompt.test.ts

@@ -0,0 +1,135 @@
+import { describe, expect, it, vi } from 'vitest';
+import {
+  buildUserMessageContentWithAttachments,
+  loadChatAttachments,
+} from '../../../src/lib/chatAttachmentPrompt.js';
+
+const pdfBytes = new Uint8Array([0x25, 0x50, 0x44, 0x46]);
+const docxText = 'Unit 3 lesson plan content';
+
+vi.mock('officeparser', () => ({
+  OfficeParser: {
+    parseOffice: vi.fn(async () => ({
+      toText: () => docxText,
+    })),
+  },
+}));
+
+vi.mock('../../../src/lib/googleCloudStorage.js', () => ({
+  bucket: {
+    file: vi.fn((path: string) => ({
+      download: vi.fn(async () => {
+        if (path.includes('missing')) {
+          throw new Error('not found');
+        }
+        if (path.endsWith('.docx')) {
+          return [Buffer.from('fake-docx-bytes')];
+        }
+        return [Buffer.from(pdfBytes)];
+      }),
+    })),
+  },
+}));
+
+describe('buildUserMessageContentWithAttachments', () => {
+  it('returns plain text when there are no attachments', () => {
+    expect(buildUserMessageContentWithAttachments('Hello', [])).toBe('Hello');
+  });
+
+  it('includes PDFs as inline file parts with bytes', () => {
+    const content = buildUserMessageContentWithAttachments('Review this', [
+      {
+        id: '1',
+        name: 'essay.pdf',
+        type: 'application/pdf',
+        path: 'essay.pdf',
+        data: pdfBytes,
+      },
+    ]);
+
+    expect(content).toEqual([
+      { type: 'text', text: 'Review this' },
+      {
+        type: 'file',
+        data: pdfBytes,
+        mediaType: 'application/pdf',
+        filename: 'essay.pdf',
+      },
+    ]);
+  });
+
+  it('includes extracted docx text in the message body', () => {
+    const content = buildUserMessageContentWithAttachments('Summarize', [
+      {
+        id: '2',
+        name: 'plan.docx',
+        type: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+        path: 'plan.docx',
+        extractedText: docxText,
+      },
+    ]);
+
+    expect(content).toBe(
+      `Summarize\n\n[Content of plan.docx]:\n${docxText}`,
+    );
+  });
+
+  it('lists unsupported file types when extraction failed', () => {
+    const content = buildUserMessageContentWithAttachments('Here', [
+      {
+        id: '3',
+        name: 'clip.mp4',
+        type: 'video/mp4',
+        path: 'clip.mp4',
+      },
+    ]);
+
+    expect(content).toBe(
+      'Here\n[Attached files (content not extracted): clip.mp4 (video/mp4)]',
+    );
+  });
+});
+
+describe('loadChatAttachments', () => {
+  it('loads bytes and extracts office document text', async () => {
+    const resolved = await loadChatAttachments([
+      {
+        id: '1',
+        name: 'plan.docx',
+        type: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+        path: 'uploads/plan.docx',
+      },
+    ]);
+
+    expect(resolved[0]?.data).toBeDefined();
+    expect(resolved[0]?.extractedText).toBe(docxText);
+  });
+
+  it('loads pdf bytes without extracted text', async () => {
+    const resolved = await loadChatAttachments([
+      {
+        id: '2',
+        name: 'scan.pdf',
+        type: 'application/pdf',
+        path: 'uploads/scan.pdf',
+      },
+    ]);
+
+    expect(resolved[0]?.data).toEqual(pdfBytes);
+    expect(resolved[0]?.extractedText).toBeUndefined();
+  });
+
+  it('skips download for video attachments', async () => {
+    const resolved = await loadChatAttachments([
+      {
+        id: '3',
+        name: 'clip.mp4',
+        type: 'video/mp4',
+        path: 'uploads/clip.mp4',
+      },
+    ]);
+
+    expect(resolved[0]?.data).toBeUndefined();
+    expect(resolved[0]?.extractedText).toBeUndefined();
+  });
+});

package/tests/unit/lib/inferDocumentFormat.test.ts

@@ -30,4 +30,14 @@ describe('inferDocumentFormat', () => {
   it('defaults to pdf for generic handouts', () => {
     expect(inferDocumentFormat('Reading Guide', 'Chapter summary', 'pdf')).toBe('pdf');
   });
+
+  it('overrides pdf when content mentions a flowchart', () => {
+    expect(
+      inferDocumentFormat(
+        'Photosynthesis Overview',
+        'Include a flowchart of the light reactions.',
+        'pdf',
+      ),
+    ).toBe('png');
+  });
 });

package/tests/unit/pipelines/labChatPrompt.test.ts

@@ -17,8 +17,8 @@ describe("buildLabChatAgentSystemPrompt", () => {
 });
 
 describe("buildLabChatAgentMessages", () => {
-  test("folds the class context into the system prompt, not the message list", () => {
-    const { system, messages } = buildLabChatAgentMessages({
+  test("folds the class context into the system prompt, not the message list", async () => {
+    const { system, messages } = await buildLabChatAgentMessages({
       context: "Topic context",
       classContext: "CLASS: Biology",
       recentMessages: [
@@ -41,8 +41,8 @@ describe("buildLabChatAgentMessages", () => {
     ]);
   });
 
-  test("sorts timestamped history into chronological order before prompt assembly", () => {
-    const { messages } = buildLabChatAgentMessages({
+  test("sorts timestamped history into chronological order before prompt assembly", async () => {
+    const { messages } = await buildLabChatAgentMessages({
       context: "Topic context",
       classContext: "CLASS: Biology",
       recentMessages: [
@@ -76,8 +76,8 @@ describe("buildLabChatAgentMessages", () => {
     });
   });
 
-  test("keeps original order among messages without timestamps", () => {
-    const { messages } = buildLabChatAgentMessages({
+  test("keeps original order among messages without timestamps", async () => {
+    const { messages } = await buildLabChatAgentMessages({
       context: "Topic context",
       classContext: "CLASS: Biology",
       recentMessages: [