diff options
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", |