Skip to content

Commit dd2a1f6

Browse files
committed
fix: add logger with category and update telemetry import for bedrock provider
1 parent 5105228 commit dd2a1f6

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

docs/docs/providers/safety/remote_bedrock.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ AWS Bedrock safety provider for content moderation using AWS's safety services.
1919
| `aws_access_key_id` | `str \| None` | No | | The AWS access key to use. Default use environment variable: AWS_ACCESS_KEY_ID |
2020
| `aws_secret_access_key` | `str \| None` | No | | The AWS secret access key to use. Default use environment variable: AWS_SECRET_ACCESS_KEY |
2121
| `aws_session_token` | `str \| None` | No | | The AWS session token to use. Default use environment variable: AWS_SESSION_TOKEN |
22-
| `region_name` | `str \| None` | No | | The default AWS Region to use, for example, us-west-1 or us-west-2.Default use environment variable: AWS_DEFAULT_REGION |
22+
| `region_name` | `str \| None` | No | us-east-2 | The default AWS Region to use, for example, us-west-1 or us-west-2.Default use environment variable: AWS_DEFAULT_REGION |
2323
| `profile_name` | `str \| None` | No | | The profile name that contains credentials to use.Default use environment variable: AWS_PROFILE |
2424
| `total_max_attempts` | `int \| None` | No | | An integer representing the maximum number of attempts that will be made for a single request, including the initial attempt. Default use environment variable: AWS_MAX_ATTEMPTS |
2525
| `retry_mode` | `str \| None` | No | | A string representing the type of retries Boto3 will perform.Default use environment variable: AWS_RETRY_MODE |

llama_stack/providers/remote/inference/bedrock/bedrock.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,14 @@
1818
OpenAIEmbeddingsRequestWithExtraBody,
1919
OpenAIEmbeddingsResponse,
2020
)
21+
from llama_stack.core.telemetry.tracing import get_current_span
22+
from llama_stack.log import get_logger
2123
from llama_stack.providers.utils.inference.openai_mixin import OpenAIMixin
22-
from llama_stack.providers.utils.telemetry.tracing import get_current_span
2324

2425
from .config import BedrockConfig
2526

27+
logger = get_logger(name=__name__, category="inference::bedrock")
28+
2629

2730
class BedrockInferenceAdapter(OpenAIMixin):
2831
"""

0 commit comments

Comments
 (0)