fix(langchain): try/catch langchain import in service; Only langchain is installed with the [langchain] extra (#190)

This commit is contained in:
TomTom101
2024-05-31 10:19:27 +02:00
parent b19243ab75
commit 4b39efeee3
3 changed files with 17 additions and 10 deletions

View File

@@ -36,7 +36,7 @@ try:
except ModuleNotFoundError as e:
logger.exception(
"You need to `pip install langchain langchain-openai langchain-community` for this example. Also, be sure to set `OPENAI_API_KEY` in the environment variable."
"In order to run this example you need to `pip install pipecat-ai[langchain] langchain-community langchain-openai. Also, be sure to set `OPENAI_API_KEY` in the environment variable."
)
raise Exception(f"Missing module: {e}")
@@ -94,12 +94,12 @@ async def main(room_url: str, token):
pipeline = Pipeline(
[
transport.input(), # Transport user input
tma_in, # User responses
lc, # Langchain
tts, # TTS
transport.output(), # Transport bot output
tma_out, # Assistant spoken responses
transport.input(), # Transport user input
tma_in, # User responses
lc, # Langchain
tts, # TTS
transport.output(), # Transport bot output
tma_out, # Assistant spoken responses
]
)