add retry plugin (#195)

coderabbitai · Apr 18, 2023 · 94fc36e · 94fc36e
1 parent d9cbf24
commit 94fc36e
Show file tree

Hide file tree

Showing 4 changed files with 123 additions and 4 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -29,6 +29,7 @@
     "@actions/github": "^5.1.1",
     "@dqbd/tiktoken": "^1.0.6",
     "@octokit/action": "^5.0.2",
+    "@octokit/plugin-retry": "^4.1.3",
     "@octokit/plugin-throttling": "^5.0.1",
     "minimatch": "^9.0.0",
     "node-fetch": "^3.3.1",

diff --git a/src/octokit.ts b/src/octokit.ts
@@ -1,11 +1,12 @@
 import * as core from '@actions/core'
 import {Octokit} from '@octokit/action'
+import {retry} from '@octokit/plugin-retry'
 import {throttling} from '@octokit/plugin-throttling'
 
 const token = core.getInput('token') || process.env.GITHUB_TOKEN
 
-const ThrottlingOctokit = Octokit.plugin(throttling)
-export const octokit = new ThrottlingOctokit({
+const RetryAndThrottlingOctokit = Octokit.plugin(throttling, retry)
+export const octokit = new RetryAndThrottlingOctokit({
   auth: `token ${token}`,
   throttle: {
     onRateLimit: (
@@ -28,5 +29,9 @@ Retry count: ${retryCount}
       )
       return true
     }
+  },
+  retry: {
+    doNotRetry: ['429'],
+    maxRetries: 10
   }
 })