Update utils.py

utils.py

@@ -34,6 +34,12 @@ example_selector = SemanticSimilarityExampleSelector.from_examples(
 
 db = SQLDatabase.from_uri("sqlite:///attendance_system.db")
 
+def query_as_list(db, query):
+    res = db.run(query)
+    res = [el for sub in ast.literal_eval(res) for el in sub if el]
+    res = [re.sub(r"\b\d+\b", "", string).strip() for string in res]
+    return list(set(res))
+
 employee = query_as_list(db, "SELECT FullName FROM Employee")
 
 vector_db = Chroma.from_texts(employee, OpenAIEmbeddings(model="text-embedding-3-small", api_key=os.environ['API_KEY']))
@@ -47,12 +53,6 @@ retriever_tool = create_retriever_tool(
 )
 
 
-def query_as_list(db, query):
-    res = db.run(query)
-    res = [el for sub in ast.literal_eval(res) for el in sub if el]
-    res = [re.sub(r"\b\d+\b", "", string).strip() for string in res]
-    return list(set(res))
-
 
 def get_answer(user_query):