В Psalm скоро появятся условные типы как в TypeScript.

/**
 * @template T as string|list<string>
 * @psalm-param T $name
 * @psalm-return (T is string ? Connection : array<string, Connection>)
 */
public function getConnection($name) {
    // ...
}

Поиграться можно тут: https://psalm.dev/r/5fb0ea4228 .
Источник: https://twitter.com/psalmphp/status/1241744682494971904 .
Подробнее в issue: https://github.com/vimeo/psalm/issues/3005 .