Spaces:
Paused
Paused
Mirrowel
commited on
Commit
·
b76364c
1
Parent(s):
da8d5f4
fix(client): sync key acquisition logging between stream/non-stream
Browse files
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 |
)
|