Skip to content

Commit

Permalink
LLM OCP API: Add to RegistrationContext
Browse files Browse the repository at this point in the history
Signed-off-by: Marcel Klehr <mklehr@gmx.net>
  • Loading branch information
marcelklehr committed Jul 7, 2023
1 parent 6fc4cb6 commit 82d3b00
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 1 deletion.
21 changes: 21 additions & 0 deletions lib/private/AppFramework/Bootstrap/RegistrationContext.php
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@
use OCP\Calendar\Resource\IBackend as IResourceBackend;
use OCP\Calendar\Room\IBackend as IRoomBackend;
use OCP\Collaboration\Reference\IReferenceProvider;
use OCP\LanguageModel\ILanguageModelProvider;
use OCP\SpeechToText\ISpeechToTextProvider;
use OCP\Talk\ITalkBackend;
use OCP\Translation\ITranslationProvider;
Expand Down Expand Up @@ -115,6 +116,9 @@ class RegistrationContext {
/** @var ServiceRegistration<ISpeechToTextProvider>[] */
private $speechToTextProviders = [];

/** @var ServiceRegistration<ILanguageModelProvider>[] */
private $languageModelProviders = [];

/** @var ServiceRegistration<ICustomTemplateProvider>[] */
private $templateProviders = [];

Expand Down Expand Up @@ -262,6 +266,12 @@ public function registerSpeechToTextProvider(string $providerClass): void {
$providerClass
);
}
public function registerLanguageModelProvider(string $providerClass): void {
$this->context->registerLanguageModelProvider(
$this->appId,
$providerClass
);
}

public function registerTemplateProvider(string $providerClass): void {
$this->context->registerTemplateProvider(
Expand Down Expand Up @@ -429,6 +439,10 @@ public function registerSpeechToTextProvider(string $appId, string $class): void
$this->speechToTextProviders[] = new ServiceRegistration($appId, $class);
}

public function registerLanguageModelProvider(string $appId, string $class): void {
$this->languageModelProviders[] = new ServiceRegistration($appId, $class);
}

public function registerTemplateProvider(string $appId, string $class): void {
$this->templateProviders[] = new ServiceRegistration($appId, $class);
}
Expand Down Expand Up @@ -707,6 +721,13 @@ public function getSpeechToTextProviders(): array {
return $this->speechToTextProviders;
}

/**
* @return ServiceRegistration<ILanguageModelProvider>[]
*/
public function getLanguageModelProviders(): array {
return $this->languageModelProviders;
}

/**
* @return ServiceRegistration<ICustomTemplateProvider>[]
*/
Expand Down
2 changes: 1 addition & 1 deletion lib/private/LanguageModel/LanguageModelManager.php
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ public function getProviders(): array {

$this->providers = [];

foreach ($context->getSpeechToTextProviders() as $providerServiceRegistration) {
foreach ($context->getLanguageModelProviders() as $providerServiceRegistration) {
$class = $providerServiceRegistration->getService();
try {
$this->providers[$class] = $this->serverContainer->get($class);
Expand Down

0 comments on commit 82d3b00

Please sign in to comment.