bedrockagentcore

type AccessToken struct {
	Expires  time.Time `json:"expires"`
	Token    string    `json:"token"`
	Account  string    `json:"account"`
	Region   string    `json:"region"`
	Workload string    `json:"workload"`
	UserID   string    `json:"userId,omitempty"`
}

type ConfigProvider interface {
	GetAgentCoreSettings() Settings
	GetAgentCoreEndpointURL() string
}

type ContainerRuntime interface {
	CreateAndStart(ctx context.Context, spec container.Spec) (string, error)
	StopAndRemove(ctx context.Context, containerID string) error
	Exec(ctx context.Context, containerID string, cmd []string) (container.ExecResult, error)
}

type Control interface {
	Runtime(sc Scope, ref, qualifier string) (RuntimeInfo, error)
	Memory(sc Scope, id string) (MemoryInfo, bool)
	Tool(sc Scope, kind ToolKind, id string) (ToolInfo, bool)
	WorkloadIdentity(sc Scope, name string) bool
	Provider(sc Scope, kind ProviderKind, name string) (ProviderInfo, bool)
	SecretValue(arn, jsonKey string) (string, bool)
}

type EngineConfig struct {
	Runtime ContainerRuntime
	// Ports hands out host ports for runtime containers; nil picks free OS ports.
	Ports *portalloc.Allocator
	// HTTP reaches agent containers; nil uses an isolated client.
	HTTP HTTPDoer
	// Host is the address published container ports are reached on; default 127.0.0.1.
	Host string
	// EndpointURL is the gopherstack URL as seen from containers (AWS_ENDPOINT_URL).
	EndpointURL string
	// RegistryHost, when set, replaces the host of ECR image URIs so images are pulled from the emulated registry.
	RegistryHost string
	// PythonImage is the code interpreter sandbox image.
	PythonImage string
	// StartTimeout bounds boot until a runtime container answers its health check.
	StartTimeout time.Duration
	PollInterval time.Duration
	StopTimeout  time.Duration
	// IdleTimeout applies to runtimes that set no idleRuntimeSessionTimeout.
	IdleTimeout time.Duration
}

type Event struct {
	Timestamp   time.Time      `json:"timestamp"`
	Metadata    map[string]any `json:"metadata,omitempty"`
	Branch      map[string]any `json:"branch,omitempty"`
	Account     string         `json:"account"`
	Region      string         `json:"region"`
	MemoryID    string         `json:"memoryId"`
	ActorID     string         `json:"actorId"`
	SessionID   string         `json:"sessionId"`
	EventID     string         `json:"eventId"`
	ClientToken string         `json:"clientToken,omitempty"`
	Payload     []any          `json:"payload"`
	Seq         uint64         `json:"seq"`
}

type HTTPDoer interface {
	Do(req *http.Request) (*http.Response, error)
}

type Handler struct {
	Backend *InMemoryBackend

	DefaultRegion string
	// contains filtered or unexported fields
}

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

type MemoryInfo struct {
	ID         string
	Strategies []StrategyInfo
	ExpiryDays int
}

type MemoryRecord struct {
	Created    time.Time      `json:"created"`
	Metadata   map[string]any `json:"metadata,omitempty"`
	Account    string         `json:"account"`
	Region     string         `json:"region"`
	MemoryID   string         `json:"memoryId"`
	ID         string         `json:"id"`
	StrategyID string         `json:"strategyId,omitempty"`
	Text       string         `json:"text"`
	Namespaces []string       `json:"namespaces"`
	Seq        uint64         `json:"seq"`
}

type Provider struct{}

type ProviderInfo struct {
	Config       map[string]any
	Name         string
	Vendor       string
	SecretARN    string
	SecretJSONID string
}

type ProviderKind string

type RuntimeInfo struct {
	Env          map[string]string
	ID           string
	ImageURI     string
	Protocol     string
	IdleSeconds  int
	MaxLifetimeS int
}

type Scope struct{ Account, Region string }

type Settings struct {
	Engine string `` //nolint:lll // config struct tags are intentionally verbose
	/* 218-byte string literal not displayed */
	RegistryHost string `` //nolint:lll // config struct tags are intentionally verbose
	/* 230-byte string literal not displayed */
	PythonImage string `` //nolint:lll // config struct tags are intentionally verbose
	/* 170-byte string literal not displayed */
	IdleTimeout time.Duration `` //nolint:lll // config struct tags are intentionally verbose
	/* 215-byte string literal not displayed */
}

type StrategyInfo struct{ ID, Name, Type string }

type ToolInfo struct {
	ID   string
	Mode string
}

type ToolKind string

type ToolSession struct {
	Created    time.Time `json:"created"`
	Updated    time.Time `json:"updated"`
	LastActive time.Time `json:"lastActive"`
	Config     doc       `json:"config,omitempty"`
	Account    string    `json:"account"`
	Region     string    `json:"region"`
	Kind       ToolKind  `json:"kind"`
	ToolID     string    `json:"toolId"`
	ID         string    `json:"id"`
	Name       string    `json:"name,omitempty"`
	Status     string    `json:"status"`
	Token      string    `json:"clientToken,omitempty"`
	TimeoutSec int       `json:"timeoutSec"`
}