(fix) request_timeout mapped to timeout for openai

This commit is contained in:
ishaan-jaff 2023-11-21 13:22:21 -08:00
parent 20b129b7ad
commit 589cdce8d3

View file

@ -338,8 +338,8 @@ def completion(
if mock_response: if mock_response:
return mock_completion(model, messages, stream=stream, mock_response=mock_response) return mock_completion(model, messages, stream=stream, mock_response=mock_response)
if timeout is None: if timeout is None:
timeout = 600 # set timeout for 10 minutes by default timeout = kwargs.get("request_timeout", None) or 600 # set timeout for 10 minutes by default
timeout = float(timeout) timeout = float(timeout)
try: try:
if base_url is not None: if base_url is not None: