klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Lint and formatting fixes (#554)

Browse Source 
Co-authored-by: cachho <admin@ch-webdev.com>
Co-authored-by: Taranjeet Singh <reachtotj@gmail.com>
This commit is contained in:
Dev Khant
2023-09-06 04:24:19 +05:30
committed by GitHub
parent 6481b555b4
commit 129242534d
13 changed files with 33 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
embedchain/bots/__init__.py
View File

@@ -1,4 +1,5 @@
from embedchain.bots.poe import PoeBot
from embedchain.bots.whatsapp import WhatsAppBot
from embedchain.bots.poe import PoeBot # noqa: F401
from embedchain.bots.whatsapp import WhatsAppBot # noqa: F401
# TODO: fix discord import
# from embedchain.bots.discord import DiscordBot
# from embedchain.bots.discord import DiscordBot

17
embedchain/bots/whatsapp.py
View File

@@ -1,4 +1,5 @@
import argparse
import importlib
import logging
import signal
import sys
@@ -11,8 +12,14 @@ from .base import BaseBot
@register_deserializable
class WhatsAppBot(BaseBot):
def __init__(self):
from flask import Flask, request
from twilio.twiml.messaging_response import MessagingResponse
try:
self.flask = importlib.import_module("flask")
self.twilio = importlib.import_module("twilio")
except ModuleNotFoundError:
raise ModuleNotFoundError(
"The required dependencies for WhatsApp are not installed. "
'Please install with `pip install --upgrade "embedchain[whatsapp]"`'
) from None
super().__init__()
def handle_message(self, message):
@@ -41,7 +48,7 @@ class WhatsAppBot(BaseBot):
return response
def start(self, host="0.0.0.0", port=5000, debug=True):
app = Flask(__name__)
app = self.flask.Flask(__name__)
def signal_handler(sig, frame):
logging.info("\nGracefully shutting down the WhatsAppBot...")
@@ -51,9 +58,9 @@ class WhatsAppBot(BaseBot):
@app.route("/chat", methods=["POST"])
def chat():
incoming_message = request.values.get("Body", "").lower()
incoming_message = self.flask.request.values.get("Body", "").lower()
response = self.handle_message(incoming_message)
twilio_response = MessagingResponse()
twilio_response = self.twilio.twiml.messaging_response.MessagingResponse()
twilio_response.message(response)
return str(twilio_response)