bge-reranker-base

Cloudflare ホスト

embedding モデルと異なり、reranker は質問と文書を入力として受け取り、embedding ではなく類似度を直接出力します。クエリとパッセージを入力すると関連度スコアが得られます。このスコアは sigmoid 関数で [0,1] の浮動小数点値に変換できます。

モデル情報
単価	$0.0031 per M input tokens

使い方


export interface Env {
	AI: Ai;
}

export default {
	async fetch(request, env): Promise<Response> {
		const query = 'Which one is cooler?'
		const contexts = [
			{
				text: 'a cyberpunk lizzard'
			},
			{
				text: 'a cyberpunk cat'
			}
		];

		const response = await env.AI.run('@cf/baai/bge-reranker-base', { query, contexts });

		return Response.json(response);
	},
} satisfies ExportedHandler<Env>;


import os
import requests

ACCOUNT_ID = "your-account-id"
AUTH_TOKEN = os.environ.get("CLOUDFLARE_AUTH_TOKEN")

response = requests.post(
  f"https://api.cloudflare.com/client/v4/accounts/{ACCOUNT_ID}/ai/run/@cf/baai/bge-reranker-base",
    headers={"Authorization": f"Bearer {AUTH_TOKEN}"},
    json={
	  "query": "Which one is better?",
      "contexts": [
        {"text": "a cyberpunk lizzard"},
		    {"text": "a cyberpunk car"},
      ]
    }
)
result = response.json()
print(result)


curl https://api.cloudflare.com/client/v4/accounts/$CLOUDFLARE_ACCOUNT_ID/ai/run/@cf/baai/bge-reranker-base \
  -X POST \
  -H "Authorization: Bearer $CLOUDFLARE_AUTH_TOKEN" \
  -d '{ "query": "Which one is better?", "contexts": [{ "text": "a cyberpunk lizzard" }, {"text": "a cyberpunk cat"}]}'

パラメーター

query

string必須minLength: 1提供したコンテキストに対して行うクエリです。

top_k

integerminimum: 1最高スコアから返す結果の件数です。

▶contexts[]

array必須提供したコンテキストの一覧です。応答が参照するため、配列内のインデックスが重要です。

▶response[]

array

API スキーマ（Raw）

Input

Output