feat: playground rerank

2024-10-14 10:27:46 +08:00
parent ecc683a6bd
commit 4c3dddc486
31 changed files with 1579 additions and 140 deletions
@@ -4,6 +4,8 @@ export const CHAT_API = '/v1-openai/chat/completions';

 export const OPENAI_MODELS = '/v1-openai/models';

+export const RERANKER_API = '/rerank';
+
 export async function execChatCompletions(params: any) {
  return request(`${CHAT_API}`, {
    method: 'POST',
@@ -17,3 +19,19 @@ export const queryModelsList = async (params: any) => {
    params
  });
 };
+
+export const rerankerQuery = async (
+  params: {
+    model: string;
+    query: string;
+    top_n: number;
+    documents: string[];
+  },
+  options?: any
+) => {
+  return request(`${RERANKER_API}`, {
+    method: 'POST',
+    data: params,
+    cancelToken: options?.cancelToken
+  });
+};