23 lines
554 B
Python
23 lines
554 B
Python
from pydantic import BaseModel
|
|
from typing import Optional, List, Any
|
|
from datetime import datetime
|
|
|
|
class ScanRequest(BaseModel):
|
|
image: str # base64
|
|
|
|
class DocumentResponse(BaseModel):
|
|
id: int
|
|
title: Optional[str]
|
|
category: Optional[str]
|
|
extracted_text: Optional[str]
|
|
summary: Optional[str]
|
|
extracted_data: Optional[Any]
|
|
risk_alerts: Optional[Any]
|
|
tags: Optional[Any]
|
|
file_size: Optional[int]
|
|
created_at: datetime
|
|
|
|
class DocumentListResponse(BaseModel):
|
|
documents: List[DocumentResponse]
|
|
total: int
|