Przeglądaj źródła

server : relay error messages (#4131)

SoftwareRenderer 2 lat temu
rodzic
commit
936c79b227
2 zmienionych plików z 15 dodań i 0 usunięć
  1. 4 0
      examples/server/public/completion.js
  2. 11 0
      examples/server/server.cpp

+ 4 - 0
examples/server/public/completion.js

@@ -94,6 +94,10 @@ export async function* llama(prompt, params = {}, config = {}) {
               break;
             }
           }
+          if (result.error) {
+            result.error = JSON.parse(result.error);
+            console.error(`llama.cpp error: ${result.error.content}`);
+          }
         }
       }
     }

+ 11 - 0
examples/server/server.cpp

@@ -2368,6 +2368,17 @@ int main(int argc, char **argv)
                                     break;
                                 }
                             } else {
+                                const std::string str =
+                                "error: " +
+                                result.result_json.dump(-1, ' ', false, json::error_handler_t::replace) +
+                                "\n\n";
+                                LOG_VERBOSE("data stream", {
+                                    { "to_send", str }
+                                });
+                                if (!sink.write(str.c_str(), str.size()))
+                                {
+                                    return false;
+                                }
                                 break;
                             }
                         }