MCPcopy Create free account
hub / github.com/Mercury7353/PyBench / __exit__

Method __exit__

llms/api_key_pool.py:283–298  ·  view source on GitHub ↗

Put the resource back in the pool.

(self, exc_type, exc_value, exc_traceback)

Source from the content-addressed store, hash-verified

281 return self.queue[self._index].client_args
282
283 def __exit__(self, exc_type, exc_value, exc_traceback) -> bool:
284 """Put the resource back in the pool."""
285
286 if exc_value is None:
287 self._index = (self._index + 1) % len(self)
288 return True
289 else:
290 if (
291 exc_type is RateLimitError
292 and exc_value.status_code == 429
293 and judge_quota_exceeded(exc_value.message)
294 ):
295 self._remove_cur_arg()
296 else:
297 self._index = (self._index + 1) % len(self)
298 return False
299
300
301def judge_quota_exceeded(message: str) -> bool:

Callers

nothing calls this directly

Calls 2

judge_quota_exceededFunction · 0.85
_remove_cur_argMethod · 0.80

Tested by

no test coverage detected