Currently, there's no way to register an entity with a specific name, it automatically just defaults to the name of the class. I know there's a hacky way around it by changing name, but it would be so much cleaner to have a parameter for it (like they do in dotnet sdk).
Suggestion -
def add_entity(self, fn: task.Entity, entity_name: str = None) -> str: