calltask

type Ending struct {
	Status  string
	Message string
	Result  json.RawMessage
	Error   json.RawMessage
}

type Options struct {
	Store Store
	// Retention is how long a task is kept after it is created; zero is
	// DefaultRetention.
	Retention time.Duration
	// Heartbeat is how often running calls are stamped; zero is
	// DefaultHeartbeat.
	Heartbeat time.Duration
	// PollInterval is the polling interval a client is handed; zero is
	// DefaultPollInterval.
	PollInterval time.Duration
	// Now is the clock; nil is time.Now.
	Now func() time.Time
}

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

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

type Store interface {
	// Create inserts a working task.
	Create(ctx context.Context, t Task) error
	// Get reads one task, or errTaskNotFound.
	Get(ctx context.Context, id string) (*Task, error)
	// Finish ends a working task; a task that already ended is left as it
	// is.
	Finish(ctx context.Context, id string, end Ending) error
	// RequestCancel records a request to stop a working task.
	RequestCancel(ctx context.Context, id string) error
	// Heartbeat stamps the working tasks ids name as alive, and returns the
	// ones among them a cancel was requested for.
	Heartbeat(ctx context.Context, ids []string) ([]string, error)
	// FailStale fails a working task whose heartbeat is older than before,
	// reporting whether it did.
	FailStale(ctx context.Context, id string, before time.Time, message string, rpcErr json.RawMessage) (bool, error)
}

type Task struct {
	ID            string
	ToolName      string
	UserID        string
	Status        string
	StatusMessage string
	// Result is the tool's result in its wire form, on a completed task.
	Result json.RawMessage
	// Error is the JSON-RPC error, on a failed task.
	Error           json.RawMessage
	CreatedAt       time.Time
	UpdatedAt       time.Time
	HeartbeatAt     time.Time
	CancelRequested bool
	ExpiresAt       time.Time
}