api

type AbortRequest struct {
	SessionID string
	Reason    string
}

type AbortResponse struct{}

type AgentEvent struct {
	Type            AgentEventType
	SessionID       string
	DriverSessionID string
	State           AgentState
	Turn            int
	Message         *AgentMessage
	Assistant       *ai.AssistantMessage
	ToolName        string
	ToolCallID      string
	ToolResult      *AgentToolResult
	Delta           string
	ControlMessage  string
	ErrorMessage    string
	At              time.Time
	Metadata        map[string]any
}

type AgentEventType string

type AgentMessage struct {
	Turn      int
	Message   ai.Message
	CreatedAt time.Time
}

type AgentMessageRecord struct {
	Turn      int
	Role      string
	Content   json.RawMessage
	CreatedAt time.Time
}

type AgentService interface {
	CreateSession(ctx context.Context, req *CreateAgentSessionRequest) (*CreateAgentSessionResponse, error)
	GetSession(ctx context.Context, req *GetAgentSessionRequest) (*GetAgentSessionResponse, error)
	ListSessions(ctx context.Context, req *ListAgentSessionsRequest) (*ListAgentSessionsResponse, error)
	SendMessage(ctx context.Context, req *SendMessageRequest) (EventReceiver, error)
	Continue(ctx context.Context, req *ContinueRequest) (EventReceiver, error)
	Steer(ctx context.Context, req *SteerRequest) (*SteerResponse, error)
	FollowUp(ctx context.Context, req *FollowUpRequest) (*FollowUpResponse, error)
	Abort(ctx context.Context, req *AbortRequest) (*AbortResponse, error)
	SubscribeEvents(ctx context.Context, req *SubscribeEventsRequest) (EventReceiver, error)
	ResumeSession(ctx context.Context, req *ResumeSessionRequest) (*ResumeSessionResponse, error)
	DestroySession(ctx context.Context, req *DestroyAgentSessionRequest) (*DestroyAgentSessionResponse, error)
	Close() error
}

type AgentSessionSummary struct {
	SessionID string
	State     string
	CreatedAt time.Time
}

type AgentState string

type AgentToolResult struct {
	Content    []ai.ContentBlock
	SnapshotID string
	ExitCode   *int
	IsError    bool
	Metadata   map[string]any
}

type ContinueRequest struct {
	SessionID string
}

type CreateAgentSessionRequest struct {
	SessionConfig
}

type CreateAgentSessionResponse struct {
	SessionID string
	State     string
}

type DestroyAgentSessionRequest struct {
	SessionID string
}

type DestroyAgentSessionResponse struct{}

type EventReceiver interface {
	Recv() (*AgentEvent, error)
	Close() error
}

type FollowUpRequest struct {
	SessionID string
	Message   string
}

type FollowUpResponse struct{}

type GetAgentSessionRequest struct {
	SessionID string
}

type GetAgentSessionResponse struct {
	SessionID       string
	State           string
	Metrics         SessionMetrics
	ConversationLen int
}

type ListAgentSessionsRequest struct {
	Labels map[string]string
}

type ListAgentSessionsResponse struct {
	Sessions []AgentSessionSummary
}

type ResumeSessionRequest struct {
	SessionConfig

	SchemaVersion   int
	ConversationLog []AgentMessageRecord
}

type ResumeSessionResponse struct {
	SessionID       string
	State           string
	ConversationLen int
	Warnings        []string
}

type SendMessageRequest struct {
	SessionID string
	Message   string
}

type SessionConfig struct {
	SessionID    string
	Driver       string
	Model        string
	Provider     string
	SystemPrompt string
	Tools        []string
	Metadata     map[string]any

	ToolEnvironment ToolEnvironmentConfig
}

type SessionMetrics struct {
	TurnsStarted      uint64
	TurnsCompleted    uint64
	MessagesAppended  uint64
	ToolCallsStarted  uint64
	ToolCallsFinished uint64
	Errors            uint64
}

type SteerRequest struct {
	SessionID string
	Message   string
}

type SteerResponse struct{}

type SubscribeEventsRequest struct {
	SessionID string
}

type ToolEnvironmentConfig struct {
	Type ToolEnvironmentType

	SandboxHostAddr  string
	SandboxSessionID string

	LocalRootDir string
}

type ToolEnvironmentType string