phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix linting

Browse source
This commit is contained in:
ishaan-jaff 2023-10-04 16:03:01 -07:00
parent 7e9aae363d
commit e9160a1485
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/utils.py
View file

@ -3761,7 +3761,6 @@ class APIRequest:
response = await litellm.acompletion( response = await litellm.acompletion(
**self.request_json **self.request_json
) )
print(response)
logging.info(f"Completed request #{self.task_id}") logging.info(f"Completed request #{self.task_id}")
except Exception as e: except Exception as e:
logging.warning( logging.warning(
@ -3771,7 +3770,7 @@ class APIRequest:
error = e error = e
print(f"got exception {e}") print(f"got exception {e}")
if "Rate limit" in str(e): if "Rate limit" in str(e):
status_tracker.time_of_last_rate_limit_error = time.time() status_tracker.time_of_last_rate_limit_error = int(time.time())
status_tracker.num_rate_limit_errors += 1 status_tracker.num_rate_limit_errors += 1
status_tracker.num_api_errors -= ( status_tracker.num_api_errors -= (
1 # rate limit errors are counted separately 1 # rate limit errors are counted separately
@ -3812,7 +3811,7 @@ class APIRequest:
f.write(json_string + "\n") f.write(json_string + "\n")
class RateLimitHandler(): class RateLimitManager():
def __init__(self, max_tokens_per_minute, max_requests_per_minute): def __init__(self, max_tokens_per_minute, max_requests_per_minute):
self.max_tokens_per_minute = max_tokens_per_minute self.max_tokens_per_minute = max_tokens_per_minute
self.max_requests_per_minute = max_requests_per_minute self.max_requests_per_minute = max_requests_per_minute
@ -3823,8 +3822,8 @@ class RateLimitHandler():
self, self,
requests_filepath: str = "", requests_filepath: str = "",
jobs: list = [], jobs: list = [],
save_filepath: str = None, save_filepath: Optional[str] = None,
api_key: str = os.getenv("OPENAI_API_KEY"), api_key: Optional[str] = os.getenv("OPENAI_API_KEY"),
max_requests_per_minute: float = 3_000 * 0.5, max_requests_per_minute: float = 3_000 * 0.5,
max_tokens_per_minute: float = 250_000 * 0.5, max_tokens_per_minute: float = 250_000 * 0.5,
token_encoding_name: str = "cl100k_base", token_encoding_name: str = "cl100k_base",
@ -3851,7 +3850,7 @@ class RateLimitHandler():
request_header = {"Authorization": f"Bearer {api_key}"} request_header = {"Authorization": f"Bearer {api_key}"}
# initialize trackers # initialize trackers
queue_of_requests_to_retry = asyncio.Queue() queue_of_requests_to_retry = asyncio.Queue() # type: ignore
task_id_generator = ( task_id_generator = (
self.task_id_generator_function() self.task_id_generator_function()
) # generates integer IDs of 1, 2, 3, ... ) # generates integer IDs of 1, 2, 3, ...