File tree Expand file tree Collapse file tree 1 file changed +24
-6
lines changed Expand file tree Collapse file tree 1 file changed +24
-6
lines changed Original file line number Diff line number Diff line change @@ -145,16 +145,34 @@ const handlePoolEvaluationQuestions = async (
145
145
return questions . map ( question => question . question ) ;
146
146
}
147
147
148
- const [ evalError , evaluationQuestions ] = await catchError (
149
- requestEvaluationQuestions ( poolMetadata )
150
- ) ;
148
+ // Retry logic
149
+ let retries = 5 ;
150
+ let evalError ;
151
+ let evaluationQuestions : undefined | PromptEvaluationQuestions ;
152
+
153
+ while ( retries > 0 ) {
154
+ [ evalError , evaluationQuestions ] = await catchError (
155
+ requestEvaluationQuestions ( poolMetadata )
156
+ ) ;
151
157
152
- if ( evalError != null || evaluationQuestions == null ) {
158
+ if ( evalError == null && evaluationQuestions != null ) {
159
+ break ;
160
+ }
161
+
162
+ retries -- ;
163
+ if ( retries > 0 ) {
164
+ logger . warn (
165
+ `Retrying evaluation question request. Attempts remaining: ${ retries } `
166
+ ) ;
167
+ }
168
+ }
169
+
170
+ if ( evaluationQuestions === undefined ) {
153
171
logger . error (
154
- `Error requesting evaluation questions: ${ evalError ?. message } `
172
+ `Error requesting evaluation questions after 5 attempts : ${ evalError ?. message } `
155
173
) ;
156
174
throw new Error (
157
- `Error requesting evaluation questions ${ evalError ?. message } `
175
+ `Error requesting evaluation questions: ${ evalError ?. message } `
158
176
) ;
159
177
}
160
178
You can’t perform that action at this time.
0 commit comments