Update utils.py

utils.py

@@ -50,6 +50,26 @@ for sheet_name in xls.sheet_names:
     
     i += 1
 
+def extract_question_type(llm, query):
+    messages = [
+    (
+        "system",
+        """
+        You are an AI assistant that determines if a question seeks any specific information about any item or any table or a generic information of the database. \
+    Answer only "specific information" or "generic information".""",
+    ),
+    ("human", query),
+    ]
+    output = llm.invoke(messages)
+    pred = output.content
+    
+    if 'generic' in pred:
+        return 'generic'
+    elif 'specific' in pred:
+        return 'specific'
+    else:
+        return 'unknown'
+        
 class EmbeddingsSearch:
     def __init__(self, metadata_df, emb_model):