36 lines
980 B
Python
36 lines
980 B
Python
"""AWS Bedrock provider profile."""
|
|
|
|
from providers import register_provider
|
|
from providers.base import ProviderProfile
|
|
|
|
|
|
class BedrockProfile(ProviderProfile):
|
|
"""AWS Bedrock — no REST /v1/models endpoint; uses AWS SDK."""
|
|
|
|
def fetch_models(
|
|
self,
|
|
*,
|
|
api_key: str | None = None,
|
|
timeout: float = 8.0,
|
|
) -> list[str] | None:
|
|
"""Bedrock model listing requires AWS SDK, not a REST call."""
|
|
return None
|
|
|
|
|
|
bedrock = BedrockProfile(
|
|
name="bedrock",
|
|
aliases=("aws", "aws-bedrock", "amazon-bedrock", "amazon"),
|
|
api_mode="bedrock_converse",
|
|
env_vars=(), # AWS SDK credentials — not env vars
|
|
base_url="https://bedrock-runtime.us-east-1.amazonaws.com",
|
|
auth_type="aws_sdk",
|
|
)
|
|
|
|
register_provider(bedrock)
|
|
|
|
|
|
def register(ctx):
|
|
"""Plugin entry point — delegates to the inner hermes_agent_bedrock package."""
|
|
from hermes_agent_bedrock import register as _inner_register
|
|
_inner_register(ctx)
|