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 (
145145 return questions . map ( question => question . question ) ;
146146 }
147147
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+ ) ;
151157
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 ) {
153171 logger . error (
154- `Error requesting evaluation questions: ${ evalError ?. message } `
172+ `Error requesting evaluation questions after 5 attempts : ${ evalError ?. message } `
155173 ) ;
156174 throw new Error (
157- `Error requesting evaluation questions ${ evalError ?. message } `
175+ `Error requesting evaluation questions: ${ evalError ?. message } `
158176 ) ;
159177 }
160178
You can’t perform that action at this time.
0 commit comments