Skip to content

Commit 625752e

Browse files
committed
refactor(gen): remove mistral errors
1 parent 62f8ab1 commit 625752e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

bigcodebench/gen/util/mistral_request.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
import time
2+
13
from mistralai.client import MistralClient
24
from mistralai.models.chat_completion import ChatMessage
35

@@ -7,6 +9,7 @@ def make_auto_request(client: MistralClient, *args, **kwargs) -> ChatMessage:
79
try:
810
ret = client.chat(*args, **kwargs)
911
except Exception as e:
12+
print("Unknown error. Waiting...")
1013
print(e)
1114
time.sleep(1)
1215
return ret

0 commit comments

Comments
 (0)