From 794355463138d5750b7f997e655fc9826ec62111 Mon Sep 17 00:00:00 2001 From: Max Chis Date: Fri, 26 Sep 2025 20:24:18 -0400 Subject: [PATCH] Add additional attributes for agency search --- src/api/endpoints/search/agency/models/response.py | 4 ++++ src/api/endpoints/search/agency/query.py | 2 ++ src/db/models/impl/agency/sqlalchemy.py | 6 +++++- 3 files changed, 11 insertions(+), 1 deletion(-) diff --git a/src/api/endpoints/search/agency/models/response.py b/src/api/endpoints/search/agency/models/response.py index c7ed4460..1b6b82d5 100644 --- a/src/api/endpoints/search/agency/models/response.py +++ b/src/api/endpoints/search/agency/models/response.py @@ -1,7 +1,11 @@ from pydantic import BaseModel +from src.db.models.impl.agency.enums import AgencyType, JurisdictionType + class AgencySearchResponse(BaseModel): agency_id: int agency_name: str + jurisdiction_type: JurisdictionType | None + agency_type: AgencyType location_display_name: str diff --git a/src/api/endpoints/search/agency/query.py b/src/api/endpoints/search/agency/query.py index 8169c9a7..9476e039 100644 --- a/src/api/endpoints/search/agency/query.py +++ b/src/api/endpoints/search/agency/query.py @@ -32,6 +32,8 @@ async def run(self, session: AsyncSession) -> list[AgencySearchResponse]: select( Agency.agency_id, Agency.name.label("agency_name"), + Agency.jurisdiction_type, + Agency.agency_type, LocationExpandedView.display_name.label("location_display_name") ) ) diff --git a/src/db/models/impl/agency/sqlalchemy.py b/src/db/models/impl/agency/sqlalchemy.py index 20cd5f12..c8a19a56 100644 --- a/src/db/models/impl/agency/sqlalchemy.py +++ b/src/db/models/impl/agency/sqlalchemy.py @@ -26,7 +26,11 @@ class Agency( county = Column(String, nullable=True) locality = Column(String, nullable=True) agency_type = enum_column(AgencyType, name="agency_type_enum") - jurisdiction_type = enum_column(JurisdictionType, name="jurisdiction_type_enum") + jurisdiction_type = enum_column( + JurisdictionType, + name="jurisdiction_type_enum", + nullable=True, + ) # Relationships automated_suggestions = relationship("AgencyIDSubtaskSuggestion")