Signed-off-by: Julien Veyssier <julien-nc@posteo.net>pull/56618/head
parent
c4da3cca34
commit
9f78ccbfd0
@ -0,0 +1,98 @@ |
||||
<?php |
||||
|
||||
/** |
||||
* SPDX-FileCopyrightText: 2025 Nextcloud GmbH and Nextcloud contributors |
||||
* SPDX-License-Identifier: AGPL-3.0-or-later |
||||
*/ |
||||
|
||||
declare(strict_types=1); |
||||
|
||||
namespace OCA\Testing\TaskProcessing; |
||||
|
||||
use OCA\Testing\AppInfo\Application; |
||||
use OCP\AppFramework\Services\IAppConfig; |
||||
use OCP\TaskProcessing\Exception\ProcessingException; |
||||
use OCP\TaskProcessing\ISynchronousProvider; |
||||
use OCP\TaskProcessing\TaskTypes\TextToTextChat; |
||||
use RuntimeException; |
||||
|
||||
class FakeTextToTextChatProvider implements ISynchronousProvider { |
||||
|
||||
public function __construct( |
||||
protected IAppConfig $appConfig, |
||||
) { |
||||
} |
||||
|
||||
public function getId(): string { |
||||
return Application::APP_ID . '-text2textchat'; |
||||
} |
||||
|
||||
public function getName(): string { |
||||
return 'Fake text2text chat task processing provider'; |
||||
} |
||||
|
||||
public function getTaskTypeId(): string { |
||||
return TextToTextChat::ID; |
||||
} |
||||
|
||||
public function getExpectedRuntime(): int { |
||||
return 1; |
||||
} |
||||
|
||||
public function getInputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getInputShapeDefaults(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalInputShape(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalInputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalInputShapeDefaults(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalOutputShape(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOutputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalOutputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function process(?string $userId, array $input, callable $reportProgress): array { |
||||
if ($this->appConfig->getAppValueBool('fail-' . $this->getId())) { |
||||
throw new ProcessingException('Failing as set by AppConfig'); |
||||
} |
||||
|
||||
if (!isset($input['system_prompt']) || !is_string($input['system_prompt'])) { |
||||
throw new RuntimeException('Invalid system prompt'); |
||||
} |
||||
|
||||
if (!isset($input['input']) || !is_string($input['input'])) { |
||||
throw new RuntimeException('Invalid input message'); |
||||
} |
||||
|
||||
if (!isset($input['history']) || !is_array($input['history'])) { |
||||
throw new RuntimeException('Invalid message history'); |
||||
} |
||||
|
||||
return [ |
||||
'output' => 'This is a fake response message: ' |
||||
. "\n\n- System prompt: " . $input['system_prompt'] |
||||
. "\n- Input message: " . $input['input'] |
||||
. "\n- Message history:\n" . count($input['history']) . ' messages', |
||||
]; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue