[Online Server] Chat Api for streaming and not streaming response (#5470)

* fix bugs

* fix bugs

* fix api server

* fix api server

* add chat api and test

* del request.n
This commit is contained in:
Jianghai
2024-04-07 14:45:43 +08:00
committed by CjhHa1
parent de378cd2ab
commit c064032865
7 changed files with 326 additions and 19 deletions

View File

@@ -1,3 +1,8 @@
from typing import Any, Optional
from pydantic import BaseModel
# make it singleton
class NumericIDGenerator:
_instance = None
@@ -14,3 +19,18 @@ class NumericIDGenerator:
id_generator = NumericIDGenerator()
class ChatMessage(BaseModel):
role: str
content: Any
class DeltaMessage(BaseModel):
role: Optional[str] = None
content: Optional[Any] = None
class ChatCompletionResponseStreamChoice(BaseModel):
index: int
message: DeltaMessage