We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b7a52c2 commit 843e891Copy full SHA for 843e891
oneping/server.py
@@ -41,7 +41,7 @@ class GenerateRequest(BaseModel):
41
model: str | None = None
42
system: str | None = None
43
prefill: str | None = None
44
- prediction: str | None = None
+ prediction: dict[str, str] | None = None
45
max_tokens: int | None = None
46
history: list[dict[str, str]] | None = None
47
@@ -60,7 +60,7 @@ class GenerateRequest(BaseModel):
60
async def chat(genreq: GenerateRequest):
61
data = genreq.model_dump(exclude_none=True)
62
patch = patch_payload(data)
63
- if genreq.stream:
+ if patch.get('stream', False):
64
ret = stream_api(**kwargs, **patch)
65
return StreamingResponse(ret, media_type='text/plain')
66
else:
0 commit comments