Function list_model_schemas

backend/app/services/model/model_schema.py:143–174

List model schemas. Only one in `offset`, `after` and `before` can be used at the same time. :param limit: the maximum number of model schemas to return. :param offset: the offset of model schemas to return. :param provider_id: the provider id to filter by. :param type: the

(
    limit: int,
    offset: Optional[int],
    provider_id: Optional[str],
    type: Optional[ModelType],
)

async def list_model_schemas(
    limit: int,
    offset: Optional[int],
    provider_id: Optional[str],
    type: Optional[ModelType],
) -> Tuple[List[ModelSchema], int, bool]:
    """
    List model schemas.
    Only one in `offset`, `after` and `before` can be used at the same time.

    :param limit: the maximum number of model schemas to return.
    :param offset: the offset of model schemas to return.
    :param provider_id: the provider id to filter by.
    :param type: the model type to filter by.
    :return: a tuple of (model schemas, total count, has more)
    """

    # Filter by provider_id and type
    filtered_schemas = [
        schema
        for schema in _model_schemas
        if (provider_id is None or schema.provider_id == provider_id) and (type is None or schema.type == type.value or schema.type == ModelType.WILDCARD.value)
    ]

    # Paginate
    end_index = offset + limit
    page = filtered_schemas[offset:end_index]

    # Check if there's more
    has_more = end_index < len(filtered_schemas)

    return page, len(filtered_schemas), has_more


def get_provider(provider_id: str) -> Optional[Provider]: