Chris4K commited on
Commit
29786ae
1 Parent(s): acb34d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -20
app.py CHANGED
@@ -145,30 +145,18 @@ def get_conversational_rag_chain(retriever_chain):
145
 
146
  return create_retrieval_chain(retriever_chain, stuff_documents_chain)
147
 
148
- def get_response(user_input):
149
- retriever_chain = get_context_retriever_chain(st.session_state.vector_store)
150
- conversation_rag_chain = get_conversational_rag_chain(retriever_chain)
151
-
152
- response = conversation_rag_chain.invoke({
153
- "chat_history": st.session_state.chat_history,
154
- "input": user_query
155
- })
156
 
157
  return response
158
 
159
- app = FastAPI()
160
 
161
- @app.post("/generate/")
162
- def generate(user_input):
163
- retriever_chain = get_context_retriever_chain(st.session_state.vector_store)
164
- conversation_rag_chain = get_conversational_rag_chain(retriever_chain)
165
-
166
- response = conversation_rag_chain.invoke({
167
- "chat_history": [],
168
- "input": user_query
169
- })
170
-
171
- return response
172
 
173
  ###################
174
 
@@ -210,6 +198,18 @@ def get_response(user_input):
210
  return last_part
211
 
212
 
 
 
 
 
 
 
 
 
 
 
 
 
213
  def history_to_dialog_format(chat_history: list[str]):
214
  dialog = []
215
  if len(chat_history) > 0:
 
145
 
146
  return create_retrieval_chain(retriever_chain, stuff_documents_chain)
147
 
148
+ #def get_response(user_input):
149
+ # retriever_chain = get_context_retriever_chain(st.session_state.vector_store)
150
+ # conversation_rag_chain = get_conversational_rag_chain(retriever_chain)
151
+ #
152
+ # response = conversation_rag_chain.invoke({
153
+ # "chat_history": st.session_state.chat_history,
154
+ # "input": user_query
155
+ # })
156
 
157
  return response
158
 
 
159
 
 
 
 
 
 
 
 
 
 
 
 
160
 
161
  ###################
162
 
 
198
  return last_part
199
 
200
 
201
+ ###############
202
+
203
+ from fastapi import FastAPI
204
+
205
+ app = FastAPI()
206
+
207
+ @app.post("/generate/")
208
+ def generate(user_input):
209
+ print("----yuhu -----")
210
+ return get_response(user_input)
211
+ ##################
212
+
213
  def history_to_dialog_format(chat_history: list[str]):
214
  dialog = []
215
  if len(chat_history) > 0: