agentrun

type ArtifactBaseline struct {
	Ref string
}

type AttachmentIngestor interface {
	IngestAttachments(ctx context.Context, req *agentrundomain.RunRequest)
}

type EventContext struct {
	OrgID uint
	// TODO: 梳理 WorkerID、AssistantID、AssistantPKID 的使用场景，收敛和清晰化
	WorkerID          uint
	SessionID         string
	AssistantID       string
	AssistantPKID     uint
	TraceID           string
	RequestID         string
	TaskID            string
	RunID             string
	ParentID          string
	ReplyToMessageIDs []string
	ClientIP          string
}

type Finalization struct {
	Result *agentrundomain.RunResult
	Events []messaging.RunEventBody
}

type Finalizer interface {
	FinalizeRequired(
		ctx context.Context,
		run *PreparedRun,
		runtimeResult *agent.ExecutionResult,
		snapshot JournalSnapshot,
	) (*Finalization, error)
	PostRunBestEffort(
		ctx context.Context,
		run *PreparedRun,
		result *agentrundomain.RunResult,
		snapshot JournalSnapshot,
	)
}

type Journal interface {
	Record(ctx context.Context, event RunEventDraft) error
	Snapshot() JournalSnapshot
}

type JournalFactory interface {
	New(
		req *agentrundomain.RunRequest,
		eventContext EventContext,
		publisher RunEventPublisher,
	) Journal
}

type JournalSnapshot struct {
	ToolCalls    []agentrundomain.ToolCallRecord
	Usage        *agentrundomain.Usage
	MessageCount int
	ToolFailures int
	ToolNames    []string
	Events       []messaging.RunEventRecord
}

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

type PlanPublishError struct {
	Phase string
	Err   error
}

type PlanPublisher interface {
	Publish(ctx context.Context, event agent.NodeEvent) (*messaging.RunEventBody, error)
}

type PlanPublisherConfig struct {
	ServerAddr string
	OrgID      uint
	AuthToken  string
	HTTPClient *http.Client
}

type PreparedRun struct {
	Request   *agentrundomain.RunRequest
	Execution agent.ExecutionRequest
	Workspace WorkspacePreparation
	Baseline  ArtifactBaseline
}

type Preparer interface {
	Prepare(ctx context.Context, req *agentrundomain.RunRequest) (*PreparedRun, func(), error)
}

type ProviderSessionBinding struct {
	InternalSessionID string
	Provider          string
	ProviderSessionID string
	WorkDir           string
	AssistantID       string
	Status            string
	LastError         string
}

type ProviderSessionKey struct {
	InternalSessionID string
	Provider          string
	WorkDir           string
	AssistantID       string
}

type ProviderSessionStore interface {
	GetProviderSession(ctx context.Context, key ProviderSessionKey) (*ProviderSessionBinding, error)
	UpsertProviderSession(ctx context.Context, binding *ProviderSessionBinding) error
	MarkProviderSessionFailed(ctx context.Context, key ProviderSessionKey, reason string) error
}

type RunEventDraft struct {
	OccurredAt time.Time
	Body       messaging.RunEventBody
}

type RunEventPublisher interface {
	PublishRunEvent(ctx context.Context, event messaging.RunEvent) error
}

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

type ToolProvider interface {
	ToolsFor(
		req *agentrundomain.RunRequest,
		workspace WorkspacePreparation,
	) ([]agent.Tool, error)
}

type WorkspaceManager interface {
	PrepareWorkspace(
		ctx context.Context,
		req agentworkspace.TaskWorkspaceRequest,
	) (WorkspacePreparation, error)
}

type WorkspacePreparation struct {
	WorkDir              string
	RepoDir              string
	TaskDir              string
	ArtifactManifestPath string
	PreRunTreeSHA        string // Git tree SHA captured before agent execution
}