bedrockagent

package
v1.1.4 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jul 20, 2026 License: MIT Imports: 21 Imported by: 0

README

Bedrock Agent

Parity grade: A · SDK aws-sdk-go-v2/service/bedrockagent@v1.54.0 · last audited 2026-07-12 (05e127fa13a618837560e0b6a56098937fc1cae4)

Coverage

Metric Value
Operations audited 74 (73 ok, 1 partial)
Feature families 3 (3 ok)
Known gaps 2
Deferred items 2
Resource leaks clean
Known gaps
  • "CreateAgentActionGroup ignores the {agentVersion} path parameter and always stores the action group under DRAFT, even if a client (unusually) targets a different version in the URL. Real AWS action groups can only be created against DRAFT anyway, so this is unlikely to be hit by a real client, but a strict client sending a non-DRAFT version would get silently redirected instead of a validation error. Not fixed this sweep (low risk, would need a new exact-match AWS error string to validate against). (bd: TODO — file gopherstack-bedrockagent-actiongroup-version)"
  • "ValidateFlowDefinition always returns zero validation errors regardless of the definition passed — acceptable for a permissive emulator (the op still reads real state and returns the AWS-accurate empty-array shape); not a disguised no-op flag, just an easy target if flow-definition validation logic is ever wanted."
Deferred
  • "KBDocument/DataSource nested configuration blobs (dataSourceConfiguration, vectorIngestionConfiguration, knowledgeBaseConfiguration, storageConfiguration, actionGroupExecutor, apiSchema, functionSchema, guardrailConfiguration, memoryConfiguration, promptOverrideConfiguration) are passed through as opaque map[string]any/JSON blobs rather than typed + validated against the SDK's nested shape unions. This is consistent with how this service already treats them; deep-shape validation of these blobs was out of scope this sweep."
  • "IngestionJob does not model the statistics field AWS returns (documents scanned/indexed/failed counts) — every ingestion job here always indexes 0 documents worth of statistics; not audited for wire-accuracy this sweep."

More

Documentation

Overview

Package bedrockagent provides a local stub for the Amazon Bedrock Agent service.

Index

Constants

This section is empty.

Variables

View Source 
var (
	// ErrNotFound is returned when a requested resource does not exist.
	ErrNotFound = awserr.New("ResourceNotFoundException", awserr.ErrNotFound)
	// ErrAlreadyExists is returned when a resource with the given name already exists.
	ErrAlreadyExists = awserr.New("ConflictException", awserr.ErrAlreadyExists)
	// ErrValidation is returned for invalid request parameters.
	ErrValidation = awserr.New("ValidationException", awserr.ErrInvalidParameter)
)
View Source 
var ErrNilAppContext = errors.New("bedrockagent: AppContext must not be nil")

ErrNilAppContext is returned when a nil AppContext is passed to Provider.Init.

Functions

This section is empty.

Types

type ActionGroupConfig 

type ActionGroupConfig struct {
	ActionGroupExecutor map[string]any
	APISchema           map[string]any
	FunctionSchema      map[string]any
	ActionGroupName     string
	Description         string
	ActionGroupState    string
}

ActionGroupConfig holds fields for creating or updating an AgentActionGroup.

type ActionGroupSummary 

type ActionGroupSummary struct {
	ActionGroupID    string `json:"actionGroupId"`
	ActionGroupName  string `json:"actionGroupName"`
	ActionGroupState string `json:"actionGroupState"`
	Description      string `json:"description,omitempty"`
}

ActionGroupSummary is used in list responses.

type Agent 

type Agent struct {
	CreatedAt                   time.Time         `json:"createdAt"`
	UpdatedAt                   time.Time         `json:"updatedAt"`
	PreparedAt                  *time.Time        `json:"preparedAt,omitempty"`
	Tags                        map[string]string `json:"tags,omitempty"`
	Guardrail                   map[string]any    `json:"guardrailConfiguration,omitempty"`
	Memory                      map[string]any    `json:"memoryConfiguration,omitempty"`
	PromptOverrideConfiguration map[string]any    `json:"promptOverrideConfiguration"`
	AgentID                     string            `json:"agentId"`
	AgentARN                    string            `json:"agentArn"`
	AgentName                   string            `json:"agentName"`
	AgentVersion                string            `json:"agentVersion"`
	AgentStatus                 string            `json:"agentStatus"`
	Collaboration               string            `json:"agentCollaboration"`
	Description                 string            `json:"description,omitempty"`
	FoundationModel             string            `json:"foundationModel,omitempty"`
	Instruction                 string            `json:"instruction,omitempty"`
	RoleARN                     string            `json:"agentResourceRoleArn,omitempty"`
	IdleSessionTTLInSeconds     int               `json:"idleSessionTTLInSeconds"`
}

Agent represents a Bedrock Agent.

type AgentActionGroup 

type AgentActionGroup struct {
	CreatedAt           time.Time      `json:"createdAt"`
	UpdatedAt           time.Time      `json:"updatedAt"`
	ActionGroupExecutor map[string]any `json:"actionGroupExecutor,omitempty"`
	APISchema           map[string]any `json:"apiSchema,omitempty"`
	FunctionSchema      map[string]any `json:"functionSchema,omitempty"`
	ActionGroupID       string         `json:"actionGroupId"`
	ActionGroupName     string         `json:"actionGroupName"`
	AgentID             string         `json:"agentId"`
	AgentVersion        string         `json:"agentVersion"`
	ActionGroupState    string         `json:"actionGroupState"`
	Description         string         `json:"description,omitempty"`
}

AgentActionGroup is an action group attached to an agent version.

type AgentAlias 

type AgentAlias struct {
	CreatedAt            time.Time         `json:"createdAt"`
	UpdatedAt            time.Time         `json:"updatedAt"`
	Tags                 map[string]string `json:"tags,omitempty"`
	AgentAliasID         string            `json:"agentAliasId"`
	AgentAliasARN        string            `json:"agentAliasArn"`
	AgentAliasName       string            `json:"agentAliasName"`
	AgentAliasStatus     string            `json:"agentAliasStatus"`
	AgentID              string            `json:"agentId"`
	Description          string            `json:"description,omitempty"`
	RoutingConfiguration []AliasRouting    `json:"routingConfiguration"`
}

AgentAlias routes traffic to a specific agent version.

type AgentAliasSummary 

type AgentAliasSummary struct {
	AgentAliasID     string `json:"agentAliasId"`
	AgentAliasName   string `json:"agentAliasName"`
	AgentAliasStatus string `json:"agentAliasStatus"`
	Description      string `json:"description,omitempty"`
}

AgentAliasSummary is used in list responses.

type AgentCollaborator 

type AgentCollaborator struct {
	CreatedAt                time.Time      `json:"createdAt"`
	UpdatedAt                time.Time      `json:"updatedAt"`
	AgentDescriptor          map[string]any `json:"agentDescriptor,omitempty"`
	AgentID                  string         `json:"agentId"`
	AgentVersion             string         `json:"agentVersion"`
	CollaboratorID           string         `json:"collaboratorId"`
	CollaboratorName         string         `json:"collaboratorName"`
	CollaborationInstruction string         `json:"collaborationInstruction,omitempty"`
	RelayConversationHistory string         `json:"relayConversationHistory,omitempty"`
	CollaboratorStatus       string         `json:"collaboratorStatus"`
}

AgentCollaborator links two agents for multi-agent collaboration.

type AgentConfig 

type AgentConfig struct {
	Tags                    map[string]string
	Guardrail               map[string]any
	Memory                  map[string]any
	AgentName               string
	Collaboration           string
	Description             string
	FoundationModel         string
	Instruction             string
	RoleARN                 string
	IdleSessionTTLInSeconds int
}

AgentConfig holds fields for creating or updating an Agent.

type AgentKnowledgeBase 

type AgentKnowledgeBase struct {
	CreatedAt       time.Time `json:"createdAt"`
	UpdatedAt       time.Time `json:"updatedAt"`
	AgentID         string    `json:"agentId"`
	AgentVersion    string    `json:"agentVersion"`
	KnowledgeBaseID string    `json:"knowledgeBaseId"`
	KBState         string    `json:"knowledgeBaseState"`
	Description     string    `json:"description,omitempty"`
}

AgentKnowledgeBase is the association between an agent and a knowledge base.

type AgentSummary 

type AgentSummary struct {
	UpdatedAt   time.Time `json:"updatedAt"`
	AgentID     string    `json:"agentId"`
	AgentName   string    `json:"agentName"`
	AgentStatus string    `json:"agentStatus"`
	Description string    `json:"description,omitempty"`
}

AgentSummary is the condensed agent representation used in list responses.

type AgentVersion 

type AgentVersion struct {
	CreatedAt       time.Time `json:"createdAt"`
	UpdatedAt       time.Time `json:"updatedAt"`
	AgentID         string    `json:"agentId"`
	AgentARN        string    `json:"agentArn"`
	AgentName       string    `json:"agentName"`
	AgentStatus     string    `json:"agentStatus"`
	AgentVersion    string    `json:"agentVersion"`
	Description     string    `json:"description,omitempty"`
	FoundationModel string    `json:"foundationModel,omitempty"`
	Instruction     string    `json:"instruction,omitempty"`
	RoleARN         string    `json:"agentResourceRoleArn,omitempty"`
}

AgentVersion holds a snapshot version of an agent.

type AgentVersionSummary 

type AgentVersionSummary struct {
	UpdatedAt    time.Time `json:"updatedAt"`
	AgentName    string    `json:"agentName"`
	AgentStatus  string    `json:"agentStatus"`
	AgentVersion string    `json:"agentVersion"`
	Description  string    `json:"description,omitempty"`
}

AgentVersionSummary is used in list-agent-versions responses.

type AliasConfig 

type AliasConfig struct {
	Tags                 map[string]string
	AliasName            string
	Description          string
	RoutingConfiguration []AliasRouting
}

AliasConfig holds fields for creating or updating an AgentAlias.

type AliasRouting 

type AliasRouting struct {
	AgentVersion string `json:"agentVersion"`
}

AliasRouting maps an alias to an agent version.

type CollaboratorConfig 

type CollaboratorConfig struct {
	AgentDescriptor          map[string]any
	CollaboratorName         string
	CollaborationInstruction string
	RelayConversationHistory string
}

CollaboratorConfig holds fields for an AgentCollaborator.

type DataSource 

type DataSource struct {
	CreatedAt               time.Time      `json:"createdAt"`
	UpdatedAt               time.Time      `json:"updatedAt"`
	DataSourceConfiguration map[string]any `json:"dataSourceConfiguration,omitempty"`
	VectorIngestionConfig   map[string]any `json:"vectorIngestionConfiguration,omitempty"`
	DataSourceID            string         `json:"dataSourceId"`
	KnowledgeBaseID         string         `json:"knowledgeBaseId"`
	Name                    string         `json:"name"`
	DataSourceStatus        string         `json:"dataSourceStatus"`
	Description             string         `json:"description,omitempty"`
	DataDeletionPolicy      string         `json:"dataDeletionPolicy,omitempty"`
}

DataSource is a knowledge base data source.

type DataSourceConfig 

type DataSourceConfig struct {
	DataSourceConfiguration map[string]any
	VectorIngestionConfig   map[string]any
	Name                    string
	Description             string
	DataDeletionPolicy      string
}

DataSourceConfig holds fields for creating or updating a DataSource.

type DataSourceSummary 

type DataSourceSummary struct {
	UpdatedAt        time.Time `json:"updatedAt"`
	DataSourceID     string    `json:"dataSourceId"`
	KnowledgeBaseID  string    `json:"knowledgeBaseId"`
	Name             string    `json:"name"`
	DataSourceStatus string    `json:"dataSourceStatus"`
	Description      string    `json:"description,omitempty"`
}

DataSourceSummary is used in list responses.

type Flow 

type Flow struct {
	CreatedAt   time.Time         `json:"createdAt"`
	UpdatedAt   time.Time         `json:"updatedAt"`
	Tags        map[string]string `json:"tags,omitempty"`
	Definition  map[string]any    `json:"definition,omitempty"`
	FlowID      string            `json:"id"`
	FlowARN     string            `json:"arn"`
	Name        string            `json:"name"`
	Status      string            `json:"status"`
	Description string            `json:"description,omitempty"`
	RoleARN     string            `json:"executionRoleArn,omitempty"`
	Version     string            `json:"version"`
}

Flow is a Bedrock prompt flow.

type FlowAlias 

type FlowAlias struct {
	CreatedAt            time.Time          `json:"createdAt"`
	UpdatedAt            time.Time          `json:"updatedAt"`
	Tags                 map[string]string  `json:"tags,omitempty"`
	AliasID              string             `json:"id"`
	AliasARN             string             `json:"arn"`
	FlowID               string             `json:"flowId"`
	Name                 string             `json:"name"`
	Description          string             `json:"description,omitempty"`
	RoutingConfiguration []FlowAliasRouting `json:"routingConfiguration,omitempty"`
}

FlowAlias routes traffic to a specific flow version.

type FlowAliasConfig 

type FlowAliasConfig struct {
	Tags                 map[string]string
	Name                 string
	Description          string
	RoutingConfiguration []FlowAliasRouting
}

FlowAliasConfig holds fields for creating or updating a FlowAlias.

type FlowAliasRouting 

type FlowAliasRouting struct {
	FlowVersion string `json:"flowVersion"`
}

FlowAliasRouting maps a flow alias to a specific flow version.

type FlowAliasSummary 

type FlowAliasSummary struct {
	CreatedAt   time.Time `json:"createdAt"`
	UpdatedAt   time.Time `json:"updatedAt"`
	AliasID     string    `json:"id"`
	AliasARN    string    `json:"arn"`
	FlowID      string    `json:"flowId"`
	Name        string    `json:"name"`
	Description string    `json:"description,omitempty"`
}

FlowAliasSummary is used in list responses.

type FlowConfig 

type FlowConfig struct {
	Tags        map[string]string
	Definition  map[string]any
	Name        string
	Description string
	RoleARN     string
}

FlowConfig holds fields for creating or updating a Flow.

type FlowSummary 

type FlowSummary struct {
	UpdatedAt   time.Time `json:"updatedAt"`
	FlowID      string    `json:"id"`
	Name        string    `json:"name"`
	Status      string    `json:"status"`
	Description string    `json:"description,omitempty"`
	Version     string    `json:"version"`
}

FlowSummary is used in list responses.

type FlowValidationError 

type FlowValidationError struct {
	Message  string `json:"message"`
	Severity string `json:"severity"`
}

FlowValidationError is a flow definition validation error.

type FlowVersion 

type FlowVersion struct {
	CreatedAt   time.Time      `json:"createdAt"`
	Definition  map[string]any `json:"definition,omitempty"`
	FlowARN     string         `json:"arn"`
	FlowID      string         `json:"id"`
	Name        string         `json:"name"`
	Status      string         `json:"status"`
	Version     string         `json:"version"`
	Description string         `json:"description,omitempty"`
}

FlowVersion is a snapshot of a flow.

type FlowVersionSummary 

type FlowVersionSummary struct {
	CreatedAt   time.Time `json:"createdAt"`
	Arn         string    `json:"arn"`
	FlowID      string    `json:"id"`
	Name        string    `json:"name"`
	Status      string    `json:"status"`
	Version     string    `json:"version"`
	Description string    `json:"description,omitempty"`
}

FlowVersionSummary is used in list responses.

type Handler 

type Handler struct {
	Backend       StorageBackend
	AccountID     string
	DefaultRegion string
}

Handler is the HTTP handler for the Bedrock Agent REST API.

func NewHandler 

func NewHandler(backend StorageBackend) *Handler

NewHandler creates a new Bedrock Agent handler.

func (*Handler) ChaosOperations 

func (h *Handler) ChaosOperations() []string

ChaosOperations returns all operations.

func (*Handler) ChaosRegions 

func (h *Handler) ChaosRegions() []string

ChaosRegions returns the supported regions.

func (*Handler) ChaosServiceName 

func (h *Handler) ChaosServiceName() string

ChaosServiceName returns the chaos service name.

func (*Handler) ExtractOperation 

func (h *Handler) ExtractOperation(c *echo.Context) string

ExtractOperation determines the operation name from the request.

func (*Handler) ExtractResource 

func (h *Handler) ExtractResource(c *echo.Context) string

ExtractResource extracts an agent or flow ID from the request path.

func (*Handler) GetSupportedOperations 

func (h *Handler) GetSupportedOperations() []string

GetSupportedOperations returns the list of supported operations.

func (*Handler) Handler 

func (h *Handler) Handler() echo.HandlerFunc

Handler returns the Echo handler function.

func (*Handler) MatchPriority 

func (h *Handler) MatchPriority() int

MatchPriority returns routing priority.

func (*Handler) Name 

func (h *Handler) Name() string

Name returns the service name.

func (*Handler) Reset 

func (h *Handler) Reset()

Reset clears handler state (delegates to backend).

func (*Handler) Restore 

func (h *Handler) Restore(ctx context.Context, data []byte) error

Restore implements persistence.Persistable by delegating to the backend.

func (*Handler) RouteMatcher 

func (h *Handler) RouteMatcher() service.Matcher

RouteMatcher returns a function matching Bedrock Agent requests.

func (*Handler) Snapshot 

func (h *Handler) Snapshot(ctx context.Context) []byte

Snapshot implements persistence.Persistable by delegating to the backend. Handler previously had no Snapshot/Restore of its own -- and neither did InMemoryBackend -- so cli.go's generic setupPersistence (which type-asserts the registered service.Registerable, i.e. the Handler, for a Snapshot/Restore pair) never picked BedrockAgent up at all: dead wiring, with no persistence underneath it either. This delegation (matching the codecommit/codepipeline/cleanrooms pattern) is what wires BedrockAgent into persistence for the first time.

type InMemoryBackend 

type InMemoryBackend struct {
	// contains filtered or unexported fields
}

InMemoryBackend implements StorageBackend with in-memory maps, isolated by region.

Phase 3.3 datalayer conversion: every map[string]*T resource collection is a *store.Table[T] registered on b.registry, with a *store.Index[T] added where a parent-scoped List/bulk-delete previously did a linear key-prefix scan (see store_setup.go's registerAllTables). Resources with a real, globally-unique identity field (agents, knowledgeBases, flows, prompts) are registered directly by that ID; resources whose ID is only unique within a parent (agent/flow/knowledge-base versions, action groups, aliases, collaborators, associations, data sources, ingestion jobs, KB documents) keep the same "parent/child[/grandchild]" composite key the original map already used, matching the composite-key rule in .claude/memories/parity-principles.md.

agentsByName, kbsByName, flowsByName, and promptsByName are left as plain map[string]string reverse-lookup caches (name -> ID): their values are bare strings, not *T, so there is nothing for store.Table to key on. agentVersionCtrs, flowVersionCtrs, and promptVersionCtrs (map[string]int) are likewise left raw for the same reason. tags (map[string]map[string]string) is the one remaining grouping map with a non-*T value.

func NewInMemoryBackend 

func NewInMemoryBackend(region, accountID string) *InMemoryBackend

NewInMemoryBackend creates and initialises an InMemoryBackend.

func (*InMemoryBackend) AssociateAgentCollaborator 

func (b *InMemoryBackend) AssociateAgentCollaborator(
	_ context.Context, agentID, agentVersion string, cfg CollaboratorConfig,
) (*AgentCollaborator, error)

AssociateAgentCollaborator creates a collaborator association.

func (*InMemoryBackend) AssociateAgentKnowledgeBase 

func (b *InMemoryBackend) AssociateAgentKnowledgeBase(
	_ context.Context, agentID, agentVersion, kbID, description, kbState string,
) (*AgentKnowledgeBase, error)

AssociateAgentKnowledgeBase creates an agent–KB association.

func (*InMemoryBackend) CreateAgent 

func (b *InMemoryBackend) CreateAgent(ctx context.Context, cfg AgentConfig) (*Agent, error)

CreateAgent creates a new agent.

func (*InMemoryBackend) CreateAgentActionGroup 

func (b *InMemoryBackend) CreateAgentActionGroup(
	_ context.Context, agentID string, cfg ActionGroupConfig,
) (*AgentActionGroup, error)

CreateAgentActionGroup creates an action group for an agent version.

func (*InMemoryBackend) CreateAgentAlias 

func (b *InMemoryBackend) CreateAgentAlias(
	ctx context.Context, agentID string, cfg AliasConfig,
) (*AgentAlias, error)

CreateAgentAlias creates an alias for an agent.

func (*InMemoryBackend) CreateAgentVersion 

func (b *InMemoryBackend) CreateAgentVersion(
	_ context.Context, agentID, description string,
) (*AgentVersion, error)

CreateAgentVersion creates a numbered snapshot of an agent.

Note: real Bedrock Agents has no CreateAgentVersion wire operation -- numbered agent versions are created as a side effect of CreateAgentAlias when called with no routingConfiguration (see the doAgentAlias helper in CreateAgentAlias below, which calls newAgentVersionLocked directly while already holding b.mu). This method stays exported on InMemoryBackend/ StorageBackend for internal/programmatic use (e.g. tests seeding a version directly) but is deliberately not reachable from any HTTP route.

func (*InMemoryBackend) CreateDataSource 

func (b *InMemoryBackend) CreateDataSource(
	_ context.Context, kbID string, cfg DataSourceConfig,
) (*DataSource, error)

CreateDataSource creates a data source in a knowledge base.

func (*InMemoryBackend) CreateFlow 

func (b *InMemoryBackend) CreateFlow(ctx context.Context, cfg FlowConfig) (*Flow, error)

CreateFlow creates a new flow.

func (*InMemoryBackend) CreateFlowAlias 

func (b *InMemoryBackend) CreateFlowAlias(
	ctx context.Context, flowID string, cfg FlowAliasConfig,
) (*FlowAlias, error)

CreateFlowAlias creates a flow alias.

func (*InMemoryBackend) CreateFlowVersion 

func (b *InMemoryBackend) CreateFlowVersion(
	_ context.Context, flowID, description string,
) (*FlowVersion, error)

CreateFlowVersion creates a numbered snapshot of a flow.

func (*InMemoryBackend) CreateKnowledgeBase 

func (b *InMemoryBackend) CreateKnowledgeBase(
	ctx context.Context, cfg KnowledgeBaseConfig,
) (*KnowledgeBase, error)

CreateKnowledgeBase creates a new knowledge base.

func (*InMemoryBackend) CreatePrompt 

func (b *InMemoryBackend) CreatePrompt(ctx context.Context, cfg PromptConfig) (*Prompt, error)

CreatePrompt creates a new prompt.

func (*InMemoryBackend) CreatePromptVersion 

func (b *InMemoryBackend) CreatePromptVersion(
	_ context.Context, promptID, description string,
) (*PromptVersion, error)

CreatePromptVersion creates a versioned snapshot of a prompt.

func (*InMemoryBackend) DeleteAgent 

func (b *InMemoryBackend) DeleteAgent(_ context.Context, agentID string) error

DeleteAgent deletes an agent.

Note: this only removes the agent itself, its name-lookup entry, its versions (agentVersions), and its version counter -- it does NOT clean up actionGroups, agentAliases, or agentCollaborators for the deleted agent (the pre-Phase-3.3 map version keyed agentCollaborators by "agentID/"+ agentVersion and called delete(b.agentCollaborators, agentID) here, which never matched any real key -- a no-op cleanup bug preserved as-is by this conversion; see the DeleteAgent doc in .claude/memories for the byte-for-byte preservation rule).

func (*InMemoryBackend) DeleteAgentActionGroup 

func (b *InMemoryBackend) DeleteAgentActionGroup(
	_ context.Context, agentID, agentVersion, actionGroupID string,
) error

DeleteAgentActionGroup deletes an action group.

func (*InMemoryBackend) DeleteAgentAlias 

func (b *InMemoryBackend) DeleteAgentAlias(_ context.Context, agentID, aliasID string) error

DeleteAgentAlias deletes an agent alias.

func (*InMemoryBackend) DeleteAgentVersion 

func (b *InMemoryBackend) DeleteAgentVersion(
	_ context.Context, agentID, agentVersion string,
) error

DeleteAgentVersion deletes an agent version. See the not-found precedence note on GetAgentVersion.

func (*InMemoryBackend) DeleteDataSource 

func (b *InMemoryBackend) DeleteDataSource(_ context.Context, kbID, dsID string) error

DeleteDataSource deletes a data source.

func (*InMemoryBackend) DeleteFlow 

func (b *InMemoryBackend) DeleteFlow(_ context.Context, flowID string) error

DeleteFlow deletes a flow.

func (*InMemoryBackend) DeleteFlowAlias 

func (b *InMemoryBackend) DeleteFlowAlias(_ context.Context, flowID, aliasID string) error

DeleteFlowAlias deletes a flow alias.

func (*InMemoryBackend) DeleteFlowVersion 

func (b *InMemoryBackend) DeleteFlowVersion(_ context.Context, flowID, flowVersion string) error

DeleteFlowVersion deletes a flow version.

func (*InMemoryBackend) DeleteKnowledgeBase 

func (b *InMemoryBackend) DeleteKnowledgeBase(_ context.Context, kbID string) error

DeleteKnowledgeBase deletes a knowledge base.

func (*InMemoryBackend) DeleteKnowledgeBaseDocuments 

func (b *InMemoryBackend) DeleteKnowledgeBaseDocuments(
	_ context.Context, kbID, dsID string, docIDs []string,
) ([]KBDocumentDetail, error)

DeleteKnowledgeBaseDocuments deletes documents from a knowledge base data source.

func (*InMemoryBackend) DeletePrompt 

func (b *InMemoryBackend) DeletePrompt(_ context.Context, promptID string) error

DeletePrompt deletes a prompt.

func (*InMemoryBackend) DeletePromptVersion 

func (b *InMemoryBackend) DeletePromptVersion(
	_ context.Context, promptID, version string,
) error

DeletePromptVersion deletes a prompt version.

func (*InMemoryBackend) DisassociateAgentCollaborator 

func (b *InMemoryBackend) DisassociateAgentCollaborator(
	_ context.Context, agentID, agentVersion, collaboratorID string,
) error

DisassociateAgentCollaborator removes a collaborator.

func (*InMemoryBackend) DisassociateAgentKnowledgeBase 

func (b *InMemoryBackend) DisassociateAgentKnowledgeBase(
	_ context.Context, agentID, agentVersion, kbID string,
) error

DisassociateAgentKnowledgeBase removes an agent–KB association.

func (*InMemoryBackend) GetAgent 

func (b *InMemoryBackend) GetAgent(_ context.Context, agentID string) (*Agent, error)

GetAgent returns an agent by ID.

func (*InMemoryBackend) GetAgentActionGroup 

func (b *InMemoryBackend) GetAgentActionGroup(
	_ context.Context, agentID, agentVersion, actionGroupID string,
) (*AgentActionGroup, error)

GetAgentActionGroup returns an action group.

func (*InMemoryBackend) GetAgentAlias 

func (b *InMemoryBackend) GetAgentAlias(_ context.Context, agentID, aliasID string) (*AgentAlias, error)

GetAgentAlias returns an agent alias.

func (*InMemoryBackend) GetAgentCollaborator 

func (b *InMemoryBackend) GetAgentCollaborator(
	_ context.Context, agentID, agentVersion, collaboratorID string,
) (*AgentCollaborator, error)

GetAgentCollaborator returns a collaborator by ID.

func (*InMemoryBackend) GetAgentKnowledgeBase 

func (b *InMemoryBackend) GetAgentKnowledgeBase(
	_ context.Context, agentID, agentVersion, kbID string,
) (*AgentKnowledgeBase, error)

GetAgentKnowledgeBase returns an agent–KB association.

func (*InMemoryBackend) GetAgentVersion 

func (b *InMemoryBackend) GetAgentVersion(
	_ context.Context, agentID, agentVersion string,
) (*AgentVersion, error)

GetAgentVersion returns a specific agent version.

Not-found precedence note: the pre-Phase-3.3 map checked for the presence of the per-agent inner map (created lazily by the first CreateAgentVersion call and never removed except by DeleteAgent) to decide between "agent not found" and "agent version not found". That inner-map presence check is not reproducible with a flat store.Table + secondary Index (an Index prunes a group the moment its last member is deleted, see pkgs/store's Index.remove), so this checks b.agents.Has(agentID) instead -- identical result in every case except the one where every version of a still-existing agent has been deleted via DeleteAgentVersion, where the pre-conversion code returned "agent not found" (arguably itself a mislabeled error) and this returns "agent version not found".

func (*InMemoryBackend) GetDataSource 

func (b *InMemoryBackend) GetDataSource(_ context.Context, kbID, dsID string) (*DataSource, error)

GetDataSource returns a data source.

func (*InMemoryBackend) GetFlow 

func (b *InMemoryBackend) GetFlow(_ context.Context, flowID string) (*Flow, error)

GetFlow returns a flow.

func (*InMemoryBackend) GetFlowAlias 

func (b *InMemoryBackend) GetFlowAlias(_ context.Context, flowID, aliasID string) (*FlowAlias, error)

GetFlowAlias returns a flow alias.

func (*InMemoryBackend) GetFlowVersion 

func (b *InMemoryBackend) GetFlowVersion(
	_ context.Context, flowID, flowVersion string,
) (*FlowVersion, error)

GetFlowVersion returns a flow version. See the not-found precedence note on GetAgentVersion in the Agent version CRUD section above -- the same b.flows.Has(flowID)-instead-of-inner-map-presence reasoning applies here.

func (*InMemoryBackend) GetIngestionJob 

func (b *InMemoryBackend) GetIngestionJob(
	_ context.Context, kbID, dsID, jobID string,
) (*IngestionJob, error)

GetIngestionJob returns an ingestion job.

func (*InMemoryBackend) GetKnowledgeBase 

func (b *InMemoryBackend) GetKnowledgeBase(_ context.Context, kbID string) (*KnowledgeBase, error)

GetKnowledgeBase returns a knowledge base.

func (*InMemoryBackend) GetKnowledgeBaseDocuments 

func (b *InMemoryBackend) GetKnowledgeBaseDocuments(
	_ context.Context, kbID, dsID string, docIDs []string,
) ([]KBDocumentDetail, error)

GetKnowledgeBaseDocuments retrieves document details.

func (*InMemoryBackend) GetPrompt 

func (b *InMemoryBackend) GetPrompt(_ context.Context, promptID string) (*Prompt, error)

GetPrompt returns a prompt.

func (*InMemoryBackend) GetPromptVersion 

func (b *InMemoryBackend) GetPromptVersion(
	_ context.Context, promptID, version string,
) (*PromptVersion, error)

GetPromptVersion returns a specific prompt version. See the not-found precedence note on GetAgentVersion in the Agent version CRUD section above -- the same b.prompts.Has(promptID)-instead-of-inner-map-presence reasoning applies here.

func (*InMemoryBackend) IngestKnowledgeBaseDocuments 

func (b *InMemoryBackend) IngestKnowledgeBaseDocuments(
	_ context.Context, kbID, dsID string, docs []KBDocument,
) ([]KBDocumentDetail, error)

IngestKnowledgeBaseDocuments ingests documents into a knowledge base data source.

func (*InMemoryBackend) ListAgentActionGroups 

func (b *InMemoryBackend) ListAgentActionGroups(
	_ context.Context, agentID, agentVersion string, maxResults int, nextToken string,
) ([]*ActionGroupSummary, string, error)

ListAgentActionGroups returns all action groups for an agent version.

func (*InMemoryBackend) ListAgentAliases 

func (b *InMemoryBackend) ListAgentAliases(
	_ context.Context, agentID string, maxResults int, nextToken string,
) ([]*AgentAliasSummary, string, error)

ListAgentAliases returns paginated alias summaries for an agent.

func (*InMemoryBackend) ListAgentCollaborators 

func (b *InMemoryBackend) ListAgentCollaborators(
	_ context.Context, agentID, agentVersion string, maxResults int, nextToken string,
) ([]*AgentCollaborator, string, error)

ListAgentCollaborators returns paginated collaborators.

func (*InMemoryBackend) ListAgentKnowledgeBases 

func (b *InMemoryBackend) ListAgentKnowledgeBases(
	_ context.Context, agentID, agentVersion string, maxResults int, nextToken string,
) ([]*AgentKnowledgeBase, string, error)

ListAgentKnowledgeBases returns paginated agent–KB associations.

func (*InMemoryBackend) ListAgentVersions 

func (b *InMemoryBackend) ListAgentVersions(
	_ context.Context, agentID string, maxResults int, nextToken string,
) ([]*AgentVersionSummary, string, error)

ListAgentVersions returns paginated agent version summaries.

func (*InMemoryBackend) ListAgents 

func (b *InMemoryBackend) ListAgents(
	_ context.Context, maxResults int, nextToken string,
) ([]*AgentSummary, string, error)

ListAgents returns a paginated list of agent summaries.

func (*InMemoryBackend) ListDataSources 

func (b *InMemoryBackend) ListDataSources(
	_ context.Context, kbID string, maxResults int, nextToken string,
) ([]*DataSourceSummary, string, error)

ListDataSources returns paginated data source summaries.

func (*InMemoryBackend) ListFlowAliases 

func (b *InMemoryBackend) ListFlowAliases(
	_ context.Context, flowID string, maxResults int, nextToken string,
) ([]*FlowAliasSummary, string, error)

ListFlowAliases returns paginated flow alias summaries.

func (*InMemoryBackend) ListFlowVersions 

func (b *InMemoryBackend) ListFlowVersions(
	_ context.Context, flowID string, maxResults int, nextToken string,
) ([]*FlowVersionSummary, string, error)

ListFlowVersions returns paginated flow version summaries.

func (*InMemoryBackend) ListFlows 

func (b *InMemoryBackend) ListFlows(
	_ context.Context, maxResults int, nextToken string,
) ([]*FlowSummary, string, error)

ListFlows returns paginated flow summaries.

func (*InMemoryBackend) ListIngestionJobs 

func (b *InMemoryBackend) ListIngestionJobs(
	_ context.Context, kbID, dsID string, maxResults int, nextToken string,
) ([]*IngestionJob, string, error)

ListIngestionJobs returns paginated ingestion job summaries.

func (*InMemoryBackend) ListKnowledgeBaseDocuments 

func (b *InMemoryBackend) ListKnowledgeBaseDocuments(
	_ context.Context, kbID, dsID string, maxResults int, nextToken string,
) ([]KBDocumentDetail, string, error)

ListKnowledgeBaseDocuments returns paginated document details.

func (*InMemoryBackend) ListKnowledgeBases 

func (b *InMemoryBackend) ListKnowledgeBases(
	_ context.Context, maxResults int, nextToken string,
) ([]*KnowledgeBaseSummary, string, error)

ListKnowledgeBases returns paginated knowledge base summaries.

func (*InMemoryBackend) ListPrompts 

func (b *InMemoryBackend) ListPrompts(
	_ context.Context, maxResults int, nextToken string,
) ([]*PromptSummary, string, error)

ListPrompts returns paginated prompt summaries.

func (*InMemoryBackend) ListTagsForResource 

func (b *InMemoryBackend) ListTagsForResource(
	_ context.Context, resourceARN string,
) (map[string]string, error)

ListTagsForResource returns tags for a resource ARN.

func (*InMemoryBackend) PrepareAgent 

func (b *InMemoryBackend) PrepareAgent(_ context.Context, agentID string) (*Agent, error)

PrepareAgent transitions agent to PREPARED status.

func (*InMemoryBackend) PrepareFlow 

func (b *InMemoryBackend) PrepareFlow(_ context.Context, flowID string) (*Flow, error)

PrepareFlow transitions a flow to prepared status.

func (*InMemoryBackend) Reset 

func (b *InMemoryBackend) Reset()

Reset clears all backend state (used in tests).

func (*InMemoryBackend) Restore 

func (b *InMemoryBackend) Restore(ctx context.Context, data []byte) error

Restore deserializes backend state from a snapshot. It implements persistence.Persistable.

func (*InMemoryBackend) Snapshot 

func (b *InMemoryBackend) Snapshot(ctx context.Context) []byte

Snapshot serializes the backend state to JSON. It implements persistence.Persistable.

func (*InMemoryBackend) StartIngestionJob 

func (b *InMemoryBackend) StartIngestionJob(
	_ context.Context, kbID, dsID, description string,
) (*IngestionJob, error)

StartIngestionJob creates and starts a new ingestion job.

func (*InMemoryBackend) StopIngestionJob 

func (b *InMemoryBackend) StopIngestionJob(
	_ context.Context, kbID, dsID, jobID string,
) (*IngestionJob, error)

StopIngestionJob stops an ingestion job.

func (*InMemoryBackend) TagResource 

func (b *InMemoryBackend) TagResource(
	_ context.Context, resourceARN string, tags map[string]string,
) error

TagResource adds or updates tags on a resource.

func (*InMemoryBackend) UntagResource 

func (b *InMemoryBackend) UntagResource(
	_ context.Context, resourceARN string, tagKeys []string,
) error

UntagResource removes tags from a resource.

func (*InMemoryBackend) UpdateAgent 

func (b *InMemoryBackend) UpdateAgent(_ context.Context, agentID string, cfg AgentConfig) (*Agent, error)

UpdateAgent updates an existing agent.

func (*InMemoryBackend) UpdateAgentActionGroup 

func (b *InMemoryBackend) UpdateAgentActionGroup(
	_ context.Context, agentID, agentVersion, actionGroupID string, cfg ActionGroupConfig,
) (*AgentActionGroup, error)

UpdateAgentActionGroup updates an action group.

func (*InMemoryBackend) UpdateAgentAlias 

func (b *InMemoryBackend) UpdateAgentAlias(
	_ context.Context, agentID, aliasID string, cfg AliasConfig,
) (*AgentAlias, error)

UpdateAgentAlias updates an agent alias.

func (*InMemoryBackend) UpdateAgentCollaborator 

func (b *InMemoryBackend) UpdateAgentCollaborator(
	_ context.Context, agentID, agentVersion, collaboratorID string, cfg CollaboratorConfig,
) (*AgentCollaborator, error)

UpdateAgentCollaborator updates a collaborator.

func (*InMemoryBackend) UpdateAgentKnowledgeBase 

func (b *InMemoryBackend) UpdateAgentKnowledgeBase(
	_ context.Context, agentID, agentVersion, kbID, description, kbState string,
) (*AgentKnowledgeBase, error)

UpdateAgentKnowledgeBase updates an agent–KB association.

func (*InMemoryBackend) UpdateDataSource 

func (b *InMemoryBackend) UpdateDataSource(
	_ context.Context, kbID, dsID string, cfg DataSourceConfig,
) (*DataSource, error)

UpdateDataSource updates a data source.

func (*InMemoryBackend) UpdateFlow 

func (b *InMemoryBackend) UpdateFlow(_ context.Context, flowID string, cfg FlowConfig) (*Flow, error)

UpdateFlow updates a flow.

func (*InMemoryBackend) UpdateFlowAlias 

func (b *InMemoryBackend) UpdateFlowAlias(
	_ context.Context, flowID, aliasID string, cfg FlowAliasConfig,
) (*FlowAlias, error)

UpdateFlowAlias updates a flow alias.

func (*InMemoryBackend) UpdateKnowledgeBase 

func (b *InMemoryBackend) UpdateKnowledgeBase(
	_ context.Context, kbID string, cfg KnowledgeBaseConfig,
) (*KnowledgeBase, error)

UpdateKnowledgeBase updates a knowledge base.

func (*InMemoryBackend) UpdatePrompt 

func (b *InMemoryBackend) UpdatePrompt(
	_ context.Context, promptID string, cfg PromptConfig,
) (*Prompt, error)

UpdatePrompt updates a prompt.

func (*InMemoryBackend) ValidateFlowDefinition 

func (b *InMemoryBackend) ValidateFlowDefinition(
	_ context.Context, _ map[string]any,
) ([]FlowValidationError, error)

ValidateFlowDefinition validates a flow definition (stub - always passes).

type IngestionJob 

type IngestionJob struct {
	StartedAt       time.Time `json:"startedAt"`
	UpdatedAt       time.Time `json:"updatedAt"`
	IngestionJobID  string    `json:"ingestionJobId"`
	KnowledgeBaseID string    `json:"knowledgeBaseId"`
	DataSourceID    string    `json:"dataSourceId"`
	Status          string    `json:"status"`
	Description     string    `json:"description,omitempty"`
}

IngestionJob is a knowledge base data ingestion job.

type KBDocument 

type KBDocument struct {
	Metadata map[string]any
	Content  map[string]any
	DocID    string
}

KBDocument is a knowledge base document for ingestion.

type KBDocumentDetail 

type KBDocumentDetail struct {
	DocumentID      string `json:"documentId"`
	KnowledgeBaseID string `json:"knowledgeBaseId"`
	DataSourceID    string `json:"dataSourceId"`
	Status          string `json:"status"`
}

KBDocumentDetail is the status of a knowledge base document operation.

type KnowledgeBase 

type KnowledgeBase struct {
	CreatedAt            time.Time         `json:"createdAt"`
	UpdatedAt            time.Time         `json:"updatedAt"`
	Tags                 map[string]string `json:"tags,omitempty"`
	KBConfiguration      map[string]any    `json:"knowledgeBaseConfiguration,omitempty"`
	StorageConfiguration map[string]any    `json:"storageConfiguration,omitempty"`
	KnowledgeBaseID      string            `json:"knowledgeBaseId"`
	KnowledgeBaseARN     string            `json:"knowledgeBaseArn"`
	Name                 string            `json:"name"`
	Status               string            `json:"status"`
	Description          string            `json:"description,omitempty"`
	RoleARN              string            `json:"roleArn,omitempty"`
}

KnowledgeBase is a Bedrock Knowledge Base.

type KnowledgeBaseConfig 

type KnowledgeBaseConfig struct {
	Tags                 map[string]string
	KBConfiguration      map[string]any
	StorageConfiguration map[string]any
	Name                 string
	Description          string
	RoleARN              string
}

KnowledgeBaseConfig holds fields for creating or updating a KnowledgeBase.

type KnowledgeBaseSummary 

type KnowledgeBaseSummary struct {
	UpdatedAt       time.Time `json:"updatedAt"`
	KnowledgeBaseID string    `json:"knowledgeBaseId"`
	Name            string    `json:"name"`
	Status          string    `json:"status"`
	Description     string    `json:"description,omitempty"`
}

KnowledgeBaseSummary is used in list responses.

type Prompt 

type Prompt struct {
	CreatedAt      time.Time         `json:"createdAt"`
	UpdatedAt      time.Time         `json:"updatedAt"`
	Tags           map[string]string `json:"tags,omitempty"`
	PromptID       string            `json:"id"`
	PromptARN      string            `json:"arn"`
	Name           string            `json:"name"`
	Description    string            `json:"description,omitempty"`
	DefaultVariant string            `json:"defaultVariant,omitempty"`
	Version        string            `json:"version"`
	Variants       []map[string]any  `json:"variants,omitempty"`
}

Prompt is a Bedrock Prompt resource.

type PromptConfig 

type PromptConfig struct {
	Tags           map[string]string
	Name           string
	Description    string
	DefaultVariant string
	Variants       []map[string]any
}

PromptConfig holds fields for creating or updating a Prompt.

type PromptSummary 

type PromptSummary struct {
	CreatedAt   time.Time `json:"createdAt"`
	UpdatedAt   time.Time `json:"updatedAt"`
	PromptID    string    `json:"id"`
	PromptARN   string    `json:"arn"`
	Name        string    `json:"name"`
	Description string    `json:"description,omitempty"`
	Version     string    `json:"version"`
}

PromptSummary is used in list responses.

type PromptVersion 

type PromptVersion struct {
	CreatedAt   time.Time        `json:"createdAt"`
	PromptARN   string           `json:"arn"`
	PromptID    string           `json:"id"`
	Name        string           `json:"name"`
	Version     string           `json:"version"`
	Description string           `json:"description,omitempty"`
	Variants    []map[string]any `json:"variants,omitempty"`
}

PromptVersion is an immutable snapshot of a prompt.

type Provider 

type Provider struct{}

Provider implements service.Provider for the Bedrock Agent service.

func (*Provider) Init 

func (p *Provider) Init(ctx *service.AppContext) (service.Registerable, error)

Init initialises the Bedrock Agent backend and handler.

func (*Provider) Name 

func (p *Provider) Name() string

Name returns the provider name.

type StorageBackend 

type StorageBackend interface {
	// Snapshot and Restore implement persistence.Persistable. Handler
	// delegates to them (see persistence.go) so cli.go's generic
	// setupPersistence picks BedrockAgent up. Before Phase 3.3, BedrockAgent
	// had no persistence at all: neither Handler nor InMemoryBackend
	// implemented Snapshot/Restore, so state never survived a restart. This
	// is the dead-wiring fix in scope for the Phase 3.3 datalayer
	// conversion, built from scratch following the codecommit/cleanrooms
	// pattern (see persistence.go).
	Snapshot(ctx context.Context) []byte
	Restore(ctx context.Context, data []byte) error

	// Agent operations.
	CreateAgent(ctx context.Context, cfg AgentConfig) (*Agent, error)
	GetAgent(ctx context.Context, agentID string) (*Agent, error)
	UpdateAgent(ctx context.Context, agentID string, cfg AgentConfig) (*Agent, error)
	DeleteAgent(ctx context.Context, agentID string) error
	ListAgents(ctx context.Context, maxResults int, nextToken string) ([]*AgentSummary, string, error)
	PrepareAgent(ctx context.Context, agentID string) (*Agent, error)

	// Agent version operations.
	CreateAgentVersion(ctx context.Context, agentID, description string) (*AgentVersion, error)
	GetAgentVersion(ctx context.Context, agentID, agentVersion string) (*AgentVersion, error)
	DeleteAgentVersion(ctx context.Context, agentID, agentVersion string) error
	ListAgentVersions(
		ctx context.Context, agentID string, maxResults int, nextToken string,
	) ([]*AgentVersionSummary, string, error)

	// Agent action group operations.
	CreateAgentActionGroup(
		ctx context.Context, agentID string, cfg ActionGroupConfig,
	) (*AgentActionGroup, error)
	GetAgentActionGroup(
		ctx context.Context, agentID, agentVersion, actionGroupID string,
	) (*AgentActionGroup, error)
	UpdateAgentActionGroup(
		ctx context.Context, agentID, agentVersion, actionGroupID string, cfg ActionGroupConfig,
	) (*AgentActionGroup, error)
	DeleteAgentActionGroup(
		ctx context.Context, agentID, agentVersion, actionGroupID string,
	) error
	ListAgentActionGroups(
		ctx context.Context, agentID, agentVersion string, maxResults int, nextToken string,
	) ([]*ActionGroupSummary, string, error)

	// Agent alias operations.
	CreateAgentAlias(ctx context.Context, agentID string, cfg AliasConfig) (*AgentAlias, error)
	GetAgentAlias(ctx context.Context, agentID, agentAliasID string) (*AgentAlias, error)
	UpdateAgentAlias(ctx context.Context, agentID, agentAliasID string, cfg AliasConfig) (*AgentAlias, error)
	DeleteAgentAlias(ctx context.Context, agentID, agentAliasID string) error
	ListAgentAliases(
		ctx context.Context, agentID string, maxResults int, nextToken string,
	) ([]*AgentAliasSummary, string, error)

	// Agent collaborator operations.
	AssociateAgentCollaborator(
		ctx context.Context, agentID, agentVersion string, cfg CollaboratorConfig,
	) (*AgentCollaborator, error)
	GetAgentCollaborator(
		ctx context.Context, agentID, agentVersion, collaboratorID string,
	) (*AgentCollaborator, error)
	UpdateAgentCollaborator(
		ctx context.Context, agentID, agentVersion, collaboratorID string, cfg CollaboratorConfig,
	) (*AgentCollaborator, error)
	DisassociateAgentCollaborator(
		ctx context.Context, agentID, agentVersion, collaboratorID string,
	) error
	ListAgentCollaborators(
		ctx context.Context, agentID, agentVersion string, maxResults int, nextToken string,
	) ([]*AgentCollaborator, string, error)

	// Knowledge base operations.
	CreateKnowledgeBase(ctx context.Context, cfg KnowledgeBaseConfig) (*KnowledgeBase, error)
	GetKnowledgeBase(ctx context.Context, kbID string) (*KnowledgeBase, error)
	UpdateKnowledgeBase(ctx context.Context, kbID string, cfg KnowledgeBaseConfig) (*KnowledgeBase, error)
	DeleteKnowledgeBase(ctx context.Context, kbID string) error
	ListKnowledgeBases(ctx context.Context, maxResults int, nextToken string) ([]*KnowledgeBaseSummary, string, error)

	// Agent–knowledge base association operations.
	AssociateAgentKnowledgeBase(
		ctx context.Context, agentID, agentVersion, kbID, description, kbState string,
	) (*AgentKnowledgeBase, error)
	GetAgentKnowledgeBase(
		ctx context.Context, agentID, agentVersion, kbID string,
	) (*AgentKnowledgeBase, error)
	UpdateAgentKnowledgeBase(
		ctx context.Context, agentID, agentVersion, kbID, description, kbState string,
	) (*AgentKnowledgeBase, error)
	DisassociateAgentKnowledgeBase(
		ctx context.Context, agentID, agentVersion, kbID string,
	) error
	ListAgentKnowledgeBases(
		ctx context.Context, agentID, agentVersion string, maxResults int, nextToken string,
	) ([]*AgentKnowledgeBase, string, error)

	// Data source operations.
	CreateDataSource(ctx context.Context, kbID string, cfg DataSourceConfig) (*DataSource, error)
	GetDataSource(ctx context.Context, kbID, dataSourceID string) (*DataSource, error)
	UpdateDataSource(ctx context.Context, kbID, dataSourceID string, cfg DataSourceConfig) (*DataSource, error)
	DeleteDataSource(ctx context.Context, kbID, dataSourceID string) error
	ListDataSources(
		ctx context.Context, kbID string, maxResults int, nextToken string,
	) ([]*DataSourceSummary, string, error)

	// Ingestion job operations.
	StartIngestionJob(ctx context.Context, kbID, dataSourceID, description string) (*IngestionJob, error)
	GetIngestionJob(ctx context.Context, kbID, dataSourceID, ingestionJobID string) (*IngestionJob, error)
	StopIngestionJob(ctx context.Context, kbID, dataSourceID, ingestionJobID string) (*IngestionJob, error)
	ListIngestionJobs(
		ctx context.Context, kbID, dataSourceID string, maxResults int, nextToken string,
	) ([]*IngestionJob, string, error)

	// Flow operations.
	CreateFlow(ctx context.Context, cfg FlowConfig) (*Flow, error)
	GetFlow(ctx context.Context, flowID string) (*Flow, error)
	UpdateFlow(ctx context.Context, flowID string, cfg FlowConfig) (*Flow, error)
	DeleteFlow(ctx context.Context, flowID string) error
	ListFlows(ctx context.Context, maxResults int, nextToken string) ([]*FlowSummary, string, error)
	PrepareFlow(ctx context.Context, flowID string) (*Flow, error)
	ValidateFlowDefinition(ctx context.Context, definition map[string]any) ([]FlowValidationError, error)

	// Flow version operations.
	CreateFlowVersion(ctx context.Context, flowID, description string) (*FlowVersion, error)
	GetFlowVersion(ctx context.Context, flowID, flowVersion string) (*FlowVersion, error)
	DeleteFlowVersion(ctx context.Context, flowID, flowVersion string) error
	ListFlowVersions(
		ctx context.Context, flowID string, maxResults int, nextToken string,
	) ([]*FlowVersionSummary, string, error)

	// Flow alias operations.
	CreateFlowAlias(ctx context.Context, flowID string, cfg FlowAliasConfig) (*FlowAlias, error)
	GetFlowAlias(ctx context.Context, flowID, aliasID string) (*FlowAlias, error)
	UpdateFlowAlias(ctx context.Context, flowID, aliasID string, cfg FlowAliasConfig) (*FlowAlias, error)
	DeleteFlowAlias(ctx context.Context, flowID, aliasID string) error
	ListFlowAliases(
		ctx context.Context, flowID string, maxResults int, nextToken string,
	) ([]*FlowAliasSummary, string, error)

	// Prompt operations.
	CreatePrompt(ctx context.Context, cfg PromptConfig) (*Prompt, error)
	GetPrompt(ctx context.Context, promptID string) (*Prompt, error)
	UpdatePrompt(ctx context.Context, promptID string, cfg PromptConfig) (*Prompt, error)
	DeletePrompt(ctx context.Context, promptID string) error
	ListPrompts(ctx context.Context, maxResults int, nextToken string) ([]*PromptSummary, string, error)

	// Prompt version operations.
	CreatePromptVersion(ctx context.Context, promptID, description string) (*PromptVersion, error)
	GetPromptVersion(ctx context.Context, promptID, version string) (*PromptVersion, error)
	DeletePromptVersion(ctx context.Context, promptID, version string) error

	// Knowledge base document operations.
	IngestKnowledgeBaseDocuments(
		ctx context.Context, kbID, dataSourceID string, docs []KBDocument,
	) ([]KBDocumentDetail, error)
	GetKnowledgeBaseDocuments(
		ctx context.Context, kbID, dataSourceID string, docIDs []string,
	) ([]KBDocumentDetail, error)
	DeleteKnowledgeBaseDocuments(
		ctx context.Context, kbID, dataSourceID string, docIDs []string,
	) ([]KBDocumentDetail, error)
	ListKnowledgeBaseDocuments(
		ctx context.Context, kbID, dataSourceID string, maxResults int, nextToken string,
	) ([]KBDocumentDetail, string, error)

	// Tagging operations.
	ListTagsForResource(ctx context.Context, resourceARN string) (map[string]string, error)
	TagResource(ctx context.Context, resourceARN string, tags map[string]string) error
	UntagResource(ctx context.Context, resourceARN string, tagKeys []string) error
}

StorageBackend defines all persistence operations for the Bedrock Agent service.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.