Combine context.plans and context.plan_functions into a single dictionary

Making context.plans hold a dataclass e.g.
```python
@dataclass
class PlanWithMetadata:
plan_function: PlanGenerator
params:
metadata:
```

prevents having to maintain two dictionaries but also means we can easily extract the serialisable parts of the plan to send.