Update main.py

main.py

@@ -30,6 +30,25 @@ class PostRequest(BaseModel):
         extra = Extra.allow
 
 
+class FileRequest(BaseModel):
+    size: int = (100,)
+    parent_id: Optional[str] = (None,)
+    next_page_token: Optional[str] = (None,)
+    additional_filters: Optional[Dict[str, Any]] = (None,)
+
+    class Config:
+        extra = Extra.allow
+
+
+class OfflineRequest(BaseModel):
+    file_url: str
+    parent_id: Optional[str] = None
+    name: Optional[str] = None
+
+    class Config:
+        extra = Extra.allow
+
+
 security = HTTPBearer()
 # SECRET_TOKEN = "SECRET_TOKEN"
 SECRET_TOKEN = os.getenv("SECRET_TOKEN")
@@ -121,22 +140,17 @@ async def home(request: Request):
 
 
 @app.get("/files")
-async def get_files(
-    size: int = 100,
-    parent_id: Optional[str] = None,
-    next_page_token: Optional[str] = None,
-    additional_filters: Optional[Dict[str, Any]] = None,
-):
+async def get_files(item: FileRequest):
     return await THUNDERX_CLIENT.file_list(
-        size, parent_id, next_page_token, additional_filters
+        item.size, item.parent_id, item.next_page_token, item.additional_filters
     )
 
 
 @app.get("/offline")
-async def offline(
-    file_url: str, parent_id: Optional[str] = None, name: Optional[str] = None
-):
-    return await THUNDERX_CLIENT.offline_download(file_url, parent_id, name)
+async def offline(item: OfflineRequest):
+    return await THUNDERX_CLIENT.offline_download(
+        item.file_url, item.parent_id, item.name
+    )
 
 
 @app.get("/userinfo")