Function aggregateOpenAIResponses

ai/ai.ts:1949–2016

(
  payloads: any[],
)

const aggregateOpenAIResponses = (
  payloads: any[],
): {
  text: string;
  images: AIImage[];
  sources: Array<{ url: string; title?: string }>;
} => {
  const deltaTexts: string[] = [];
  const deltaImages: AIImage[] = [];
  let fallbackText = "";
  let fallbackImages: AIImage[] = [];
  const sources: Array<{ url: string; title?: string }> = [];
  const seenSources = new Set<string>();

  const appendSources = (entries: Array<{ url: string; title?: string }>) => {
    for (const entry of entries) {
      if (seenSources.has(entry.url)) continue;
      seenSources.add(entry.url);
      sources.push(entry);
    }
  };

  for (const payload of payloads) {
    const choice = payload?.choices?.[0];

    if (choice?.delta?.content !== undefined) {
      const parsedDelta = parseOpenAIChatResponse({
        choices: [{ message: { content: choice.delta.content } }],
      });
      if (parsedDelta.text && parsedDelta.text !== "AI 回复为空") {
        deltaTexts.push(parsedDelta.text);
      }
      if (parsedDelta.images.length > 0) {
        deltaImages.push(...parsedDelta.images);
      }
    }

    const fallbackContent =
      choice?.message?.content ?? choice?.content ?? payload?.content;
    if (fallbackContent !== undefined) {
      const parsedFallback = parseOpenAIChatResponse({
        choices: [{ message: { content: fallbackContent } }],
      });
      if (parsedFallback.text && parsedFallback.text !== "AI 回复为空") {
        fallbackText = parsedFallback.text;
      }
      if (parsedFallback.images.length > 0) {
        fallbackImages = parsedFallback.images;
      }
    } else if (typeof choice?.text === "string" && choice.text.trim()) {
      fallbackText = choice.text.trim();
      fallbackImages = [];
    } else if (typeof payload?.text === "string" && payload.text.trim()) {
      fallbackText = payload.text.trim();
      fallbackImages = [];
    }

    appendSources(collectOpenAISources(payload));
  }