expose error types

This commit is contained in:
Krrish Dholakia 2023-08-07 11:32:01 -07:00
parent 5e564ef5de
commit 1a4044091a
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