Update langchain_bot.py

langchain_bot.py

@@ -53,4 +53,37 @@ memory = ConversationBufferWindowMemory(
     output_key="answer", #output key 出力ののキー名
     k=8, #saved conversation number 保持する会話の履歴数
     return_messages=True, #get chat list チャット履歴をlistで取得する場合はTrue
-)
+)
+
+
+# RAG conversation chain （RAG用）会話chainの設定
+chain = ConversationalRetrievalChain.from_llm(
+    llm=LLM,
+    retriever=retriever,
+    combine_docs_chain_kwargs={'prompt': QUESTION_PROMPT}, # プロンプトをセット
+    chain_type="stuff", # 検索した文章の処理方法
+    memory=memory  # メモリーをセット
+)
+
+def invoke_question_time(chain,question):
+    
+    start_time = time.time()
+    response = chain.invoke({"question": question})
+    end_time = time.time()
+    print(response["answer"])
+    print("responce time:", end_time - start_time, "seconds")
+    return response, end_time - start_time
+
+#Test main
+def main():
+   
+    response,_=invoke_question_time(chain, "Hello what is precious plastic ? ")
+    time.sleep(30)
+    response,_=invoke_question_time(chain, """I live in the UK and want so start an extruder work shop. 
+                                    What is needed ? What should safety issues might this have and where could i buy equipment could i buy in the UK to help me
+                                """)
+    time.sleep(30)
+    response,_=invoke_question_time(chain, "Hello what is precious plastic ? ")
+
+if __name__ ==" __main__":
+    main()