Function tryOpenAIRerank

src/context/reranker.ts:121–150

OpenAI/Cohere-style reranker: POST /v1/rerank {model, query, documents} → {results:[{index,relevance_score}]}.

(
  baseUrl: string,
  model: string,
  query: string,
  documents: string[],
  signal?: AbortSignal,
)

/** OpenAI/Cohere-style reranker: POST /v1/rerank {model, query, documents} → {results:[{index,relevance_score}]}. */
async function tryOpenAIRerank(
  baseUrl: string,
  model: string,
  query: string,
  documents: string[],
  signal?: AbortSignal,
): Promise<number[] | null> {
  try {
    const res = await proxyFetch(`${baseUrl.replace(/\/$/, '')}/v1/rerank`, {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ model, query, documents, top_n: documents.length }),
      signal,
    });
    if (!res.ok) return null;
    const data: any = await res.json();
    const arr = data?.results;
    if (!Array.isArray(arr)) return null;
    const scores = new Array<number>(documents.length).fill(0);
    for (const r of arr) {
      const idx = r.index;
      const score = r.relevance_score ?? r.score;
      if (typeof idx === 'number' && typeof score === 'number') scores[idx] = score;
    }
    return scores;
  } catch (e: any) {
    logger.debug('OpenAI-style rerank not available', { err: e?.message });
    return null;
  }
}