From ca63356a99407f59dc6f064474ecb984c99a0852 Mon Sep 17 00:00:00 2001 From: Shogo Nakamoto Date: Thu, 31 Jul 2025 15:15:34 +0900 Subject: [PATCH] feat: allow overriding bedrock provider endpoint (#2540) --- lua/avante/providers/bedrock.lua | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/lua/avante/providers/bedrock.lua b/lua/avante/providers/bedrock.lua index dc5d7c1..f0664e2 100644 --- a/lua/avante/providers/bedrock.lua +++ b/lua/avante/providers/bedrock.lua @@ -127,11 +127,18 @@ function M:parse_curl_args(prompt_opts) session_token = awsCreds.session_token end - local endpoint = string.format( - "https://bedrock-runtime.%s.amazonaws.com/model/%s/invoke-with-response-stream", - region, - provider_conf.model - ) + local endpoint + if provider_conf.endpoint then + -- Use custom endpoint if provided + endpoint = provider_conf.endpoint + else + -- Default to AWS Bedrock endpoint + endpoint = string.format( + "https://bedrock-runtime.%s.amazonaws.com/model/%s/invoke-with-response-stream", + region, + provider_conf.model + ) + end local headers = { ["Content-Type"] = "application/json",