Querying the knowledge base

const API = 'https://api.subsidia.protypa.fr'
const headers = {
  Authorization: 'Bearer ' + process.env.SUBSIDIA_API_KEY,
  'Content-Type': 'application/json',
}

async function answer(question: string) {
  const { matches } = await fetch(API + '/v1/knowledge/query', {
    method: 'POST',
    headers,
    body: JSON.stringify({ query: question, topK: 6 }),
  }).then((r) => r.json())

  if (matches.length === 0) return { answer: 'The documents do not cover this.', sources: [] }

  const passages = matches
    .map((m: any, i: number) => '[' + (i + 1) + '] ' + m.sourceName + (m.page ? ' p.' + m.page : '') + '\n' + m.content)
    .join('\n\n')

  const res = await fetch(API + '/v1/chat/completions', {
    method: 'POST',
    headers,
    body: JSON.stringify({
      model: 'pulse-auto',
      messages: [
        {
          role: 'system',
          content:
            'Answer using ONLY the numbered passages below. Cite each claim as [n]. ' +
            'If the passages do not contain the answer, reply exactly: The documents do not cover this.\n\n' + passages,
        },
        { role: 'user', content: question },
      ],
    }),
  })
  const proofId = res.headers.get('x-pulse-proof')
  const completion = await res.json()
  return { answer: completion.choices[0].message.content, sources: matches, proofId }
}