From c3c61593b3d04aaf1b0ae79293913451d771733f Mon Sep 17 00:00:00 2001
From: Amrit Krishnan <amrit110@gmail.com>
Date: Tue, 12 Nov 2024 13:39:07 -0500
Subject: [PATCH] Add rerank attribute to the Query dataclass

---
 health_rec/api/data.py | 3 +++
 1 file changed, 3 insertions(+)

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):