File size: 1,352 Bytes
7893226
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
be38df9
 
 
 
 
7893226
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
// frontend/src/api.js
export async function runSearch(query, topK, answerStyle, handlers) {
  const res = await fetch('/api/search', {
    method: 'POST',
    headers: { 'Content-Type': 'application/json' },
    body: JSON.stringify({ query, top_k: topK, answer_style: answerStyle }),
  });
  const { jobId } = await res.json();
  const es = new EventSource(`/api/jobs/${jobId}/events`);

  es.addEventListener('stage', (e) => handlers.onStage?.(JSON.parse(e.data)));
  es.addEventListener('papers', (e) => handlers.onPapers?.(JSON.parse(e.data).papers));
  es.addEventListener('answer_token', (e) => handlers.onAnswerToken?.(JSON.parse(e.data).text));
  es.addEventListener('consensus', (e) => handlers.onConsensus?.(JSON.parse(e.data)));
  es.addEventListener('blocked', (e) => { handlers.onBlocked?.(JSON.parse(e.data)); es.close(); });
  es.addEventListener('result', () => { handlers.onDone?.(); es.close(); });
  es.addEventListener('error', (e) => {
    try { handlers.onError?.(JSON.parse(e.data)); } catch { handlers.onError?.({ message: 'connection lost' }); }
    es.close();
  });
  return es;
}

export async function fetchUmapSample() {
  const res = await fetch('/api/umap-sample');
  return (await res.json()).points;
}

export async function fetchStats() {
  const res = await fetch('/api/stats');
  return (await res.json()).total;
}