Remove unused models.
This commit is contained in:
@@ -84,36 +84,3 @@ class SummarizationRequest(BaseModel):
|
||||
raise ValueError(f"text exceeds maximum length of {max_length}")
|
||||
return v
|
||||
|
||||
|
||||
# ==================== Response Models ====================
|
||||
class EntityOutput(BaseModel):
|
||||
entity: str
|
||||
span: str
|
||||
start: int
|
||||
end: int
|
||||
score: float
|
||||
|
||||
|
||||
class RelationOutput(BaseModel):
|
||||
source: EntityOutput
|
||||
relation: str
|
||||
target: EntityOutput
|
||||
score: float
|
||||
|
||||
|
||||
class SummaryOutput(BaseModel):
|
||||
text: str
|
||||
start: int
|
||||
end: int
|
||||
score: float
|
||||
|
||||
# ==================== Error Response Model ====================
|
||||
class ErrorDetail(BaseModel):
|
||||
code: str
|
||||
message: str
|
||||
details: Dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
|
||||
class ErrorResponse(BaseModel):
|
||||
error: ErrorDetail
|
||||
|
||||
|
||||
Reference in New Issue
Block a user