moriire commited on
Commit
cb52748
1 Parent(s): 7801839

Update app/llm.py

Browse files
Files changed (1) hide show
  1. app/llm.py +2 -2
app/llm.py CHANGED
@@ -36,10 +36,10 @@ class RagChat:
36
  vectorstore = Chroma.from_documents(documents=self.agent, embedding=GPT4AllEmbeddings())
37
  return vectorstore
38
 
39
- def prompt_template(self, question):
40
  chain = (
41
  RunnablePassthrough.assign(context=RunnablePick("context") | format_docs)
42
- | rag_prompt
43
  | llm
44
  | StrOutputParser()
45
  )
 
36
  vectorstore = Chroma.from_documents(documents=self.agent, embedding=GPT4AllEmbeddings())
37
  return vectorstore
38
 
39
+ def chat(self, question):
40
  chain = (
41
  RunnablePassthrough.assign(context=RunnablePick("context") | format_docs)
42
+ | rag_prompt_llama
43
  | llm
44
  | StrOutputParser()
45
  )