diff --git a/src/main/reranker/BaseReranker.ts b/src/main/reranker/BaseReranker.ts index 9a40dfbdf8..83d241fe85 100644 --- a/src/main/reranker/BaseReranker.ts +++ b/src/main/reranker/BaseReranker.ts @@ -86,7 +86,7 @@ export default abstract class BaseReranker { return data.output.results } else if (provider === 'voyageai') { return data.data - } else if (provider === 'mis-tei') { + } else if (provider?.includes('tei')) { return data.map((item: any) => { return { index: item.index,