diff --git a/engine/commands/chat_completion_cmd.cc b/engine/commands/chat_completion_cmd.cc
index 8d06ce4bc..9929321ee 100644
--- a/engine/commands/chat_completion_cmd.cc
+++ b/engine/commands/chat_completion_cmd.cc
@@ -97,7 +97,9 @@ void ChatCompletionCmd::Exec(const std::string& host, int port,
       std::string user_input = std::move(msg);
       if (user_input.empty()) {
         std::cout << "> ";
-        std::getline(std::cin, user_input);
+        if (!std::getline(std::cin, user_input)) {
+          break;
+        }
       }
       if (user_input == kExitChat) {
         break;