tasks

type Caller interface {
	Call(ctx context.Context, method string, params, result any) error
}

type CancelParams struct {
	TaskID string `json:"taskId"`
}

type Capability struct{}

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

type DetailedTask struct {
	Task
	// InputRequests is set while Status is input_required.
	InputRequests protocol.InputRequests `json:"inputRequests,omitempty"`
	// Result is the stored tool result (including resultType) once completed.
	Result json.RawMessage `json:"result,omitempty"`
	// Error is set once failed.
	Error *protocol.Error `json:"error,omitempty"`
}

type GetParams struct {
	TaskID string `json:"taskId"`
}

type HandlerFor[In any] func(ctx context.Context, tc *Context, in In) (*protocol.CallToolResult, error)

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

type OnInputFunc func(ctx context.Context, requests protocol.InputRequests) (protocol.InputResponses, error)

type Options struct {
	// PollIntervalMs is advertised on every task (default 500).
	PollIntervalMs int64
	// TTL is how long finished tasks stay retrievable (default 5 minutes).
	TTL time.Duration
	// Reject makes task tools fail with -32021 for clients that did not
	// declare the tasks capability on the request, instead of falling back to
	// synchronous execution.
	Reject bool
}

type Status string

type Store interface {
	// Put creates or replaces a task snapshot.
	Put(ctx context.Context, t *DetailedTask) error
	// Get returns a copy of the task, or ErrNotFound.
	Get(ctx context.Context, taskID string) (*DetailedTask, error)
	// Delete removes a task; deleting an unknown task is not an error.
	Delete(ctx context.Context, taskID string) error
}

type Task struct {
	protocol.WithMeta
	TaskID        string `json:"taskId"`
	Status        Status `json:"status"`
	StatusMessage string `json:"statusMessage,omitempty"`
	CreatedAt     string `json:"createdAt"`
	LastUpdatedAt string `json:"lastUpdatedAt"`
	// TTLMs is how long the server retains the task; null means indefinitely.
	// The field is required on the wire.
	TTLMs          *int64 `json:"ttlMs"`
	PollIntervalMs int64  `json:"pollIntervalMs,omitempty"`
}

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

type UpdateParams struct {
	TaskID         string                  `json:"taskId"`
	InputResponses protocol.InputResponses `json:"inputResponses"`
}