From 34eed5cf3bec24fcefdb133c97092a22eb2832d8 Mon Sep 17 00:00:00 2001 From: ahmed531998 Date: Thu, 20 Apr 2023 05:18:14 +0200 Subject: [PATCH] janet --- NLU.py | 2 +- ResponseGenerator.py | 2 +- main.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/NLU.py b/NLU.py index 5fcdf0f..d39bc51 100644 --- a/NLU.py +++ b/NLU.py @@ -89,7 +89,7 @@ class NLU: Query -> coref resolution & intent extraction -> if intents are not confident or if query is ambig -> rewrite query and recheck -> if still ambig, ask a clarifying question """ if utterance.lower() in ["help", "list resources", "list papers", "list datasets", "list topics"]: - return {"modified_query": utterance, "intent": "COMMAND", "entities": [], "is_offensive": False, "is_clear": True} + return {"modified_query": utterance.lower(), "intent": "COMMAND", "entities": [], "is_offensive": False, "is_clear": True} self.to_process = utterance diff --git a/ResponseGenerator.py b/ResponseGenerator.py index 4188c81..6c2da8c 100644 --- a/ResponseGenerator.py +++ b/ResponseGenerator.py @@ -247,7 +247,7 @@ class ResponseGenerator: elif action == "command": if utterance == "help": - return self.gen_response(action="Help", name=name, vrename=vrename) + return self.gen_response(action="getHelp", vrename=vrename) elif utterance == "list resources": return self.gen_response(action="listResources", vrename=vrename) elif utterance == "list papers": diff --git a/main.py b/main.py index 5de38d5..09b04ad 100644 --- a/main.py +++ b/main.py @@ -130,7 +130,7 @@ def predict(): rec.generate_recommendations(users[token]['username'], new_user_interests, new_vre_material) dm.update(state) action = dm.next_action() - response = rg.gen_response(action=action, utterance=state['modified_query'], state=dm.get_recent_state(), consec_history=dm.get_consec_history(), chitchat_history=dm.get_chitchat_history(), vrename=vre.name) + response = rg.gen_response(action=action, utterance=state['modified_query'], state=dm.get_recent_state(), consec_history=dm.get_consec_history(), chitchat_history=dm.get_chitchat_history(), vrename=vre.name, username=users[token]['username'], name=users[token]['name'].split()[0]) message = {"answer": response, "query": text, "cand": "candidate", "history": dm.get_consec_history(), "modQuery": state['modified_query']} if state['intent'] == "QA": response = response.split("_______ \n The answer is: ")[1]