diff --git a/embedchain/bots/discord.py b/embedchain/bots/discord.py index d2984982d6..2d2d482e45 100644 --- a/embedchain/bots/discord.py +++ b/embedchain/bots/discord.py @@ -2,14 +2,21 @@ import logging import os -import discord -from discord import app_commands -from discord.ext import commands - from embedchain.helper.json_serializable import register_deserializable from .base import BaseBot +try: + import discord + from discord import app_commands + from discord.ext import commands +except ModuleNotFoundError: + raise ModuleNotFoundError( + "The required dependencies for Discord are not installed." + 'Please install with `pip install "embedchain[discord]"`' + ) from None + + intents = discord.Intents.default() intents.message_content = True client = discord.Client(intents=intents) diff --git a/embedchain/bots/poe.py b/embedchain/bots/poe.py index 58e82bfb76..ebd71824eb 100644 --- a/embedchain/bots/poe.py +++ b/embedchain/bots/poe.py @@ -3,12 +3,17 @@ import os from typing import List, Optional -from fastapi_poe import PoeBot, run - from embedchain.helper.json_serializable import register_deserializable from .base import BaseBot +try: + from fastapi_poe import PoeBot, run +except ModuleNotFoundError: + raise ModuleNotFoundError( + "The required dependencies for Poe are not installed." 'Please install with `pip install "embedchain[poe]"`' + ) from None + def start_command(): parser = argparse.ArgumentParser(description="EmbedChain PoeBot command line interface")