expose error types

This commit is contained in:
Krrish Dholakia 2023-08-07 11:32:01 -07:00
parent 46fe03120e
commit 31c7e0a4a9
5 changed files with 5 additions and 4 deletions

View file

@ -1,10 +1,10 @@
from openai.error import AuthenticationError, InvalidRequestError, RateLimitError, OpenAIError
# from openai.error import AuthenticationError, InvalidRequestError, RateLimitError, OpenAIError
import os
import sys
import traceback
sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
from litellm import embedding, completion, AuthenticationError, InvalidRequestError, RateLimitError, ServiceUnavailableError, OpenAIError
from concurrent.futures import ThreadPoolExecutor
import pytest