1 files changed, 3 insertions, 3 deletions
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
 );