From 20cada3d0ecfb3a63bd24138fa1ad329c778c8d1 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 27 Jan 2025 08:13:50 +0800
Subject: feat:  ernie migrates to v2 api (#1130)

---
 src/client/openai_compatible.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'src/client/openai_compatible.rs')

diff --git a/src/client/openai_compatible.rs b/src/client/openai_compatible.rs
index ce1eea3..30bc464 100644
--- a/src/client/openai_compatible.rs
+++ b/src/client/openai_compatible.rs
@@ -79,7 +79,11 @@ fn prepare_rerank(self_: &OpenAICompatibleClient, data: &RerankData) -> Result<R
     let api_key = self_.get_api_key().ok();
     let api_base = get_api_base_ext(self_)?;
 
-    let url = format!("{api_base}/rerank");
+    let url = if self_.name().starts_with("ernie") {
+        format!("{api_base}/rerankers")
+    } else {
+        format!("{api_base}/rerank")
+    };
 
     let body = generic_build_rerank_body(data, &self_.model);
 
@@ -149,7 +153,7 @@ pub fn generic_build_rerank_body(data: &RerankData, model: &Model) -> Value {
         "query": query,
         "documents": documents,
     });
-    if model.client_name() == "voyageai" {
+    if model.client_name().starts_with("voyageai") {
         body["top_k"] = (*top_n).into()
     } else {
         body["top_n"] = (*top_n).into()
-- 
cgit v1.2.3