[Serve] Support generics for DeploymentHandle type hints #52654
Description
Description
It would be great to support generics for DeploymentHandle type hints so linters can catch incorrect remote call args / response handling.
Use case
It would be great to support generics for DeploymentHandle type hints so linters can catch incorrect remote call args / response handling.
For example take this code:
@serve.deployment
@serve.ingress(fastapi_app)
class DeploymentA:
def __init__(
self, deployment_b: DeploymentHandle, deployment_c: DeploymentHandle
) -> None:
self.deployment_b = deployment_b
self.deployment_c = deployment_cIt would be nice if it can be specified more granularly as:
@serve.deployment
@serve.ingress(fastapi_app)
class DeploymentA:
def __init__(
self, deployment_b: DeploymentHandle[B], deployment_c: DeploymentHandle[C]
) -> None:
self.deployment_b = deployment_b
self.deployment_c = deployment_c