Optimize assistant context by requested themes
This commit is contained in:
@@ -98,6 +98,28 @@ def test_geo_assistant_recognizes_dutch_historical_questions(question: str) -> N
|
||||
assert GeoAssistantService.history_requested(question) is True
|
||||
|
||||
|
||||
def test_geo_assistant_limits_explicit_cross_domain_question_to_requested_themes() -> None:
|
||||
themes = GeoAssistantService.requested_themes(
|
||||
"Geef een profiel met ruimtebeslag, open ruimte, bevolking, bereikbaarheid, voorzieningen en bodem."
|
||||
)
|
||||
|
||||
assert themes == {"space_occupation", "open_space", "population", "accessibility", "services", "soil"}
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("question", "expected"),
|
||||
[
|
||||
("Hoe evolueerden bevolking en bosoppervlakte?", {"population", "forest"}),
|
||||
("Toon wegen, waterlopen en overstromingen.", {"roads", "water", "flood_hazard"}),
|
||||
("Welke bodemtypes en landbouwteelten komen voor?", {"soil", "agriculture"}),
|
||||
("Vat de belangrijkste gebiedsmetingen samen.", None),
|
||||
("Welke officiële bronnen zijn beschikbaar?", None),
|
||||
],
|
||||
)
|
||||
def test_geo_assistant_theme_selection_preserves_general_overviews(question: str, expected: set[str] | None) -> None:
|
||||
assert GeoAssistantService.requested_themes(question) == expected
|
||||
|
||||
|
||||
def test_geo_assistant_discloses_estimated_population_values() -> None:
|
||||
metrics = [
|
||||
AssistantContextMetric(
|
||||
|
||||
Reference in New Issue
Block a user