fix: stream defaults to true, return request ID

This commit is contained in:
Jun Siang Cheah 2024-06-02 17:09:15 +01:00
parent 207e25035a
commit 24c35c308d

View File

@ -993,7 +993,7 @@ async def generate_chat_completion(
def stream_content():
try:
if payload.get("stream", None):
if payload.get("stream", True):
yield json.dumps({"id": request_id, "done": False}) + "\n"
for chunk in r.iter_content(chunk_size=8192):
@ -1522,7 +1522,7 @@ async def deprecated_proxy(
if path == "generate":
data = json.loads(body.decode("utf-8"))
if not ("stream" in data and data["stream"] == False):
if data.get("stream", True):
yield json.dumps({"id": request_id, "done": False}) + "\n"
elif path == "chat":