def METHOD_NAME(cls):
    METHOD_NAME = super().METHOD_NAME()
    METHOD_NAME.update({
        'top_k': NumberField(
            value_type=int, min_value=1, max_value=1000, default=5, optional=True,
            description='Specifies the number of closest context embeddings to check.'
        ),
    })
    return METHOD_NAME