]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
dolly : update error print behavior (#337)
authorJakob Frick <redacted>
Tue, 4 Jul 2023 13:26:57 +0000 (14:26 +0100)
committerGitHub <redacted>
Tue, 4 Jul 2023 13:26:57 +0000 (16:26 +0300)
examples/dolly-v2/main.cpp

index b95c5b1c6a11d2bd6983c737b83cb7d588f318a7..4d497b016f4c5d7d9daddc4b4f629c341173f551 100644 (file)
@@ -785,7 +785,7 @@ std::string execute_prompt(
 int setup_port(const int port) {
     int sockfd = socket(AF_INET, SOCK_STREAM, 0);
     if (sockfd < 0) {
-        std::cerr << "Failed to create socket\n";
+        fprintf(stderr, "%s: Failed to create new socket\n", __func__);
         return -1;
     }
 
@@ -797,12 +797,12 @@ int setup_port(const int port) {
     servaddr.sin_port = htons(port);
 
     if (bind(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0) {
-        std::cerr << "Failed to bind to port\n";
+        fprintf(stderr, "%s: Failed to bind to port %i\n", __func__, port);
         return -1;
     }
 
     if (listen(sockfd, 10) < 0) {
-        std::cerr << "Failed to listen on socket\n";
+        fprintf(stderr, "%s: Failed to listen to socket on port %i\n", __func__, port);
         return -1;
     }
     return sockfd;
@@ -810,7 +810,7 @@ int setup_port(const int port) {
 
 std::string read_from_port(int sockfd, int clientfd) {
     if (clientfd < 0) {
-        std::cerr << "Failed to accept new connection\n";
+        fprintf(stderr, "%s: Failed to accept new connection\n", __func__);
         return "";
     }
 
@@ -818,7 +818,7 @@ std::string read_from_port(int sockfd, int clientfd) {
     std::memset(buffer, 0, sizeof(buffer));
 
     if (read(clientfd, buffer, sizeof(buffer)) < 0) {
-        std::cerr << "Failed to read from client\n";
+        fprintf(stderr, "%s: Failed to read from client\n", __func__);
     } else {
         std::cout << "Received: " << buffer;
         return std::string(buffer);
@@ -915,11 +915,11 @@ int main(int argc, char ** argv) {
 #if defined(DOLLY_INTERACTIVE_PORT)
             if (params.interactive_port != -1) {
                 if (write(clientfd, response.c_str(), response.size()) < 0) {
-                    std::cerr << "Failed to write to client\n";
+                    fprintf(stderr, "%s: Failed to write answer '%s' to client\n", __func__, response.c_str());
                 }
 
                 if (close(clientfd) < 0) {
-                    std::cerr << "Failed to close client socket\n";
+                    fprintf(stderr, "%s: Failed to close client socket\n", __func__);
                 }
             } else
 #endif
@@ -953,7 +953,7 @@ int main(int argc, char ** argv) {
 
 #if defined(DOLLY_INTERACTIVE_PORT)
     if (params.interactive_port != -1 && close(sockfd) < 0) {
-        std::cerr << "Failed to close server socket\n";
+        fprintf(stderr, "%s: Failed to close server socket\n", __func__);
     }
 #endif