diff --git a/lib/public/LanguageModel/FreePromptTask.php b/lib/public/LanguageModel/FreePromptTask.php
index ac8ba6638a193..3de215784ab70 100644
--- a/lib/public/LanguageModel/FreePromptTask.php
+++ b/lib/public/LanguageModel/FreePromptTask.php
@@ -39,7 +39,7 @@ final class FreePromptTask extends AbstractLanguageModelTask {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function visitProvider($provider): string {
+	public function visitProvider(ILanguageModelProvider $provider): string {
 		return $provider->prompt($this->getInput());
 	}
 
@@ -47,7 +47,7 @@ public function visitProvider($provider): string {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function canUseProvider($provider): bool {
+	public function canUseProvider(ILanguageModelProvider $provider): bool {
 		return true;
 	}
 
diff --git a/lib/public/LanguageModel/HeadlineTask.php b/lib/public/LanguageModel/HeadlineTask.php
index 789ad22665635..e66c8893d8301 100644
--- a/lib/public/LanguageModel/HeadlineTask.php
+++ b/lib/public/LanguageModel/HeadlineTask.php
@@ -41,7 +41,7 @@ final class HeadlineTask extends AbstractLanguageModelTask {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function visitProvider($provider): string {
+	public function visitProvider(ILanguageModelProvider $provider): string {
 		if (!$this->canUseProvider($provider)) {
 			throw new \RuntimeException('HeadlineTask#visitProvider expects IHeadlineProvider');
 		}
@@ -52,7 +52,7 @@ public function visitProvider($provider): string {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function canUseProvider($provider): bool {
+	public function canUseProvider(ILanguageModelProvider $provider): bool {
 		return $provider instanceof IHeadlineProvider;
 	}
 
diff --git a/lib/public/LanguageModel/ILanguageModelTask.php b/lib/public/LanguageModel/ILanguageModelTask.php
index f5793632f582c..09d7507662261 100644
--- a/lib/public/LanguageModel/ILanguageModelTask.php
+++ b/lib/public/LanguageModel/ILanguageModelTask.php
@@ -62,18 +62,20 @@ interface ILanguageModelTask extends \JsonSerializable {
 	];
 
 	/**
-	 * @param T $provider
+	 * @psalm-param T $provider
+	 * @param ILanguageModelProvider $provider
 	 * @return string
 	 * @since 28.0.0
 	 */
-	public function visitProvider($provider): string;
+	public function visitProvider(ILanguageModelProvider $provider): string;
 
 	/**
-	 * @param T $provider
+	 * @psalm-param T $provider
+	 * @param ILanguageModelProvider $provider
 	 * @return bool
 	 * @since 28.0.0
 	 */
-	public function canUseProvider($provider): bool;
+	public function canUseProvider(ILanguageModelProvider $provider): bool;
 
 
 	/**
diff --git a/lib/public/LanguageModel/SummaryTask.php b/lib/public/LanguageModel/SummaryTask.php
index 4d7f9e0813f57..4504cdff7503c 100644
--- a/lib/public/LanguageModel/SummaryTask.php
+++ b/lib/public/LanguageModel/SummaryTask.php
@@ -41,7 +41,7 @@ final class SummaryTask extends AbstractLanguageModelTask {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function visitProvider($provider): string {
+	public function visitProvider(ILanguageModelProvider $provider): string {
 		if (!$this->canUseProvider($provider)) {
 			throw new \RuntimeException('SummaryTask#visitProvider expects ISummaryProvider');
 		}
@@ -52,7 +52,7 @@ public function visitProvider($provider): string {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function canUseProvider($provider): bool {
+	public function canUseProvider(ILanguageModelProvider $provider): bool {
 		return $provider instanceof ISummaryProvider;
 	}
 
diff --git a/lib/public/LanguageModel/TopicsTask.php b/lib/public/LanguageModel/TopicsTask.php
index 513241240a223..ccf5d512a8a97 100644
--- a/lib/public/LanguageModel/TopicsTask.php
+++ b/lib/public/LanguageModel/TopicsTask.php
@@ -41,7 +41,7 @@ final class TopicsTask extends AbstractLanguageModelTask {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function visitProvider($provider): string {
+	public function visitProvider(ILanguageModelProvider $provider): string {
 		if (!$this->canUseProvider($provider)) {
 			throw new \RuntimeException('TopicsTask#visitProvider expects ITopicsProvider');
 		}
@@ -52,7 +52,7 @@ public function visitProvider($provider): string {
 	 * @inheritDoc
 	 * @since 28.0.0
 	 */
-	public function canUseProvider($provider): bool {
+	public function canUseProvider(ILanguageModelProvider $provider): bool {
 		return $provider instanceof ITopicsProvider;
 	}