upstream/nextcloud
1
0
Fork 0
mirror of https://github.com/nextcloud/server.git synced 2026-04-15 22:11:17 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 27

LLM OCP API: s/getAvailableTasks/getAvailableTaskClasses/

Browse source 
Signed-off-by: Marcel Klehr <mklehr@gmx.net>
This commit is contained in:
Marcel Klehr 2023-07-07 11:40:19 +02:00
parent a28d8fac2f
commit d56286b8ac
3 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lib/private/LanguageModel/LanguageModelManager.php
View file

@ -102,7 +102,7 @@ class LanguageModelManager implements ILanguageModelManager {
/**
* @inheritDoc
*/
public function getAvailableTasks(): array {
public function getAvailableTaskClasses(): array {
$tasks = [];
foreach ($this->getProviders() as $provider) {
$tasks[FreePromptTask::class] = true;
@ -123,11 +123,11 @@ class LanguageModelManager implements ILanguageModelManager {
* @inheritDoc
*/
public function getAvailableTaskTypes(): array {
return array_map(fn ($taskClass) => $taskClass::TYPE, $this->getAvailableTasks());
return array_map(fn ($taskClass) => $taskClass::TYPE, $this->getAvailableTaskClasses());
}
public function canHandleTask(ILanguageModelTask $task): bool {
return count(array_filter($this->getAvailableTasks(), fn ($class) => $task instanceof $class)) > 0;
return count(array_filter($this->getAvailableTaskClasses(), fn ($class) => $task instanceof $class)) > 0;
}
/**

2
lib/public/LanguageModel/ILanguageModelManager.php
View file

@ -45,7 +45,7 @@ interface ILanguageModelManager {
* @return string[]
* @since 28.0.0
*/
public function getAvailableTasks(): array;
public function getAvailableTaskClasses(): array;
/**
* @return string[]

12
tests/lib/LanguageModel/LanguageModelManagerTest.php
View file

@ -163,7 +163,7 @@ class LanguageModelManagerTest extends \Test\TestCase {
public function testShouldNotHaveAnyProviders() {
$this->registrationContext->expects($this->any())->method('getLanguageModelProviders')->willReturn([]);
$this->assertCount(0, $this->languageModelManager->getAvailableTasks());
$this->assertCount(0, $this->languageModelManager->getAvailableTaskClasses());
$this->assertCount(0, $this->languageModelManager->getAvailableTaskTypes());
$this->assertFalse($this->languageModelManager->hasProviders());
$this->expectException(PreConditionNotMetException::class);
@ -174,7 +174,7 @@ class LanguageModelManagerTest extends \Test\TestCase {
$this->registrationContext->expects($this->any())->method('getLanguageModelProviders')->willReturn([
new ServiceRegistration('test', TestVanillaLanguageModelProvider::class)
]);
$this->assertCount(1, $this->languageModelManager->getAvailableTasks());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskClasses());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskTypes());
$this->assertTrue($this->languageModelManager->hasProviders());
$this->assertEquals('Hello Free Prompt', $this->languageModelManager->runTask(new FreePromptTask('Hello', 'test', null)));
@ -189,7 +189,7 @@ class LanguageModelManagerTest extends \Test\TestCase {
$this->registrationContext->expects($this->any())->method('getLanguageModelProviders')->willReturn([
new ServiceRegistration('test', TestVanillaLanguageModelProvider::class)
]);
$this->assertCount(1, $this->languageModelManager->getAvailableTasks());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskClasses());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskTypes());
$this->assertTrue($this->languageModelManager->hasProviders());
@ -259,7 +259,7 @@ class LanguageModelManagerTest extends \Test\TestCase {
new ServiceRegistration('test', TestVanillaLanguageModelProvider::class),
new ServiceRegistration('test', TestFullLanguageModelProvider::class),
]);
$this->assertCount(3, $this->languageModelManager->getAvailableTasks());
$this->assertCount(3, $this->languageModelManager->getAvailableTaskClasses());
$this->assertCount(3, $this->languageModelManager->getAvailableTaskTypes());
$this->assertTrue($this->languageModelManager->hasProviders());
@ -287,7 +287,7 @@ class LanguageModelManagerTest extends \Test\TestCase {
$this->registrationContext->expects($this->any())->method('getLanguageModelProviders')->willReturn([
new ServiceRegistration('test', TestFailingLanguageModelProvider::class),
]);
$this->assertCount(1, $this->languageModelManager->getAvailableTasks());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskClasses());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskTypes());
$this->assertTrue($this->languageModelManager->hasProviders());
@ -354,7 +354,7 @@ class LanguageModelManagerTest extends \Test\TestCase {
$this->registrationContext->expects($this->any())->method('getLanguageModelProviders')->willReturn([
new ServiceRegistration('test', TestVanillaLanguageModelProvider::class)
]);
$this->assertCount(1, $this->languageModelManager->getAvailableTasks());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskClasses());
$this->assertCount(1, $this->languageModelManager->getAvailableTaskTypes());
$this->assertTrue($this->languageModelManager->hasProviders());
$task = new FreePromptTask('Hello', 'test', null);