def update_model(app, message, **extra):
"""
Signal hook to be called after training is completed.
Reloads ml models and synonyms.
:param app:
:param message:
:param extra:
:return:
"""
sentence_classifier.load(PATH)
synonyms = get_synonyms()
global entity_extraction
entity_extraction = EntityExtractor(synonyms)
app.logger.info("Intent Model updated")
def update_model(app, message, **extra):
"""
Signal hook to be called after training is completed.
Reloads ml models and synonyms.
:param app:
:param message:
:param extra:
:return:
"""
global sentence_classifier
sentence_classifier = EmbeddingIntentClassifier.load(app.config["MODELS_DIR"])
synonyms = get_synonyms()
global entity_extraction
entity_extraction = EntityExtractor(synonyms)
app.logger.info("Intent Model updated")
def update_model(app,botId, **extra):
"""
Signal hook to be called after training is completed.
Reloads ml models and synonyms.
:param app:
:param message:
:param extra:
:return:
"""
PATH = "{}/{}".format(app.config["MODELS_DIR"],
botId+ ".model")
sentence_classifier.load(PATH)
synonyms = get_synonyms()
global entity_extraction
entity_extraction = EntityExtractor(synonyms)
app.logger.info("Intent Model updated")
def update_model(app, message, **extra):
"""
Signal hook to be called after training is completed.
Reloads ml models and synonyms.
:param app:
:param message:
:param extra:
:return:
"""
global sentence_classifier
sentence_classifier = EmbeddingIntentClassifier.load(
app.config["MODELS_DIR"], app.config["USE_WORD_VECTORS"])
synonyms = get_synonyms()
global entity_extraction
entity_extraction = EntityExtractor(synonyms)
app.logger.info("Intent Model updated")
from app.endpoint.utils import get_synonyms, SilentUndefined, split_sentence, call_api
endpoint = Blueprint('api', __name__, url_prefix='/api')
# Loading ML Models at app startup
from app.nlu.intent_classifer import IntentClassifier
with app.app_context():
PATH = "{}/{}".format(app.config["MODELS_DIR"],
app.config["INTENT_MODEL_NAME"])
sentence_classifier = IntentClassifier()
sentence_classifier.load(PATH)
synonyms = get_synonyms()
entity_extraction = EntityExtractor(synonyms)
app.logger.info("Intent Model loaded.")
# Request Handler
@endpoint.route('/v1', methods=['POST'])
def api():
"""
Endpoint to converse with chatbot.
Chat context is maintained by exchanging the payload between client and bot.
sample input/output payload =>
{
"currentNode": "",
def api():
"""
Endpoint to converse with chatbot.
Chat context is maintained by exchanging the payload between client and bot.
sample input/output payload =>
{
"currentNode": "",
"complete": false,
"parameters": [],
"extractedParameters": {},
"missingParameters": [],
"intent": {
},
"context": {},
"input": "hello",
"speechResponse": [
]
}
:param json:
:return json:
"""
request_json = request.get_json(silent=True)
result_json = request_json
botId = result_json["botId"]
session.__setattr__('botId',botId)
PATH = "{}/{}".format(app.config["MODELS_DIR"],
botId+".model")
sentence_classifier.load(PATH)
synonyms = get_synonyms()
entity_extraction = EntityExtractor(synonyms)
app.logger.info("Intent Model loaded.")
print(result_json)
if request_json:
context = {}
context["context"] = request_json["context"]
if request_json['role']:
session.__setattr__('role', request_json["role"])
print(role)
if app.config["DEFAULT_WELCOME_INTENT_NAME"] in request_json.get(
"input"):
intent = Intent.objects(
intentId=app.config["DEFAULT_WELCOME_INTENT_NAME"]).first()
result_json["complete"] = True
result_json["intent"]["intentId"] = intent.intentId
result_json["intent"]["id"] = str(intent.id)
result_json["input"] = request_json.get("input")
template = Template(
intent.speechResponse,
undefined=SilentUndefined)
result_json["speechResponse"] = split_sentence(template.render(**context))
logger.info(request_json.get("input"), extra=result_json)
return build_response.build_json(result_json)
intent_id, confidence = predict(request_json.get("input"))
intent = Intent.objects.get(id=ObjectId(intent_id))
print("Intent",intent)
if intent.parameters:
parameters = intent.parameters
else:
parameters = []
if ((request_json.get("complete") is None) or (
request_json.get("complete") is True)):
result_json["intent"] = {
"name": intent.name,
"confidence": confidence,
"id": str(intent.id)
}
if parameters:
# Extract NER entities
extracted_parameters = entity_extraction.predict(
intent_id, request_json.get("input"))
missing_parameters = []
result_json["missingParameters"] = []
result_json["extractedParameters"] = {}
result_json["parameters"] = []
for parameter in parameters:
result_json["parameters"].append({
"name": parameter.name,
"type": parameter.type,
"required": parameter.required
})
if parameter.required:
if parameter.name not in extracted_parameters.keys():
result_json["missingParameters"].append(
parameter.name)
missing_parameters.append(parameter)
result_json["extractedParameters"] = extracted_parameters
if missing_parameters:
result_json["complete"] = False
current_node = missing_parameters[0]
result_json["currentNode"] = current_node["name"]
result_json["speechResponse"] = split_sentence(current_node["prompt"])
else:
result_json["complete"] = True
context["parameters"] = extracted_parameters
else:
result_json["complete"] = True
elif request_json.get("complete") is False:
if "cancel" not in intent.name:
intent_id = request_json["intent"]["id"]
intent = Intent.objects.get(id=ObjectId(intent_id))
extracted_parameter = entity_extraction.replace_synonyms({
request_json.get("currentNode"): request_json.get("input")
})
# replace synonyms for entity values
result_json["extractedParameters"].update(extracted_parameter)
result_json["missingParameters"].remove(
request_json.get("currentNode"))
if len(result_json["missingParameters"]) == 0:
result_json["complete"] = True
context = {}
context["parameters"] = result_json["extractedParameters"]
context["context"] = request_json["context"]
else:
missing_parameter = result_json["missingParameters"][0]
result_json["complete"] = False
current_node = [
node for node in intent.parameters if missing_parameter in node.name][0]
result_json["currentNode"] = current_node.name
result_json["speechResponse"] = split_sentence(current_node.prompt)
else:
result_json["currentNode"] = None
result_json["missingParameters"] = []
result_json["parameters"] = {}
result_json["intent"] = {}
result_json["complete"] = True
if result_json["complete"]:
if intent.apiTrigger:
isJson = False
parameters = result_json["extractedParameters"]
headers = intent.apiDetails.get_headers()
app.logger.info("headers %s"%headers)
url_template = Template(
intent.apiDetails.url, undefined=SilentUndefined)
rendered_url = url_template.render(**context)
if intent.apiDetails.isJson:
isJson = True
request_template = Template(
intent.apiDetails.jsonData, undefined=SilentUndefined)
parameters = json.loads(request_template.render(**context))
try:
result = call_api(rendered_url,
intent.apiDetails.requestType,headers,
parameters, isJson)
except Exception as e:
app.logger.warn("API call failed", e)
result_json["speechResponse"] = ["Service is not available. Please try again later."]
else:
context["result"] = result
template = Template(
intent.speechResponse, undefined=SilentUndefined)
result_json["speechResponse"] = split_sentence(template.render(**context))
else:
context["result"] = {}
template = Template(intent.speechResponse,
undefined=SilentUndefined)
result_json["speechResponse"] = split_sentence(template.render(**context))
logger.info(request_json.get("input"), extra=result_json)
return build_response.build_json(result_json)
else:
return abort(400)