diff options
| author | sigoden <sigoden@gmail.com> | 2024-09-24 07:42:24 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-24 07:42:24 +0800 |
| commit | 912773c25a113f49c5df63cd3a8086d38c75103e (patch) | |
| tree | 7d653edf095f0949d1eeb2f83ec3e5616972c3b1 /src/client/bedrock.rs | |
| parent | 00c4a6e421f01590ffc3fa5601e93d5ec755fca7 (diff) | |
| download | aichat-912773c25a113f49c5df63cd3a8086d38c75103e.tar.gz | |
refactor: embeddings/rerank fn accept ref data (#878)
Diffstat (limited to 'src/client/bedrock.rs')
| -rw-r--r-- | src/client/bedrock.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/client/bedrock.rs b/src/client/bedrock.rs index 25c666c..f7b3019 100644 --- a/src/client/bedrock.rs +++ b/src/client/bedrock.rs @@ -98,7 +98,7 @@ impl BedrockClient { fn embeddings_builder( &self, client: &ReqwestClient, - data: EmbeddingsData, + data: &EmbeddingsData, ) -> Result<RequestBuilder> { let access_key_id = self.get_access_key_id()?; let secret_access_key = self.get_secret_access_key()?; @@ -173,7 +173,7 @@ impl Client for BedrockClient { async fn embeddings_inner( &self, client: &ReqwestClient, - data: EmbeddingsData, + data: &EmbeddingsData, ) -> Result<EmbeddingsOutput> { let builder = self.embeddings_builder(client, data)?; embeddings(builder).await |