Skip to content

Commit 1186d59

Browse files
committed
use intent_id insted of object id for models,response json etc
1 parent 7ece5ba commit 1186d59

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

app/endpoint/controllers.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,8 @@ def api():
6161
intent = Intent.objects(
6262
intentId=app.config["DEFAULT_WELCOME_INTENT_NAME"]).first()
6363
result_json["complete"] = True
64-
result_json["intent"]["intentId"] = intent.intentId
65-
result_json["intent"]["id"] = str(intent.id)
64+
result_json["intent"]["object_id"] = str(intent.id)
65+
result_json["intent"]["id"] = str(intent.intentId)
6666
result_json["input"] = request_json.get("input")
6767
template = Template(
6868
intent.speechResponse,
@@ -74,7 +74,7 @@ def api():
7474

7575
intent_id, confidence,suggetions = predict(request_json.get("input"))
7676
app.logger.info("Suggetions => %s"%suggetions)
77-
intent = Intent.objects.get(id=ObjectId(intent_id))
77+
intent = Intent.objects.get(intentId=intent_id)
7878

7979
if intent.parameters:
8080
parameters = intent.parameters
@@ -84,9 +84,9 @@ def api():
8484
if ((request_json.get("complete") is None) or (
8585
request_json.get("complete") is True)):
8686
result_json["intent"] = {
87-
"name": intent.name,
87+
"object_id": str(intent.id),
8888
"confidence": confidence,
89-
"id": str(intent.id)
89+
"id": str(intent.intentId)
9090
}
9191

9292
if parameters:
@@ -127,7 +127,7 @@ def api():
127127
elif request_json.get("complete") is False:
128128
if "cancel" not in intent.name:
129129
intent_id = request_json["intent"]["id"]
130-
intent = Intent.objects.get(id=ObjectId(intent_id))
130+
intent = Intent.objects.get(intentId=intent_id)
131131

132132
extracted_parameter = entity_extraction.replace_synonyms({
133133
request_json.get("currentNode"): request_json.get("input")

app/intents/models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def get_headers(self):
4343
class Intent(Document):
4444
name = StringField(max_length=100, required=True, unique=True)
4545
userDefined = BooleanField(default=True)
46-
intentId = StringField(required=True)
46+
intentId = StringField(required=True,unique=True)
4747
apiTrigger = BooleanField(required=True)
4848
apiDetails = EmbeddedDocumentField(ApiDetails)
4949
speechResponse = StringField(required=True)

app/nlu/tasks.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def train_models():
2424

2525
# train ner model for each Stories
2626
for intent in intents:
27-
train_all_ner(str(intent.id), intent.trainingData)
27+
train_all_ner(str(intent.intentId), intent.trainingData)
2828

2929
model_updated_signal.send(app,message="Training Completed.")
3030

@@ -42,7 +42,7 @@ def train_intent_classifier(intents):
4242
if example.get("text").strip() == "":
4343
continue
4444
X.append(example.get("text"))
45-
y.append(str(intent.id))
45+
y.append(str(intent.intentId))
4646

4747
intent_classifier = EmbeddingIntentClassifier()
4848
intent_classifier.train(X,y)

0 commit comments

Comments
 (0)