diff --git a/health_rec/api/data.py b/health_rec/api/data.py index 54207ec..300c1fb 100644 --- a/health_rec/api/data.py +++ b/health_rec/api/data.py @@ -198,12 +198,15 @@ class Query(BaseModel): The latitude coordinate of the user. radius : Optional[float] The radius of the search. + rerank : Optional[bool] + Whether to rerank the recommendations. """ query: str latitude: Optional[float] = Field(default=None) longitude: Optional[float] = Field(default=None) radius: Optional[float] = Field(default=None) + rerank: Optional[bool] = Field(default=False) class RefineRequest(BaseModel):