agentruntime

type AgentExecutionResult struct {
	Agent           string
	Model           string
	Steps           int
	ToolCalls       int
	MemoryWrites    int
	EstimatedTokens int
	TokensUsed      int
	TokenSource     string
	Duration        time.Duration
	Output          string
	LastEvent       string
	Events          []string
	StepEvents      []AgentStepEvent
}

type AgentJobStore interface {
	Get(ctx context.Context, name string) (resources.Task, bool, error)
	SetAgentJobInput(ctx context.Context, name string, input map[string]string, agent, messageID string) error
	SetAgentJobResult(ctx context.Context, name string, result *resources.AgentJobResult) error
	GetAgentJobResult(ctx context.Context, name string) (*resources.AgentJobResult, error)
	ClearAgentJobFields(ctx context.Context, name string) error
}

type AgentMessage struct {
	MessageID      string `json:"message_id,omitempty"`
	IdempotencyKey string `json:"idempotency_key,omitempty"`
	TaskID         string `json:"task_id,omitempty"`
	Attempt        int    `json:"attempt,omitempty"`
	System         string `json:"system,omitempty"`
	Namespace      string `json:"namespace,omitempty"`
	FromAgent      string `json:"from_agent,omitempty"`
	ToAgent        string `json:"to_agent,omitempty"`
	BranchID       string `json:"branch_id,omitempty"`
	ParentBranchID string `json:"parent_branch_id,omitempty"`
	Type           string `json:"type,omitempty"`
	Payload        string `json:"payload,omitempty"`
	Timestamp      string `json:"timestamp,omitempty"`
	TraceID        string `json:"trace_id,omitempty"`
	ParentID       string `json:"parent_id,omitempty"`
	DelegateOf     string `json:"delegate_of,omitempty"`
}

type AgentMessageBus interface {
	Publish(context.Context, AgentMessage) (AgentMessage, error)
	Consume(context.Context, AgentMessageSubscription, AgentMessageHandler) error
	Close() error
}

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

type AgentMessageConsumerOptions struct {
	WorkerID            string
	Namespace           string
	RefreshEvery        time.Duration
	DedupeWindow        time.Duration
	ConsumerDelay       time.Duration
	LeaseExtendDuration time.Duration
	Executor            *TaskExecutor
	Tools               ToolResourceLookup
	Roles               AgentRoleLookup
	ToolPermissions     ToolPermissionLookup
	IsolatedToolRuntime ToolRuntime
	WasmToolRuntime     ToolRuntime
	CliToolConfig       CLIToolRuntimeConfig
	SecretResolver      SecretResolver
	McpSessionManager   *McpSessionManager
	McpServerStore      McpServerLookup
	Extensions          Extensions
	Memories            MemoryResourceLookup
	MemoryBackends      *PersistentMemoryBackendRegistry
	ModelEndpoints      resources.ModelEndpointLookup
	ToolApprovals       ToolApprovalUpserter
	TaskApprovals       TaskApprovalUpserter
	Policies            AgentPolicyLookup
	ContextAdapters     ContextAdapterGetter
	KubernetesToolRT    ToolRuntime
	A2AToolRuntime      ToolRuntime
	AgentK8sRuntime     *KubernetesAgentRuntime
	OnStepEvent         func(taskName, namespace string, evt AgentStepEvent)
	DebugLogger         *log.Logger
}

type AgentMessageDelivery interface {
	Message() AgentMessage
	Ack(context.Context) error
	Nack(context.Context, bool) error
	NackWithDelay(context.Context, time.Duration) error
	ExtendLease(context.Context, time.Duration) error
}

type AgentMessageHandler func(context.Context, AgentMessageDelivery) error

type AgentMessageSubscription struct {
	Namespace string
	Agent     string
	Durable   string
}

type AgentPolicyLookup interface {
	List(ctx context.Context) ([]resources.AgentPolicy, error)
}

type AgentRegistry interface {
	List(ctx context.Context) ([]resources.Agent, error)
	Get(ctx context.Context, name string) (resources.Agent, bool, error)
}

type AgentRoleLookup interface {
	Get(ctx context.Context, name string) (resources.AgentRole, bool, error)
}

type AgentStepEvent struct {
	Timestamp           string
	Type                string
	Step                int
	Tool                string
	Message             string
	ErrorCode           string
	ErrorReason         string
	Retryable           *bool
	ToolContractVersion string
	ToolRequestID       string
	ToolAttempt         int
	LatencyMS           int64
	Tokens              int
	InputTokens         int
	OutputTokens        int
	UsageSource         string
	ToolAuthProfile     string
	ToolAuthSecretRef   string
}

type AgentSystemRegistry interface {
	Get(ctx context.Context, name string) (resources.AgentSystem, bool, error)
}

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

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

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

type AnthropicModelGatewayConfig struct {
	APIKey           string
	BaseURL          string
	DefaultModel     string
	AnthropicVersion string
	MaxTokens        int
	Timeout          time.Duration
	HTTPClient       *http.Client
}

type AuditEvent struct {
	Timestamp    string            `json:"timestamp"`
	Component    string            `json:"component,omitempty"`
	Action       string            `json:"action"`
	Outcome      string            `json:"outcome"`
	Namespace    string            `json:"namespace,omitempty"`
	ResourceKind string            `json:"resource_kind,omitempty"`
	ResourceName string            `json:"resource_name,omitempty"`
	Principal    string            `json:"principal,omitempty"`
	Message      string            `json:"message,omitempty"`
	Metadata     map[string]string `json:"metadata,omitempty"`
}

type AuditSink interface {
	RecordAudit(ctx context.Context, event AuditEvent)
}

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

type AuthResult struct {
	Headers map[string]string
	EnvVars map[string]string
	Profile string
}

type AuthorizeResult struct {
	Verdict string
	Reason  string
	Details map[string]string
}

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

type AzureOpenAIModelGatewayConfig struct {
	APIKey            string
	BaseURL           string
	DefaultDeployment string
	APIVersion        string
	Timeout           time.Duration
	HTTPClient        *http.Client
}

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

type BedrockModelGatewayConfig struct {
	Region          string
	AccessKeyID     string
	SecretAccessKey string
	SessionToken    string
	Profile         string
	BaseURL         string
	DefaultModel    string
	MaxTokens       int
	Timeout         time.Duration
}

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

type CLICommandRunner interface {
	Run(ctx context.Context, command string, args []string, stdin string, env []string, dir string) (stdout string, stderr string, exitCode int, err error)
}

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

type CLIToolRuntimeConfig struct {
	AllowedCommands []string
	MaxArgvLength   int
}

type Capability struct {
	ID          string `json:"id"`
	Enabled     bool   `json:"enabled"`
	Description string `json:"description,omitempty"`
	Source      string `json:"source,omitempty"`
}

type CapabilityProvider interface {
	Capabilities(ctx context.Context) CapabilitySnapshot
}

type CapabilitySnapshot struct {
	GeneratedAt  string       `json:"generated_at"`
	Capabilities []Capability `json:"capabilities"`
}

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

type ChatMessage struct {
	Role       string // "system", "user", "assistant", "tool"
	Content    string
	ToolCallID string         // role="tool": the ID of the tool call this result answers
	ToolCalls  []ChatToolCall // role="assistant": tool calls the model made this turn
	IsError    bool           // role="tool": true when this tool result represents a failure
}

type ChatToolCall struct {
	ID           string
	Name         string
	Input        string
	ProviderName string
}

type Chunk struct {
	Index  int    `json:"index"`
	Text   string `json:"text"`
	Offset int    `json:"offset"`
}

type ContainerCommandRunner interface {
	Run(ctx context.Context, binary string, args []string, stdin string, env map[string]string) (stdout string, stderr string, err error)
}

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

type ContainerToolRuntimeConfig struct {
	RuntimeBinary string
	Image         string
	Network       string
	Memory        string
	CPUs          string
	PidsLimit     int
	User          string
	Shell         string
}

type ContextAdapterDeps struct {
	Tools          ToolResourceLookup
	Isolated       ToolRuntime
	Wasm           ToolRuntime
	SecretResolver SecretResolver
	Cli            CLIToolRuntimeConfig
	McpMgr         *McpSessionManager
	McpStore       McpServerLookup
}

type ContextAdapterGetter interface {
	Get(ctx context.Context, key string) (resources.ContextAdapter, bool, error)
}

type ContextAdapterHook interface {
	AdaptContext(ctx context.Context, input map[string]string) (map[string]string, error)
}

type EmbeddingProvider interface {
	Embed(ctx context.Context, texts []string) ([][]float32, error)
	Dimensions() int
}

type EnvSecretResolver struct {
	Prefix string
}

type EvalScoreResult struct {
	Score     *float64
	Pass      *bool
	Reasoning string
	Error     string
}

type EvalScorer struct {
	Gateway ModelGateway
}

type ExecutionEngine interface {
	Execute(ctx context.Context, agent resources.Agent, input map[string]string) (AgentExecutionResult, error)
}

type Extensions struct {
	Metering     MeteringSink
	Audit        AuditSink
	Capabilities CapabilityProvider
}

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

type GRPCDialer interface {
	DialContext(ctx context.Context, target string, opts ...grpc.DialOption) (*grpc.ClientConn, error)
}

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

type GovernedToolApprovalContext struct {
	Getter    func(key string) (resources.ToolApproval, bool, error)
	TaskKey   string
	MessageID string
}

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

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

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

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

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

type KubernetesAgentConfig struct {
	Namespace      string
	ServiceAccount string
	Image          string
	JobTTLSeconds  int32
	DefaultMemory  string
	DefaultCPU     string
	EnvConfigMap   string
	EnvSecretName  string
}

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

type KubernetesJobClient interface {
	CreateJob(ctx context.Context, namespace string, job *batchv1.Job) (*batchv1.Job, error)
	WatchJob(ctx context.Context, namespace, name string) (watch.Interface, error)
	GetJob(ctx context.Context, namespace, name string) (*batchv1.Job, error)
	GetPodLogs(ctx context.Context, namespace, podName string) (string, error)
	ListPods(ctx context.Context, namespace, jobName string) ([]corev1.Pod, error)
	DeleteJob(ctx context.Context, namespace, name string) error
}

type KubernetesSecretClient interface {
	GetSecret(ctx context.Context, namespace, name string) (map[string][]byte, error)
}

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

type KubernetesToolConfig struct {
	Namespace      string
	ServiceAccount string
	DefaultImage   string
	JobTTLSeconds  int32
}

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

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

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

type McpCapabilities struct {
	Tools *McpToolCapability `json:"tools,omitempty"`
}

type McpContent struct {
	Type string `json:"type"`
	Text string `json:"text,omitempty"`
}

type McpInitResult struct {
	ProtocolVersion string          `json:"protocolVersion"`
	ServerInfo      McpServerInfo   `json:"serverInfo"`
	Capabilities    McpCapabilities `json:"capabilities"`
}

type McpServerInfo struct {
	Name    string `json:"name"`
	Version string `json:"version,omitempty"`
}

type McpServerLookup interface {
	Get(ctx context.Context, name string) (resources.McpServer, bool, error)
}

type McpSession struct {
	Transport  McpTransport
	InitResult *McpInitResult
	ServerName string
	// contains filtered or unexported fields
}

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

type McpToolCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type McpToolDefinition struct {
	Name        string         `json:"name"`
	Description string         `json:"description,omitempty"`
	InputSchema map[string]any `json:"inputSchema,omitempty"`
}

type McpToolResult struct {
	Content []McpContent `json:"content"`
	IsError bool         `json:"isError,omitempty"`
}

type McpTransport interface {
	Initialize(ctx context.Context) (*McpInitResult, error)
	ListTools(ctx context.Context) ([]McpToolDefinition, error)
	CallTool(ctx context.Context, name string, arguments map[string]any) (*McpToolResult, error)
	Close() error
}

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

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

type MemoryProviderConfig struct {
	Type           string
	Provider       string
	EmbeddingModel string
	Endpoint       string
	AuthToken      string
	Options        map[string]string
	Embedder       EmbeddingProvider
}

type MemoryProviderFactory func(cfg MemoryProviderConfig) (PersistentMemoryBackend, error)

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

type MemoryResourceLookup interface {
	Get(ctx context.Context, name string) (resources.Memory, bool, error)
}

type MemorySearchResult struct {
	Key   string  `json:"key"`
	Value string  `json:"value"`
	Score float64 `json:"score,omitempty"`
}

type MemoryStore interface {
	Put(key, value string)
	Get(key string) (string, bool)
	Snapshot() map[string]string
}

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

type MeteringEvent struct {
	Timestamp       string            `json:"timestamp"`
	Component       string            `json:"component,omitempty"`
	Type            string            `json:"type"`
	Namespace       string            `json:"namespace,omitempty"`
	Task            string            `json:"task,omitempty"`
	System          string            `json:"system,omitempty"`
	Agent           string            `json:"agent,omitempty"`
	Worker          string            `json:"worker,omitempty"`
	Attempt         int               `json:"attempt,omitempty"`
	MessageID       string            `json:"message_id,omitempty"`
	Status          string            `json:"status,omitempty"`
	TokensUsed      int               `json:"tokens_used,omitempty"`
	TokensEstimated int               `json:"tokens_estimated,omitempty"`
	ToolCalls       int               `json:"tool_calls,omitempty"`
	Metadata        map[string]string `json:"metadata,omitempty"`
}

type MeteringSink interface {
	RecordMetering(ctx context.Context, event MeteringEvent)
}

type MockModelGateway struct{}

type MockToolClient struct{}

type ModelEndpointLookup interface {
	Get(ctx context.Context, name string) (resources.ModelEndpoint, bool, error)
}

type ModelGateway interface {
	Complete(ctx context.Context, req ModelRequest) (ModelResponse, error)
}

type ModelGatewayConfig struct {
	Provider     string
	APIKey       string
	BaseURL      string
	DefaultModel string
	Options      map[string]string
	Timeout      time.Duration
	HTTPClient   *http.Client
	// AllowPrivate permits outbound gateway requests to trusted local/private
	// model endpoints, including loopback, RFC 1918 / ULA, and CGNAT
	// addresses. Only honored when HTTPClient is nil (otherwise the caller is
	// responsible for the supplied client's egress policy).
	AllowPrivate bool
}

type ModelGatewayError struct {
	StatusCode int
	Provider   string
	Message    string
}

type ModelProviderPlugin interface {
	Name() string
	Aliases() []string
	RequiresAPIKey() bool
	BuildGateway(cfg ModelGatewayConfig) (ModelGateway, error)
}

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

type ModelRequest struct {
	Model             string
	ModelRef          string
	FallbackModelRefs []string
	Namespace         string
	Agent             string
	Prompt            string
	Step              int
	Tools             []string
	ToolSchemas       map[string]ToolSchemaInfo
	Context           map[string]string
	Messages          []ChatMessage
	OutputSchema      map[string]any
}

type ModelResponse struct {
	Content   string
	Done      bool
	ToolCalls []ModelToolCall
	Usage     ModelUsage
}

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

type ModelRouterConfig struct {
	Endpoints       ModelEndpointLookup
	Secrets         SecretResourceLookup
	SecretEnvPrefix string
}

type ModelToolCall struct {
	ID           string
	Name         string
	Input        string
	ProviderName string
}

type ModelUsage struct {
	InputTokens  int
	OutputTokens int
	TotalTokens  int
	Source       string
}

type ModuleSource int

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

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

type OCIArtifactPuller interface {
	Pull(ctx context.Context, ref string, destPath string, auth *OCIAuth) error
}

type OCIAuth struct {
	Username string
	Password string
}

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

type OllamaModelGatewayConfig struct {
	BaseURL      string
	DefaultModel string
	Timeout      time.Duration
	HTTPClient   *http.Client
}

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

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

type OpenAIModelGatewayConfig struct {
	APIKey        string
	RequireAPIKey bool
	BaseURL       string
	DefaultModel  string
	Timeout       time.Duration
	HTTPClient    *http.Client
}

type OrlojTaskStore interface {
	Get(ctx context.Context, name string) (resources.Task, bool, error)
	Upsert(ctx context.Context, item resources.Task) (resources.Task, error)
	ListPaged(ctx context.Context, limit, offset int, namespace string) ([]resources.Task, error)
}

type OrlojToolConfig struct {
	ParentNamespace string
	ParentTaskName  string
	CurrentDepth    int
	MaxDepth        int
	MaxChildren     int
}

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

type PersistentMemoryBackend interface {
	Put(ctx context.Context, key, value string) error
	Get(ctx context.Context, key string) (string, bool, error)
	Search(ctx context.Context, query string, topK int) ([]MemorySearchResult, error)
	List(ctx context.Context, prefix string) ([]MemorySearchResult, error)
	Ping(ctx context.Context) error
}

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

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

type PgvectorOptions struct {
	Table     string // table name, default "orloj_memory"
	Dimension int    // vector dimension override; auto-detected when 0
}

type ReActExecutionEngine struct {
	OnStepEvent func(AgentStepEvent)
	// contains filtered or unexported fields
}

type RegistryAuthResolver func(ctx context.Context, resolver SecretResolver, secretRef string) (*registryCredentials, error)

type RetryRequestError struct {
	Delay time.Duration
	Err   error
}

type ReviewRequestPayload struct {
	Content        string `json:"content,omitempty"`
	Feedback       string `json:"feedback,omitempty"`
	PreviousOutput string `json:"previous_output,omitempty"`
	CheckpointID   string `json:"checkpoint_id,omitempty"`
	Cycle          int    `json:"cycle,omitempty"`
	RequestedBy    string `json:"requested_by,omitempty"`
	Supersedes     string `json:"supersedes,omitempty"`
}

type SecretResolver interface {
	Resolve(ctx context.Context, secretRef string) (string, error)
}

type SecretResourceLookup interface {
	Get(ctx context.Context, name string) (resources.Secret, bool, error)
}

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

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

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

type StdioMcpTransportConfig struct {
	Command string
	Args    []string
	Env     []string
	OnClose func() // optional callback invoked after the process is killed
}

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

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

type StreamableHTTPMcpTransportConfig struct {
	Endpoint     string
	Headers      map[string]string
	Client       HTTPDoer
	AllowPrivate bool // permit connections to private/internal IPs
}

type TaskApprovalUpserter interface {
	Upsert(ctx context.Context, item resources.TaskApproval) (resources.TaskApproval, error)
	Get(ctx context.Context, key string) (resources.TaskApproval, bool, error)
}

type TaskExecutor struct {
	OnStepEvent func(AgentStepEvent)
	// contains filtered or unexported fields
}

type TaskStateStore interface {
	Get(ctx context.Context, name string) (resources.Task, bool, error)
	Upsert(ctx context.Context, item resources.Task) (resources.Task, error)
	AppendLog(ctx context.Context, name, message string) error
}

type ToolApprovalRequiredError struct {
	Tool   string
	Input  string
	Reason string
}

type ToolApprovalUpserter interface {
	Upsert(ctx context.Context, item resources.ToolApproval) (resources.ToolApproval, error)
	Get(ctx context.Context, key string) (resources.ToolApproval, bool, error)
}

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

type ToolCallAuthorizer interface {
	Authorize(tool string, spec resources.ToolSpec) (*AuthorizeResult, error)
}

type ToolCapabilityRegistry interface {
	Resolve(tool string) (resources.ToolSpec, bool)
}

type ToolClient = ToolRuntime

type ToolContractExecutor interface {
	Execute(ctx context.Context, req ToolExecutionRequest) (ToolExecutionResponse, error)
}

type ToolError struct {
	Status    string
	Code      string
	Reason    string
	Retryable bool
	Message   string
	Details   map[string]string
	Cause     error
}

type ToolExecutionAuth struct {
	Profile   string   `json:"profile,omitempty"`
	SecretRef string   `json:"secret_ref,omitempty"`
	Scopes    []string `json:"scopes,omitempty"`
}

type ToolExecutionFailure struct {
	Code      string            `json:"code,omitempty"`
	Reason    string            `json:"reason,omitempty"`
	Retryable bool              `json:"retryable"`
	Message   string            `json:"message,omitempty"`
	Details   map[string]string `json:"details,omitempty"`
}

type ToolExecutionOutput struct {
	Result string `json:"result,omitempty"`
}

type ToolExecutionRequest struct {
	ToolContractVersion string                    `json:"tool_contract_version,omitempty"`
	RequestID           string                    `json:"request_id,omitempty"`
	TaskID              string                    `json:"task_id,omitempty"`
	Namespace           string                    `json:"namespace,omitempty"`
	Agent               string                    `json:"agent,omitempty"`
	Tool                ToolExecutionRequestTool  `json:"tool,omitempty"`
	Input               map[string]string         `json:"input,omitempty"`
	InputRaw            string                    `json:"input_raw,omitempty"`
	Runtime             ToolExecutionRuntime      `json:"runtime,omitempty"`
	Auth                ToolExecutionAuth         `json:"auth,omitempty"`
	Trace               ToolExecutionTraceContext `json:"trace,omitempty"`
	Attempt             int                       `json:"attempt,omitempty"`
}

type ToolExecutionRequestTool struct {
	Name         string   `json:"name,omitempty"`
	Operation    string   `json:"operation,omitempty"`
	Capabilities []string `json:"capabilities,omitempty"`
	RiskLevel    string   `json:"risk_level,omitempty"`
}

type ToolExecutionResponse struct {
	ToolContractVersion string                    `json:"tool_contract_version,omitempty"`
	RequestID           string                    `json:"request_id,omitempty"`
	Status              string                    `json:"status,omitempty"`
	Output              ToolExecutionOutput       `json:"output,omitempty"`
	Usage               ToolExecutionUsage        `json:"usage,omitempty"`
	Trace               ToolExecutionTraceContext `json:"trace,omitempty"`
	Error               *ToolExecutionFailure     `json:"error,omitempty"`
}

type ToolExecutionRuntime struct {
	Mode         string `json:"mode,omitempty"`
	TimeoutMS    int    `json:"timeout_ms,omitempty"`
	MaxAttempts  int    `json:"max_attempts,omitempty"`
	Backoff      string `json:"backoff,omitempty"`
	MaxBackoffMS int    `json:"max_backoff_ms,omitempty"`
	Jitter       bool   `json:"jitter,omitempty"`
}

type ToolExecutionTraceContext struct {
	TraceID string `json:"trace_id,omitempty"`
	SpanID  string `json:"span_id,omitempty"`
}

type ToolExecutionUsage struct {
	DurationMS int64 `json:"duration_ms,omitempty"`
	Attempt    int   `json:"attempt,omitempty"`
}

type ToolIsolationBackendFactory func(options ToolIsolationBackendOptions) (ToolRuntime, error)

type ToolIsolationBackendOptions struct {
	Mode                string
	ContainerConfig     ContainerToolRuntimeConfig
	SecretResolver      SecretResolver
	WASMConfig          WASMToolRuntimeConfig
	WASMExecutorFactory WASMToolExecutorFactory
	McpSessionManager   *McpSessionManager
	McpServerStore      McpServerLookup
	KubernetesConfig    KubernetesToolConfig
}

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

type ToolPermissionLookup interface {
	List(ctx context.Context) ([]resources.ToolPermission, error)
}

type ToolResourceLookup interface {
	Get(ctx context.Context, name string) (resources.Tool, bool, error)
}

type ToolRuntime interface {
	Call(ctx context.Context, tool string, input string) (string, error)
}

type ToolSchemaInfo struct {
	Description string
	InputSchema map[string]any
}

type ToolSchemaResolver interface {
	ResolveToolSchemas(toolNames []string) map[string]ToolSchemaInfo
}

type UnsupportedIsolatedToolRuntime struct{}

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

type WASMModuleResolverConfig struct {
	CacheDir       string
	AllowPrivate   bool
	SecretResolver SecretResolver
	OCIPuller      OCIArtifactPuller
}

type WASMToolExecuteRequest struct {
	Namespace       string
	Tool            string
	Input           string
	Capabilities    []string
	RiskLevel       string
	Runtime         WASMToolRuntimeConfig
	AuthProfile     string
	AuthHeaders     map[string]string
	ImagePullSecret string
}

type WASMToolExecuteResponse struct {
	Output string
}

type WASMToolExecutor interface {
	Execute(ctx context.Context, req WASMToolExecuteRequest) (WASMToolExecuteResponse, error)
}

type WASMToolExecutorFactory interface {
	Build(ctx context.Context, cfg WASMToolRuntimeConfig) (WASMToolExecutor, error)
}

type WASMToolModuleReqAuth struct {
	Profile string            `json:"profile,omitempty"`
	Headers map[string]string `json:"headers,omitempty"`
}

type WASMToolModuleReqRuntime struct {
	Entrypoint     string `json:"entrypoint,omitempty"`
	MaxMemoryBytes int64  `json:"max_memory_bytes,omitempty"`
	Fuel           uint64 `json:"fuel,omitempty"`
	EnableWASI     bool   `json:"enable_wasi"`
}

type WASMToolModuleRequest struct {
	ContractVersion string                   `json:"contract_version,omitempty"`
	Namespace       string                   `json:"namespace,omitempty"`
	Tool            string                   `json:"tool,omitempty"`
	Input           string                   `json:"input,omitempty"`
	Capabilities    []string                 `json:"capabilities,omitempty"`
	RiskLevel       string                   `json:"risk_level,omitempty"`
	Runtime         WASMToolModuleReqRuntime `json:"runtime,omitempty"`
	Auth            WASMToolModuleReqAuth    `json:"auth,omitempty"`
}

type WASMToolModuleResponse struct {
	ContractVersion string                `json:"contract_version,omitempty"`
	Status          string                `json:"status,omitempty"`
	Output          string                `json:"output,omitempty"`
	Error           *ToolExecutionFailure `json:"error,omitempty"`
}

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