github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-29 18:08:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix linters, update notebook

Browse Source
This commit is contained in:
mateusz.wosinski 2023-09-06 10:22:42 +02:00
parent 882a588264
commit 7b7bea5424
4 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/extras/integrations/tools/eleven_labs_tts.ipynb
View File

@ -108,7 +108,7 @@
"metadata": {},
"outputs": [],
"source": [
"tts.stream(text_to_speak)"
"tts.stream_speech(text_to_speak)"
]
},
{

4
libs/langchain/langchain/agents/load_tools.py
View File

@ -286,8 +286,8 @@ def _get_dataforseo_api_search_json(**kwargs: Any) -> BaseTool:
return DataForSeoAPISearchResults(api_wrapper=DataForSeoAPIWrapper(**kwargs))
def _get_eleven_labs_text2speech() -> BaseTool:
return ElevenLabsText2SpeechTool()
def _get_eleven_labs_text2speech(**kwargs: Any) -> BaseTool:
return ElevenLabsText2SpeechTool(**kwargs)
_EXTRA_LLM_TOOLS: Dict[

2
libs/langchain/langchain/tools/eleven_labs/__init__.py
View File

@ -2,4 +2,4 @@
from langchain.tools.eleven_labs.text2speech import ElevenLabsText2SpeechTool
__all__ = [ElevenLabsText2SpeechTool]
__all__ = ["ElevenLabsText2SpeechTool"]

9
libs/langchain/langchain/tools/eleven_labs/text2speech.py
View File

@ -1,6 +1,7 @@
import tempfile
from typing import Dict, Union
from typing import Dict, Optional, Union
from langchain.callbacks.manager import CallbackManagerForToolRun
from langchain.pydantic_v1 import root_validator
from langchain.tools.base import BaseTool
from langchain.tools.eleven_labs.models import ElevenLabsModel
@ -45,7 +46,9 @@ class ElevenLabsText2SpeechTool(BaseTool):
f.write(speech)
return f.name
def _run(self, query: str) -> str:
def _run(
self, query: str, run_manager: Optional[CallbackManagerForToolRun] = None
) -> str:
"""Use the tool."""
try:
speech_file = self._text2speech(query)
@ -60,7 +63,7 @@ class ElevenLabsText2SpeechTool(BaseTool):
elevenlabs.play(speech)
def stream(self, query: str) -> None:
def stream_speech(self, query: str) -> None:
"""Stream the text as speech as it is generated.
Play the text in your speakers."""
speech_stream = elevenlabs.generate(text=query, model=self.model, stream=True)