Mirrowel commited on
Commit
b76364c
·
1 Parent(s): da8d5f4

fix(client): sync key acquisition logging between stream/non-stream

Browse files
Files changed (1) hide show
  1. src/rotator_library/client.py +2 -1
src/rotator_library/client.py CHANGED
@@ -346,8 +346,9 @@ class RotatingClient:
346
  if not keys_to_try:
347
  break
348
 
 
349
  current_key = await self.usage_manager.acquire_key(
350
- available_keys=keys_to_try,
351
  model=model,
352
  deadline=deadline
353
  )
 
346
  if not keys_to_try:
347
  break
348
 
349
+ lib_logger.info(f"Acquiring key for model {model}. Tried keys: {len(tried_keys)}/{len(keys_for_provider)}")
350
  current_key = await self.usage_manager.acquire_key(
351
+ available_keys=keys_to_try,
352
  model=model,
353
  deadline=deadline
354
  )