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="""