File tree Expand file tree Collapse file tree 11 files changed +16
-16
lines changed
docs/docs/providers/inference
providers/remote/inference/bedrock
tests/unit/providers/inference Expand file tree Collapse file tree 11 files changed +16
-16
lines changed Original file line number Diff line number Diff line change @@ -22,6 +22,6 @@ AWS Bedrock inference provider using OpenAI compatible endpoint.
2222## Sample Configuration
2323
2424``` yaml
25- api_key : ${env.AWS_BEDROCK_API_KEY :=}
25+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
2626region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
2727` ` `
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ providers:
4747 - provider_id : bedrock
4848 provider_type : remote::bedrock
4949 config :
50- api_key : ${env.AWS_BEDROCK_API_KEY :=}
50+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
5151 region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
5252 - provider_id : ${env.NVIDIA_API_KEY:+nvidia}
5353 provider_type : remote::nvidia
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ providers:
4747 - provider_id : bedrock
4848 provider_type : remote::bedrock
4949 config :
50- api_key : ${env.AWS_BEDROCK_API_KEY :=}
50+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
5151 region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
5252 - provider_id : ${env.NVIDIA_API_KEY:+nvidia}
5353 provider_type : remote::nvidia
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ providers:
4747 - provider_id : bedrock
4848 provider_type : remote::bedrock
4949 config :
50- api_key : ${env.AWS_BEDROCK_API_KEY :=}
50+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
5151 region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
5252 - provider_id : ${env.NVIDIA_API_KEY:+nvidia}
5353 provider_type : remote::nvidia
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ providers:
4747 - provider_id : bedrock
4848 provider_type : remote::bedrock
4949 config :
50- api_key : ${env.AWS_BEDROCK_API_KEY :=}
50+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
5151 region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
5252 - provider_id : ${env.NVIDIA_API_KEY:+nvidia}
5353 provider_type : remote::nvidia
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ providers:
4747 - provider_id : bedrock
4848 provider_type : remote::bedrock
4949 config :
50- api_key : ${env.AWS_BEDROCK_API_KEY :=}
50+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
5151 region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
5252 - provider_id : ${env.NVIDIA_API_KEY:+nvidia}
5353 provider_type : remote::nvidia
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ providers:
4747 - provider_id : bedrock
4848 provider_type : remote::bedrock
4949 config :
50- api_key : ${env.AWS_BEDROCK_API_KEY :=}
50+ api_key : ${env.AWS_BEARER_TOKEN_BEDROCK :=}
5151 region_name : ${env.AWS_DEFAULT_REGION:=us-east-2}
5252 - provider_id : ${env.NVIDIA_API_KEY:+nvidia}
5353 provider_type : remote::nvidia
Original file line number Diff line number Diff line change @@ -37,7 +37,7 @@ class BedrockInferenceAdapter(OpenAIMixin):
3737 """
3838
3939 config : BedrockConfig
40- provider_data_api_key_field : str = "aws_bedrock_api_key "
40+ provider_data_api_key_field : str = "aws_bearer_token_bedrock "
4141
4242 def get_base_url (self ) -> str :
4343 """Get base URL for OpenAI client."""
@@ -111,7 +111,7 @@ async def openai_chat_completion(
111111 logger .error (f"AWS Bedrock authentication token expired: { error_msg } " )
112112 raise ValueError (
113113 "AWS Bedrock authentication failed: Bearer token has expired. "
114- "The AWS_BEDROCK_API_KEY environment variable contains an expired pre-signed URL. "
114+ "The AWS_BEARER_TOKEN_BEDROCK environment variable contains an expired pre-signed URL. "
115115 "Please refresh your token by generating a new pre-signed URL with AWS credentials. "
116116 "Refer to AWS Bedrock documentation for details on OpenAI-compatible endpoints."
117117 ) from e
Original file line number Diff line number Diff line change 1212
1313
1414class BedrockProviderDataValidator (BaseModel ):
15- aws_bedrock_api_key : str | None = Field (
15+ aws_bearer_token_bedrock : str | None = Field (
1616 default = None ,
17- description = "API key for Amazon Bedrock" ,
17+ description = "API Key (Bearer token) for Amazon Bedrock" ,
1818 )
1919
2020
@@ -27,6 +27,6 @@ class BedrockConfig(RemoteInferenceProviderConfig):
2727 @classmethod
2828 def sample_run_config (cls , ** kwargs ):
2929 return {
30- "api_key" : "${env.AWS_BEDROCK_API_KEY :=}" ,
30+ "api_key" : "${env.AWS_BEARER_TOKEN_BEDROCK :=}" ,
3131 "region_name" : "${env.AWS_DEFAULT_REGION:=us-east-2}" ,
3232 }
Original file line number Diff line number Diff line change @@ -40,8 +40,8 @@ def test_api_key_from_header_overrides_config():
4040 """Test API key from request header overrides config via client property"""
4141 config = BedrockConfig (api_key = "config-key" , region_name = "us-east-1" )
4242 adapter = BedrockInferenceAdapter (config = config )
43- adapter .provider_data_api_key_field = "aws_bedrock_api_key "
44- adapter .get_request_provider_data = MagicMock (return_value = SimpleNamespace (aws_bedrock_api_key = "header-key" ))
43+ adapter .provider_data_api_key_field = "aws_bearer_token_bedrock "
44+ adapter .get_request_provider_data = MagicMock (return_value = SimpleNamespace (aws_bearer_token_bedrock = "header-key" ))
4545
4646 # The client property is where header override happens (in OpenAIMixin)
4747 assert adapter .client .api_key == "header-key"
You can’t perform that action at this time.
0 commit comments