1 files changed, 28 insertions, 1 deletions
diff --git a/ATRI/plugins/setu/models.py b/ATRI/plugins/setu/models.py
index 7144f27..48459f0 100644
--- a/ATRI/plugins/setu/models.py
+++ b/ATRI/plugins/setu/models.py
@@ -1,6 +1,33 @@
+from typing import List
 from pydantic import BaseModel
 
 
+class Urls(BaseModel):
+    original: str
+
+
+class Datum(BaseModel):
+    pid: int
+    p: int
+    uid: int
+    title: str
+    author: str
+    r18: bool
+    width: int
+    height: int
+    tags: List[str]
+    ext: str
+    aiType: int
+    uploadDate: int
+    urls: Urls
+
+
+class LoliconResponse(BaseModel):
+    error: str
+    data: List[Datum]
+
+
 class SetuInfo(BaseModel):
     title: str
-    pid: str
+    pid: int
+    url: str