fix: sometimes openai stream will stop but it's not failure

plateaukao · Nov 22, 2024 · d0d7562 · d0d7562
1 parent 1bb64f6
commit d0d7562
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/app/src/main/java/info/plateaukao/einkbro/preference/ConfigManager.kt b/app/src/main/java/info/plateaukao/einkbro/preference/ConfigManager.kt
@@ -238,7 +238,7 @@ class ConfigManager(
     var gptUserPromptForWebPage by StringPreference(
         sp,
         K_GPT_USER_PROMPT_WEB_PAGE,
-        "Summarize in 300 words:"
+        "Summarize in 50 words:"
     )
     var imageApiKey by StringPreference(sp, K_IMAGE_API_KEY, "")
     var gptModel by StringPreference(sp, K_GPT_MODEL, "gpt-3.5-turbo")

diff --git a/app/src/main/java/info/plateaukao/einkbro/service/OpenAiRepository.kt b/app/src/main/java/info/plateaukao/einkbro/service/OpenAiRepository.kt
@@ -102,7 +102,12 @@ class OpenAiRepository : KoinComponent {
 
             override fun onFailure(eventSource: EventSource, t: Throwable?, response: Response?) {
                 super.onFailure(eventSource, t, response)
-                failureAction()
+                if(response?.code == 200) {
+                    doneAction()
+                    this@OpenAiRepository.eventSource = null
+                } else {
+                    failureAction()
+                }
             }
         })
     }