xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 05:25:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: AttributeError in OpenAI-compatible server (#3018)

Browse Source
This commit is contained in:
Jae-Won Chung 2024-02-29 01:04:07 -05:00 committed by GitHub
parent 01a5d18a53
commit a6d471c759
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/entrypoints/openai/protocol.py
View File

@ -57,7 +57,7 @@ class UsageInfo(BaseModel):
class ChatCompletionRequest(BaseModel):
model: str
messages: Union[str, List[Dict[str, str]]]
messages: List[Dict[str, str]]
temperature: Optional[float] = 0.7
top_p: Optional[float] = 1.0
n: Optional[int] = 1

2
vllm/entrypoints/openai/serving_chat.py
View File

@ -80,7 +80,7 @@ class OpenAIServingChat(OpenAIServing):
if request.add_generation_prompt:
return self.response_role
else:
return request.messages[-1].role
return request.messages[-1]["role"]
async def chat_completion_stream_generator(
self, request: ChatCompletionRequest,