From 483fe257d926804975ee49e0074f9d9e5d643fe1 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 1 May 2023 21:33:57 -0700 Subject: [PATCH] bump timeout (#3939) --- langchain/chat_models/openai.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/langchain/chat_models/openai.py b/langchain/chat_models/openai.py index ca851f6d..faa95606 100644 --- a/langchain/chat_models/openai.py +++ b/langchain/chat_models/openai.py @@ -3,7 +3,7 @@ from __future__ import annotations import logging import sys -from typing import Any, Callable, Dict, List, Mapping, Optional, Tuple +from typing import Any, Callable, Dict, List, Mapping, Optional, Tuple, Union from pydantic import Extra, Field, root_validator from tenacity import ( @@ -120,8 +120,8 @@ class ChatOpenAI(BaseChatModel): """Holds any model parameters valid for `create` call not explicitly specified.""" openai_api_key: Optional[str] = None openai_organization: Optional[str] = None - request_timeout: int = 120 - """Timeout in seconds for the OpenAPI request.""" + request_timeout: Optional[Union[float, Tuple[float, float]]] = None + """Timeout for requests to OpenAI completion API. Default is 600 seconds.""" max_retries: int = 6 """Maximum number of retries to make when generating.""" streaming: bool = False