fix(shirelang): add exception handling for LlmProvider streaming output

#60
phodal · Aug 26, 2024 · a8272b2 · a8272b2
1 parent da44e85
commit a8272b2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/shirelang/src/main/kotlin/com/phodal/shirelang/run/flow/ShireConversationService.kt b/shirelang/src/main/kotlin/com/phodal/shirelang/run/flow/ShireConversationService.kt
@@ -76,11 +76,15 @@ class ShireConversationService(val project: Project) {
         val finalPrompt = prompt.toString()
         if (consoleView != null) {
             runBlocking {
-                LlmProvider.provider(project)
-                    ?.stream(finalPrompt, "", true)
-                    ?.collect {
-                        consoleView.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
-                    }
+                try {
+                    LlmProvider.provider(project)
+                        ?.stream(finalPrompt, "", true)
+                        ?.collect {
+                            consoleView.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
+                        }
+                } catch (e: Exception) {
+                    consoleView.print(e.message ?: "Error", ConsoleViewContentType.ERROR_OUTPUT)
+                }
             }
         }
     }

diff --git a/shirelang/src/main/kotlin/com/phodal/shirelang/run/flow/ShireProcessProcessor.kt b/shirelang/src/main/kotlin/com/phodal/shirelang/run/flow/ShireProcessProcessor.kt
@@ -103,10 +103,13 @@ class ShireProcessProcessor(val project: Project) {
             if (consoleView == null) return
 
             runBlocking {
-                LlmProvider.provider(project)?.stream(result.shireOutput, "Shirelang", true)
-                    ?.collect {
-                        consoleView.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
-                    }
+                try {
+                    LlmProvider.provider(project)?.stream(result.shireOutput, "Shirelang", true)?.collect {
+                            consoleView.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
+                        }
+                } catch (e: Exception) {
+                    consoleView.print(e.message ?: "Error", ConsoleViewContentType.ERROR_OUTPUT)
+                }
             }
         } else {
             if (result.nextJob == null) return