Просмотр исходного кода

server : throw an error when `slot unavailable` (#4741)

Justin Parker 2 лет назад
Родитель
Сommit
f2eb19bd8b
1 измененных файлов с 9 добавлено и 0 удалено
  1. 9 0
      examples/server/public/completion.js

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

@@ -95,6 +95,15 @@ export async function* llama(prompt, params = {}, config = {}) {
               break;
             }
           }
+          if (result.error) {
+            result.error = JSON.parse(result.error);
+            if (result.error.content.includes('slot unavailable')) {
+              // Throw an error to be caught by upstream callers
+              throw new Error('slot unavailable');
+            } else {
+              console.error(`llama.cpp error: ${result.error.content}`);
+            }
+          }
           if (result.error) {
             result.error = JSON.parse(result.error);
             console.error(`llama.cpp error: ${result.error.content}`);