From dcdcee3a744f39714503ee2b19c49b7c7b6209c9 Mon Sep 17 00:00:00 2001
From: VoidIsVoid <343750470@qq.com>
Date: Sat, 14 Sep 2024 17:36:44 +0800
Subject: [PATCH] server: add data: [DONE] to /chat/completions stream response
 (#9459)

---
 examples/server/server.cpp                    | 2 ++
 examples/server/tests/features/steps/steps.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/examples/server/server.cpp b/examples/server/server.cpp
index 73cd6aae7..14c4af3d9 100644
--- a/examples/server/server.cpp
+++ b/examples/server/server.cpp
@@ -2993,6 +2993,8 @@ int main(int argc, char ** argv) {
                 }, [&](json error_data) {
                     server_sent_event(sink, "error", error_data);
                 });
+                static const std::string ev_done = "data: [DONE]\n\n";
+                sink.write(ev_done.data(), ev_done.size());
                 sink.done();
                 return true;
             };
diff --git a/examples/server/tests/features/steps/steps.py b/examples/server/tests/features/steps/steps.py
index 11587dd64..0f4249b13 100644
--- a/examples/server/tests/features/steps/steps.py
+++ b/examples/server/tests/features/steps/steps.py
@@ -1020,6 +1020,8 @@ async def oai_chat_completions(user_prompt,
                             event_data = line.split(': ', 1)
                             assert event_data[0] == 'data', f'Bad event code received: ```{event_data}```'
                             chunk_raw = event_data[1]
+                            if chunk_raw == '[DONE]':
+                                break
 
                             chunk = json.loads(chunk_raw)
                             assert len(chunk['choices']) == 1, f"no choices provided, line ```{line}```"