Update functions.py

functions.py

@@ -43,7 +43,7 @@ async def handle_userinput(user_question, custom_graph):
         )
 
         # Retrieve the documents from the graph's state (if available)
-        docs = state_dict.get("documents", [])
+        docs = state("documents")
         
         # Display the retrieved documents in the sidebar
         with st.sidebar: