Browse Source

fix(TextToImage): Refactor scheduling mechanism

pull/58117/head
Marcel Klehr 7 months ago
committed by backportbot[bot]
parent
commit
d379130a05
  1. 6
      core/Controller/TextToImageApiController.php
  2. 6
      lib/public/TextToImage/Task.php

6
core/Controller/TextToImageApiController.php

@ -81,6 +81,12 @@ class TextToImageApiController extends OCSController {
if (strlen($input) > 64_000) {
return new DataResponse(['message' => $this->l->t('Input text is too long')], Http::STATUS_PRECONDITION_FAILED);
}
if ($numberOfImages > 12) {
return new DataResponse(['message' => $this->l->t('Cannot generate more than 12 images')], Http::STATUS_PRECONDITION_FAILED);
}
if ($numberOfImages < 1) {
return new DataResponse(['message' => $this->l->t('Cannot generate less than 1 image')], Http::STATUS_PRECONDITION_FAILED);
}
$task = new Task($input, $appId, $numberOfImages, $this->userId, $identifier);
try {
try {

6
lib/public/TextToImage/Task.php

@ -68,6 +68,12 @@ final class Task implements \JsonSerializable {
protected ?string $userId,
protected ?string $identifier = '',
) {
if ($this->numberOfImages > 12) {
throw new \ValueError('Cannot generate more than 12 images');
}
if ($this->numberOfImages < 1) {
throw new \ValueError('Cannot generate less than 1 image');
}
}
/**

Loading…
Cancel
Save