LLM OCP API: Add to RegistrationContext

Signed-off-by: Marcel Klehr <[email protected]>
(cherry picked from commit 82d3b00)

lib/private/AppFramework/Bootstrap/RegistrationContext.php

@@ -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;
@@ -115,6 +116,9 @@ class RegistrationContext {
 	/** @var ServiceRegistration<ISpeechToTextProvider>[] */
 	private $speechToTextProviders = [];
 
+	/** @var ServiceRegistration<ILanguageModelProvider>[] */
+	private $languageModelProviders = [];
+
 	/** @var ServiceRegistration<ICustomTemplateProvider>[] */
 	private $templateProviders = [];
 
@@ -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(
@@ -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);
 	}
@@ -707,6 +721,13 @@ public function getSpeechToTextProviders(): array {
 		return $this->speechToTextProviders;
 	}
 
+	/**
+	 * @return ServiceRegistration<ILanguageModelProvider>[]
+	 */
+	public function getLanguageModelProviders(): array {
+		return $this->languageModelProviders;
+	}
+
 	/**
 	 * @return ServiceRegistration<ICustomTemplateProvider>[]
 	 */

lib/private/LanguageModel/LanguageModelManager.php

@@ -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);