From 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 25 Jun 2024 07:39:35 +0800
Subject: feat: support rag-dedicated clients (jina and voyageai) (#645)

---
 src/client/ernie.rs | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

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

diff --git a/src/client/ernie.rs b/src/client/ernie.rs
index 428d263..64c1820 100644
--- a/src/client/ernie.rs
+++ b/src/client/ernie.rs
@@ -1,4 +1,5 @@
 use super::access_token::*;
+use super::rag_dedicated::*;
 use super::*;
 
 use anyhow::{anyhow, bail, Context, Result};
@@ -220,15 +221,11 @@ struct EmbeddingsResBodyEmbedding {
 async fn rerank(builder: RequestBuilder) -> Result<RerankOutput> {
     let data: Value = builder.send().await?.json().await?;
     maybe_catch_error(&data)?;
-    let res_body: RerankResBody = serde_json::from_value(data).context("Invalid rerank data")?;
+    let res_body: RagDedicatedRerankResBody =
+        serde_json::from_value(data).context("Invalid rerank data")?;
     Ok(res_body.results)
 }
 
-#[derive(Deserialize)]
-struct RerankResBody {
-    results: RerankOutput,
-}
-
 fn build_chat_completions_body(data: ChatCompletionsData, model: &Model) -> Value {
     let ChatCompletionsData {
         mut messages,
-- 
cgit v1.2.3