diff --git a/server/src/infra/repositories/machine-learning.repository.ts b/server/src/infra/repositories/machine-learning.repository.ts index 1ed887082a..457ab54b51 100644 --- a/server/src/infra/repositories/machine-learning.repository.ts +++ b/server/src/infra/repositories/machine-learning.repository.ts @@ -18,6 +18,12 @@ export class MachineLearningRepository implements IMachineLearningRepository { private async post(url: string, input: TextModelInput | VisionModelInput, config: ModelConfig): Promise { const formData = await this.getFormData(input, config); const res = await fetch(`${url}/predict`, { method: 'POST', body: formData }); + if (res.status >= 400) { + throw new Error( + `Request ${config.modelType ? `for ${config.modelType.replace('-', ' ')} ` : ''}` + + `failed with status ${res.status}: ${res.statusText}`, + ); + } return res.json(); }