-@ssi_variable(register, name='choose_cite', patch_response=[add_never_cache_headers])
-def choose_cite_tag(request, book_id=None, tag_ids=None):
- cite = get_or_choose_cite(request, book_id, tag_ids)
- return cite.pk if cite is not None else None
+@register.simple_tag(takes_context=True)
+def choose_cite(context, book_id=None, tag_ids=None):
+ request = context['request']
+ return get_or_choose_cite(request, book_id, tag_ids)