From 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 25 Jun 2024 07:39:35 +0800 Subject: feat: support rag-dedicated clients (jina and voyageai) (#645) --- src/client/openai_compatible.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/client/openai_compatible.rs') diff --git a/src/client/openai_compatible.rs b/src/client/openai_compatible.rs index 789604f..132b40a 100644 --- a/src/client/openai_compatible.rs +++ b/src/client/openai_compatible.rs @@ -1,5 +1,5 @@ -use super::cohere::*; use super::openai::*; +use super::rag_dedicated::*; use super::*; use anyhow::Result; @@ -90,7 +90,7 @@ impl OpenAICompatibleClient { let api_key = self.get_api_key().ok(); let api_base = self.get_api_base_ext()?; - let body = cohere_build_rerank_body(data, &self.model); + let body = rag_dedicated_build_rerank_body(data, &self.model); let url = format!("{api_base}/rerank"); @@ -131,5 +131,5 @@ impl_client_trait!( openai_chat_completions, openai_chat_completions_streaming, openai_embeddings, - cohere_rerank + rag_dedicated_rerank ); -- cgit v1.2.3