From e37cdab0c6fd1e924522ab3cb720908d0c02e226 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Sat, 12 Oct 2024 00:36:28 -0700 Subject: [PATCH] Fix ignore_eos (#1645) --- python/sglang/srt/openai_api/protocol.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/openai_api/protocol.py b/python/sglang/srt/openai_api/protocol.py index 3d7d450c9..ff4c62f00 100644 --- a/python/sglang/srt/openai_api/protocol.py +++ b/python/sglang/srt/openai_api/protocol.py @@ -170,8 +170,8 @@ class CompletionRequest(BaseModel): # Extra parameters for SRT backend only and will be ignored by OpenAI models. regex: Optional[str] = None json_schema: Optional[str] = None - ignore_eos: Optional[bool] = False - min_tokens: Optional[int] = 0 + ignore_eos: bool = False + min_tokens: int = 0 repetition_penalty: Optional[float] = 1.0 stop_token_ids: Optional[List[int]] = Field(default_factory=list) @@ -275,6 +275,7 @@ class ChatCompletionRequest(BaseModel): min_tokens: Optional[int] = 0 repetition_penalty: Optional[float] = 1.0 stop_token_ids: Optional[List[int]] = Field(default_factory=list) + ignore_eos: bool = False class ChatMessage(BaseModel):