Update llm/oai.py
Browse files- llm/oai.py +1 -0
llm/oai.py
CHANGED
|
@@ -7,6 +7,7 @@ from pprint import pformat
|
|
| 7 |
from typing import Dict, Iterator, List, Optional, Literal, Union
|
| 8 |
|
| 9 |
import openai
|
|
|
|
| 10 |
|
| 11 |
if openai.__version__.startswith('0.'):
|
| 12 |
from openai.error import OpenAIError # noqa
|
|
|
|
| 7 |
from typing import Dict, Iterator, List, Optional, Literal, Union
|
| 8 |
|
| 9 |
import openai
|
| 10 |
+
from openai import OpenAIError, RateLimitError
|
| 11 |
|
| 12 |
if openai.__version__.startswith('0.'):
|
| 13 |
from openai.error import OpenAIError # noqa
|