mirror of
https://github.com/k8sgpt-ai/k8sgpt.git
synced 2025-08-19 08:17:04 +00:00
21bc76e5b7
Signed-off-by: rkarthikr <38294804+rkarthikr@users.noreply.github.com> Co-authored-by: Alex Jones <alexsimonjones@gmail.com>
19 lines
692 B
Go
19 lines
692 B
Go
package ai
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/bedrock"
|
|
"github.com/aws/aws-sdk-go-v2/service/bedrockruntime"
|
|
)
|
|
|
|
// BedrockManagementAPI defines the interface for Bedrock management operations
|
|
type BedrockManagementAPI interface {
|
|
GetInferenceProfile(ctx context.Context, params *bedrock.GetInferenceProfileInput, optFns ...func(*bedrock.Options)) (*bedrock.GetInferenceProfileOutput, error)
|
|
}
|
|
|
|
// BedrockRuntimeAPI defines the interface for Bedrock runtime operations
|
|
type BedrockRuntimeAPI interface {
|
|
InvokeModel(ctx context.Context, params *bedrockruntime.InvokeModelInput, optFns ...func(*bedrockruntime.Options)) (*bedrockruntime.InvokeModelOutput, error)
|
|
}
|