Skip to content

Commit 04691fc

Browse files
committed
fix: fix int autodev llm server error issue
1 parent f81e64f commit 04691fc

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

java/src/main/kotlin/cc/unitmesh/idea/context/JavaCodeModifier.kt

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -60,12 +60,17 @@ open class JavaCodeModifier : CodeModifier {
6060

6161
val newTestMethod = ReadAction.compute<PsiMethod, Throwable> {
6262
val psiElementFactory = PsiElementFactory.getInstance(project)
63-
val methodCode = psiElementFactory.createMethodFromText(code, rootElement)
64-
if (rootElement.findMethodsByName(methodCode.name, false).isNotEmpty()) {
65-
log.error("Method already exists in the class: ${methodCode.name}")
63+
try {
64+
val methodCode = psiElementFactory.createMethodFromText(code, rootElement)
65+
if (rootElement.findMethodsByName(methodCode.name, false).isNotEmpty()) {
66+
log.error("Method already exists in the class: ${methodCode.name}")
67+
}
68+
69+
methodCode
70+
} catch (e: Throwable) {
71+
log.error("Failed to create method from text: $code", e)
72+
return@compute null
6673
}
67-
68-
methodCode
6974
}
7075

7176
WriteCommandAction.runWriteCommandAction(project) {

0 commit comments

Comments
 (0)