99 lines
2.3 KiB
PHP
99 lines
2.3 KiB
PHP
<?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',
|
|
];
|
|
}
|
|
}
|