Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-16 06:13:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix litellm openai imports (#13307)

Browse Source
This commit is contained in:
Krish Dholakia 2023-11-13 17:55:10 -08:00 committed by GitHub
parent 1c67db4c18
commit 5a920e14c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/langchain/langchain/chat_models/litellm.py
View File

@ -62,14 +62,13 @@ def _create_retry_decorator(
] = None, ] = None,
) -> Callable[[Any], Any]: ) -> Callable[[Any], Any]:
"""Returns a tenacity retry decorator, preconfigured to handle PaLM exceptions""" """Returns a tenacity retry decorator, preconfigured to handle PaLM exceptions"""
import openai import litellm
errors = [ errors = [
openai.error.Timeout, litellm.Timeout,
openai.error.APIError, litellm.APIError,
openai.error.APIConnectionError, litellm.APIConnectionError,
openai.error.RateLimitError, litellm.RateLimitError,
openai.error.ServiceUnavailableError,
] ]
return create_base_retry_decorator( return create_base_retry_decorator(
error_types=errors, max_retries=llm.max_retries, run_manager=run_manager error_types=errors, max_retries=llm.max_retries, run_manager=run_manager