1つの問いを、出典付きリサーチブリーフに変える

const BASE = "https://openapi.felo.ai";
const headers = {
  Authorization: `Bearer ${process.env.FELO_API_KEY}`,
  "Content-Type": "application/json",
};

async function felo<T>(path: string, body?: unknown): Promise<T> {
  const res = await fetch(`${BASE}${path}`, {
    method: body ? "POST" : "GET",
    headers,
    body: body ? JSON.stringify(body) : undefined,
  });
  if (!res.ok) throw new Error(`${res.status}: ${await res.text()}`);
  return res.json() as Promise<T>;
}

type ChatData = {
  data: { answer: string; resources: { link: string; title: string }[] };
};

export async function answer(question: string) {
  const chat = await felo<ChatData>("/v2/chat", { query: question });
  return { answer: chat.data.answer, sources: chat.data.resources.slice(0, 3) };
}

type ExtractData = { data: { content: string } };

const pages = await Promise.all(
  sources.map((source) =>
    felo<ExtractData>("/v2/web/extract", {
      url: source.link,
      output_format: "markdown",
      with_readability: true,
      with_cache: true,
    }).then((page) => ({ ...source, content: page.data.content })),
  ),
);

type XData = {
  data: {
    tweets: {
      content: string;
      author: { username: string };
      created_at: string;
    }[];
  };
};

const since = Math.floor((Date.now() - 7 * 24 * 60 * 60 * 1000) / 1000);

const social = await felo<XData>("/v2/x/tweet/search", {
  query: question,
  limit: 10,
  since_time: since,
});

type LiveDocData = { data: { short_id: string } };

const doc = await felo<LiveDocData>("/v2/livedocs", {
  name: `Research: ${question.slice(0, 60)}`,
  description: "Sources collected while answering the question",
});

await felo(`/v2/livedocs/${doc.data.short_id}/resources/urls`, {
  urls: sources.map((source) => ({ url: source.link, title: source.title })),
});

console.log(doc.data.short_id); // keep this for later queries

type Retrieved = {
  data: { title: string; content: string; score: number }[];
};

async function ask(livedoc: string, query: string) {
  const hits = await felo<Retrieved>(
    `/v2/livedocs/${livedoc}/resources/retrieve`,
    { query },
  );
  return hits.data.filter((hit) => hit.score >= 0.7);
}