diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-05 06:06:11 +0000 |
|---|---|---|
| committer | sigoden <sigoden@gmail.com> | 2024-06-05 06:06:11 +0000 |
| commit | b17719457bb098b6572671dcb5cc60a0241c4b8f (patch) | |
| tree | c533a9191619b2ad298b57932858afb23152fc67 /src/client/qianwen.rs | |
| parent | eacc88f04a1129c74ab3c0ad56ae3ec0159a486e (diff) | |
| download | aichat-b17719457bb098b6572671dcb5cc60a0241c4b8f.tar.gz | |
chore: format
Diffstat (limited to 'src/client/qianwen.rs')
| -rw-r--r-- | src/client/qianwen.rs | 16 |
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) } |
