summaryrefslogtreecommitdiffstats
path: root/src/client/qianwen.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-05 06:06:11 +0000
committersigoden <sigoden@gmail.com>2024-06-05 06:06:11 +0000
commitb17719457bb098b6572671dcb5cc60a0241c4b8f (patch)
treec533a9191619b2ad298b57932858afb23152fc67 /src/client/qianwen.rs
parenteacc88f04a1129c74ab3c0ad56ae3ec0159a486e (diff)
downloadaichat-b17719457bb098b6572671dcb5cc60a0241c4b8f.tar.gz
chore: format
Diffstat (limited to 'src/client/qianwen.rs')
-rw-r--r--src/client/qianwen.rs16
1 files changed, 10 insertions, 6 deletions
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<EmbeddingsOutput> {
+async fn embeddings(builder: RequestBuilder) -> Result<EmbeddingsOutput> {
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)
}