Fix query response model

superagent-ai · Jan 16, 2024 · 9ce12ed · 9ce12ed
1 parent 67adf00
commit 9ce12ed
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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]