fix broken bedrock inference provider (#151)

This commit is contained in:
moritalous 2024-09-30 12:17:58 +09:00 committed by GitHub
parent 2f096ca509
commit 2bd785354d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -94,4 +94,15 @@ def available_providers() -> List[ProviderSpec]:
provider_data_validator="llama_stack.providers.adapters.safety.together.TogetherProviderDataValidator",
),
),
remote_provider_spec(
api=Api.inference,
adapter=AdapterSpec(
adapter_id="bedrock",
pip_packages=[
"boto3"
],
module="llama_stack.providers.adapters.inference.bedrock",
config_class="llama_stack.providers.adapters.inference.bedrock.BedrockConfig",
),
),
]