瀏覽代碼

fix: stream defaults to true, return request ID

Jun Siang Cheah 11 月之前
父節點
當前提交
24c35c308d
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      backend/apps/ollama/main.py

+ 2 - 2
backend/apps/ollama/main.py

@@ -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":