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/cohere.rs | |
| parent | 00c4a6e421f01590ffc3fa5601e93d5ec755fca7 (diff) | |
| download | aichat-912773c25a113f49c5df63cd3a8086d38c75103e.tar.gz | |
refactor: embeddings/rerank fn accept ref data (#878)
Diffstat (limited to 'src/client/cohere.rs')
| -rw-r--r-- | src/client/cohere.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/client/cohere.rs b/src/client/cohere.rs index 64263b7..f471e7a 100644 --- a/src/client/cohere.rs +++ b/src/client/cohere.rs @@ -57,7 +57,7 @@ fn prepare_chat_completions( Ok(request_data) } -fn prepare_embeddings(self_: &CohereClient, data: EmbeddingsData) -> Result<RequestData> { +fn prepare_embeddings(self_: &CohereClient, data: &EmbeddingsData) -> Result<RequestData> { let api_key = self_.get_api_key()?; let api_base = self_ .get_api_base() @@ -83,7 +83,7 @@ fn prepare_embeddings(self_: &CohereClient, data: EmbeddingsData) -> Result<Requ Ok(request_data) } -fn prepare_rerank(self_: &CohereClient, data: RerankData) -> Result<RequestData> { +fn prepare_rerank(self_: &CohereClient, data: &RerankData) -> Result<RequestData> { let api_key = self_.get_api_key()?; let api_base = self_ .get_api_base() |