diff options
| author | sigoden <sigoden@gmail.com> | 2025-02-06 19:20:48 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-02-06 19:20:48 +0800 |
| commit | a491820ab5ff9bb58e62fd25d88f31ab53a81db9 (patch) | |
| tree | 86a84d7176a5fb16bce9b195808e952d181a8146 /src/client/bedrock.rs | |
| parent | 4812e446ee30467bd3ecce983a826f7649de7110 (diff) | |
| download | aichat-a491820ab5ff9bb58e62fd25d88f31ab53a81db9.tar.gz | |
feat: support model alias (#1150)
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 435aa57..ebc0154 100644 --- a/src/client/bedrock.rs +++ b/src/client/bedrock.rs @@ -46,7 +46,7 @@ impl BedrockClient { let region = self.get_region()?; let host = format!("bedrock-runtime.{region}.amazonaws.com"); - let model_name = &self.model.name(); + let model_name = &self.model.real_name(); let uri = if data.stream { format!("/model/{model_name}/converse-stream") @@ -95,7 +95,7 @@ impl BedrockClient { let region = self.get_region()?; let host = format!("bedrock-runtime.{region}.amazonaws.com"); - let uri = format!("/model/{}/invoke", self.model.name()); + let uri = format!("/model/{}/invoke", self.model.real_name()); let input_type = match data.query { true => "search_query", |