Update functions.py

functions.py

@@ -43,8 +43,8 @@ def setup_vector_db():
 
     nist_chunks = text_splitter.split_documents(nist_doc)
 
-    # embeddings_small = AzureOpenAIEmbeddings(azure_deployment="text-embedding-3-small")
-    embeddings_small = OpenAIEmbeddings(model="text-embedding-3-small")
+    embeddings_small = AzureOpenAIEmbeddings(azure_deployment="text-embedding-3-small")
+    # embeddings_small = OpenAIEmbeddings(model="text-embedding-3-small")
 
     qdrant_client = QdrantClient(":memory:") # set Qdrant DB and its location (in-memory)
 
@@ -69,8 +69,8 @@ def setup_vector_db():
 
 # define a global variable to store the retriever object
 retriever = setup_vector_db()
-# qa_gpt4_llm = AzureChatOpenAI(azure_deployment="gpt-4o-mini", temperature=0) # GPT-4o-mini model
-qa_gpt4_llm = ChatOpenAI(model="gpt-4o-mini", temperature=0) # GPT-4o-mini model
+qa_gpt4_llm = AzureChatOpenAI(azure_deployment="gpt-4o-mini", temperature=0) # GPT-4o-mini model
+# qa_gpt4_llm = ChatOpenAI(model="gpt-4o-mini", temperature=0) # GPT-4o-mini model
 
 # define a template for the RAG model
 rag_template = """