mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
from typing import Generic, TypeVar
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
from pydantic.generics import GenericModel
|
||
|
|
|
||
|
|
GenericBaseModel = TypeVar("GenericBaseModel", bound=BaseModel)
|
||
|
|
|
||
|
|
|
||
|
|
class CursorPaginatedResults(GenericModel, Generic[GenericBaseModel]):
|
||
|
|
"""
|
||
|
|
Cursor-paginated results
|
||
|
|
Generic must be a Pydantic model
|
||
|
|
"""
|
||
|
|
|
||
|
|
limit: int = Field(..., description="Limit of items to get")
|
||
|
|
has_more: bool = Field(..., description="Whether there are more items available")
|
||
|
|
items: list[GenericBaseModel] = Field(..., description="Items")
|
||
|
|
|
||
|
|
|
||
|
|
class OffsetPaginatedResults(GenericModel, Generic[GenericBaseModel]):
|
||
|
|
"""
|
||
|
|
Offset-paginated results
|
||
|
|
Generic must be a Pydantic model
|
||
|
|
"""
|
||
|
|
|
||
|
|
limit: int = Field(description="Limit of items to get")
|
||
|
|
offset: int = Field(description="Offset from which to retrieve items")
|
||
|
|
total: int = Field(description="Total number of items in result")
|
||
|
|
items: list[GenericBaseModel] = Field(description="Items")
|
||
|
|
|
||
|
|
|
||
|
|
class PaginatedResults(GenericModel, Generic[GenericBaseModel]):
|
||
|
|
"""
|
||
|
|
Paginated results
|
||
|
|
Generic must be a Pydantic model
|
||
|
|
"""
|
||
|
|
|
||
|
|
page: int = Field(description="Current Page")
|
||
|
|
pages: int = Field(description="Total number of pages")
|
||
|
|
per_page: int = Field(description="Number of items per page")
|
||
|
|
total: int = Field(description="Total number of items in result")
|
||
|
|
items: list[GenericBaseModel] = Field(description="Items")
|