Upload 4 files

app.py

@@ -0,0 +1,14 @@
+
+import gradio as gr 
+import os 
+from langchain_bot import chain 
+
+chat_history=[]
+def chat_messages(prompt,chat_history):
+    response = chain.invoke({"question":prompt})
+    return response['answer']
+
+gr.ChatInterface(
+    chat_messages
+    ).launch(debug=True)
+

langchain_bot.py

@@ -0,0 +1,39 @@
+import os 
+import json
+from langchain_openai import ChatOpenAI
+from langchain_core.prompts import MessagesPlaceholder
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.runnables import RunnablePassthrough
+from langchain.chains.conversational_retrieval.base import ConversationalRetrievalChain
+from langchain.memory.buffer_window import ConversationBufferWindowMemory
+from langchain_core.prompts import PromptTemplate
+from langchain.memory import VectorStoreRetrieverMemo
+### Contextualize question ###
+from langchain.chains import create_history_aware_retriever, create_retrieval_chain
+
+
+settings= json.load(open("system.json","r"))
+
+
+from upstash_vector import Index
+from langchain_community.vectorstores.upstash import UpstashVectorStore
+index = Index(os.environ["UPSTASH_VECTOR_REST_URL"],os.environ["UPSTASH_VECTOR_REST_TOKEN"])
+vectorStore = UpstashVectorStore(
+    embedding=True, index=index,
+)
+retriever = vectorStore.as_retriever(search_kwargs={"k": settings["k"]})
+#LLM setup
+LLM= ChatOpenAI(model=settings["model"], temperature=settings["temp"])
+
+#Setup prompt template 
+QUESTION_PROMPT = PromptTemplate(
+    template=settings["prompt_temp"], # プロンプトテンプレートをセット
+    input_variables=["context", "question"] # プロンプトに挿入する変数
+)
+# Conversation memory
+memory = ConversationBufferWindowMemory(
+    memory_key=settings["MEMORY_KEY"], #  Memory key メモリーのキー名
+    output_key="answer", #output key 出力ののキー名
+    k=8, #saved conversation number 保持する会話の履歴数
+    return_messages=True, #get chat list チャット履歴をlistで取得する場合はTrue
+)

system.json

@@ -0,0 +1,18 @@
+[{
+    "model":"gpt-3.5-turbo-1106",
+    "temp":0.2,
+    "MEMORY_KEY":"chat_history",
+    "k":20,
+    "prompt_temp":"""
+    You are an AI Chatbot from precious Plastic your job is to answer Question about recycling plastic.
+    You can return links to in the answer as well as image if you want
+    Us the following context to help in answering the Question.
+    
+    ------
+    {context}
+    ------
+    Question: {question}
+    Do not:
+     ・ Do not make thing up that you do not know, if you dont know, say that you dont know  \
+    """
+}]