Skip to content

feat: AWS Bedrock guardrails, provisioned throughput & invocation logging SDK-compat - #210

Merged
NitinKumar004 merged 2 commits into
developmentfrom
feat/aws-bedrock-management-sdk-compat
Jun 11, 2026
Merged

feat: AWS Bedrock guardrails, provisioned throughput & invocation logging SDK-compat#210
NitinKumar004 merged 2 commits into
developmentfrom
feat/aws-bedrock-management-sdk-compat

Conversation

@NitinKumar004

Copy link
Copy Markdown
Collaborator

Closes #207.

Summary

Extends the AWS Bedrock SDK-compat handler (built in #160) with the remaining control-plane management surface, end-to-end compatible with the real aws-sdk-go-v2/service/bedrock client.

What's included

  • Guardrails: CreateGuardrail, GetGuardrail, ListGuardrails, UpdateGuardrail, DeleteGuardrail (synchronous READY state, DRAFT version, name/ARN lookup).
  • Provisioned model throughput: CreateProvisionedModelThroughput, GetProvisionedModelThroughput, ListProvisionedModelThroughputs, DeleteProvisionedModelThroughput (synchronous InService, model resolved against the foundation/custom catalog).
  • Model invocation logging: Put/Get/DeleteModelInvocationLoggingConfiguration (S3 + CloudWatch delivery config).
  • Added across all layers: driver interface, portable wrapper, in-memory provider (management.go), and restJson1 server handler routing (/guardrails, /provisioned-model-throughput[s], /logging/modelinvocations).

Tests

  • Provider-level and portable-level unit tests.
  • SDK roundtrip tests driving the real bedrock client: guardrail lifecycle, provisioned-throughput lifecycle, and invocation-logging put/get/delete.

go build ./..., go test ./..., and golangci-lint run all pass.

…gging

Extend the Bedrock SDK-compat handler beyond the model catalog + runtime with
the remaining control-plane management surface: guardrails CRUD, provisioned
model throughput lifecycle, and model-invocation logging configuration. Real
aws-sdk-go-v2/service/bedrock clients work end-to-end against the in-memory
backend.
A rename via UpdateGuardrail stored the guardrail under its old name key, so
lookups by the new name failed. Re-key (delete old, set new) so name lookups
stay correct after a rename.
@NitinKumar004
NitinKumar004 merged commit 616e815 into development Jun 11, 2026
@NitinKumar004
NitinKumar004 deleted the feat/aws-bedrock-management-sdk-compat branch July 25, 2026 13:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

feat: AWS Bedrock guardrails, provisioned throughput & invocation logging SDK-compat

1 participant