feat: support rerank (#620)

1 files changed, 4 insertions, 6 deletions

diff --git a/src/client/reka.rs b/src/client/reka.rs
index 46f9118..2e9b88f 100644
--- a/src/client/reka.rs
+++ b/src/client/reka.rs
@@ -43,11 +43,7 @@ impl RekaClient {
     }
 }
 
-impl_client_trait!(
-    RekaClient,
-    chat_completions,
-    chat_completions_streaming
-);
+impl_client_trait!(RekaClient, chat_completions, chat_completions_streaming);
 
 async fn chat_completions(builder: RequestBuilder) -> Result<ChatCompletionsOutput> {
     let res = builder.send().await?;
@@ -113,7 +109,9 @@ fn build_chat_completions_body(data: ChatCompletionsData, model: &Model) -> Valu
 }
 
 fn extract_chat_completions(data: &Value) -> Result<ChatCompletionsOutput> {
-    let text = data["responses"][0]["message"]["content"].as_str().ok_or_else(|| anyhow!("Invalid response data: {data}"))?;
+    let text = data["responses"][0]["message"]["content"]
+        .as_str()
+        .ok_or_else(|| anyhow!("Invalid response data: {data}"))?;
 
     let output = ChatCompletionsOutput {
         text: text.to_string(),