From abc588daac6053ec2edbdcde3f5a2dc5eb7d50b8 Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 21 Jun 2024 06:00:26 +0800 Subject: feat: support rerank (#620) --- src/client/reka.rs | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) (limited to 'src/client/reka.rs') 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 { 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 { - 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(), -- cgit v1.2.3