switched from openai to azure openai

app.py

@@ -9,21 +9,21 @@ from dotenv import load_dotenv
 import arxiv
 import pinecone
 from langchain.embeddings.openai import OpenAIEmbeddings
+from langchain.embeddings.azure_openai import AzureOpenAIEmbeddings
+from langchain.chat_models import ChatOpenAI, AzureChatOpenAI
 from langchain.embeddings import CacheBackedEmbeddings
 from langchain.storage import LocalFileStore, InMemoryStore
 from utils.store import index_documents, search_and_index
 from utils.chain import create_chain
 from langchain.vectorstores import Pinecone
-from langchain.chat_models import ChatOpenAI
+
 from langchain.schema.runnable import RunnableSequence
 from langchain.schema import format_document
 from pprint import pprint
 from langchain_core.vectorstores import VectorStoreRetriever
 import langchain
 from langchain.cache import InMemoryCache
-from langchain_core.messages.human import HumanMessage
 from langchain.memory import ConversationBufferMemory
-from chainlit import make_async
 
 load_dotenv()
 YOUR_API_KEY = os.environ["PINECONE_API_KEY"]
@@ -42,14 +42,16 @@ async def start_chat():
     }
 
     await cl.Message(
-        content="What would you like to learn about today? 😊"
+        content="Hi, I am here to help you learn about a topic, what would you like to learn about today? 😊"
     ).send()
 
     # create an embedder through a cache interface (locally) (on start)
     store = InMemoryStore()
 
-    core_embeddings_model = OpenAIEmbeddings(
-        api_key=os.environ['OPENAI_API_KEY']
+    core_embeddings_model = AzureOpenAIEmbeddings(
+        api_key=os.environ['AZURE_OPENAI_API_KEY'],
+        azure_deployment="text-embedding-ada-002",
+        azure_endpoint=os.environ['AZURE_OPENAI_ENDPOINT']
     )
 
     embedder = CacheBackedEmbeddings.from_bytes_store(
@@ -71,13 +73,13 @@ async def start_chat():
             dimension=1536
         )
     index = pinecone.GRPCIndex(INDEX_NAME)
-    
-    # setup your ChatOpenAI model (on start)
-    llm = ChatOpenAI(
-        model=settings['model'],
+
+    llm = AzureChatOpenAI(
         temperature=settings['temperature'],
         max_tokens=settings['max_tokens'],
-        api_key=os.environ["OPENAI_API_KEY"],
+        api_key=os.environ['AZURE_OPENAI_API_KEY'],
+        azure_deployment="gpt-35-turbo-16k",
+        api_version="2023-07-01-preview",
         streaming=True
     )
 
@@ -89,7 +91,6 @@ async def start_chat():
     os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
 
     # setup memory
-    
     memory = ConversationBufferMemory(memory_key="chat_history")
 
     tools = {
@@ -102,7 +103,7 @@ async def start_chat():
     cl.user_session.set("settings", settings)
     cl.user_session.set("first_run", False)
 
-@cl.on_settings_update
+
 @cl.on_message  # marks a function that should be run each time the chatbot receives a message from a user
 async def main(message: cl.Message):
     settings = cl.user_session.get("settings")