chore: add log

unit-mesh · Aug 18, 2023 · dc5d14c · dc5d14c
1 parent 4d5e573
commit dc5d14c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomLLMProvider.kt b/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomLLMProvider.kt
@@ -86,6 +86,7 @@ class CustomLLMProvider(val project: Project) : LLMProvider {
             }, BackpressureStrategy.BUFFER)
 
         try {
+            logger.warn("Starting to stream:")
             return callbackFlow {
                 withContext(Dispatchers.IO) {
                     sseFlowable
@@ -94,6 +95,7 @@ class CustomLLMProvider(val project: Project) : LLMProvider {
                             if (engineFormat.isNotEmpty()) {
                                 val chunk: String = JsonPath.parse(sse!!.data)?.read(engineFormat)
                                     ?: throw Exception("Failed to parse chunk")
+                                logger.warn(" $chunk")
                                 trySend(chunk)
                             } else {
                                 val result: ChatCompletionResult =