runtime

type Adapter interface {
	Kind() string
	Execute(ctx *ExecutionContext) (*Result, error)
}

type ArtifactWriter interface {
	WriteArtifact(ctx context.Context, artifact agentcore.AgentRunArtifact) error
}

type EinoAgenticModelFactory struct {
	Model    einomodel.AgenticModel
	Provider string
}

type EinoChatModelFactory struct {
	Model einomodel.ToolCallingChatModel
}

type EinoProviderFactory struct {
	AnthropicAPIKey   string
	AnthropicBaseURL  string
	OpenAIAPIKey      string
	OpenAIBaseURL     string
	OpenRouterAPIKey  string
	OpenRouterBaseURL string
	DefaultProvider   string
	DefaultModel      string
	MaxTokens         int
}

type Event = sdk.Event

type EventSink interface {
	Emit(ctx context.Context, event Event)
}

type ExecutionContext struct {
	ModelCredentials          *modelauth.Manager
	Context                   context.Context
	AppID                     string
	Agent                     *agentcore.Agent
	Run                       *agentcore.AgentRun
	Store                     agentcore.Store
	TargetContext             *host.TargetContext
	WorkspaceLease            *agentcore.WorkspaceLease
	AllowedTools              map[string]bool
	Tools                     *tools.Registry
	WorkspaceManager          tools.WorkspaceManager
	SkillRefs                 []agentcore.SkillRef
	SkillDefinitions          []skills.Definition
	AvailableSkillRefs        []agentcore.SkillRef
	AvailableSkillDefinitions []skills.Definition
	SkillInstructions         string
	SkillPolicy               skills.Policy
	UsesSplitSkills           bool
	StagedSkillRoot           string
	ArtifactWriter            ArtifactWriter
	InteractionBroker         InteractionBroker
	EventSink                 EventSink
	MCPBrokerURL              string
	MCPBrokerToken            string
}

type InteractionBroker interface {
	RequestInteraction(ctx context.Context, interaction agentcore.AgentRunInteraction) error
}

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

type NativeBlock struct {
	Type           string          `json:"type"`
	Text           string          `json:"text,omitempty"`
	ToolCallID     string          `json:"tool_call_id,omitempty"`
	ToolName       string          `json:"tool_name,omitempty"`
	Input          json.RawMessage `json:"input,omitempty"`
	Output         string          `json:"output,omitempty"`
	IsError        bool            `json:"is_error,omitempty"`
	FinishRejected bool            `json:"-"`
}

type NativeConfig struct {
	Reviewer     *TypeSafeReviewer
	ModelFactory NativeModelFactory
	MaxToolSteps int
}

type NativeContextPolicy struct {
	Enabled          bool  `json:"enabled"`
	ContextWindow    int   `json:"context_window,omitempty"`
	InputLimit       int   `json:"input_limit,omitempty"`
	MaxOutputTokens  int   `json:"max_output_tokens,omitempty"`
	TriggerTokens    int   `json:"trigger_tokens,omitempty"`
	KeepRecentTokens int   `json:"keep_recent_tokens,omitempty"`
	UserAnchorTokens int   `json:"user_anchor_tokens,omitempty"`
	SummaryTokens    int   `json:"summary_tokens,omitempty"`
	SafetyTokens     int   `json:"safety_tokens,omitempty"`
	MaxTotalTokens   int64 `json:"max_total_tokens,omitempty"`
}

type NativeMessage struct {
	ProviderState    *NativeProviderState `json:"provider_state,omitempty"`
	Role             string               `json:"role"`
	Content          string               `json:"content,omitempty"`
	ReasoningContent string               `json:"reasoning_content,omitempty"`
	Blocks           []NativeBlock        `json:"blocks,omitempty"`
	ContextSummary   bool                 `json:"context_summary,omitempty"`
	// Provenance is checkpoint metadata; it is not a provider role. Empty is
	// retained for old checkpoints and must not be upgraded to known human.
	Provenance string `json:"provenance,omitempty"`
}

type NativeModel interface {
	Generate(ctx context.Context, req NativeModelRequest) (*NativeModelResponse, error)
}

type NativeModelFactory interface {
	ResolveNativeModel(ctx context.Context, execCtx *ExecutionContext, definitions []tools.Definition) (NativeModel, error)
}

type NativeModelRequest struct {
	SystemPrompt string             `json:"system_prompt,omitempty"`
	Messages     []NativeMessage    `json:"messages"`
	Tools        []tools.Definition `json:"tools,omitempty"`
	Step         int                `json:"step"`
}

type NativeModelResponse struct {
	Message      NativeMessage         `json:"message"`
	Usage        NativeUsage           `json:"usage,omitempty"`
	Continuation *ProviderContinuation `json:"continuation,omitempty"`
	Incomplete   bool                  `json:"incomplete,omitempty"`
}

type NativeModelStream interface {
	Recv() (*NativeModelResponse, error)
	Close()
}

type NativeProviderState struct {
	Provider  string          `json:"provider"`
	Reasoning json.RawMessage `json:"reasoning"`
}

type NativeStreamingModel interface {
	Stream(ctx context.Context, req NativeModelRequest) (NativeModelStream, error)
}

type NativeSummaryModel interface {
	Summarize(context.Context, string, int) (*NativeModelResponse, error)
}

type NativeUsage struct {
	InputTokens           int64 `json:"input_tokens,omitempty"`
	CachedInputTokens     int64 `json:"cached_input_tokens,omitempty"`
	OutputTokens          int64 `json:"output_tokens,omitempty"`
	ReasoningOutputTokens int64 `json:"reasoning_output_tokens,omitempty"`
}

type ProviderCapability struct {
	Protocols                []string `json:"protocols,omitempty"`
	Controls                 []string `json:"controls,omitempty"`
	TranscriptContinuation   bool     `json:"transcript_continuation"`
	LosslessResponseReplay   bool     `json:"lossless_response_replay"`
	AuthModes                []string `json:"auth_modes"`
	RunCredentialsConfigured bool     `json:"run_credentials_configured"`
	Name                     string   `json:"name"`
	Configured               bool     `json:"configured"`
	DefaultModel             string   `json:"default_model"`
	BaseURLOverridden        bool     `json:"base_url_overridden"`
}

type ProviderContinuation struct {
	Provider           string `json:"provider,omitempty"`
	ResponseID         string `json:"response_id,omitempty"`
	PreviousResponseID string `json:"previous_response_id,omitempty"`
	AfterSequenceNo    int    `json:"after_sequence_no,omitempty"`
}

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

type Result struct {
	AssistantMessage      string
	AssistantMessageID    string
	OutputSummary         json.RawMessage
	WaitForApproval       bool
	AwaitingInput         bool
	AwaitingAuth          bool
	MessagesPersisted     bool
	TurnFinished          bool
	TurnOutcome           string
	CompletionCorrections int
}

type TypeSafeReviewer struct {
	APIKey              string
	Model               string
	EvaluatedModel      string
	AskedThreshold      float64
	HazardThreshold     float64
	EscalationThreshold float64
	AutoApprove         bool
	Client              *http.Client
}