From 3a3388375be05758d5f5574cce1d0f8eb7bdd604 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 4 Dec 2024 21:03:59 +0800 Subject: refactor: improve retrieve model (#1036) - check the model type while retrieve model - select chat/reranker model even if it is missed in client models - find predefined-models for openai-compatible client with startsWith - remove client::ApiType --- src/client/bedrock.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/client/bedrock.rs') diff --git a/src/client/bedrock.rs b/src/client/bedrock.rs index 081a65b..7cd289c 100644 --- a/src/client/bedrock.rs +++ b/src/client/bedrock.rs @@ -67,7 +67,7 @@ impl BedrockClient { let body = build_chat_completions_body(data, &self.model)?; let mut request_data = RequestData::new("", body); - self.patch_request_data(&mut request_data, ApiType::ChatCompletions); + self.patch_request_data(&mut request_data); let RequestData { url: _, headers, @@ -118,7 +118,7 @@ impl BedrockClient { }); let mut request_data = RequestData::new("", body); - self.patch_request_data(&mut request_data, ApiType::Embeddings); + self.patch_request_data(&mut request_data); let RequestData { url: _, headers, -- cgit v1.2.3