Sbnos commited on
Commit
36ccd09
·
verified ·
1 Parent(s): ddbcdf4

update, because last one not working

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -3,10 +3,7 @@ import os
3
  from langchain_community.vectorstores import Chroma
4
  from langchain_community.embeddings import HuggingFaceBgeEmbeddings
5
  from langchain_community.llms import Together
6
- from langchain import hub
7
- from operator import itemgetter
8
- from langchain.chains import LLMChain
9
- from langchain.chains import ConversationalRetrievalChain
10
  from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder, PromptTemplate
11
  from langchain.memory import ConversationBufferMemory
12
  from langchain_community.chat_message_histories import StreamlitChatMessageHistory
@@ -98,12 +95,15 @@ def app():
98
  Question: {question}"""
99
  ANSWER_PROMPT = ChatPromptTemplate.from_template(answer_template)
100
 
 
 
 
101
  conversational_qa_chain = ConversationalRetrievalChain(
102
  retriever=retriever,
103
  memory=memory,
104
  combine_docs_chain=_combine_documents,
105
- condense_question_chain=LLMChain(llm=llmc, prompt=CONDENSE_QUESTION_PROMPT),
106
- qa_chain=LLMChain(llm=llm, prompt=ANSWER_PROMPT)
107
  )
108
 
109
  st.header("Ask Away!")
 
3
  from langchain_community.vectorstores import Chroma
4
  from langchain_community.embeddings import HuggingFaceBgeEmbeddings
5
  from langchain_community.llms import Together
6
+ from langchain.chains import ConversationalRetrievalChain, LLMChain
 
 
 
7
  from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder, PromptTemplate
8
  from langchain.memory import ConversationBufferMemory
9
  from langchain_community.chat_message_histories import StreamlitChatMessageHistory
 
95
  Question: {question}"""
96
  ANSWER_PROMPT = ChatPromptTemplate.from_template(answer_template)
97
 
98
+ question_generator_chain = LLMChain(llm=llmc, prompt=CONDENSE_QUESTION_PROMPT)
99
+ qa_chain = LLMChain(llm=llm, prompt=ANSWER_PROMPT)
100
+
101
  conversational_qa_chain = ConversationalRetrievalChain(
102
  retriever=retriever,
103
  memory=memory,
104
  combine_docs_chain=_combine_documents,
105
+ question_generator=question_generator_chain,
106
+ qa_chain=qa_chain
107
  )
108
 
109
  st.header("Ask Away!")