diff --git a/models/query.py b/models/query.py
index ed88c932..aaab3776 100644
--- a/models/query.py
+++ b/models/query.py
@@ -9,6 +9,12 @@ class RequestPayload(BaseModel):
     index_name: str
 
 
+class ResponseData(BaseModel):
+    content: str
+    file_url: str
+    page_label: str
+
+
 class ResponsePayload(BaseModel):
     success: bool
-    data: List
+    data: List[ResponseData]