app/lib/search/follow-up.mjs

Functions

Imports

It imports no other module of this repo. Imported by app/enhance/ask.ts, app/lib/search/ask-prompt.mjs.

Source

1// Not in ask-prompt.mjs: the client imports this, and importing from there shipped the system prompt
2// in the client bundle (tree-shaking did not remove it), which would make answerLeaksPrompt moot.
3
4/** A marker, not "the last line": splitting on position would eat a sentence when no follow-up came. */
5export const FOLLOW_UP_MARKER = "NEXT:";
6
7/**
8 * The marker must start a line (mid-sentence it is prose), and is searched from the end, since the
9 * follow-up is asked for last.
10 *
11 * @param {string} raw
12 * @returns {{ answer: string, followUp: string | null }}
13 */
14export function splitFollowUp(raw) {
15  const lines = raw.split("\n");
16  for (let i = lines.length - 1; i >= 0; i -= 1) {
17    /* The ?? "" only satisfies noUncheckedIndexedAccess. */
18    const line = (lines[i] ?? "").trim();
19    if (!line.startsWith(FOLLOW_UP_MARKER)) continue;
20    const question = line.slice(FOLLOW_UP_MARKER.length).trim();
21    const answer = lines.slice(0, i).join("\n").trimEnd();
22    return { answer, followUp: question.length > 0 ? question : null };
23  }
24  return { answer: raw, followUp: null };
25}