From a491820ab5ff9bb58e62fd25d88f31ab53a81db9 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 6 Feb 2025 19:20:48 +0800 Subject: feat: support model alias (#1150) --- 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 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", -- cgit v1.2.3