TYPE_CHECKING added

This commit is contained in:
mateusz.wosinski 2023-09-07 19:56:53 +02:00
parent 868db99b17
commit ff1c6de86c

View File

@ -1,5 +1,5 @@
import tempfile import tempfile
from typing import Dict, Optional, Union from typing import Dict, Optional, Union, TYPE_CHECKING
from langchain.callbacks.manager import CallbackManagerForToolRun from langchain.callbacks.manager import CallbackManagerForToolRun
from langchain.pydantic_v1 import root_validator from langchain.pydantic_v1 import root_validator
@ -7,13 +7,14 @@ from langchain.tools.base import BaseTool
from langchain.tools.eleven_labs.models import ElevenLabsModel from langchain.tools.eleven_labs.models import ElevenLabsModel
from langchain.utils import get_from_dict_or_env from langchain.utils import get_from_dict_or_env
try: if TYPE_CHECKING:
import elevenlabs try:
import elevenlabs
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"elevenlabs is not installed. " "Run `pip install elevenlabs` to install." "elevenlabs is not installed. " "Run `pip install elevenlabs` to install."
) )
class ElevenLabsText2SpeechTool(BaseTool): class ElevenLabsText2SpeechTool(BaseTool):