From b17719457bb098b6572671dcb5cc60a0241c4b8f Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 5 Jun 2024 06:06:11 +0000 Subject: chore: format --- src/client/qianwen.rs | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) (limited to 'src/client/qianwen.rs') diff --git a/src/client/qianwen.rs b/src/client/qianwen.rs index c34e409..908cd1e 100644 --- a/src/client/qianwen.rs +++ b/src/client/qianwen.rs @@ -18,7 +18,7 @@ const CHAT_COMPLETIONS_API_URL: &str = const CHAT_COMPLETIONS_API_URL_VL: &str = "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation"; -const EMBEDDINGS_API_URL: &str = +const EMBEDDINGS_API_URL: &str = "https://dashscope.aliyuncs.com/api/v1/services/embeddings/text-embedding/text-embedding"; #[derive(Debug, Clone, Deserialize, Default)] @@ -249,13 +249,17 @@ fn build_chat_completions_body(data: ChatCompletionsData, model: &Model) -> Resu Ok((body, has_upload)) } -async fn embeddings( - builder: RequestBuilder, -) -> Result { +async fn embeddings(builder: RequestBuilder) -> Result { let data: Value = builder.send().await?.json().await?; maybe_catch_error(&data)?; - let res_body: EmbeddingsResBody = serde_json::from_value(data).context("Invalid request data")?; - let output = res_body.output.embeddings.into_iter().map(|v| v.embedding).collect(); + let res_body: EmbeddingsResBody = + serde_json::from_value(data).context("Invalid request data")?; + let output = res_body + .output + .embeddings + .into_iter() + .map(|v| v.embedding) + .collect(); Ok(output) } -- cgit v1.2.3