switched back to langchain

app.py

@@ -74,8 +74,8 @@ def generate(
 ) -> Iterator[str]:
     
     documents = load_documents("documents/blockchain")
-    session_state["document_db"] = prepare_documents(documents)
-    context, sources = get_context_sources(session_state["original_message"], session_state["document_db"])
+    DB = prepare_documents(documents)
+    context, sources = get_context_sources(message, DB)
     
     conversation = []
     context_message = f"{INSTRUCTION_2}\n\nContesto: {context}\n\nDomanda dell'utente: {message}"