lrtherond commited on
Commit
5191bb0
1 Parent(s): 9065fba

Prompt revision

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,7 +77,7 @@ llm = HuggingFaceEndpoint(
77
 
78
  prompt = PromptTemplate(
79
  template=template,
80
- input_variables=["history", "context", "question"],
81
  )
82
 
83
  memory = ConversationBufferWindowMemory(
@@ -94,7 +94,7 @@ rag_chain = RetrievalQA.from_chain_type(
94
  retriever=vector_store.as_retriever(search_kwargs={'k': 4}),
95
  chain_type_kwargs={
96
  "prompt": prompt,
97
- "memory": memory,
98
  },
99
  )
100
 
 
77
 
78
  prompt = PromptTemplate(
79
  template=template,
80
+ input_variables=["context", "question"]
81
  )
82
 
83
  memory = ConversationBufferWindowMemory(
 
94
  retriever=vector_store.as_retriever(search_kwargs={'k': 4}),
95
  chain_type_kwargs={
96
  "prompt": prompt,
97
+ # "memory": memory,
98
  },
99
  )
100