[FIR IDE] Fixed getting symbols for lambda literal

idea/idea-frontend-api/src/org/jetbrains/kotlin/idea/frontend/api/symbols/KtSymbolProvider.kt

@@ -27,6 +27,11 @@ abstract class KtSymbolProvider : KtAnalysisSessionComponent() {
             literalExpression?.let(::getAnonymousObjectSymbol) ?: getClassOrObjectSymbol(psi)
         }
         is KtPropertyAccessor -> getPropertyAccessorSymbol(psi)
+        is KtFunctionLiteral -> {
+            val literalExpression = psi.parent as? KtLambdaExpression
+            check(literalExpression != null) { "Unexpected parent for KtFunctionLiteral" }
+            getAnonymousFunctionSymbol(literalExpression)
+        }
         else -> error("Cannot build symbol for ${psi::class}")
     }
 

idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/FirModuleResolveStateImpl.kt

@@ -127,6 +127,7 @@ internal class FirModuleResolveStateImpl(
         val firDeclaration = FirElementFinder.findElementIn<FirDeclaration>(container) { firDeclaration ->
             when (val realPsi = firDeclaration.realPsi) {
                 is KtObjectLiteralExpression -> realPsi.objectDeclaration == ktDeclaration
+                is KtFunctionLiteral -> realPsi.parent == ktDeclaration
                 else -> realPsi == ktDeclaration
             }
         }