from forestadmin.datasource_toolkit.context.agent_context import AgentCustomizationContext
from forestadmin.datasource_toolkit.decorators.chart.result_builder import ResultBuilder
from forestadmin.datasource_toolkit.interfaces.chart import Chart
async def books_by_author_country_chart_fn(
context: AgentCustomizationContext, result_builder: ResultBuilder
) -> Chart:
return result_builder.distribution({
"validated": 100,
"rejected": 100,
"to_validate": 100,
})
agent.add_chart("booksByAuthorCountry", books_by_author_country_chart_fn)