summaryrefslogtreecommitdiffstats
path: root/src/client/bedrock.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-12-04 21:03:59 +0800
committerGitHub <noreply@github.com>2024-12-04 21:03:59 +0800
commit3a3388375be05758d5f5574cce1d0f8eb7bdd604 (patch)
treecc8e19e218c20f28a04feca33850fce4bb2cbe98 /src/client/bedrock.rs
parent7d42fe9429f75d195f865b07cef10d040d5397f2 (diff)
downloadaichat-3a3388375be05758d5f5574cce1d0f8eb7bdd604.tar.gz
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
Diffstat (limited to 'src/client/bedrock.rs')
-rw-r--r--src/client/bedrock.rs4
1 files changed, 2 insertions, 2 deletions
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,