Commit 24c35c30 authored by Jun Siang Cheah's avatar Jun Siang Cheah
Browse files

fix: stream defaults to true, return request ID

parent 207e2503
...@@ -993,7 +993,7 @@ async def generate_chat_completion( ...@@ -993,7 +993,7 @@ async def generate_chat_completion(
def stream_content(): def stream_content():
try: try:
if payload.get("stream", None): if payload.get("stream", True):
yield json.dumps({"id": request_id, "done": False}) + "\n" yield json.dumps({"id": request_id, "done": False}) + "\n"
for chunk in r.iter_content(chunk_size=8192): for chunk in r.iter_content(chunk_size=8192):
...@@ -1522,7 +1522,7 @@ async def deprecated_proxy( ...@@ -1522,7 +1522,7 @@ async def deprecated_proxy(
if path == "generate": if path == "generate":
data = json.loads(body.decode("utf-8")) 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" yield json.dumps({"id": request_id, "done": False}) + "\n"
elif path == "chat": elif path == "chat":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment