From e20f4cb0feb2e71eb0b060acb00b0f7580b46d1b Mon Sep 17 00:00:00 2001 From: Nitzan Shwartz <83908750+NitzanShwartz@users.noreply.github.com> Date: Sat, 18 Jan 2025 22:42:18 +0200 Subject: [PATCH] docs: correct types for example (#5188) --- docs/source/build-with-bentoml/iotypes.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/build-with-bentoml/iotypes.rst b/docs/source/build-with-bentoml/iotypes.rst index c6fcdedea86..e6da8829dec 100644 --- a/docs/source/build-with-bentoml/iotypes.rst +++ b/docs/source/build-with-bentoml/iotypes.rst @@ -38,7 +38,7 @@ Python's standard types such as strings, integers, floats, booleans, lists, and class LanguageModel: @bentoml.api def generate( - self, prompt: int = Field(description="The prompt text"), + self, prompt: str = Field(description="The prompt text"), temperature: float = Field(default=0.0, description="A sampling temperature between 0 and 2"), max_tokens: int = Field(default=1000, description="max tokens to use"), ) -> Generator[str, None, None]: