refactor(context): remove @NotNull annotations and add error handling #…

…324

- Remove redundant @NotNull annotations from MethodContextProvider.
- Add try-catch block in ClassContext to handle potential exceptions when retrieving method signatures.

core/src/main/kotlin/cc/unitmesh/devti/context/ClassContext.kt

@@ -22,7 +22,11 @@ class ClassContext(
     }
 
     private fun getMethodSignatures(): List<String> = methods.mapNotNull {
-        MethodContextProvider(false, gatherUsages = false).from(it).signature
+        try {
+            MethodContextProvider(false, gatherUsages = false).from(it).signature
+        } catch (e: Exception) {
+            null
+        }
     }
 
     override fun format(): String {

core/src/main/kotlin/cc/unitmesh/devti/context/MethodContextProvider.kt

@@ -5,24 +5,20 @@ import cc.unitmesh.devti.context.builder.MethodContextBuilder
 import com.intellij.lang.Language
 import com.intellij.lang.LanguageExtension
 import com.intellij.psi.PsiElement
-import org.jetbrains.annotations.NotNull
 
 class MethodContextProvider(private val includeClassContext: Boolean, private val gatherUsages: Boolean):
     LLMCodeContextProvider<PsiElement> {
-    @NotNull
     private val languageExtension: LanguageExtension<MethodContextBuilder> =
         LanguageExtension("cc.unitmesh.methodContextBuilder")
 
-    @NotNull
     private var providers: List<MethodContextBuilder> = emptyList()
 
     init {
         val registeredLanguages = Language.getRegisteredLanguages()
         providers = registeredLanguages.mapNotNull(languageExtension::forLanguage)
     }
 
-    @NotNull
-    override fun from(@NotNull psiElement: PsiElement): MethodContext {
+    override fun from(psiElement: PsiElement): MethodContext {
         val iterator = providers.iterator()
         while (iterator.hasNext()) {
             val provider = iterator.next()