본문으로 건너뛰기

함수: rerank()

function rerank<TAdapter, TDocument>(options): Promise<RerankResult<TDocument>>;

정의 위치: packages/ai/src/activities/rerank/index.ts:152

rerank 활동입니다. 쿼리와의 관련성에 따라 문서 순서를 다시 정렬합니다.

타입 매개변수

TAdapter

TAdapter extends RerankAdapter&lt;string, RerankProviderOptions&lt;TAdapter>>

TDocument

TDocument extends string | object = string

매개변수

options

RerankActivityOptions&lt;TAdapter, TDocument>

반환값

Promise&lt;RerankResult&lt;TDocument>>

예제

기본 순위 재지정

import { rerank } from '@tanstack/ai'
import { cohereRerank } from '@tanstack/ai-cohere'

const { ranking, rerankedDocuments } = await rerank({
adapter: cohereRerank('rerank-v3.5'),
query: 'talk about rain',
documents: ['sunny day at the beach', 'rainy afternoon in the city'],
topN: 2,
})

console.log(rerankedDocuments[0]) // 'rainy afternoon in the city'

객체 문서의 순위 재지정

const { ranking } = await rerank({
adapter: cohereRerank('rerank-v3.5'),
query: 'best laptop for travel',
documents: [
{ id: 1, text: 'A heavy gaming desktop' },
{ id: 2, text: 'A lightweight ultrabook with all-day battery' },
],
})

// ranking[0].document is the original object, fully typed.
console.log(ranking[0].document.id)