Update app.py

app.py

@@ -28,6 +28,9 @@ import langchain
 import os
 OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
 
+
+
+
 # 嵌入模型
 #embeddings = OpenAIEmbeddings()
 embeddings = HuggingFaceEmbeddings(model_name="BAAI/bge-large-en")
@@ -35,16 +38,19 @@ embeddings = HuggingFaceEmbeddings(model_name="BAAI/bge-large-en")
 # 加载数据
 #docsearch = FAISS.from_texts(texts, embeddings)
 docsearch = FAISS.load_local("./faiss_index", embeddings)
-chain = load_qa_chain(OpenAI(temperature=0,model_name="gpt-3.5-turbo", verbose=True), chain_type="stuff",verbose=True)
 
+#chain = load_qa_chain(OpenAI(temperature=0,model_name="gpt-3.5-turbo",prompt=chat_prompt), chain_type="stuff",verbose=True)
 
 
-template="您是回答所有ANSYS软件使用查询的得力助手,如果所问的内容不在范围内，请回答您提的问题不在本知识库内，请重新提问. {input_language} to {output_language}."
+
+template="您是回答ANSYS软件使用查询的得力助手,所有回复必需用中文 {input_language} to {output_language}."
 system_message_prompt = SystemMessagePromptTemplate.from_template(template)
 human_template="{text}"
 human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
 
-#prompt = "您是回答所有ANSYS软件使用查询的得力助手,如果所问的内容不在范围内，请回答您提的问题不在本知识库内，请重新提问，所有问题必需用中文回答"
+chat_prompt = ChatPromptTemplate.from_messages([system_message_prompt, human_message_prompt])
+chain = load_qa_chain(OpenAI(temperature=0,model_name="gpt-3.5-turbo",prompt=chat_prompt), chain_type="stuff",verbose=True)
+
 
 def predict(message, history):
     history_langchain_format = []