Automatically retry the OpenAI API when it errors

juntao · juntao · commit 0c8842603f74 · 2023-04-06T20:02:15.000Z
Signed-off-by: Michael Yuan &lt;michael@secondstate.io&gt;
diff --git a/Cargo.toml b/Cargo.toml
@@ -17,6 +17,6 @@ anyhow = "1"
 flowsnet-platform-sdk = "0.1.2"
 lazy_static = "1.4.0"
 regex = "1.7.1"
-openai-flows = "0.3"
+openai-flows = "0.4"
 http_req_wasi = { version = "0.10.2", features = ["wasmedge_ssl"] }
 words-count = "0.1.4"
diff --git a/src/github-pr-summary.rs b/src/github-pr-summary.rs
@@ -147,6 +147,7 @@ async fn handler(
             model: ChatModel::GPT35Turbo,
             restart: true,
             system_prompt: Some(system),
+            retry_times: 3,
         };
         let question = "The following is a GitHub patch. Please summarize the key changes and identify potential problems. Start with the most important findings.\n\n".to_string() + commit;
         if let Some(r) = chat_completion(openai_key_name, &chat_id, &question, &co) {
@@ -169,6 +170,7 @@ async fn handler(
             model: ChatModel::GPT35Turbo,
             restart: true,
             system_prompt: Some(system),
+            retry_times: 3,
         };
         let question = "Here is a set of summaries for software source code patches. Each summary starts with a ------ line. Please write an overall summary considering all the individual summary. Please present the potential issues and errors first, following by the most important findings, in your summary.\n\n".to_string() + &reviews_text;
         if let Some(r) = chat_completion(openai_key_name, &chat_id, &question, &co) {
