diff --git a/openai/api_requestor.py b/openai/api_requestor.py index 5970510434..99a392c148 100644 --- a/openai/api_requestor.py +++ b/openai/api_requestor.py @@ -7,7 +7,12 @@ from urllib.parse import urlencode, urlsplit, urlunsplit import requests -from typing_extensions import Literal + +# Literal is available from Python 3.8 +try: + from typing import Literal +except ImportError: + from typing_extensions import Literal import openai from openai import error, util, version diff --git a/setup.py b/setup.py index 0b6956ef0e..53c2b0c3a1 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ "pandas-stubs>=1.1.0.11", # Needed for type hints for mypy "openpyxl>=3.0.7", # Needed for CLI fine-tuning data preparation tool xlsx format "numpy", - "typing_extensions", # Needed for type hints for mypy + 'typing_extensions;python_version<"3.8"', # Needed for type hints for mypy ], extras_require={ "dev": ["black~=21.6b0", "pytest==6.*"],