This commit is contained in:
ahmed531998 2023-04-20 05:18:14 +02:00
parent f0955da5cc
commit 34eed5cf3b
3 changed files with 3 additions and 3 deletions

2
NLU.py
View File

@ -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

View File

@ -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":

View File

@ -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]