Update utils.py

utils.py

@@ -21,6 +21,30 @@ import ast
 import re
 
 parser = JsonOutputParser()
+llm = ChatOpenAI(model="gpt-4o-mini", temperature=0, api_key=os.environ['API_KEY'])
+example_selector = SemanticSimilarityExampleSelector.from_examples(
+    examples,
+    OpenAIEmbeddings(model="text-embedding-3-small", api_key=os.environ['API_KEY']),
+    Chroma(persist_directory="data"),
+    # Chroma,
+    k=5,
+    input_keys=["input"],
+)
+
+db = SQLDatabase.from_uri("sqlite:///attendance_system.db")
+
+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']))
+retriever = vector_db.as_retriever(search_kwargs={"k": 15})
+description = """Use to look up values to filter on. Input is an approximate spelling of the proper noun, output is \
+valid proper nouns. Use the noun most similar to the search."""
+retriever_tool = create_retriever_tool(
+    retriever,
+    name="search_proper_nouns",
+    description=description,
+)
+
 
 def query_as_list(db, query):
     res = db.run(query)