Skip to content

vllm.entrypoints.pooling

Modules:

Name Description
classify
embed
pooling
score

register_pooling_api_routers

register_pooling_api_routers(app: FastAPI)
Source code in vllm/entrypoints/pooling/__init__.py
def register_pooling_api_routers(app: FastAPI):
    from vllm.entrypoints.pooling.classify.api_router import router as classify_router
    from vllm.entrypoints.pooling.embed.api_router import router as embed_router
    from vllm.entrypoints.pooling.pooling.api_router import router as pooling_router
    from vllm.entrypoints.pooling.score.api_router import router as score_router

    app.include_router(classify_router)
    app.include_router(embed_router)
    app.include_router(score_router)
    app.include_router(pooling_router)