changes to physics.app.py

physics_bot/physics_app.py

@@ -5,9 +5,8 @@ from langchain.schema.runnable import Runnable
 from langchain.schema.runnable.config import RunnableConfig
 import chainlit as cl
 from langchain_community.embeddings import OpenAIEmbeddings
-from langchain_core.prompts import PromptTemplate
 from langchain_community.vectorstores import FAISS
-
+from langchain_core.runnables.passthrough import RunnablePassthrough
 
 @cl.on_chat_start
 async def on_chat_start():
@@ -15,9 +14,9 @@ async def on_chat_start():
     vectorstore = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
     retriever = vectorstore.as_retriever(search_kwargs={"k": 1})
 
-    model = ChatOpenAI(streaming=True)
+    chat_model = ChatOpenAI(streaming=True, model="gpt-4")
 
-    prompt = ChatPromptTemplate.from_messages(
+    prompt_template = ChatPromptTemplate.from_messages(
         [
             (
                 "system",
@@ -33,12 +32,20 @@ async def on_chat_start():
                 "Example1: “The ball rolled towards the red car.” The red car would be the reference point."
                 "Example2: “The mall is two miles north of my house.” Your house would be the reference point.",
             ),
-            ("human", "{question}"),
+            ("human", "Using this context: {context}, please answer this question: {question}"),
         ]
     )
-
-    runnable = retriever| prompt | model | StrOutputParser()
-    cl.user_session.set("runnable", runnable)
+    parser = StrOutputParser()
+    runnable_chain = (
+            {
+                "context": retriever,
+                "question": RunnablePassthrough(),
+            }
+            | prompt_template
+            | chat_model
+            | parser
+    )
+    cl.user_session.set("runnable", runnable_chain)
 
 
 @cl.on_message