Update app.py

app.py

@@ -1,13 +1,14 @@
 import os
 from typing import List
 
-from langchain.embeddings.openai import OpenAIEmbeddings
+from langchain.embeddings import CohereEmbeddings
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores.chroma import Chroma
 from langchain.chains import (
     ConversationalRetrievalChain,
 )
-from langchain.chat_models import ChatOpenAI
+from langchain.llms.fireworks import Fireworks
+from langchain.chat_models.fireworks import ChatFireworks
 from langchain.prompts.chat import (
     ChatPromptTemplate,
     SystemMessagePromptTemplate,
@@ -17,11 +18,12 @@ from langchain.docstore.document import Document
 from langchain.memory import ChatMessageHistory, ConversationBufferMemory
 from langsmith_config import setup_langsmith_config
 import openai
+import fireworks.client
 import chainlit as cl
 
-openai.api_key = os.environ["OPENAI_API_KEY"]
+FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY")
 setup_langsmith_config()
-
+os.environ["FIREWORKS_API_KEY"] = FIREWORKS_API_KEY
 text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
 
 system_template = """Use the following pieces of context to answer the users question.
@@ -78,7 +80,8 @@ async def on_chat_start():
     metadatas = [{"source": f"{i}-pl"} for i in range(len(texts))]
 
     # Create a Chroma vector store
-    embeddings = OpenAIEmbeddings()
+    embeddings = CohereEmbeddings(cohere_api_key="COHERE_API_KEY")
+
     docsearch = await cl.make_async(Chroma.from_texts)(
         texts, embeddings, metadatas=metadatas
     )