Function classifyTopicOpenAI

packages/memos-core/src/ingest/providers/openai.ts:270–311

(
  taskState: string,
  newMessage: string,
  cfg: SummarizerConfig,
  log: Logger,
)

Sub-questions, tools, methods, details of current topic = S.`;

export async function classifyTopicOpenAI(
  taskState: string,
  newMessage: string,
  cfg: SummarizerConfig,
  log: Logger,
): Promise<TopicClassifyResult> {
  const endpoint = normalizeChatEndpoint(cfg.endpoint ?? "https://api.openai.com/v1/chat/completions");
  const model = cfg.model ?? "gpt-4o-mini";
  const headers: Record<string, string> = {
    "Content-Type": "application/json",
    Authorization: `Bearer ${cfg.apiKey}`,
    ...cfg.headers,
  };

  const userContent = `TASK:\n${taskState}\n\nMSG:\n${newMessage}`;

  const resp = await fetch(endpoint, {
    method: "POST",
    headers,
    body: JSON.stringify(buildRequestBody(cfg, {
      model,
      temperature: 0,
      max_tokens: 60,
      messages: [
        { role: "system", content: TOPIC_CLASSIFIER_PROMPT },
        { role: "user", content: userContent },
      ],
    })),
    signal: AbortSignal.timeout(cfg.timeoutMs ?? 15_000),
  });

  if (!resp.ok) {
    const body = await resp.text();
    throw new Error(`OpenAI topic-classifier failed (${resp.status}): ${body}`);
  }

  const json = (await resp.json()) as { choices: Array<{ message: { content: string } }> };
  const raw = json.choices[0]?.message?.content?.trim() ?? "";
  log.debug(`Topic classifier raw: "${raw}"`);

  return parseTopicClassifyResult(raw, log);
}

const TOPIC_ARBITRATION_PROMPT = `A classifier flagged this message as possibly new topic (low confidence). Is it truly UNRELATED, or a sub-question/follow-up?
Tools/methods/details of current task = SAME. Shared entity/theme = SAME. Entirely different domain = NEW.