fix: handle EOF (#1434)

janhq · Oct 4, 2024 · 1b6e1d0 · 1b6e1d0
1 parent 4807899
commit 1b6e1d0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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;