added template

app.py

@@ -15,6 +15,8 @@ from langchain.vectorstores import Chroma
 
 from langchain.chains import RetrievalQA
 
+from langchain.prompts import ChatPromptTemplate
+
 #web_links = ["https://www.databricks.com/","https://help.databricks.com","https://docs.databricks.com","https://kb.databricks.com/","http://docs.databricks.com/getting-started/index.html","http://docs.databricks.com/introduction/index.html","http://docs.databricks.com/getting-started/tutorials/index.html","http://docs.databricks.com/machine-learning/index.html","http://docs.databricks.com/sql/index.html"]
 #loader = WebBaseLoader(web_links)
 #documents = loader.load()
@@ -40,9 +42,37 @@ def bot(history):
 
 def infer(question):
     
-    query = question
-    result = qa({"query": query})
+    #query = question
+    #result = qa({"query": query})
+    chat_template = ChatPromptTemplate.from_messages(
+        [
+            ("system", "You are a helpful Doc AI bot. Your name is {name}."),
+            ("human", "Hello, how can I install RAY?"),
+            ("ai", "To install Ray, you can use pip: \
+            pip install ray Or, if you want to install a specific version of Ray, you can use: \
+            \
+                pip install -U 'ray[version]' \
+            \
+            Where [version] is the version you want to install. \
+            If you are on Arch Linux, you can install Ray from the Arch User Repository (AUR) using an AUR helper like yay: \
+            \
+                yay -S python-ray \
+            \
+            You can also manually install the package by following the instructions on the Arch Wiki. \
+            \
+            Sources: \
+            \
+                https://docs.ray.io/en/master/ray-core/examples/gentle_walkthrough.html#install-ray \
+                https://docs.ray.io/en/master/ray-more-libs/installation.html#installing-ray-on-arch-linux \
+                https://docs.ray.io/en/master/ray-overview/installation.html#installing-ray-on-arch-linux \
+                https://docs.ray.io/en/master/installation.html#installing-ray-on-arch-linux \
+                https://docs.ray.io/en/master/installation.html#from-wheels)",
+            ("human", "{user_input}"),
+        ]
+    )
 
+    message = chat_template.format_messages(name="RAYDOC", user_input=question)
+    result = qa({"query": message})
     return result
 
 css="""