Create schemas.py

src/schemas.py

@@ -0,0 +1,25 @@
+from typing import TypedDict, Dict, Any
+
+class StorageUsage(TypedDict):
+    bytes_used: int
+    metadata_bytes_used: int
+
+
+class UploadResult(TypedDict):
+    id: str
+    metadata: Dict[str, Any] # NOTE: Should not need this field
+    storage_usage: StorageUsage
+
+class VectaraDocument(TypedDict):
+    id: str
+    metadata: Dict[str, Any]
+    # tables: list[Dict[str, Any]]
+    # parts: list[Dict[str, Any]] # NOTE: Only available when by list ID
+    # storage_usage: StorageUsage
+    # extraction_usage: Dict[str, Any]
+
+
+class VectaraDocuments(TypedDict):
+    documents: list[VectaraDocument]
+    # metadata: Dict[str, Any]
+    # page_key: str