1.0 KiB
1.0 KiB
layout
| layout |
|---|
| default |
Consumir reranker self-hosted desde Node.js
No hay integración oficial con LangChain.js, pero la API REST es muy sencilla:
async function rerank(query: string, docs: string[], topN = 3) {
const res = await fetch("http://localhost:7997/rerank", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({ query, documents: docs }),
});
const results = await res.json();
// [{ index: 2, relevance_score: 0.95 }, { index: 0, relevance_score: 0.72 }, ...]
return results
.sort((a, b) => b.relevance_score - a.relevance_score)
.slice(0, topN);
}
// Uso con documentos de LangChain
const searchResults = await vectorStore.similaritySearch(query, 10);
const ranked = await rerank(query, searchResults.map(d => d.pageContent));
const topDocs = ranked.map(r => searchResults[r.index]);
💡 Misma API para Infinity y TEI — cambiar de runtime no requiere cambiar código