From 268358bffc3a45f55ae575fb54074832fc34463e Mon Sep 17 00:00:00 2001 From: Ashim Mahara <48154590+ashim-mahara@users.noreply.github.com> Date: Wed, 7 Aug 2024 04:51:53 -0400 Subject: [PATCH] Add OpenAI request timeout (#858) * Adds the timeout kwarg for the OpenAI LLM * Don't overwrite the client level timeout * Update version --------- Co-authored-by: Agus Co-authored-by: plaguss --- src/distilabel/llms/openai.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/distilabel/llms/openai.py b/src/distilabel/llms/openai.py index b75359d569..39644e2812 100644 --- a/src/distilabel/llms/openai.py +++ b/src/distilabel/llms/openai.py @@ -247,7 +247,6 @@ async def agenerate( # type: ignore "temperature": temperature, "top_p": top_p, "stop": stop, - "timeout": 50, } if response_format is not None: