File size: 685 Bytes
8cb6e00 7c36d75 8cb6e00 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
INITIAL_CHAT_QUESTIONS = [
"What insights can I get from this data?",
"Can you summarize the dataset?",
]
INITIAL_CHART_QUESTIONS = [
"Can you provide a visualization of the data?",
]
def if_initial_chat_question(query: str) -> bool:
# Convert the query to lowercase and check if it's in the list (case insensitive)
if query.lower() in [q.lower() for q in INITIAL_CHAT_QUESTIONS]:
return True
return False
def if_initial_chart_question(query: str) -> bool:
# Convert the query to lowercase and check if it's in the list (case insensitive)
if query.lower() in [q.lower() for q in INITIAL_CHART_QUESTIONS]:
return True
return False
|