inference/completion

This commit is contained in:
Xi Yan 2024-09-11 12:32:12 -07:00
parent a7be58e4e1
commit 8b558336b4

View file

@ -170,7 +170,7 @@ class Inference(Protocol):
@webmethod(route="/inference/completion") @webmethod(route="/inference/completion")
async def completion( async def completion(
self, self,
model: str model: str,
content: InterleavedTextMedia, content: InterleavedTextMedia,
sampling_params: Optional[SamplingParams] = SamplingParams(), sampling_params: Optional[SamplingParams] = SamplingParams(),
stream: Optional[bool] = False, stream: Optional[bool] = False,